U
e5d�) � @ s. d dl Z d dlZd dlZd dlZd dlmZ G dd� d�Ze� Zd%dd�Ze� de� d d
� Z
d � � D ]Z e� e e
� qfd d
� Z
d� � D ]Ze� de e
� q�dd� Ze� de� dd� Zdd� Zdd� Zdd� Zd&d d �Ze�ee� d'd d!�Ze�ejje� d(d#d$�Zee e fD ]Z e�e 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 �key�handlerr 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_maintype�KeyError)r �msg�args�kwZ content_type�maintyper 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�||f�n|}|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__�getattr�join�__name__r ) r r r Zfull_path_for_error�typZqname�modname� full_path�namer 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||d�S )NT��decode�charset�ASCII)�errors)� get_payloadZ get_paramr) )r r, Zcontentr* r r r �get_text_content@ s r. �textc C s | j dd�S )NTr( �r- �r r r r �get_non_text_contentG s r2 z audio image video applicationc C s
| � d�S �Nr r0 r1 r r r �get_message_contentM s r4 zrfc822 external-bodyzmessage/c C s t | �d��S r3 )�bytesr- r1 r r r �%get_and_fixup_unknown_message_contentS s r6 �messagec
s� d� ||f�| d< |r�t|d d�s<| 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)�.0�header�Zmpr r �
<listcomp>d s �z _prepare_set.<locals>.<listcomp>zInvalid header: {})�policy) r! �hasattrr= Zdefectsr& �emailr, Z HeaderDefect�
ValueError�formatZfold)r r �subtype�headersr: �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-Disposition�filenameT)r: r' z
Content-ID)� set_param�items)r � dispositionrF �cid�paramsr �valuer r r �
_finalize_setp s �rM c C sV g }|d d }t dt| �|�D ]*}| ||| � }|�t�|��d�� q d�|�S )N� � r �asciir )�range�len�append�binascii�
b2a_base64r) r! )�data�max_line_lengthZ
encoded_linesZunencoded_bytes_per_line�iZthisliner r r �_encode_base64� s rY c s� | � |��� }|j� d�� � fdd�}dd� }|d kr�tdd� |D �dd �|jkr�zd
||��d�fW S tk
rz Y nX |jd kr�d ||��dd �fS ||d d
� �}t� |�d�|j�}t
� |�} t |�t | �kr�d}nd}t |�d
kr�||fS |d
k�r||��d�}
nj|d k�r,||��dd �}
nN|dk�rPt� ||��d�|j�}
n*|dk�rlt
||�|j�}
ntd�|���||
fS )NrP c s � � | �� S r �r! ��lines��linesepr r �
embedded_body� � z#_encode_text.<locals>.embedded_bodyc S s d� | �d S )N�
rZ r[ r r r � normal_body� r` z!_encode_text.<locals>.normal_bodyc s s | ]}t |�V qd S r )rR )r9 �xr r r � <genexpr>� s z _encode_text.<locals>.<genexpr>r )�default�7bit�8bit�surrogateescape�
zlatin-1�base64�quoted-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 �plain�utf-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* ZALIASES�getrM ) 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 )
N�partialz4message/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 �<module> s^ 6
� ' �
�
�