a
lj~D @ s d Z ddlZddlZddlZddlZddlZddlZddlZddlZ ddl
Z ddl Z ddl
mZ ddlmZmZmZmZmZmZmZmZ ddlmZmZ ddlm Z ddl m Z m Z ddl 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-m.Z. ddl/m0Z0 ddl1m2Z2m3Z3m4Z4 e5e6Z7e j8j9j:Z;eeG dd de?Z@e ddddZAG dd de?ZBed7 d7ZSdNe#ee httpshttpT)Zallow_redirectsN)urllibparseurlsplitr= headr r< )r r> r" netlocpathqueryfragmentrespr# r# r$ _ensure_html_responseP s rJ c C sR t t| jrt| |d tdt| |j| dddd}t| t | |S )a Access 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.
r> zGetting page %sr7 z max-age=0)ZAcceptz
Cache-Control)r8 )
r r filenamerJ loggerdebugr r9 r r< )r r> rI r# r# r$ _get_html_response` s
rO )r8 r c C s2 | r.d| v r.t | d \}}d|v r.|d S dS )z=Determine if we have any encoding information in our headers.r5 charsetN)cgiZ parse_header)r8 r' paramsr# r# r$ _get_encoding_from_headers s
rS )documentpage_urlr c C s. | dD ] }|d}|dur
| S q
|S )a Determine 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)findallr9 )rT rU baserV r# r# r$ _determine_base_url s
rY )partr c C s t jt j| S )zP
Clean a "part" of a URL path (i.e. after splitting on "@" characters).
)rA rB quoteunquoterZ r# r# r$ _clean_url_path_part s r^ c C s t jt j| S )z
Clean the first part of a URL path that corresponds to a local
filesystem path (i.e. the first part after splitting on "@" characters).
)rA r: Z pathname2url url2pathnamer] r# r# r$ _clean_file_url_path s
r` z(@|%2F))rF
is_local_pathr c C s^ |r
t }nt}t| }g }tt|dgD ]$\}}||| || q.d |S )z*
Clean the path portion of a URL.
r6 )
r` r^ _reserved_chars_resplitr itertoolschainappendupperjoin)rF ra Z
clean_funcpartsZ
cleaned_partsZto_cleanZreservedr# r# r$ _clean_url_path s
rj c C s6 t j| }|j }t|j|d}t j|j|dS )z
Make sure a link is fully quoted.
For example, if ' ' occurs in the URL, it will be replaced with "%20",
and without double-quoting other characters.
)ra )rF )rA rB urlparserE rj rF
urlunparse_replace)r resultra rF r# r# r$ _clean_link s ro )anchorrU base_urlr c C sL | d}|sdS ttj||}| d}| d}t||||d}|S )zJ
Convert an anchor element in a simple repository page to a Link.
rV Nzdata-requires-pythonz data-yanked)Z
comes_fromZrequires_python
yanked_reason)r9 ro rA rB urljoinr )rp rU rq rV r Z pyrequirerr linkr# r# r$ _create_link_from_element s
ru c @ s: e Zd ZdddddZeedddZed d
d ZdS ) CacheablePageContentHTMLPageNpager c C s |j s
J || _d S r) )cache_link_parsingry )r, ry r# r# r$ r+ s
z CacheablePageContent.__init__)otherr c C s t |t| o | jj|jjkS r) )
isinstancetypery r )r, r{ r# r# r$ __eq__ s zCacheablePageContent.__eq__r c C s t | jjS r) )hashry r r, r# r# r$ __hash__ s z CacheablePageContent.__hash__) r/ r0 r1 r+ objectboolr~ intr r# r# r# r$ rv s rv rw )fnr c sL t jddttt d fdd t dtt d fdd }|S )
z
Given a function that parses an Iterable[Link] from an HTMLPage, cache the
function's result (keyed by CacheablePageContent), unless the HTMLPage
`page` has `page.cache_link_parsing == False`.
N)maxsize)cacheable_pager c s t | jS r) )listry )r )r r# r$ wrapper s z'with_cached_html_pages..wrapperrw rx c s | j rt| S t | S r) )rz rv r )ry r r r# r$ wrapper_wrapper s z/with_cached_html_pages..wrapper_wrapper) functools lru_cacherv r r wraps)r r r# r r$ with_cached_html_pages s
r rx c c sV t j| j| jdd}| j}t||}|dD ]"}t|||d}|du rJq.|V q.dS )zP
Parse an HTML document, and yield its anchor elements as Link objects.
F)Ztransport_encodingZnamespaceHTMLElementsz.//a)rU rq N)r rB contentencodingr rY rW ru )ry rT r rq rp rt r# r# r$ parse_links( s
r c @ s: e Zd ZdZd
eee eeddddZeddd Z dS ) rw z'Represents one page, along with its URLTN)r r r rz r c C s || _ || _|| _|| _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)r r r rz )r, r r r rz r# r# r$ r+ C s zHTMLPage.__init__r c C s
t | jS r) )r r r r# r# r$ __str__V s zHTMLPage.__str__)T)
r/ r0 r1 __doc__bytesr r2 r r+ r r# r# r# r$ rw @ s ).N)rt reasonmethr c C s |d u rt j}|d| | d S )Nz%Could not fetch URL %s: %s - skipping)rM rN )rt r r r# r# r$ _handle_get_page_failZ s r T)r4 rz r c C s t | j}t| j|| j|dS )N)r r rz )rS r8 rw r r )r4 rz r r# r# r$ _make_html_paged s
r )rt r> r c
C s |d u rt d| jddd }t|}|r@td|| d S tj|\}}}}}}|dkrt j
tj
|r|ds|d7 }tj|d}td | zt||d
}W nN ty td | Y n> ty } z"td | |j|j W Y d }~n d }~0 ty: } zt| | W Y d }~nd }~0 tyh } zt| | W Y d }~nd }~0 ty } z,d
} | t|7 } t| | tjd W Y d }~nld }~0 t j y } z t| d| W Y d }~n6d }~0 t j y t| d Y n0 t || j dS d S )Nz?_get_html_page() missing 1 required keyword argument: 'session'#r r zICannot look at %s URL %s because it does not support lookup as web pages.file/z
index.htmlz# file: URL is directory, getting %srK z`Skipping page %s because it looks like an archive, and cannot be checked by a HTTP HEAD request.ziSkipping page %s because the %s request got Content-Type: %s.The only supported Content-Type is text/htmlz4There was a problem confirming the ssl certificate: )r zconnection error: z timed out)rz )! TypeErrorr rc r% rM ZwarningrA rB rk osrF isdirr: r_ endswithrs rN rO r= r& r( r' r r r r r2 infor ConnectionErrorZTimeoutr rz )
rt r> r Z
vcs_schemer" _rF rI excr r# r# r$ _get_html_pagen s^
$$r c @ s. e Zd ZU eee ed< eee ed< dS )CollectedSources
find_links
index_urlsN)r/ r0 r1 r r r __annotations__r# r# r# r$ r s
r c @ sx e Zd ZdZeeddddZedeee d ddd Z
e e e
d
d d Zeee d
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)r> search_scoper c C s || _ || _d S r) )r r> )r, r> r r# r# r$ r+ s zLinkCollector.__init__F)r> optionssuppress_no_indexr c C s` |j g|j }|jr8|s8tdddd |D g }|jp@g }tj||d}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,c s s | ]}t |V qd S r) )r ).0r r# r# r$ z'LinkCollector.create..r r )r> r )
Z index_urlZextra_index_urlsZno_indexrM rN rh r r creater )clsr> r r r r r Zlink_collectorr# r# r$ r s"
zLinkCollector.creater c C s | j jS r) )r r r r# r# r$ r s zLinkCollector.find_links)locationr c C s t || jdS )z>
Fetch an HTML page containing package links.
rK )r r> )r, r r# r# r$
fetch_page s zLinkCollector.fetch_page) project_namecandidates_from_pager c s t fddj|D }t fddjD }ttj rdd t
||D }t | d| dg| }t
d| tt|t|d S )
Nc 3 s$ | ] }t | jjd d dV qdS )Fr Zpage_validatorZ
expand_dirrz Nr r> Zis_secure_originr locr r, r# r$ r s z0LinkCollector.collect_sources..c 3 s$ | ] }t | jjd d dV qdS )Tr Nr r r r# r$ r s c S s* g | ]"}|d ur|j d urd|j qS )Nz* )rt )r sr# r# r$
s z1LinkCollector.collect_sources..z' location(s) to search for versions of :
r ) collections OrderedDictr Zget_index_urls_locationsvaluesr rM Z isEnabledForloggingDEBUGrd re r! rN rh r r )r, r r Zindex_url_sourcesZfind_links_sourceslinesr# r r$ collect_sources s*
z LinkCollector.collect_sources)F)r/ r0 r1 r r r r+ classmethodr r r propertyr r2 r r r rw r r r r r# r# r# r$ r s( r )N)T)N)Yr rQ r r rd r r re urllib.parserA Zurllib.requestZxml.etree.ElementTreeZxmlZoptparser typingr r r r r r r r
Z pip._vendorr r Zpip._vendor.requestsr
Z pip._vendor.requests.exceptionsr r Zpip._internal.exceptionsr Zpip._internal.models.linkr Z!pip._internal.models.search_scoper Z pip._internal.network.sessionr Zpip._internal.network.utilsr Z pip._internal.utils.filetypesr Zpip._internal.utils.miscr r Zpip._internal.vcsr Zsourcesr r r Z getLoggerr/ rM ZetreeZ ElementTreeZElementZ HTMLElementr2 ZResponseHeadersr% Exceptionr& r< r= rJ rO rS rY r^ r` compile
IGNORECASErb r rj ro ru rv r r rw r r r r r r# r# r# r$ s (
2
?