관리-도구
편집 파일: cookiejar.cpython-37.pyc
B �k�a5 � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m Z m Z m Z d dl mZmZmZmZmZmZmZmZmZmZmZ d dlmZ ddlm Z m Z ddl m Z m Z dd l!m"Z"m#Z#m$Z$ d Z%ee&d f Z'G d d � d e �Z(G dd� de �Z)dS )� N)� defaultdict)� BaseCookie�Morsel� SimpleCookie) � DefaultDict�Dict�Iterable�Iterator�List�Mapping�Optional�Set�Tuple�Union�cast)�URL� )�AbstractCookieJar�ClearCookiePredicate)� is_ip_address�next_whole_second)� LooseCookies�PathLike�StrOrURL)� CookieJar�DummyCookieJarz Morsel[str]c s� e Zd ZdZe�d�Ze�d�Ze�d�Ze�dej �Z e�d�Z e j j je jjd�Ze j �d�Zd d d d d �eeeeee d f eej d d �� fdd�Ze d d�dd�Z e d d�dd�Z d6ee d d�dd�Z e!d d�dd�Z"dd �d d �Z#e$d �d d �Z%d d �d!d"�Z&e j e!e!d d#�d$d%�Z'e(� fe)e(d d&�d'd(�Z*e(� fe(ed) d*�d+d,�Z+e,e!e!ed-�d.d/��Z-e,e!e!ed0�d1d2��Z.e/e!ee j d3�d4d5��Z0� Z1S )7r 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}))�tzinfoi���FTN)�unsafe� quote_cookie�treat_as_secure_origin�loop)r r r r �returnc s� t � j|d� tt�| _t� | _|| _|| _|d kr:g }n>t |t �rP|� � g}n(t |t �rjt |�� � g}ndd� |D �}|| _ t� | _i | _| j| _y| j�� W n tk r� | j| _Y nX d S )N)r c S s* g | ]"}t |t�r t|��� n|�� �qS � )� isinstance�strr �origin)�.0�urlr" r" �D/opt/alt/python37/lib64/python3.7/site-packages/aiohttp/cookiejar.py� <listcomp>R s z&CookieJar.__init__.<locals>.<listcomp>)�super�__init__r r �_cookies�set�_host_only_cookies�_unsafe� _quote_cookier# r r% r$ �_treat_as_secure_originr �_next_expiration� _expirations�MAX_TIME� _max_time� timestamp� OverflowError�MAX_32BIT_TIME)�selfr r r r )� __class__r" r( r+ ; s, zCookieJar.__init__)� file_pathr! c C s8 t �|�}|jdd��}t�| j|tj� W d Q R X d S )N�wb)�mode)�pathlib�Path�open�pickle�dumpr, ZHIGHEST_PROTOCOL)r9 r; �fr" r" r( �save_ s zCookieJar.savec C s2 t �|�}|jdd��}t�|�| _W d Q R X d S )N�rb)r= )r> r? r@ rA �loadr, )r9 r; rC r" r" r( rF d s zCookieJar.load)� predicater! c C sJ |d kr2t � | _| j�� | j�� | j�� d S g }tj�tjj �}x^| j� � D ]P\}}xF|� � D ]:\}}||f}|| jkr�| j| |ks�||�rd|� |� qdW qRW xL|D ]D\}}||f}| j� |� || jkr�| j||f= | j| � |d � q�W t| j�� | jd�} y | jdd�tjdd� | _W n tk �rD | j| _Y nX d S )N)�defaultr )� microsecondr )�seconds)r r2 r, �clearr. r3 �datetime�now�timezone�utc�items�append�discard�pop�min�valuesr5 �replace� timedeltar7 ) r9 rG Zto_delrM �domain�cookie�nameZmorsel�keyZnext_expirationr" r" r( rK i s6 zCookieJar.clear)rX r! c s �� � �fdd�� d S )Nc s �� � | d �S )NrX )�_is_domain_match)�x)rX r9 r" r( �<lambda>� � z(CookieJar.clear_domain.<locals>.<lambda>)rK )r9 rX r" )rX r9 r( � clear_domain� s zCookieJar.clear_domainzIterator[Morsel[str]])r! c c s. | � � x | j�� D ]}|�� E d H qW d S )N)�_do_expirationr, rU )r9 �valr" r" r( �__iter__� s zCookieJar.__iter__c C s t dd� | D ��S )Nc s s | ] }d V qdS )r Nr" )r& �ir" r" r( � <genexpr>� s z$CookieJar.__len__.<locals>.<genexpr>)�sum)r9 r" r" r( �__len__� s zCookieJar.__len__c C s | � dd� � d S )Nc S s dS )NFr" )r] r" r" r( r^ � r_ z*CookieJar._do_expiration.<locals>.<lambda>)rK )r9 r" r" r( ra � s zCookieJar._do_expiration)�whenrX rZ r! c C s t | j|�| _|| j||f< d S )N)rT r2 r3 )r9 rh rX rZ r" r" r( �_expire_cookie� s zCookieJar._expire_cookie)�cookies� response_urlr! c C s� |j }| jst|�rdS t|t�r*|�� }�x�|D �]�\}}t|t�s\t� }|||< || }|d }|�d�rxd}|d= |s�|dk r�| j � ||f� | }|d< |� d�r�|dd� }||d< |r�| � ||�s�q2|d }|r�|� d��s |j }|� d�s�d}nd|d|�d�� }||d< |d } | �r�yXt| �} y tj�tjj�tj| d � } W n tk �rp | j} Y nX | �| ||� W n tk �r� d|d< Y nX n6|d } | �r�| �| �} | �r�| �| ||� nd|d < || j| |< q2W | �� dS ) zUpdate cookies.NrX �.� r �path�/zmax-age)rJ �expires) �raw_hostr/ r r# r rP r r �endswithr. �add� startswithr\ rn �rfind�intrL rM rN rO rW r7 r5 ri � ValueError� _parse_dater, ra )r9 rj rk �hostnamerZ rY �tmprX rn Zmax_ageZ delta_secondsZmax_age_expirationrp Z expire_timer" r" r( �update_cookies� sd zCookieJar.update_cookies)zBaseCookie[str]zSimpleCookie[str])� request_urlr! c C s | � � t|�}| jr t� nt� }|jp*d}t� }t�t�� |� � }W dQ R X |j dkob|| j k}x�| D ]�}|j }|d }|s�|j ||< qj| js�t|�r�qj||f| jkr�||kr�qjn| �||�s�qj| �|j|d �s�qj|r�|d r�qjtd|�|j t� ��} | �|j |j |j� | ||< qjW |S )z8Returns this jar's cookies filtered by their attributes.rm N)�https�wssrX rn Zsecurez Morsel[str])ra r r0 r r rq � contextlib�suppressrw r% �schemer1 r[ �valuer/ r r. r\ �_is_path_matchrn r �getr r- Z coded_value) r9 r| �filteredry Zrequest_originZ is_not_securerY rZ rX Zmrsl_valr" r" r( �filter_cookies� s< zCookieJar.filter_cookies)rX ry r! c C sD || kr dS |� | �sdS |dt| � � }|� d�s:dS t|� S )z0Implements domain matching adhering to RFC 6265.TFNrl )rr �lenr )rX ry � non_matchingr" r" r( r\ s zCookieJar._is_domain_match)�req_path� cookie_pathr! c C sP | � d�sd} | |krdS | � |�s(dS |�d�r6dS | t|�d� }|� d�S )z.Implements path matching adhering to RFC 6265.ro TFN)rt rr r� )r� r� r� r" r" r( r� , s zCookieJar._is_path_match)�date_strr! c C s� |sdS d}d}d}d}d } }}d} d} d} x�| j �|�D ]�} | �d�} |s~| j�| �}|r~d}dd� |�� D �\}}}q>|s�| j�| �}|r�d}t|�� �} q>|s�| j�| �}|r�d}|j dk s�t �|j } q>|s>| j �| �}|r>d}t|�� �} q>W d| k�rd k�r n n | d 7 } n$d| k�r8d k�rDn n| d 7 } d||||fk�rZdS d | k�rrdk�sxn dS | dk �s�|dk�s�|dk�s�|dk�r�dS t j | | | |||t j jd�S )z4Implements date string parsing adhering to RFC 6265.NFr �tokenTc s s | ]}t |�V qd S )N)rv )r& �sr" r" r( re W s z(CookieJar._parse_date.<locals>.<genexpr>�F �c il �E i� r � iA � �; )r )�DATE_TOKENS_RE�finditer�group�DATE_HMS_TIME_RE�match�groups�DATE_DAY_OF_MONTH_RErv � DATE_MONTH_RE� lastindex�AssertionError� DATE_YEAR_RErL rN rO )�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( rx ? s` (zCookieJar._parse_date)N)2�__name__� __module__� __qualname__�__doc__�re�compiler� r� r� �Ir� r� rL �maxrV rN rO r4 �utcfromtimestampr8 �boolr r r r �asyncio�AbstractEventLoopr+ r rD rF r rK r$ r` rc rv rg ra ri r r r{ r� � staticmethodr\ r� � classmethodrx � __classcell__r" r" )r: r( r $ sB " !N1r c s� e Zd ZdZdd�eej dd�� fdd�Zdd�d d �Ze d�d d �Z dee dd �dd�Z e dd�dd�Ze� feedd�dd�Zedd�dd�Z� ZS )r zxImplements a dummy cookie storage. It can be used with the ClientSession when no cookie processing is needed. N)r )r r! c s t � j|d� d S )N)r )r* r+ )r9 r )r: r" r( r+ � s zDummyCookieJar.__init__zIterator[Morsel[str]])r! c c s d S )Nr" )r9 r" r" r( rc � s zDummyCookieJar.__iter__c C s dS )Nr r" )r9 r" r" r( rg � s zDummyCookieJar.__len__)rG r! c C s d S )Nr" )r9 rG r" r" r( rK � s zDummyCookieJar.clear)rX r! c C s d S )Nr" )r9 rX r" r" r( r` � s zDummyCookieJar.clear_domain)rj rk r! c C s d S )Nr" )r9 rj rk r" r" r( r{ � s z DummyCookieJar.update_cookieszBaseCookie[str])r| r! c C s t � S )N)r )r9 r| r" r" r( r� � s z DummyCookieJar.filter_cookies)N)r� r� r� r� r r� r� r+ rc rv rg r rK r$ r` r r r{ r� r� r" r" )r: r( r � s r )*r� r rL �osr> rA r� � collectionsr Z http.cookiesr r r �typingr r r r r r r r r r r Zyarlr �abcr r Zhelpersr r Ztypedefsr r r �__all__r$ Z CookieItemr r r" r" r" r( �<module> s& 4 `