U
    &%ex@                     @   s  d Z ddlZddlZddlZddlZddlZddlZddlZddl	Z
ddlZ
ddlmZ ddlmZ ddlmZmZmZmZmZmZmZmZmZmZmZ ddlmZ ddlmZ ddl m!Z!m"Z" dd	l#m$Z$ dd
l%m&Z& ddl'm(Z( ddl)m*Z* ddl+m,Z, ddl-m.Z. ddl/m0Z0 ddl1m2Z2 ddl3m4Z4m5Z5m6Z6 erHddlm7Z7 ne8Z7e9e:Z;ee<e<f Z=e<ee< dddZ>G dd de?Z@eddddZAG dd de?ZBe<e*dddd ZCe<e*edd!d"ZDe=ee< d#d$d%ZEG d&d' d'ZFG d(d) d)e7ZGeGeGd*d+d,ZHeHd-ee& d.d/d0ZIG d1d- d-ZJG d2d3 d3eZKdCe&ee<e?f eed4  dd5d6d7ZLdDeeMeJd9d:d;ZNe&e*ed- d<d=d>ZOG d?d@ d@eZPG dAdB dBZQdS )EzO
The main purpose of this module is to expose LinkCollector.collect_sources().
    N)
HTMLParser)Values)TYPE_CHECKINGCallableDictIterableListMutableMapping
NamedTupleOptionalSequenceTupleUnion)requests)Response)
RetryErrorSSLError)NetworkConnectionError)Link)SearchScope)
PipSession)raise_for_status)is_archive_fileredact_auth_from_url)vcs   )CandidatesFromPage
LinkSourcebuild_source)Protocolurlreturnc                 C   s6   t jD ]*}|  |r| t| dkr|  S qdS )zgLook for VCS schemes in the URL.

    Returns the matched VCS scheme, or None if there's no match.
    z+:N)r   schemeslower
startswithlen)r"   scheme r)   \/var/www/html/Darija-Ai-API/env/lib/python3.8/site-packages/pip/_internal/index/collector.py_match_vcs_scheme7   s    

r+   c                       s&   e Zd Zeedd fddZ  ZS )_NotAPIContentN)content_typerequest_descr#   c                    s   t  || || _|| _d S N)super__init__r-   r.   )selfr-   r.   	__class__r)   r*   r1   C   s    z_NotAPIContent.__init__)__name__
__module____qualname__strr1   __classcell__r)   r)   r3   r*   r,   B   s   r,   )responser#   c                 C   s6   | j dd}| }|dr$dS t|| jjdS )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-TypeUnknown)z	text/htmlz#application/vnd.pypi.simple.v1+html#application/vnd.pypi.simple.v1+jsonN)headersgetr%   r&   r,   requestmethod)r:   r-   content_type_lr)   r)   r*   _ensure_api_headerI   s    rC   c                   @   s   e Zd ZdS )_NotHTTPN)r5   r6   r7   r)   r)   r)   r*   rD   _   s   rD   )r"   sessionr#   c                 C   sF   t j| \}}}}}|dkr$t |j| dd}t| t| dS )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.
    >   httphttpsT)allow_redirectsN)urllibparseurlsplitrD   headr   rC   )r"   rE   r(   netlocpathqueryfragmentrespr)   r)   r*   _ensure_api_responsec   s    rR   c                 C   sz   t t| jrt| |d tdt|  |j| ddddgddd	}t	| t
| td
t| |jdd |S )aY  Access 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.
    rE   zGetting page %sz, r=   z*application/vnd.pypi.simple.v1+html; q=0.1ztext/html; q=0.01z	max-age=0)AcceptzCache-Control)r>   zFetched page %s as %sr;   r<   )r   r   filenamerR   loggerdebugr   r?   joinr   rC   r>   )r"   rE   rQ   r)   r)   r*   _get_simple_responseu   s,    rY   )r>   r#   c                 C   s<   | r8d| kr8t j }| d |d< |d}|r8t|S dS )z=Determine if we have any encoding information in our headers.r;   zcontent-typecharsetN)emailmessageMessage	get_paramr8   )r>   mrZ   r)   r)   r*   _get_encoding_from_headers   s    

r`   c                   @   s:   e Zd ZdddddZeedddZed	d
dZdS )CacheablePageContentIndexContentNpager#   c                 C   s   |j s
t|| _d S r/   )cache_link_parsingAssertionErrorrd   r2   rd   r)   r)   r*   r1      s    
zCacheablePageContent.__init__)otherr#   c                 C   s   t |t| o| jj|jjkS r/   )
isinstancetyperd   r"   )r2   rh   r)   r)   r*   __eq__   s    zCacheablePageContent.__eq__r#   c                 C   s   t | jjS r/   )hashrd   r"   r2   r)   r)   r*   __hash__   s    zCacheablePageContent.__hash__)	r5   r6   r7   r1   objectboolrk   intro   r)   r)   r)   r*   ra      s   ra   c                   @   s    e Zd Zdee dddZdS )
ParseLinksrb   rc   c                 C   s   d S r/   r)   rg   r)   r)   r*   __call__   s    zParseLinks.__call__N)r5   r6   r7   r   r   rt   r)   r)   r)   r*   rs      s   rs   )fnr#   c                    sL   t jddttt d fddt  dtt d fdd	}|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#   c                    s   t  | jS r/   )listrd   )rw   )ru   r)   r*   wrapper   s    z*with_cached_index_content.<locals>.wrapperrb   rc   c                    s   | j rt| S t | S r/   )re   ra   rx   )rd   ru   ry   r)   r*   wrapper_wrapper   s    z2with_cached_index_content.<locals>.wrapper_wrapper)	functools	lru_cachera   r   r   wraps)ru   r{   r)   rz   r*   with_cached_index_content   s
    
r   rb   rc   c           
      c   s   | j  }|drTt| j}|dg D ]"}t|| j	}|dkrHq,|V  q,dS t
| j	}| jpfd}|| j| | j	}|jp|}|jD ]$}	tj|	||d}|dkrq|V  qdS )z\
    Parse a Simple API's Index Content, and yield its anchor elements as Link objects.
    r=   filesNzutf-8)page_urlbase_url)r-   r%   r&   jsonloadscontentr?   r   	from_jsonr"   HTMLLinkParserencodingfeeddecoder   anchorsfrom_element)
rd   rB   datafilelinkparserr   r"   r   anchorr)   r)   r*   parse_links   s&    





r   c                   @   s<   e Zd ZdZd
eeee eeddddZeddd	Z	dS )rb   z5Represents one response (or page), along with its URLTN)r   r-   r   r"   re   r#   c                 C   s"   || _ || _|| _|| _|| _dS )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   r"   re   )r2   r   r-   r   r"   re   r)   r)   r*   r1     s
    zIndexContent.__init__rl   c                 C   s
   t | jS r/   )r   r"   rn   r)   r)   r*   __str__  s    zIndexContent.__str__)T)
r5   r6   r7   __doc__bytesr8   r   rq   r1   r   r)   r)   r)   r*   rb     s    c                       sn   e Zd ZdZedd fddZeeeeee f  ddddZ	eeeee f  ee d	d
dZ
  ZS )r   zf
    HTMLParser that keeps the first base HREF and a list of all anchor
    elements' attributes.
    Nr!   c                    s$   t  jdd || _d | _g | _d S )NT)convert_charrefs)r0   r1   r"   r   r   )r2   r"   r3   r)   r*   r1   #  s    zHTMLLinkParser.__init__)tagattrsr#   c                 C   sH   |dkr,| j d kr,| |}|d k	rD|| _ n|dkrD| jt| d S )Nbasea)r   get_hrefr   appenddict)r2   r   r   hrefr)   r)   r*   handle_starttag*  s    
zHTMLLinkParser.handle_starttag)r   r#   c                 C   s"   |D ]\}}|dkr|  S qd S )Nr   r)   )r2   r   namevaluer)   r)   r*   r   2  s    
zHTMLLinkParser.get_href)r5   r6   r7   r   r8   r1   r   r   r   r   r   r9   r)   r)   r3   r*   r     s   "r   ).N)r   reasonmethr#   c                 C   s   |d krt j}|d| | d S )Nz%Could not fetch URL %s: %s - skipping)rV   rW   )r   r   r   r)   r)   r*   _handle_get_simple_fail9  s    r   T)r:   re   r#   c                 C   s&   t | j}t| j| jd || j|dS )Nr;   )r   r"   re   )r`   r>   rb   r   r"   )r:   re   r   r)   r)   r*   _make_index_contentC  s    
r   )r   rE   r#   c          
   
   C   s  | j ddd }t|}|r0td||  d S tj|\}}}}}}|dkrtj	
tj|r|dsv|d7 }tj|d}td| zt||d	}W nD tk
r   td
|  Y n2 tk
r } ztd| |j|j W 5 d }~X Y n tk
r$ } zt| | W 5 d }~X Y n tk
rP } zt| | W 5 d }~X Y n tk
r } z$d}	|	t|7 }	t| |	tjd W 5 d }~X Y nd tjk
r } zt| d|  W 5 d }~X Y n0 tjk
r   t| d Y nX t|| jdS d S )N#r   r   zICannot look at %s URL %s because it does not support lookup as web pages.r   /z
index.htmlz# file: URL is directory, getting %srS   z`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: )r   zconnection error: z	timed out)re   ) r"   splitr+   rV   warningrI   rJ   urlparseosrN   isdirr@   url2pathnameendswithurljoinrW   rY   rD   r,   r.   r-   r   r   r   r   r8   infor   ConnectionErrorTimeoutr   re   )
r   rE   r"   
vcs_schemer(   _rN   rQ   excr   r)   r)   r*   _get_index_contentP  sV    

  r   c                   @   s.   e Zd ZU eee  ed< eee  ed< dS )CollectedSources
find_links
index_urlsN)r5   r6   r7   r   r   r   __annotations__r)   r)   r)   r*   r     s   
r   c                   @   sx   e Zd ZdZeeddddZedeee	d ddd	Z
eee d
ddZeee dddZeeedddZdS )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)rE   search_scoper#   c                 C   s   || _ || _d S r/   )r   rE   )r2   rE   r   r)   r)   r*   r1     s    zLinkCollector.__init__F)rE   optionssuppress_no_indexr#   c                 C   sd   |j g|j }|jr8|s8tdddd |D  g }|jp@g }tj|||j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.
        zIgnoring indexes: %s,c                 s   s   | ]}t |V  qd S r/   r   ).0r"   r)   r)   r*   	<genexpr>  s     z'LinkCollector.create.<locals>.<genexpr>)r   r   no_index)rE   r   )
	index_urlextra_index_urlsr   rV   rW   rX   r   r   creater   )clsrE   r   r   r   r   r   link_collectorr)   r)   r*   r     s$    

zLinkCollector.createrl   c                 C   s   | j jS r/   )r   r   rn   r)   r)   r*   r     s    zLinkCollector.find_links)locationr#   c                 C   s   t || jdS )z>
        Fetch an HTML page containing package links.
        rS   )r   rE   )r2   r   r)   r)   r*   fetch_response  s    zLinkCollector.fetch_response)project_namecandidates_from_pager#   c                    s   t  fddj|D  }t  fddjD  }ttj	rdd t
||D }t| d| dg| }td| tt|t|d	S )
Nc                 3   s$   | ]}t | jjd d dV  qdS )Fr   page_validator
expand_dirre   Nr   rE   is_secure_originr   locr   r2   r)   r*   r     s   z0LinkCollector.collect_sources.<locals>.<genexpr>c                 3   s$   | ]}t | jjd d dV  qdS )Tr   Nr   r   r   r)   r*   r     s   c                 S   s*   g | ]"}|d k	r|j d k	rd|j  qS )Nz* )r   )r   sr)   r)   r*   
<listcomp>  s    
z1LinkCollector.collect_sources.<locals>.<listcomp>z' location(s) to search for versions of :
)r   r   )collectionsOrderedDictr   get_index_urls_locationsvaluesr   rV   isEnabledForloggingDEBUG	itertoolschainr'   rW   rX   r   rx   )r2   r   r   index_url_sourcesfind_links_sourceslinesr)   r   r*   collect_sources  s&    


zLinkCollector.collect_sources)F)r5   r6   r7   r   r   r   r1   classmethodr   rq   r   propertyr   r8   r   r   r   rb   r   r   r   r   r)   r)   r)   r*   r     s(   	 !r   )N)T)Rr   r   email.messager[   r|   r   r   r   r   urllib.parserI   urllib.requesthtml.parserr   optparser   typingr   r   r   r   r   r	   r
   r   r   r   r   pip._vendorr   Zpip._vendor.requestsr   Zpip._vendor.requests.exceptionsr   r   pip._internal.exceptionsr   pip._internal.models.linkr   !pip._internal.models.search_scoper   pip._internal.network.sessionr   pip._internal.network.utilsr   pip._internal.utils.filetypesr   pip._internal.utils.miscr   pip._internal.vcsr   sourcesr   r   r   r    rp   	getLoggerr5   rV   r8   ResponseHeadersr+   	Exceptionr,   rC   rD   rR   rY   r`   ra   rs   r   r   rb   r   r   rq   r   r   r   r   r)   r)   r)   r*   <module>   sv   4
? 

  =