U
e5d) @ s. d dl Z d dlZd dlZd dlZd dlmZ G dd dZe Zd%ddZe de d d
Z
d D ]Z e e e
qfd d
Z
d D ]Ze de e
qdd Ze de dd Zdd Zdd Zdd Zd&d d Zeee d'd d!Zeejje d(d#d$Zee e fD ]Z ee e qdS )) 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.8/email/contentmanager.py__init__ s zContentManager.__init__c C s || j |< d S r )r )r keyhandlerr r r add_get_handler
s z ContentManager.add_get_handlerc O s| | }|| jkr(| j| |f||S | }|| 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 r )r )r Ztypekeyr r r r add_set_handler s z ContentManager.add_set_handlerc O s> | dkrtd| ||}| |||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 }t |jD ]}|| jkr*| j| S |j}t|dd}|rNd||fn|}|d kr^|}|| jkrv| j| S || jkr| j| S |j}|| jkr| j| S qd | 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}| dd}|j||dS )NTdecodecharsetASCII)errors) get_payloadZ get_paramr) )r r, Zcontentr* r r r get_text_content@ s r. textc C s | j ddS )NTr( r- r r r r get_non_text_contentG s r2 z audio image video applicationc C s
| dS Nr r0 r1 r r r get_message_contentM s r4 zrfc822 external-bodyzmessage/c C s t | dS r3 )bytesr- r1 r r r %get_and_fixup_unknown_message_contentS s r6 messagec
s d ||f| d< |rt|d ds<| j fdd|D }z(|D ] }|jrV|jd || |j< qBW n@ tjjk
r } z td |j
| jd|W 5 d }~X Y nX d S ) N/z Content-Typer r& c s g | ]} j |g qS r )Zheader_factoryZheader_source_parse).0headerZmpr r
d s z _prepare_set..zInvalid header: {})policy) r! hasattrr= Zdefectsr& emailr, Z HeaderDefect
ValueErrorformatZfold)r r subtypeheadersr: excr r; r _prepare_set_ s$
rE c C sx |d kr|d k rd}|d k r$|| d< |d k r>| j d|ddd |d k rN|| d< |d k rt| D ]\}}| || q^d S )NZ
attachmentzContent-DispositionfilenameT)r: r' z
Content-ID) set_paramitems)r dispositionrF cidparamsr valuer r r
_finalize_setp s rM c C sV g }|d d }t dt| |D ]*}| ||| }|t|d q d|S )N r asciir )rangelenappendbinascii
b2a_base64r) r! )datamax_line_lengthZ
encoded_linesZunencoded_bytes_per_lineiZthisliner r r _encode_base64 s rY c s | | }|j d fdd}dd }|d krtdd |D dd |jkrzd
||dfW S tk
rz Y nX |jd krd ||dd fS ||d d
}t |d|j}t
|} t |t | krd}nd}t |d
kr||fS |d
kr||d}
nj|d kr,||dd }
nN|dkrPt ||d|j}
n*|dkrlt
|||j}
ntd|||
fS )NrP c s | S r r! lineslinesepr r
embedded_body z#_encode_text..embedded_bodyc S s d | d S )N
rZ r[ r r r normal_body r` z!_encode_text..normal_bodyc s s | ]}t |V qd S r )rR )r9 xr r r s z _encode_text..r )default7bit8bitsurrogateescape
zlatin-1base64quoted-printablez$Unknown content transfer encoding {})encode
splitlinesr^ maxrW r) UnicodeDecodeErrorZcte_typer Z body_encoderT rU rR rY r@ rA ) stringr* cter= r\ r_ rb ZsniffZsniff_qpZ sniff_base64rV r r] r _encode_text sD
rr plainutf-8c
C sd t | d|| t|||| j\}}
| |
| jdtjj||dd || d< t | |||| d S )Nr/ r* T)r' Content-Transfer-Encoding)
rE rr r= set_payloadrG r? r* ZALIASESgetrM ) r rp rB r* rq rI rF rJ rK rC Zpayloadr r r set_text_content s
rx rfc822c C s |dkrt d|dkr@|dkr.t d||d kr:dn|}n0|dkrd|dkr^t d |d
}n |d krpd
}t| d || | |g || d < t| |||| d S )
Npartialz4message/partial is not supported for Message objectsry )Nrf rg binaryz*message/rfc822 parts do not support cte={}rg z
external-body)Nrf z1message/external-body parts do not support cte={}rf r7 ru )r@ rA rE rv rM ) r r7 rB rq rI rF rJ rK rC r r r set_message_content s( r| rj c
C s t | ||| |dkr(t|| jjd}nN|dkrNtj|dddd}|d}n(|dkrb|d n|d krv|dd
}| | || d < t | |||| d S ) Nrj )rW rk FT)istextr: Z quotetabsrP rf )rg r{ rh ru )
rE rY r= rW rT Zb2a_qpr) rl rv rM )
r rV r rB rq rI rF rJ rK rC r r r set_bytes_content s
r~ )r' )rs rt NNNNNN)ry NNNNNN)rj NNNNN) rT Z
email.charsetr? Z
email.messageZ email.errorsr r Zraw_data_managerr. r
r2 splitr r4 rB r6 rE rM rY rr rx r strr| r7 ZMessager~ r5 bytearray
memoryviewr# r r r r s^ 6
'