3
�gaz
� @ sf d dl mZ d dlmZ ddlmZmZ ddlmZ G dd� d�Z G dd � d e �Z
G d
d � d e �Z d S )
� )�Iterable)�urlsplit� )�
split_cookies�parse_content_type_header)�cached_propertyc @ sv e Zd ZdZdd� Zeee d�dd�Zeee d�dd�Z e
e d �d
d ��Z e
e d �d d
��Ze
e d �dd��ZdS )� HTTPMessagez!Abstract class for HTTP messages.c C s
|| _ d S )N)�_orig)�selfZorig� r � /usr/lib/python3.6/models.py�__init__ s zHTTPMessage.__init__)�
chunk_size�returnc C s t �dS )z!Return an iterator over the body.N)�NotImplementedError)r
r r r r � iter_body s zHTTPMessage.iter_bodyc C s t �dS )z@Return an iterator over the body yielding (`line`, `line_feed`).N)r )r
r r r r �
iter_lines s zHTTPMessage.iter_lines)r c C s t �dS )z*Return a `str` with the message's headers.N)r )r
r r r �headers s zHTTPMessage.headersc C s t | j�\}}|jdd�S )N�charset� )r � content_type�get)r
�ctZparamsr r r �encoding s zHTTPMessage.encodingc C s& | j jjdd�}t|t�s"|j� }|S )z Return the message content type.z Content-Typer )r r r �
isinstance�str�decode)r
r r r r r s
zHTTPMessage.content_typeN)�__name__�
__module__� __qualname__�__doc__r
�intr �bytesr r �propertyr r r r r r r r r r s r c @ s. e Zd ZdZd
dd�Zdd� Zedd� �Zd S ) � HTTPResponsez,A :class:`requests.models.Response` wrapper.r c C s | j j|d�S )N)r )r Z iter_content)r
r r r r r , s zHTTPResponse.iter_bodyc C s dd� | j j|�D �S )Nc s s | ]}|d fV qdS )�
Nr )�.0�liner r r � <genexpr>0 s z*HTTPResponse.iter_lines.<locals>.<genexpr>)r r )r
r r r r r / s zHTTPResponse.iter_linesc C s� y| j jjj}W n tk
r( d}Y nX ddddd�| }| j }d|� d|j� d|j� �}|g}|jd d
� |jj � D �� |jd d
� t
|jj d ��D �� d
j |�S )N� z0.9z1.0z1.1�2)� �
r) � zHTTP/� c s s$ | ] }|d dkrdj |�V qdS )r z
Set-Cookiez: N)�join)r& �headerr r r r( E s z'HTTPResponse.headers.<locals>.<genexpr>c s s | ]}d |� �V qdS )z Set-Cookie: Nr )r& Zcookier r r r( J s z
Set-Cookiez
)
r �rawZ_original_response�version�AttributeErrorZ status_code�reason�extendr �itemsr r r/ )r
Z raw_versionr2 �originalZ status_liner r r r r 3 s&
zHTTPResponse.headersN)r )r r r r r r r# r r r r r r$ ) s
r$ c @ s8 e Zd ZdZdd� Zdd� Zedd� �Zedd � �Zd
S ) � HTTPRequestz+A :class:`requests.models.Request` wrapper.c c s | j V d S )N)�body)r
r r r r r S s zHTTPRequest.iter_bodyc c s | j dfV d S )N� )r9 )r
r r r r r V s zHTTPRequest.iter_linesc C s� t | jj�}dj| jj|jp d|jr0d|j� �ndd�}t| jj�}d| jjkrd|j j
d�d
|d< d d
� |j � D �}|j d |� d j
|�j� }|S )Nz {method} {path}{query} HTTP/1.1�/�?r )�method�path�queryZHost�@r c S s0 g | ](\}}|� d t |t�r |n|j� � ��qS )z: )r r r )r& �name�valuer r r �
<listcomp>h s z'HTTPRequest.headers.<locals>.<listcomp>r z
���)r r �url�formatr= r> r? �dictr Znetloc�splitr6 �insertr/ �strip)r
rE Z request_liner r r r r Y s zHTTPRequest.headersc C s" | j j}t|t�r|j� }|p dS )Nr: )r r9 r r �encode)r
r9 r r r r9 p s
zHTTPRequest.bodyN) r r r r r r r# r r9 r r r r r8 P s
r8 N) Ztypingr Z urllib.parser Zutilsr r �compatr r r$ r8 r r r r �<module> s !'