3
�gag � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZm Z m
Z
ddl m Z ddl
mZ ddlmZmZ d d
lmZ d d lmZmZ d d lmZ d
Zee Zejd�Z ddgZ ee e e e dd�dd�Z G dd� de�Z dS )z(
Persistent, JSON-serialized sessions.
� N)� SimpleCookie)�Path)�Iterable�Optional�Union)�urlsplit)�AuthBase)�RequestsCookieJar�
create_cookie� )�RequestHeadersDict)�BaseConfigDict�DEFAULT_CONFIG_DIR)�plugin_managerZsessionsz^[a-zA-Z0-9_.-]+$zContent-zIf-�Session)�
config_dir� session_name�host�url�returnc C sp t jj|krt jj|�}nB|p0t|�jjd�d }|s:d}|jdd�}| t | |� d� }t |�}|j
� |S )N�@r Z localhost�:�_z.json���) �os�path�sep�
expanduserr Znetloc�split�replace�SESSIONS_DIR_NAMEr �load)r r r r r ZhostnameZsession� r" � /usr/lib/python3.6/sessions.py�get_httpie_session s r$ c s� e Zd ZdZdZeeef d�� fdd� Ze d�dd�Z
e e d �d
d ��Z e e
d �d d
��Zeje
d�dd
��Ze ee d �dd��Zejed�dd��Zee d�dd�Z� ZS )r z https://httpie.io/docs#sessionszHTTPie session file)r c s6 t � jt|�d� i | d< i | d< d d d d�| d< d S )N)r �headers�cookies)�type�username�password�auth)�super�__init__r )�selfr )� __class__r" r# r, : s zSession.__init__)�request_headersc C s� | j }x�|j� j� D ]�\}}|dkr&qt|�tk r:|j� }|j� dkrR|jd�rRq|j� dkr�x*t|�j� D ]\}}d|j i| d |< qlW ||= qx(t
D ]}|j� j|j� �r�P q�W |||< qW t |�| d< dS )zq
Update the session headers with the request ones while ignoring
certain name prefixes.
Nz
user-agentzHTTPie/�cookie�valuer& r% ) r% �copy�itemsr' �str�decode�lower�
startswithr r1 � SESSION_IGNORED_HEADER_PREFIXES�dict)r- r/ r% �namer1 Z cookie_nameZmorsel�prefixr" r" r# �update_headersD s$
zSession.update_headers)r c C s t | d �S )Nr% )r )r- r" r" r# r% d s zSession.headersc C sF t � }x2| d j� D ]"\}}|jt||jd�f|�� qW |j� |S )Nr& r1 )r r3 Z
set_cookier
�popZclear_expired_cookies)r- �jarr: Z cookie_dictr" r" r# r& h s zSession.cookies)r> c sB ddddg}i | d< x(|D ] � � fdd�|D �| d � j < qW d S )Nr1 r ZsecureZexpiresr& c s i | ]}t � |�|�qS r" )�getattr)�.0Zattname)r0 r" r# �
<dictcomp>w s z#Session.cookies.<locals>.<dictcomp>)r: )r- r> Z stored_attrsr" )r0 r# r& q s
c C s� | j dd �}| s |d r d S tj|d �� }d d d�}y|d |_W n& tk
rn |d |d d�}Y n,X |jr�ddlm} ||j�}|j|j d�}|j
f |�S ) Nr* r' )r( r) �raw_authr( r) r )�
parse_auth) �getr Zget_auth_pluginrB �KeyErrorZ
auth_parseZ cli.argtypesrC �keyr1 Zget_auth)r- r* ZpluginZ credentialsrC Zparsedr" r" r# r* | s
z Session.auth)r* c C s ddh|j � kst�|| d< d S )Nr' rB r* )�keys�AssertionError)r- r* r" r" r# r* � s )�namesc C s( x"|D ]}|| d kr| d |= qW d S )Nr& r" )r- rI r: r" r" r# �remove_cookies� s
zSession.remove_cookies)�__name__�
__module__� __qualname__ZhelpurlZaboutr r4 r r, r r< �propertyr% r r& �setterr r r* r9 r rJ �
__classcell__r" r" )r. r# r 6 s
)!�__doc__r �reZ http.cookiesr Zpathlibr Ztypingr r r Z urllib.parser Z
requests.authr Zrequests.cookiesr r
Z cli.dictsr �configr
r Zplugins.registryr r ZDEFAULT_SESSIONS_DIR�compileZVALID_SESSION_NAME_PATTERNr8 r4 r$ r r" r" r" r# �<module> s*