U
    ;g
C                     @   s,  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	Z	d dl
Z
d dlmZ d dlmZmZ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 m!Z! ddl"m#Z# dd	l$m%Z%m&Z&m'Z' d
Z(ee)df Z*dj+Z,dj+Z-dZ.G dd de Z/G dd de Z0dS )    N)defaultdict)
BaseCookieMorselSimpleCookie)DefaultDictDictIterableIteratorListMappingOptionalSetTupleUnioncast)URL   )AbstractCookieJarClearCookiePredicate)is_ip_address)LooseCookiesPathLikeStrOrURL)	CookieJarDummyCookieJarMorsel[str]z{}/{}z{1}.{0}d   c                	       s  e Zd ZdZedZedZedZedej	Z
edZeejjjejjd d Zzeee W n8 eefk
r   ed	ZY n ek
r   d
ZY nX ed Zdddddeeeeee df e e!j" dd fddZ#e$ddddZ%e$ddddZ&d8e e' ddddZ(e)ddddZ*dddd Z+edd!d"Z,ddd#d$Z-ee.e)e)e)f  dd%d&d'Z/e0e)e)e)dd(d)d*Z1e2 fe3e2dd+d,d-Z4e2 fe2d.d/d0d1Z5e6e)e)ed2d3d4Z7e8e)e e d5d6d7Z9  Z:S )9r   z/Implements cookie storage adhering to RFC 6265.z][\x09\x20-\x2F\x3B-\x40\x5B-\x60\x7B-\x7E]*(?P<token>[\x00-\x08\x0A-\x1F\d:a-zA-Z\x7F-\xFF]+)z(\d{1,2}):(\d{1,2}):(\d{1,2})z	(\d{1,2})zG(jan)|(feb)|(mar)|(apr)|(may)|(jun)|(jul)|(aug)|(sep)|(oct)|(nov)|(dec)z	(\d{2,4}))tzinfor   )	i           ;   r!   r"   r"   iFTN)unsafequote_cookietreat_as_secure_originloop)r#   r$   r%   r&   returnc                   s   t  j|d tt| _tt| _t | _|| _	|| _
|d krDg }n>t|trZ| g}n(t|trtt| g}ndd |D }|| _g | _i | _d S )Nr&   c                 S   s*   g | ]"}t |trt| n| qS  )
isinstancestrr   origin).0urlr)   r)   Q/opt/bot/vods_filmes/script/venv/lib/python3.8/site-packages/aiohttp/cookiejar.py
<listcomp>o   s   z&CookieJar.__init__.<locals>.<listcomp>)super__init__r   r   _cookiesdict_morsel_cacheset_host_only_cookies_unsafe_quote_cookier*   r   r,   r+   _treat_as_secure_origin_expire_heap_expirations)selfr#   r$   r%   r&   	__class__r)   r/   r2   V   s*    

zCookieJar.__init__)	file_pathr'   c              	   C   s8   t |}|jdd}t| j|tj W 5 Q R X d S )Nwbmode)pathlibPathopenpickledumpr3   HIGHEST_PROTOCOLr=   r@   fr)   r)   r/   savew   s    
zCookieJar.savec              	   C   s2   t |}|jdd}t|| _W 5 Q R X d S )NrbrB   )rD   rE   rF   rG   loadr3   rJ   r)   r)   r/   rN   |   s    
zCookieJar.load	predicater'   c                    sv   d kr>j   j  j  j  j  d S t  fddj D }|rr| d S )Nc                    sT   g | ]L\\}}}|  D ]6\}}|||f  jkrBj  ksJ|r qqS r)   )itemsr<   )r-   domainpathcookienamemorselkeynowrP   r=   r)   r/   r0      s   

 z#CookieJar.clear.<locals>.<listcomp>)	r;   clearr3   r5   r7   r<   timerQ   _delete_cookies)r=   rP   to_delr)   rW   r/   rZ      s    





zCookieJar.clearrR   r'   c                    s     fdd d S )Nc                    s     | d S )NrR   )_is_domain_match)xrR   r=   r)   r/   <lambda>       z(CookieJar.clear_domain.<locals>.<lambda>)rZ   r=   rR   r)   ra   r/   clear_domain   s    zCookieJar.clear_domainIterator[Morsel[str]]r'   c                 c   s*   |    | j D ]}| E d H  qd S N)_do_expirationr3   values)r=   valr)   r)   r/   __iter__   s    zCookieJar.__iter__c                 C   s   t dd | j D S )zReturn number of cookies.

        This function does not iterate self to avoid unnecessary expiration
        checks.
        c                 s   s   | ]}t | V  qd S rh   )lenrj   )r-   rT   r)   r)   r/   	<genexpr>   s     z$CookieJar.__len__.<locals>.<genexpr>)sumr3   rj   r=   r)   r)   r/   __len__   s    zCookieJar.__len__c                    s   t  j }sdS |tkrN|t  jd krN fdd jD  _t j t }g } jr jd \}}||krxqt j  j||krZ|	| qZ|r 
| dS )zRemove expired cookies.N   c                    s(   g | ] } j |d  |d kr|qS )r   r   )r<   get)r-   entryrp   r)   r/   r0      s   z,CookieJar._do_expiration.<locals>.<listcomp>r   )rm   r;    _MIN_SCHEDULED_COOKIE_EXPIRATIONr<   heapqheapifyr[   heappoprs   appendr\   )r=   Zexpire_heap_lenrY   r]   when
cookie_keyr)   rp   r/   ri      s*    
zCookieJar._do_expiration)r]   r'   c                 C   sd   |D ]Z\}}}| j ||f | j||f |d  | j||f |d  | j|||fd  qd S rh   )r7   discardr3   popr5   r<   )r=   r]   rR   rS   rU   r)   r)   r/   r\      s
    zCookieJar._delete_cookies)rz   rR   rS   rU   r'   c                 C   s>   |||f}| j ||krd S t| j||f || j |< d S rh   )r<   rs   rv   heappushr;   )r=   rz   rR   rS   rU   r{   r)   r)   r/   _expire_cookie   s
    
zCookieJar._expire_cookiecookiesresponse_urlr'   c              	   C   s  |j }| jst|rdS t|tr*| }|D ]\}}t|tsXt }|||< || }|d }|rz|d dkrzd}|d= |s|dk	r| j	||f | }|d< |r|d dkr|dd }||d< |r| 
||sq.|d }|r|d d	kr(|j}|d	s
d	}nd	|d|d	  }||d< |d	}|d
  }	rz0t|	}
tt |
 | j}| |||| W n tk
r   d|d
< Y nX n8|d  }r| | }r| |||| nd|d< ||f}| j| ||kr.|| j| |< | j| |d q.|   dS )zUpdate cookies.NrR   r"   . r   r   rS   /zmax-ageexpires)raw_hostr8   r   r*   r   rQ   r   r   r7   addr_   rS   
startswithrfindrstripintminr[   MAX_TIMEr   
ValueError_parse_dater3   rs   r5   r}   ri   )r=   r   r   hostnamerU   rT   tmprR   rS   max_ageZdelta_secondsZmax_age_expirationr   expire_timerX   r)   r)   r/   update_cookies   s\    


zCookieJar.update_cookiesBaseCookie[str]request_urlr'   c              	   C   s  | j rt nt }| js|S |   | js.|S t|}|jp>d}|jdk}|r| jrt }t	
t | }W 5 Q R X || jk}| jd  D ]}|j||j< qt|r| js|S |f}ntt|dt}t|jdt}t||}	t|j}
|	D ]}| j|  D ]\}}|d }||f| jkr@||kr@qt|d |
krVq|rj|d rjq|| j| kr| j| | ||< qtd	||jt  }|!|j|j|j" || j| |< |||< qq |S )
z8Returns this jar's cookies filtered by their attributes.r   )httpswss)r   r   r   r   rR   rS   securer   )#r9   r   r   r3   ri   r   r   schemer:   
contextlibsuppressr   r,   rj   valuerX   r   r8   	itertools
accumulatereversedsplit_FORMAT_DOMAIN_REVERSEDrS   _FORMAT_PATHproductrm   rQ   r7   r5   r   rs   r   r6   coded_value)r=   r   filteredr   Zis_not_secureZrequest_origincdomainspathspairsZpath_lenprU   rT   rR   Zmrsl_valr)   r)   r/   filter_cookies,  sZ    



 
zCookieJar.filter_cookies)rR   r   r'   c                 C   sD   || krdS | | sdS |dt|   }| ds:dS t| S )z0Implements domain matching adhering to RFC 6265.TFNr   )endswithrm   r   )rR   r   Znon_matchingr)   r)   r/   r_   s  s    

zCookieJar._is_domain_match)date_strr'   c                 C   s  |sdS d}d}d}d}d } }}d}	d}
d}| j |D ]}|d}|s|| j|}|r|d}dd | D \}}}q<|s| j|}|rd}t| }	q<|s| j|}|rd}|j	dk	st
|j	}
q<|s<| j|}|r<d}t| }q<d|  krd	krn n
|d
7 }n$d|  kr4dkr@n n|d7 }d||||fkrVdS d|	  krndkstn dS |dk s|dks|dks|dkrdS t||
|	|||dddf	S )z4Implements date string parsing adhering to RFC 6265.NFr   tokenTc                 s   s   | ]}t |V  qd S rh   )r   )r-   sr)   r)   r/   rn     s     z(CookieJar._parse_date.<locals>.<genexpr>F   c   il  E   i  r   r   iA  r    r!   r"   )DATE_TOKENS_REfinditergroupDATE_HMS_TIME_REmatchgroupsDATE_DAY_OF_MONTH_REr   DATE_MONTH_RE	lastindexAssertionErrorDATE_YEAR_REcalendartimegm)clsr   Z
found_timeZ	found_dayZfound_monthZ
found_yearhourminuteseconddaymonthyearZtoken_matchr   Z
time_matchZ	day_matchZmonth_matchZ
year_matchr)   r)   r/   r     s^    

(zCookieJar._parse_date)N);__name__
__module____qualname____doc__recompiler   r   r   Ir   r   r   datetimemaxreplacetimezoneutc	timestampr   r   r   r[   gmtimeOSErrorr   OverflowErrorZSUB_MAX_TIMEboolr   r   r
   r   asyncioAbstractEventLoopr2   r   rL   rN   r   rZ   r+   re   rl   rq   ri   r   r\   floatr   r   r   r   r   staticmethodr_   classmethodr   __classcell__r)   r)   r>   r/   r   1   s\   



!+JGr   c                       s   e Zd ZdZddeej dd fddZddd	d
Ze	dddZ
dee ddddZeddddZe feeddddZeddddZ  ZS )r   zxImplements a dummy cookie storage.

    It can be used with the ClientSession when no cookie processing is needed.

    Nr(   )r&   r'   c                   s   t  j|d d S )Nr(   )r1   r2   )r=   r&   r>   r)   r/   r2     s    zDummyCookieJar.__init__rf   rg   c                 c   s   d S rh   r)   rp   r)   r)   r/   rl     s    zDummyCookieJar.__iter__c                 C   s   dS )Nr   r)   rp   r)   r)   r/   rq     s    zDummyCookieJar.__len__rO   c                 C   s   d S rh   r)   )r=   rP   r)   r)   r/   rZ     s    zDummyCookieJar.clearr^   c                 C   s   d S rh   r)   rd   r)   r)   r/   re     s    zDummyCookieJar.clear_domainr   c                 C   s   d S rh   r)   )r=   r   r   r)   r)   r/   r     s    zDummyCookieJar.update_cookiesr   r   c                 C   s   t  S rh   )r   )r=   r   r)   r)   r/   r     s    zDummyCookieJar.filter_cookies)N)r   r   r   r   r   r   r   r2   rl   r   rq   r   rZ   r+   re   r   r   r   r   r   r)   r)   r>   r/   r     s    r   )1r   r   r   r   rv   r   osrD   rG   r   r[   collectionsr   http.cookiesr   r   r   typingr   r   r   r	   r
   r   r   r   r   r   r   Zyarlr   abcr   r   Zhelpersr   Ztypedefsr   r   r   __all__r+   Z
CookieItemformatr   r   ru   r   r   r)   r)   r)   r/   <module>   s6   4   