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)findlen)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 ).0kv)encodingr r $ s z!_header_parser..)sys version_infodecodeemailparserZ HeaderParserZparsestritems)stringr majorheadersr )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 |}ntdt|| _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| jS )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 ZdZdddZdd 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 tupleparts_find_boundary _parse_body)r# r r+ r r r r r% g s
zMultipartDecoder.__init__c C s t dd | jjdD }|d }|jdd 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..;r /Z multipartz*Unexpected mimetype in content-type: '{0}'r =boundary") r, r+ splitlowerr formatr
r r0 r r5 )r# Zct_infoZmimetypeitemattrvaluer 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 djffdd dd |jdj df}t fdd |D _d S )
Nr! s --c s t j| }t|jS )N)r) r> r r )r= Zfixed)r5 r# r r body_part s z/MultipartDecoder._parse_body..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..test_parts
c 3 s | ]}|r |V qd S )Nr )r r1 )r? rA r r r s z/MultipartDecoder._parse_body..)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 s !