U }ag@@sdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddlmZddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZddl m!Z!m"Z"dd l#m$Z$dd l%m&Z&dd l'm(Z(dd l)m*Z*dd l+m,Z,ddl-m.Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4m5Z5m6Z6erHddlm7Z7ne8Z7e9e:Z;eeGddde?Z@eddddZAGddde?ZBeZOGd?d@d@eZPGdAdBdBZQdS)EzO The main purpose of this module is to expose LinkCollector.collect_sources(). N) HTMLParser)Values) TYPE_CHECKINGCallableDictIterableListMutableMapping NamedTupleOptionalSequenceTupleUnion)requests)Response) RetryErrorSSLError)NetworkConnectionError)Link) SearchScope) PipSession)raise_for_status)is_archive_fileredact_auth_from_url)vcs)CandidatesFromPage LinkSource build_source)ProtocolurlreturncCs6tjD]*}||r|t|dkr|SqdS)zgLook for VCS schemes in the URL. Returns the matched VCS scheme, or None if there's no match. z+:N)rZschemeslower startswithlen)r"schemer(U/var/www/html/publicvenv/lib/python3.8/site-packages/pip/_internal/index/collector.py_match_vcs_scheme7s  r*cs&eZdZeeddfdd ZZS)_NotAPIContentN) content_type request_descr#cst||||_||_dSN)super__init__r,r-)selfr,r- __class__r(r)r0Csz_NotAPIContent.__init__)__name__ __module__ __qualname__strr0 __classcell__r(r(r2r)r+Bsr+)responser#cCs6|jdd}|}|dr$dSt||jjdS)z Check the Content-Type header to ensure the response contains a Simple API Response. Raises `_NotAPIContent` if the content type is not a valid content-type. Content-TypeUnknown)z text/htmlz#application/vnd.pypi.simple.v1+html#application/vnd.pypi.simple.v1+jsonN)headersgetr$r%r+requestmethod)r9r,content_type_lr(r(r)_ensure_api_headerIsrBc@s eZdZdS)_NotHTTPN)r4r5r6r(r(r(r)rC_srC)r"sessionr#cCsFtj|\}}}}}|dkr$t|j|dd}t|t|dS)z Send a HEAD request to the URL, and ensure the response contains a simple API Response. Raises `_NotHTTP` if the URL is not available for a HEAD request, or `_NotAPIContent` if the content type is not a valid content type. >httphttpsT)allow_redirectsN)urllibparseurlsplitrCheadrrB)r"rDr'netlocpathqueryfragmentrespr(r(r)_ensure_api_responsecs rQcCsztt|jrt||dtdt||j|ddddgddd }t |t |td t||j d d |S) aYAccess an Simple API response with GET, and return the response. This consists of three parts: 1. If the URL looks suspiciously like an archive, send a HEAD first to check the Content-Type is HTML or Simple API, to avoid downloading a large file. Raise `_NotHTTP` if the content type cannot be determined, or `_NotAPIContent` if it is not HTML or a Simple API. 2. Actually perform the request. Raise HTTP exceptions on network failures. 3. Check the Content-Type header to make sure we got a Simple API response, and raise `_NotAPIContent` otherwise. rDzGetting page %sz, r<z*application/vnd.pypi.simple.v1+html; q=0.1ztext/html; q=0.01z max-age=0)Acceptz Cache-Control)r=zFetched page %s as %sr:r;) rrfilenamerQloggerdebugrr>joinrrBr=)r"rDrPr(r(r)_get_simple_responseus,   rX)r=r#cCs<|r8d|kr8tj}|d|d<|d}|r8t|SdS)z=Determine if we have any encoding information in our headers.r:z content-typecharsetN)emailmessageMessage get_paramr7)r=mrYr(r(r)_get_encoding_from_headerss    r_c@s:eZdZdddddZeedddZed d d ZdS) CacheablePageContent IndexContentNpager#cCs|js t||_dSr.)cache_link_parsingAssertionErrorrcr1rcr(r(r)r0s zCacheablePageContent.__init__)otherr#cCst|t|o|jj|jjkSr.) isinstancetypercr")r1rgr(r(r)__eq__szCacheablePageContent.__eq__r#cCs t|jjSr.)hashrcr"r1r(r(r)__hash__szCacheablePageContent.__hash__) r4r5r6r0objectboolrjintrnr(r(r(r)r`sr`c@s eZdZdeedddZdS) ParseLinksrarbcCsdSr.r(rfr(r(r)__call__szParseLinks.__call__N)r4r5r6rrrsr(r(r(r)rrsrr)fnr#csLtjddtttdfdd tdttdfdd }|S) z Given a function that parses an Iterable[Link] from an IndexContent, cache the function's result (keyed by CacheablePageContent), unless the IndexContent `page` has `page.cache_link_parsing == False`. N)maxsize)cacheable_pager#cst|jSr.)listrc)rv)rtr(r)wrappersz*with_cached_index_content..wrapperrarbcs|jrt|St|Sr.)rdr`rw)rcrtrxr(r)wrapper_wrappers z2with_cached_index_content..wrapper_wrapper) functools lru_cacher`rrwraps)rtrzr(ryr)with_cached_index_contents  r~rarbc cs|j}|drTt|j}|dgD]"}t||j }|dkrHq,|Vq,dSt |j }|j pfd}| |j ||j }|jp|}|jD]$} tj| ||d}|dkrq|VqdS)z\ Parse a Simple API's Index Content, and yield its anchor elements as Link objects. r<filesNzutf-8)page_urlbase_url)r,r$r%jsonloadscontentr>r from_jsonr"HTMLLinkParserencodingfeeddecoderanchors from_element) rcrAdatafilelinkparserrr"ranchorr(r(r) parse_linkss&       rc@s<eZdZdZd eeeeeeddddZeddd Z dS) raz5Represents one response (or page), along with its URLTN)rr,rr"rdr#cCs"||_||_||_||_||_dS)am :param encoding: the encoding to decode the given content. :param url: the URL from which the HTML was downloaded. :param cache_link_parsing: whether links parsed from this page's url should be cached. PyPI index urls should have this set to False, for example. N)rr,rr"rd)r1rr,rr"rdr(r(r)r0s zIndexContent.__init__rkcCs t|jSr.)rr"rmr(r(r)__str__szIndexContent.__str__)T) r4r5r6__doc__bytesr7r rpr0rr(r(r(r)ras csneZdZdZeddfdd ZeeeeeefddddZ eeeeefeed d d Z Z S) rzf HTMLParser that keeps the first base HREF and a list of all anchor elements' attributes. Nr!cs$tjdd||_d|_g|_dS)NT)Zconvert_charrefs)r/r0r"rr)r1r"r2r(r)r0#szHTMLLinkParser.__init__)tagattrsr#cCsH|dkr,|jdkr,||}|dk rD||_n|dkrD|jt|dS)Nbasea)rget_hrefrappenddict)r1rrhrefr(r(r)handle_starttag*s  zHTMLLinkParser.handle_starttag)rr#cCs"|D]\}}|dkr|SqdS)Nrr()r1rnamevaluer(r(r)r2s  zHTMLLinkParser.get_href) r4r5r6rr7r0rr r rrr8r(r(r2r)rs"r).N)rreasonmethr#cCs|dkrtj}|d||dS)Nz%Could not fetch URL %s: %s - skipping)rUrV)rrrr(r(r)_handle_get_simple_fail9srT)r9rdr#cCs&t|j}t|j|jd||j|dS)Nr:)rr"rd)r_r=rarr")r9rdrr(r(r)_make_index_contentCs r)rrDr#c Cs|jddd}t|}|r0td||dStj|\}}}}}}|dkrtj tj |r| dsv|d7}tj|d}td|zt||d }WnDtk rtd |Yn2tk r}ztd ||j|jW5d}~XYntk r$}zt||W5d}~XYntk rP}zt||W5d}~XYntk r}z$d } | t|7} t|| tjd W5d}~XYndtjk r}zt|d|W5d}~XYn0tjk rt|dYnXt||jdSdS)N#rrzICannot look at %s URL %s because it does not support lookup as web pages.r/z index.htmlz# file: URL is directory, getting %srRz`Skipping page %s because it looks like an archive, and cannot be checked by a HTTP HEAD request.zSkipping page %s because the %s request got Content-Type: %s. The only supported Content-Types are application/vnd.pypi.simple.v1+json, application/vnd.pypi.simple.v1+html, and text/htmlz4There was a problem confirming the ssl certificate: )rzconnection error: z timed out)rd) r"splitr*rUwarningrHrIurlparseosrMisdirr? url2pathnameendswithurljoinrVrXrCr+r-r,rrrrr7inforConnectionErrorTimeoutrrd) rrDr"Z vcs_schemer'_rMrPexcrr(r(r)_get_index_contentPsV      rc@s.eZdZUeeeed<eeeed<dS)CollectedSources find_links index_urlsN)r4r5r6r r r__annotations__r(r(r(r)rs rc@sxeZdZdZeeddddZedeee dddd Z e e e d d d Zeeed ddZe eedddZdS) LinkCollectorz Responsible for collecting Link objects from all configured locations, making network requests as needed. The class's main method is its collect_sources() method. N)rD search_scoper#cCs||_||_dSr.)rrD)r1rDrr(r(r)r0szLinkCollector.__init__F)rDoptionssuppress_no_indexr#cCsd|jg|j}|jr8|s8tdddd|Dg}|jp@g}tj|||jd}t ||d}|S)z :param session: The Session to use to make requests. :param suppress_no_index: Whether to ignore the --no-index option when constructing the SearchScope object. zIgnoring indexes: %s,css|]}t|VqdSr.r).0r"r(r(r) sz'LinkCollector.create..)rrno_index)rDr) index_urlextra_index_urlsrrUrVrWrrcreater)clsrDrrrrrlink_collectorr(r(r)rs$   zLinkCollector.createrkcCs|jjSr.)rrrmr(r(r)rszLinkCollector.find_links)locationr#cCst||jdS)z> Fetch an HTML page containing package links. rR)rrD)r1rr(r(r)fetch_responseszLinkCollector.fetch_response) project_namecandidates_from_pager#cstfddjD}tfddjD}ttj rddt ||D}t |ddg|}t d|tt|t|d S) Nc 3s&|]}t|jjdddVqdS)FrZpage_validatorZ expand_dirrdrNrrDZis_secure_originrlocrrr1r(r)rs z0LinkCollector.collect_sources..c 3s&|]}t|jjdddVqdS)TrNrrrr(r)rs cSs*g|]"}|dk r|jdk rd|jqS)Nz* )r)rsr(r(r) s z1LinkCollector.collect_sources..z' location(s) to search for versions of : )rr) collections OrderedDictrZget_index_urls_locationsvaluesrrU isEnabledForloggingDEBUG itertoolschainr&rVrWrrw)r1rrZindex_url_sourcesZfind_links_sourceslinesr(rr)collect_sourcess&    zLinkCollector.collect_sources)F)r4r5r6rrrr0 classmethodrrprpropertyrr7rrr rarrrrr(r(r(r)rs(  !r)N)T)Rrr email.messagerZr{rrrr urllib.parserHurllib.requestZ html.parserroptparsertypingrrrrrr r r r r r pip._vendorrZpip._vendor.requestsrZpip._vendor.requests.exceptionsrrpip._internal.exceptionsrpip._internal.models.linkrZ!pip._internal.models.search_scoperpip._internal.network.sessionrZpip._internal.network.utilsrpip._internal.utils.filetypesrpip._internal.utils.miscrpip._internal.vcsrsourcesrrrr ro getLoggerr4rUr7ZResponseHeadersr* Exceptionr+rBrCrQrXr_r`rrr~rrarrrprrrrr(r(r(r)sv  4             ?      =