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 CookieJarDummyCookieJarc 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 @ÐAunsafeFloopNc s\ t j d | t t | _ t | _ | | _ t | j j
| _ i | _ d S)Nr )
super__init__r r _cookiesset_host_only_cookies_unsafer _looptime_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)Nmodewb)pathlibPathopenpickledumpr HIGHEST_PROTOCOL)r file_pathfr 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 ).0ir 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 itemspopappendr discardminr )
r ZnowZnext_expirationZto_delcookiesZ expirationsdomainnamewhenkeyr 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 } | ry0 t | }
| j | j j |
| | Wq5t k
rd | d <Yq5XnK | d } | r5| j | } | r+| j | j | | n
d | d <| | j | | <qE W| j d S)
zUpdate cookies.Nr; . r path/zmax-ageexpires)raw_hostr r
isinstancer r5 r r endswithr add
startswith_is_domain_matchrB rfindintr? r r
ValueError _parse_dateZ timestampr r+ )
r r: response_urlhostnamer< cookietmpr; 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 httpswssr; 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 od k n r| d 7} d | | | | f k rd Sd
| k od k n sd 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_REfinditergroupDATE_HMS_TIME_REmatchgroupsDATE_DAY_OF_MONTH_RErL
DATE_MONTH_RE lastindex DATE_YEAR_REdatetimetimezoneZutc)clsZdate_strZ
found_timeZ found_dayZ found_monthZ
found_yearZhourZminutesecondZdayZmonthZyearZ token_matchr^ Z
time_matchZ day_matchZ month_matchZ
year_matchr r r rN s` "
0 zCookieJar._parse_date) __name__
__module__ __qualname____doc__recompilerg 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 $