���Z�' � @ s� 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
d d l m Z d d l
m Z d d l m Z d d l m Z d
d f Z Gd d
� d
e � Z Gd
d � d e � Z d S)� N)� defaultdict)�Mapping)�Morsel� SimpleCookie)�ceil)�URL� )�AbstractCookieJar)�
is_ip_address� CookieJar�DummyCookieJarc s9 e Z d Z d Z e j d � Z e j d � Z e j d � Z e j d e j � Z
e j d � Z d Z d d d
d � f d d
� Z
d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z d d � Z e � d d � Z e � d d � Z e d d! � � Z e d" d# � � Z e d$ d% � � Z � S)&r 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})g @��A�unsafeF�loopNc s\ t � j d | � t t � | _ t � | _ | | _ t | j j
� � | _ i | _ d S)Nr )
�super�__init__r r �_cookies�set�_host_only_cookies�_unsafer �_loop�time�_next_expiration� _expirations)�selfr
r )� __class__� �
/cookiejar.pyr % s zCookieJar.__init__c C sH t j | � } | j d d � � } t j | j | t j � Wd QRXd S)N�mode�wb)�pathlib�Path�open�pickle�dumpr �HIGHEST_PROTOCOL)r � file_path�fr r r �save- s zCookieJar.savec C sA t j | � } | j d d � � } t j | � | _ Wd QRXd S)Nr �rb)r r r! r" �loadr )r r% r& r r r r) 2 s zCookieJar.loadc C sC | j j � | j j � t | j j � � | _ | j j � d S)N)r �clearr r r r r r )r r r r r* 7 s
zCookieJar.clearc c s7 | j � x& | j j � D] } | j � Ed Hq Wd S)N)�_do_expirationr �values)r �valr r r �__iter__= s
zCookieJar.__iter__c C s t d d � | D� � S)Nc s s | ] } d Vq d S)r Nr )�.0�ir r r � <genexpr>C s z$CookieJar.__len__.<locals>.<genexpr>)�sum)r r r r �__len__B s zCookieJar.__len__c
C s� | j j � } | j | k r" d S| j s/ d S| j } g } | j } | j } x{ | j � D]m \ \ } } } | | k r� | | j | d � | j | | f � | j j
| | f � q] t | | � } q] Wx | D]
} | | =q� Wt | � | _ d S)N)
r r r r �MAX_TIMEr �items�pop�appendr �discard�minr )
r ZnowZnext_expirationZto_del�cookiesZ expirations�domain�name�when�keyr r r r+ E s$
zCookieJar._do_expirationc C s, t | j | � | _ | | j | | f <d S)N)r9 r r )r r= r; r< r r r �_expire_cookie[ s zCookieJar._expire_cookiec
C sX | j } | j r# t | � r# d St | t � r> | j � } x | D]\ } } t | t � s} t � } | | | <| | } | d } | j d � r� d } | d =| r� | d k r� | j j
| | f � | } | d <| j d � r| d d � } | | d <| r | j | | � r qE | d } | s@| j d � r�| j
} | j d � sad } n d | d | j d � � } | | d <| d } | r�y0 t | � }
| j | j j � |
| | � Wq5t k
r�d | d <Yq5XnK | d } | r5| j | � } | r+| j | j � | | � n
d | d <| | j | | <qE W| j � d S)
zUpdate cookies.Nr; �.� r �path�/zmax-age�expires)�raw_hostr r
�
isinstancer r5 r r �endswithr �add�
startswith�_is_domain_matchrB �rfind�intr? r r �
ValueError� _parse_dateZ timestampr r+ )
r r: � response_url�hostnamer< �cookie�tmpr; rB Zmax_ageZ
delta_secondsrD Z expire_timer r r �update_cookies_ s\
zCookieJar.update_cookiesc C s= | j � t | � } t � } | j p+ d } | j d k } x� | D]� } | j } | d } | ss | j | | <qD | j r� t | � r� qD | | f | j k r� | | k r� qD n | j
| | � s� qD | j | j | d � s� qD | r� | d r� qD | j
| j t � � } | j | j | j | j � | | | <qD W| S)z8Returns this jar's cookies filtered by their attributes.rA �https�wssr; rB Zsecure)rT rU )r+ r r rE �schemer> �valuer r
r rJ �_is_path_matchrB �getr r Z coded_value) r � request_urlZfilteredrP Z
is_not_securerQ r< r; Zmrsl_valr r r �filter_cookies� s4
zCookieJar.filter_cookiesc C sX | | k r d S| j | � s# d S| d t | � � } | j d � sM d St | � S)z0Implements domain matching adhering to RFC 6265.TFNr@ )rG �lenr
)r; rP � non_matchingr r r rJ � s zCookieJar._is_domain_matchc C sn | j d � s d } | | k r% d S| j | � s8 d S| j d � rK d S| t | � d � } | j d � S)z.Implements path matching adhering to RFC 6265.rC TFN)rI rG r\ )Zreq_pathZ cookie_pathr] r r r rX � s zCookieJar._is_path_matchc C s= | s
d Sd } d } d } d } d } } } d } d }
d } x| j j | � D]� } | j d � }
| s� | j j |
� } | r� d } d d � | j � D� \ } } } qU | s� | j j |
� } | r� d } t | j � � } qU | s | j j |
� } | r d } | j }
qU | sU | j
j |
� } | rU d } t | j � � } qU Wd | k omd k n r| d
7} n& d | k o�d k n r�| d 7} d | | | | f k r�d Sd
| k o�d k n s�d S| d k s| d k s| d k s| d k rd St j | |
| | | | d t j j
�S)z4Implements date string parsing adhering to RFC 6265.NFr �tokenTc S s g | ] } t | � � q Sr )rL )r/ �sr r r �
<listcomp> s z)CookieJar._parse_date.<locals>.<listcomp>�F �c il �E i� r � iA � �; Ztzinfo)�DATE_TOKENS_RE�finditer�group�DATE_HMS_TIME_RE�match�groups�DATE_DAY_OF_MONTH_RErL �
DATE_MONTH_RE� lastindex� DATE_YEAR_RE�datetime�timezoneZutc)�clsZdate_strZ
found_timeZ found_dayZ found_monthZ
found_yearZhourZminute�secondZdayZmonthZyearZ token_matchr^ Z
time_matchZ day_matchZ month_matchZ
year_matchr r r rN � s` "
0 zCookieJar._parse_date) �__name__�
__module__� __qualname__�__doc__�re�compilerg rj rm �Irn rp r4 r r' r) r* r. r3 r+ r? r rS r[ � staticmethodrJ rX � classmethodrN r r )r r r s, I(c sm e Z d Z d Z d d � f d d � Z d d � Z d d � Z d
d � Z d d d
� Z d d � Z � S)r zxImplements a dummy cookie storage.
It can be used with the ClientSession when no cookie processing is needed.
r Nc s t � j d | � d S)Nr )r r )r r )r r r r = s zDummyCookieJar.__init__c c s d S)Nr )r r r r r. @ s zDummyCookieJar.__iter__c C s d S)Nr r )r r r r r3 D s zDummyCookieJar.__len__c C s d S)Nr )r r r r r* G s zDummyCookieJar.clearc C s d S)Nr )r r: rO r r r rS J s z DummyCookieJar.update_cookiesc C s d S)Nr )r rZ r r r r[ M s z DummyCookieJar.filter_cookies)
ru rv rw rx r r. r3 r* rS r[ r r )r r r 6 s )rq r r" ry � collectionsr �collections.abcr Z http.cookiesr r �mathr Zyarlr �abcr Zhelpersr
�__all__r r r r r r �<module> s � $