3
�P\� � @ s� d Z ddlZddlZddlmZ ddlmZ dd� ZG dd � d e �Z
G d
d � d e �Z d d
� Z G dd� de
�ZG dd� de
�ZdS )z�
requests_toolbelt.multipart.decoder
===================================
This holds all the implementation details of the MultipartDecoder
� N� )� encode_with)�CaseInsensitiveDictc C s* | j |�}| d |� | |t|� d � fS )N)�find�len)�contentZboundZpoint� r � /usr/lib/python3.6/decoder.py�_split_on_find s
r
c @ s e Zd ZdS )� ImproperBodyPartContentExceptionN)�__name__�
__module__� __qualname__r r r r r s r c @ s e Zd ZdS )� NonMultipartContentTypeExceptionN)r r
r r r r r r s r c sB t jd }|dkr | j� �} tjj� j| �j� }� fdd�|D �S )Nr � c 3 s& | ] \}}t |� �t |� �fV qd S )N)r )�.0�k�v)�encodingr r � <genexpr>$ s z!_header_parser.<locals>.<genexpr>)�sys� version_info�decode�email�parserZ HeaderParserZparsestr�items)�stringr �major�headersr )r r �_header_parser s
r c @ s$ e Zd ZdZdd� Zedd� �ZdS )�BodyParta�
The ``BodyPart`` object is a ``Response``-like interface to an individual
subpart of a multipart response. It is expected that these will
generally be created by objects of the ``MultipartDecoder`` class.
Like ``Response``, there is a ``CaseInsensitiveDict`` object named headers,
``content`` to access bytes, ``text`` to access unicode, and ``encoding``
to access the unicode codec.
c C sP || _ i }d|kr:t|d�\}| _|dkrBt|j� |�}ntd��t|�| _d S )Ns
� z$content does not contain CR-LF-CR-LF)r r
r r �lstripr r r )�selfr r r �firstr r r �__init__6 s zBodyPart.__init__c C s | j j| j�S )z'Content of the ``BodyPart`` in unicode.)r r r )r# r r r �textD s z
BodyPart.textN)r r
r �__doc__r% �propertyr& r r r r r ) s r c @ sD e Zd ZdZddd�Zdd� Zedd� �Zd d
� Ze dd d ��Z
d
S )�MultipartDecoderai
The ``MultipartDecoder`` object parses the multipart payload of
a bytestring into a tuple of ``Response``-like ``BodyPart`` objects.
The basic usage is::
import requests
from requests_toolbelt import MultipartDecoder
response = request.get(url)
decoder = MultipartDecoder.from_response(response)
for part in decoder.parts:
print(part.headers['content-type'])
If the multipart content is not from a response, basic usage is::
from requests_toolbelt import MultipartDecoder
decoder = MultipartDecoder(content, content_type)
for part in decoder.parts:
print(part.headers['content-type'])
For both these usages, there is an optional ``encoding`` parameter. This is
a string, which is the name of the unicode codec to use (default is
``'utf-8'``).
�utf-8c C s* || _ || _t� | _| j� | j|� d S )N)� content_typer �tuple�parts�_find_boundary� _parse_body)r# r r+ r r r r r% g s
zMultipartDecoder.__init__c C s� t dd� | jjd�D ��}|d }|jd�d j� dkrFtdj|���xB|dd � D ]2}t|d �\}}|j� d
krTt|jd �| j �| _
qTW d S ) Nc s s | ]}|j � V qd S )N)�strip)r �xr r r r r s z2MultipartDecoder._find_boundary.<locals>.<genexpr>�;r �/Z multipartz*Unexpected mimetype in content-type: '{0}'r �=�boundary�") r, r+ �split�lowerr �formatr
r r0 r r5 )r# Zct_infoZmimetype�item�attr�valuer r r r. q s
z MultipartDecoder._find_boundaryc C s, t |�}|| d |� kr$| |d � S | S d S )N)r )�partZboundary_markerZbm_lenr r r �_fix_first_part� s z MultipartDecoder._fix_first_partc sX dj d�jf����fdd�� dd� �|jdj d�f��}t� �fdd �|D ���_d S )
Nr! s --c s t j| � �}t|�j�S )N)r) r> r r )r= Zfixed)r5 r# r r � body_part� s z/MultipartDecoder._parse_body.<locals>.body_partc S s( | dko&| dko&| d d� dko&| dkS )Nr! s
� s --
s --r )r= r r r � test_part� s z/MultipartDecoder._parse_body.<locals>.test_parts
c 3 s | ]}�|�r� |�V qd S )Nr )r r1 )r? rA r r r � s z/MultipartDecoder._parse_body.<locals>.<genexpr>)�joinr5 r7 r, r- )r# r r- r )r? r5 r# rA r r/ � s
z MultipartDecoder._parse_bodyc C s |j }|jjdd �}| |||�S )Nz content-type)r r �get)�clsZresponser r r+ r r r �
from_response� s z MultipartDecoder.from_responseN)r* )r* ) r r
r r' r% r. � staticmethodr> r/ � classmethodrE r r r r r) J s
r) )r' r Z email.parserr �encoderr Zrequests.structuresr r
� Exceptionr r r �objectr r) r r r r �<module> s !