U 7cE@sdZddlZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZmZmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZm Z ddl!m"Z"m#Z#er@ddl$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,ddl-Z.ddl/m0Z0ddl1m2Z2ddl3m4Z4e.j5j6j7Z8e(e9e9fZ:e;e<Z=ddZ>ddZ?Gddde@ZAddZBGddde@ZCddZDdd ZEd!d"ZFd#d$ZGd%d&ZHd'd(ZId)d*ZJGd+d,d,eKZLdd6d7ZQGd8d9d9eKZRGd:d;d;eKZSdS)?zM The main purpose of this module is to expose LinkCollector.collect_links(). N) OrderedDict)html5librequests)unescape) HTTPError RetryErrorSSLError)parse)requestLink)ARCHIVE_EXTENSIONS)redact_auth_from_url)MYPY_CHECK_RUNNING) path_to_url url_to_path)is_urlvcs)CallableIterableListMutableMappingOptionalSequenceTupleUnion)Response) SearchScope) PipSessioncCs6tjD]*}||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)rschemeslower startswithlen)urlschemer%A/tmp/pip-unpacked-wheel-xvghy_sv/pip/_internal/index/collector.py_match_vcs_scheme,s  r'cCs(t|j}tD]}||rdSqdS)z2Return whether the URL looks like an archive. TF)r filenamer endswith)r#r(bad_extr%r%r&_is_url_like_archive8s   r+cseZdZfddZZS)_NotHTMLcs"tt|||||_||_dSN)superr,__init__ content_type request_desc)selfr0r1 __class__r%r&r/Dsz_NotHTML.__init__)__name__ __module__ __qualname__r/ __classcell__r%r%r3r&r,Csr,cCs.|jdd}|ds*t||jjdS)zCheck the Content-Type header to ensure the response contains HTML. Raises `_NotHTML` if the content type is not text/html. Content-Type text/htmlN)headersgetr r!r,r method)responser0r%r%r&_ensure_html_headerKsr@c@s eZdZdS)_NotHTTPN)r5r6r7r%r%r%r&rAVsrAcCsDt|\}}}}}|dkr"t|j|dd}|t|dS)zSend a HEAD request to the URL, and ensure the response contains HTML. Raises `_NotHTTP` if the URL is not available for a HEAD request, or `_NotHTML` if the content type is not text/html. >httphttpsT)allow_redirectsN) urllib_parseurlsplitrAheadraise_for_statusr@)r#sessionr$netlocpathqueryfragmentrespr%r%r&_ensure_html_responseZs rOcCsLt|rt||dtdt||j|dddd}|t||S)aAccess an HTML page 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, to avoid downloading a large file. Raise `_NotHTTP` if the content type cannot be determined, or `_NotHTML` if it is not HTML. 2. Actually perform the request. Raise HTTP exceptions on network failures. 3. Check the Content-Type header to make sure we got HTML, and raise `_NotHTML` otherwise. rIzGetting page %sr;z max-age=0)Acceptz Cache-Control)r<)r+rOloggerdebugrr=rHr@)r#rIrNr%r%r&_get_html_responseks rTcCs2|r.d|kr.t|d\}}d|kr.|dSdS)zBDetermine if we have any encoding information in our headers. r9charsetN)cgi parse_header)r<r0paramsr%r%r&_get_encoding_from_headerss  rYcCs.|dD]}|d}|dk r |Sq |S)aDetermine the HTML document's base URL. This looks for a ```` tag in the HTML document. If present, its href attribute denotes the base URL of anchor tags in the document. If there is no such tag (or if it does not have a valid href attribute), the HTML file's URL is used as the base URL. :param document: An HTML document representation. The current implementation expects the result of ``html5lib.parse()``. :param page_url: The URL of the HTML document. z.//basehrefN)findallr=)documentpage_urlbaserZr%r%r&_determine_base_urls   r_cCsPt|}|jdkr(tt|j}ntjt|jdd}t |j |dS)zMakes sure a link is fully encoded. That is, if a ' ' shows up in the link, it will be rewritten to %20 (while not over-quoting % or other characters).r:z/@)safe)rK) rEurlparserJurllib_request pathname2url url2pathnamerKquoteunquote urlunparse_replace)r#resultrKr%r%r& _clean_links   rjcCsf|d}|sdStt||}|d}|r8t|nd}|d}|rRt|}t||||d}|S)zJ Convert an anchor element in a simple repository page to a Link. rZNzdata-requires-pythonz data-yanked) comes_fromrequires_python yanked_reason)r=rjrEurljoinrr )anchorr]base_urlrZr# pyrequirermlinkr%r%r&_create_link_from_elements   rsccsVtj|j|jdd}|j}t||}|dD]"}t|||d}|dkrJq.|Vq.dS)zP Parse an HTML document, and yield its anchor elements as Link objects. F)transport_encodingnamespaceHTMLElementsz.//a)r]rpN)rr contentencodingr#r_r[rs)pager\r#rprorrr%r%r& parse_linkss  ryc@s eZdZdZddZddZdS)HTMLPagez'Represents one page, along with its URLcCs||_||_||_dS)z :param encoding: the encoding to decode the given content. :param url: the URL from which the HTML was downloaded. N)rvrwr#)r2rvrwr#r%r%r&r/s zHTMLPage.__init__cCs t|jSr-)rr#r2r%r%r&__str__!szHTMLPage.__str__N)r5r6r7__doc__r/r|r%r%r%r&rzsrzcCs|dkrtj}|d||dS)Nz%Could not fetch URL %s: %s - skipping)rRrS)rrreasonmethr%r%r&_handle_get_page_fail&srcCst|j}t|j||jdS)N)rwr#)rYr<rzrvr#)r?rwr%r%r&_make_html_page1s rc Cs|dkrtd|jddd}t|}|r@td||dSt|\}}}}}}|dkrtj t |r| ds|d7}t|d}td |zt||d }WnDtk rtd |Yn,tk r}ztd ||j|jW5d}~XYntk r0}zt||W5d}~XYntk r\}zt||W5d}~XYntk r}z$d } | t|7} t|| tjdW5d}~XYn\tjk r}zt|d|W5d}~XYn*tjk rt|dYn Xt|SdS)Nz?_get_html_page() missing 1 required keyword argument: 'session'#rzCannot look at %s URL %sfile/z index.htmlz# file: URL is directory, getting %srPzQSkipping page %s because it looks like an archive, and cannot be checked by HEAD.z.sort_pathzfile:z)Path '{0}' is ignored: it is a directory.z:Url '%s' is ignored: it is neither a file nor a directory.zQUrl '%s' is ignored. It is either a non-existing path or lacks a specific scheme.)rrKexistsr!rrrealpathlistdirjoinrrRwarningformatisfiler) locations expand_dirrr# is_local_path is_file_urlrKitemr%rr&group_locationsvsF        rc@seZdZdZddZdS)CollectedLinksa Encapsulates the return value of a call to LinkCollector.collect_links(). The return value includes both URLs to project pages containing package links, as well as individual package Link objects collected from other sources. This info is stored separately as: (1) links from the configured file locations, (2) links from the configured find_links, and (3) urls to HTML project pages, as described by the PEP 503 simple repository API. cCs||_||_||_dS)z :param files: Links from file locations. :param find_links: Links from find_links. :param project_urls: URLs to HTML project pages, as described by the PEP 503 simple repository API. Nr find_links project_urls)r2rrrr%r%r&r/s zCollectedLinks.__init__N)r5r6r7r}r/r%r%r%r&rsrc@s4eZdZdZddZeddZddZdd Zd S) LinkCollectorz Responsible for collecting Link objects from all configured locations, making network requests as needed. The class's main method is its collect_links() method. cCs||_||_dSr-) search_scoperI)r2rIrr%r%r&r/szLinkCollector.__init__cCs|jjSr-)rrr{r%r%r&rszLinkCollector.find_linkscCst||jdS)z> Fetch an HTML page containing package links. rP)rrI)r2locationr%r%r& fetch_pageszLinkCollector.fetch_pagec sj}||}t|\}}tjdd\}}ddt||D}ddjD} fddtdd|Dd d|DD} t| } d t| |g} | D]} | d | qt d | t || | d S)zFind all available links for the given project name. :return: All the Link objects (unfiltered), as a CollectedLinks object. T)rcSsg|] }t|qSr%r .0r#r%r%r& sz/LinkCollector.collect_links..cSsg|]}t|dqS)z-fr rr%r%r&rscsg|]}j|r|qSr%)rIis_secure_origin)rrrr{r%r&r s css|]}t|VqdSr-r rr%r%r& sz.LinkCollector.collect_links..css|]}t|VqdSr-r rr%r%r&r sz,{} location(s) to search for versions of {}:z* {} r)rget_index_urls_locationsrr itertoolschainrrr"rrRrSrr) r2 project_namerindex_locationsindex_file_loc index_url_loc fl_file_loc fl_url_loc file_linksfind_link_links url_locationslinesrrr%r{r& collect_linkss>       zLinkCollector.collect_linksN) r5r6r7r}r/propertyrrrr%r%r%r&rs   r)N)N)F)Tr}rVrloggingrr collectionsr pip._vendorrrpip._vendor.distlib.compatrZpip._vendor.requests.exceptionsrrrZpip._vendor.six.moves.urllibr rEr rbpip._internal.models.linkr pip._internal.utils.filetypesr pip._internal.utils.miscrpip._internal.utils.typingrpip._internal.utils.urlsrrpip._internal.vcsrrtypingrrrrrrrrxml.etree.ElementTreexmlZpip._vendor.requestsr!pip._internal.models.search_scoperpip._internal.network.sessionretree ElementTreeElementZ HTMLElementrZResponseHeaders getLoggerr5rRr'r+ Exceptionr,r@rArOrTrYr_rjrsryobjectrzrrrrrrrr%r%r%r&s^        (         3    6 <#