3
\) @ s: d dl Z d dlZd dlZd dlZd dlmZ G dd dZe Zd%ddZej de d d
Z
x d j D ]Z ej e e
qhW d d
Z
x dj D ]Zej de e
qW dd Zej de dd Zdd Zdd Zdd Zd&d d Zejee d'd d!Zejejje d(d#d$Zx ee e fD ]Z eje e q W dS )) N)
quoprimimec @ s< e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
S )ContentManagerc C s i | _ i | _d S )N) get_handlers set_handlers)self r ,/usr/lib64/python3.6/email/contentmanager.py__init__ s zContentManager.__init__c C s || j |< d S )N)r )r keyhandlerr r r add_get_handler
s z ContentManager.add_get_handlerc O s| |j }|| jkr(| j| |f||S |j }|| jkrP| j| |f||S d| jkrp| jd |f||S t|d S )N )Zget_content_typer get_content_maintypeKeyError)r msgargskwZ content_typemaintyper r r get_content s
zContentManager.get_contentc C s || j |< d S )N)r )r Ztypekeyr r r r add_set_handler s z ContentManager.add_set_handlerc O s> |j dkrtd| j||}|j |||f|| d S )NZ multipartz"set_content not valid on multipart)r TypeError_find_set_handlerZ
clear_content)r r objr r r r r r set_content s
zContentManager.set_contentc C s d }xt |jD ]}|| jkr(| j| S |j}t|dd}|rLdj||fn|}|d kr\|}|| jkrp| j| S || jkr| j| S |j}|| jkr| j| S qW d | jkr| jd S t|d S )N
__module__r
.)type__mro__r __qualname__getattrjoin__name__r ) r r r Zfull_path_for_errortypZqnamemodname full_pathnamer r r r ' s&
z ContentManager._find_set_handlerN) r! r r r r r r r r r r r r r s r replacec C s& | j dd}| jdd}|j||dS )NT)decodecharsetASCII)errors) get_payloadZ get_paramr' )r r* Zcontentr( r r r get_text_content@ s r, textc C s | j ddS )NT)r' )r+ )r r r r get_non_text_contentG s r. z audio image video applicationc C s
| j dS )Nr )r+ )r r r r get_message_contentM s r/ zrfc822 external-bodyzmessage/c C s t | jdS )Nr )bytesr+ )r r r r %get_and_fixup_unknown_message_contentS s r1 messagec s dj ||f| d< |rt|d ds<| j fdd|D }y,x&|D ] }|jrX|jd || |j< qDW W n@ tjjk
r } z tdj |j
| jd|W Y d d }~X nX d S ) N/z Content-Typer r% c s g | ]} j j|g qS r )Zheader_factoryZheader_source_parse).0header)mpr r
d s z _prepare_set..zInvalid header: {})policy) r hasattrr8 Zdefectsr% emailr* Z HeaderDefect
ValueErrorformatZfold)r r subtypeheadersr5 excr )r6 r _prepare_set_ s
r@ c C s| |d kr|d k rd}|d k r$|| d< |d k r>| j d|ddd |d k rN|| d< |d k rxx |j D ]\}}| j || q`W d S )NZ
attachmentzContent-DispositionfilenameT)r5 r& z
Content-ID) set_paramitems)r dispositionrA cidparamsr
valuer r r
_finalize_setp s rH c C sZ g }|d d }x>t dt| |D ]*}| ||| }|jtj|jd q"W dj|S )N r asciir
)rangelenappendbinascii
b2a_base64r' r )datamax_line_lengthZ
encoded_linesZunencoded_bytes_per_lineiZthisliner r r _encode_base64 s rT c s| | j |j }|jj d fdd}dd }|d kryd||jdfS tk
r\ Y nX |jdkrtdd |D |jkrd||jdd
fS ||d d }tj |jd |j}t
j |} t |t | krd
}nd}t |d kr||fS |dkr
||jd}
nj|dkr&||jdd
}
nN|dkrJtj ||jd |j}
n*|d
krft
|||j}
ntdj|||
fS )NrK c s j | S )N)r )lines)linesepr r
embedded_body s z#_encode_text..embedded_bodyc S s dj | d S )N
)r )rU r r r normal_body s z!_encode_text..normal_body7bit8bitc s s | ]}t |V qd S )N)rM )r4 xr r r s z _encode_text..surrogateescape
zlatin-1base64zquoted-printablez$Unknown content transfer encoding {})encode
splitlinesrV r' UnicodeDecodeErrorZcte_typemaxrR r Z body_encoderO rP rM rT r; r< ) stringr( cter8 rU rW rY ZsniffZsniff_qpZ sniff_base64rQ r )rV r _encode_text s@
rg plainutf-8c
C sd t | d|| t|||| j\}}
| j|
| jdtjjj||dd || d< t | |||| d S )Nr- r( T)r& zContent-Transfer-Encoding)
r@ rg r8 set_payloadrB r: r( ZALIASESgetrH ) r re r= r( rf rD rA rE rF r> Zpayloadr r r set_text_content s
rl rfc822c C s |dkrt d|dkr@|d kr.t dj||d kr:dn|}n0|dkrd|d
kr^t d j|d}n |d krpd}t| d
|| | j|g || d < t| |||| d S )Npartialz4message/partial is not supported for Message objectsrm rZ r[ binaryz*message/rfc822 parts do not support cte={}z
external-bodyz1message/external-body parts do not support cte={}r2 zContent-Transfer-Encoding)NrZ r[ ro )NrZ )r; r< r@ rj rH ) r r2 r= rf rD rA rE rF r> r r r set_message_content s$ rp r` c
C s t | ||| |dkr(t|| jjd}nN|dkrNtj|dddd}|jd}n(|dkrb|jd n|d
krv|jdd }| j| || d < t | |||| d S )Nr` )rR zquoted-printableFT)istextr5 Z quotetabsrK rZ r[ ro r^ zContent-Transfer-Encoding)r[ ro )
r@ rT r8 rR rO Zb2a_qpr' ra rj rH )
r rQ r r= rf rD rA rE rF r> r r r set_bytes_content s
rr )r& )rh ri NNNNNN)rm NNNNNN)r` NNNNN) rO Z
email.charsetr: Z
email.messageZ email.errorsr r Zraw_data_managerr, r r. splitr r/ r= r1 r@ rH rT rg rl r strrp r2 ZMessagerr r0 bytearray
memoryviewr" r r r r sD 6
'