a
i @ s d dgZ ddlZddlZddlZddlmZmZ ddlmZ ddlm Z ddl
m Z m Z ddlm
Z dd lmZ ejZd
Zed Zd d
ZdddZdd Zdd ZG dd d ZG dd deZG dd deZdS )Message EmailMessage N)BytesIOStringIO)utils)errors)Policycompat32charset)decode_bz; z[ \(\)<>@,;:\\"/\[\]\?=]c C s4 t | d\}}}|s$| d fS | | fS )N;)str partitionstrip)paramasepb r %/usr/lib64/python3.9/email/message.py _splitparam s r Tc C s |d urt |dkrt|trL| d7 } t|d |d |d }d| |f S z|d W n4 ty | d7 } t|dd}d| |f Y S 0 |st|rd | t |f S d| |f S n| S d S )
Nr * %s=%sasciizutf-8 z%s="%s")
len
isinstancetupler Zencode_rfc2231encodeUnicodeEncodeError tspecialssearchquote)r valuer% r r r _formatparam' s
r' c C s
dt | } g }d}| d||kr|d7 }| d|}|d }}|dkr|| d||| d|| 7 }|d dkrzq|| d|d }}qD|dk rt| }| d||}|dkr| || }n,| || d | |d | }|| |}q|S ) Nr
r r "z\"r =) r findcountr rstriplowerlstripappendr )sZpliststartendZinddiffifr r r _parseparamI s*
,r7 c C s4 t | tr&| d | d t| d fS t| S d S )Nr r r )r r r unquote)r& r r r
_unquotevaluec s
r9 c @ s e Zd ZefddZdd Zdcdd Zd
d Zddd d
Zdd Z dd Z
dd Z dd Z deddZ
dfddZdd Zd d Zd d Zd d! Zd"d# Zd$d% Zd&d' Zd(d) Zd*d+ Zd,d- Zd.d/ Zdgd0d1Zd2d3 Zd4d5 Z dhd6d7Z d8d9 Z d:d; Z dd? Z!d@dA Z"dBdC Z#dDdE Z$dFdG Z%didJdKZ&djdLdMZ'dkdPdQZ(dldRdSZ)dmdTdUZ*dndVdWZ+dodXdYZ,dZd[ Z-dpd\d]Z.dqd^d_Z/d`da Z0ddbl1m2Z2 dS )rr c C s: || _ g | _d | _d | _d | _d | _| _g | _d| _d S )N
text/plain) policy_headers _unixfrom_payload_charsetZpreambleZepiloguedefects
_default_type)selfr; r r r __init__~ s zMessage.__init__c C s | S N) as_stringrB r r r __str__ s zMessage.__str__Fr Nc C sJ ddl m} |d u r| jn|}t }||d||d}|j| |d | S )Nr ) GeneratorF) mangle_from_ maxheaderlenr; unixfrom)email.generatorrH r; r flattengetvalue)rB rL rJ r; rH fpgr r r rE s zMessage.as_stringc C s | S rD )as_bytesrF r r r __bytes__ s zMessage.__bytes__c C sH ddl m} |d u r| jn|}t }||d|d}|j| |d | S )Nr )BytesGeneratorF)rI r; rK )rM rT r; r rN rO )rB rL r; rT rP rQ r r r rR s zMessage.as_bytesc C s t | jtS rD )r r> listrF r r r is_multipart s zMessage.is_multipartc C s
|| _ d S rD r= )rB rL r r r set_unixfrom s zMessage.set_unixfromc C s | j S rD rW rF r r r get_unixfrom s zMessage.get_unixfromc C sD | j d u r|g| _ n,z| j | W n ty> tdY n0 d S )Nz=Attach is not valid on a message with a non-multipart payload)r> r0 AttributeError TypeError)rB payloadr r r attach s
zMessage.attachc C s | r(|rd S |d u r | jS | j| S |d urNt| jtsNtdt| j | j}t| dd }t|trt
|r| dd}|sz| |
ddd}W q ty | dd}Y q0 n0|rz| d}W n ty | d}Y n0 |s |S |d krt|S |d
krRtd | \}}|D ]}| j| | q8|S |d v rt|} t }
ztj | |
d
d |
W S tjy | Y S 0 t|tr|S |S )NzExpected list, got %szcontent-transfer-encodingr r surrogateescaper replaceraw-unicode-escapezquoted-printablebase64 )z
x-uuencodeZuuencodeZuuezx-uueT)quiet)rV r> r rU r[ typer getr. r Z_has_surrogatesr! decode get_param LookupError UnicodeErrorquopriZ decodestringr join
splitlinesr; Z
handle_defectr uurO Error) rB r5 rf r\ cteZbpayloadr& r@ ZdefectZin_fileZout_filer r r get_payload sV "
zMessage.get_payloadc C sp t |dr:|d u r || _d S t|ts.t|}||j}t |drT|dd| _n|| _|d url| | d S )Nr! rf r r^ )hasattrr> r Charsetr! output_charsetrf set_charset)rB r\ r r r r set_payload5 s
zMessage.set_payloadc C s |d u r | d d | _d S t|ts.t|}|| _d| vrH| dd d| vrf| jdd| d n| d| || kr|| j| _d| vr| }z || W nf t
y | j}|rz| dd }W n t y | |j
}Y n0 ||| _| d| Y n0 d S )
Nr MIME-Version1.0 Content-Typer: r
zContent-Transfer-Encodingr r^ ) del_paramr? r rr
add_headerZget_output_charset set_paramZ body_encoder> Zget_body_encodingr[ r! ri rs )rB r ro r\ r r r rt I s:
zMessage.set_charsetc C s | j S rD )r? rF r r r get_charsetx s zMessage.get_charsetc C s
t | jS rD )r r< rF r r r __len__ s zMessage.__len__c C s
| |S rD )re rB namer r r __getitem__ s zMessage.__getitem__c C sr | j |}|rX| }d}| jD ]4\}}| |kr"|d7 }||kr"td||q"| j| j || d S )Nr r z/There may be at most {} {} headers in a message)r; Zheader_max_countr. r<
ValueErrorformatr0 header_store_parse)rB r valZ max_countZlnamefoundkvr r r __setitem__ s zMessage.__setitem__c C s@ | }g }| jD ]"\}}| |kr|||f q|| _d S rD )r. r< r0 )rB r
newheadersr r r r r __delitem__ s zMessage.__delitem__c C s | dd | jD v S )Nc S s g | ]\}}| qS r )r. .0r r r r r
rb z(Message.__contains__..)r. r< r~ r r r __contains__ s zMessage.__contains__c c s | j D ]\}}|V qd S rD r< )rB Zfieldr& r r r __iter__ s zMessage.__iter__c C s dd | j D S )Nc S s g | ] \}}|qS r r r r r r r rb z Message.keys..r rF r r r keys s z Message.keysc s fdd j D S )Nc s g | ]\}} j ||qS r r; header_fetch_parser rF r r r s z"Message.values..r rF r rF r values s
zMessage.valuesc s fdd j D S )Nc s" g | ]\}}| j ||fqS r r r rF r r r s z!Message.items..r rF r rF r items s
z
Message.itemsc C s: | }| jD ]&\}}| |kr| j|| S q|S rD )r. r< r; r )rB r failobjr r r r r re s
z Message.getc C s | j ||f d S rD )r< r0 )rB r r& r r r set_raw s zMessage.set_rawc C s t | j S rD )iterr< copyrF r r r raw_items s zMessage.raw_itemsc C sH g }| }| jD ](\}}| |kr|| j|| q|sD|S |S rD )r. r< r0 r; r )rB r r r r r r r r get_all s zMessage.get_allc K sp g }| D ]<\}}|d u r0||dd q |t|dd| q |d ur^|d| t|| |< d S )N_-r )r r0 r_ r' insert SEMISPACErk )rB _name_valueZ_paramspartsr r r r r rz s zMessage.add_headerc C s\ | }ttt| j| jD ]0\}\}}| |kr | j||| j|< qXq t|d S rD )r. zipranger r< r; r KeyError)rB r r r5 r r r r r replace_header' s " zMessage.replace_headerc C sH t }| d|}||u r"| S t|d }|ddkrDdS |S )N content-typer /r r: )objectre get_default_typer r. r, )rB missingr& ctyper r r get_content_type: s
zMessage.get_content_typec C s | }|dd S )Nr r r splitrB r r r r get_content_maintypeR s z Message.get_content_maintypec C s | }|dd S )Nr r r r r r r get_content_subtype[ s zMessage.get_content_subtypec C s | j S rD rA rF r r r r d s zMessage.get_default_typec C s
|| _ d S rD r r r r r set_default_typem s zMessage.set_default_typec C s t }| ||}||u r |S g }t|D ]V}z$|dd\}}| }| }W n typ | }d}Y n0 |||f q*t|}|S )Nr) r r ) r re r7 r r r r0 r Z
decode_params) rB r headerr r& paramspr r r r r _get_params_preservev s
z Message._get_params_preserver Tc C s8 t }| ||}||u r |S |r0dd |D S |S d S )Nc S s g | ]\}}|t |fqS r )r9 r r r r r rb z&Message.get_params..)r r )rB r r r8 r r r r r
get_params s
zMessage.get_paramsc C sN || vr |S | ||D ]0\}}| | kr|r@t| S | S q|S rD )r r. r9 )rB r r r r8 r r r r r rg s
zMessage.get_paramrx r c C s
t |ts|r|||f}|| vr2| dkr2d}n
| |}| j||dst|s\t|||}qt|t|||g}nbd}| j||dD ]N\} }
d} | | krt|||} n t| |
|} |s| }qt|| g}q|| |kr|r| || n| |= || |< d S )Nr r: )r r r r8 )
r r r. re rg r' r rk r r ) rB r r& r requoter languager_ r Z old_param old_valueZ append_paramr r r r{ s6
zMessage.set_paramc C s || vr d S d}| j ||dD ]@\}}| | kr |sHt|||}q t|t|||g}q || |kr|| |= || |< d S )Nr r )r r. r' r rk re )rB r r r Z new_ctyper r r r r ry s
zMessage.del_paramc C s | ddkst| dkr,| d= d| d< || vr@|| |< d S | j||d}| |= || |< |dd D ]\}}| |||| qhd S )Nr r r z mime-versionrw rv r )r, r r. r r{ )rB rd r r r r r r r r set_type s zMessage.set_typec C sD t }| d|d}||u r*| d|d}||u r6|S t| S )Nfilenamecontent-dispositionr r )r rg r collapse_rfc2231_valuer )rB r r r r r r get_filename, s zMessage.get_filenamec C s, t }| d|}||u r |S t| S )Nboundary)r rg r r r- )rB r r r r r r get_boundary< s
zMessage.get_boundaryc C s t }| |d}||u r$tdg }d}|D ]:\}}| dkr\|dd| f d}q0|||f q0|s|dd| f g }| jD ]z\} }
| dkrg } |D ].\} }
|
dkr| | q| d| |
f qt| }
|| j
| |
q|| |
f q|| _d S ) Nr z No Content-Type header foundFr z"%s"Tr r ) r r r ZHeaderParseErrorr. r0 r< r rk r; r )rB r r r Z newparamsZfoundpZpkZpvr hr r r r r r r set_boundaryI s2
zMessage.set_boundaryc C s t }| d|}||u r |S t|trp|d p2d}z |d d}t||}W n ttfyn |d }Y n0 z|d W n ty | Y S 0 | S )Nr r zus-asciir r` ) r rg r r r! r rh ri r. )rB r r r ZpcharsetrR r r r get_content_charsetw s
zMessage.get_content_charsetc s fdd| D S )Nc s g | ]}| qS r )r )r partr r r r rb z(Message.get_charsets..walk)rB r r r r get_charsets s zMessage.get_charsetsc C s* | d}|d u rd S t|d }|S )Nr r )re r r. )rB r& c_dr r r get_content_disposition s
z Message.get_content_dispositionr )Fr N)FN)NF)N)N)N)Nr T)Nr T)rx TNr F)r T)rx T)N)N)N)N)3__name__
__module__ __qualname__r rC rG rE rS rR rV rX rY r] rp ru rt r| r} r r r r r r r r re r r r rz r r r r r r r r rg r{ ry r r r r r r r Zemail.iteratorsr r r r r r o sd
Z
/
"
3
.
c s e Zd Zd/ fdd Zd0 fdd Zdd Zd d
Zd d Zd1ddZh dZ dd Z
dd Z ddddZ ddddZ
dd Zd2d d Zd3d d Zd4d d!Zdd"d#d$Zd%d& Zd'd( Zd)d* Zd+d, Zd-d. Z ZS )5MIMEPartNc s( |d u rddl m} |}t | d S )Nr )default)Z email.policyr superrC )rB r; r __class__r r rC s zMIMEPart.__init__Fc s0 |d u r| j n|}|d u r |j}t |||S rD )r; Zmax_line_lengthr rE )rB rL rJ r; r r r rE s zMIMEPart.as_stringc C s | j | jjdddS )NT)utf8r; )rE r; ZclonerF r r r rG s zMIMEPart.__str__c C s | d}|d u rdS |jdkS )Nr F
attachment)re Zcontent_disposition)rB r r r r
is_attachment s
zMIMEPart.is_attachmentc c s | r d S | d\}}|dkrB||v r>|||fV d S |dksR| sVd S |dkr| D ]}| ||E d H qfd S d|v r|d|fV d }|d}|r| D ]}|d |kr|} qq|d u r| }|r|d nd }|d ur
| ||E d H d S )Nr text multipartrelatedr2
content-idr ) r r r indexrV
iter_parts
_find_bodyrg rp ) rB r preferencelistmaintypesubtypeZsubpart candidater2 Zsubpartsr r r r s6
zMIMEPart._find_bodyr htmlplainc C sB t |}d }| | |D ]$\}}||k r|}|}|dkr q>q|S )Nr )r r )rB r Z best_priobodyZprior r r r get_body s
zMIMEPart.get_body> )r r )r alternative)r r )r r c
c s" | d\}}|dks"|dkr&d S | }z | }W n tyN Y d S 0 |dkr|dkr| d}|rd}g }|D ]"}|d|krd}qz|| qz|r|E d H d S |d |E d H d S g } |D ]L}| d\}}||f| j v r|
s|| vr| | q|V qd S )
Nr r r r r2 Fr Tr ) r r rp r rZ rg re r0 pop _body_typesr )
rB r r r\ r r2 r Z attachmentsr seenr r r iter_attachments sD
zMIMEPart.iter_attachmentsc c s | r| E d H d S rD )rV rp rF r r r r C s zMIMEPart.iter_parts)content_managerc O s( |d u r| j j}|j| g|R i |S rD )r; r get_contentrB r argskwr r r r K s zMIMEPart.get_contentc O s, |d u r| j j}|j| g|R i | d S rD )r; r set_contentr r r r r P s zMIMEPart.set_contentc
C s | dkr6| }||f }||v r6td||g }g }| jD ]4\}}| drj|||f qD|||f qD|rt| | j d} || _| j
| _
| g| _
ng | _
|| _d| | d< |d ur| d| d S )Nr zCannot convert {} to {}content-r z
multipart/rx r ) r r r r r< r.
startswithr0 rd r; r> r{ )
rB r Zdisallowed_subtypesr Zexisting_subtypeZ keep_headersZ part_headersr r& r r r r _make_multipartU s.
zMIMEPart._make_multipartc C s | dd| d S )Nr )r mixedr rB r r r r make_relatedp s zMIMEPart.make_relatedc C s | dd| d S )Nr )r r r r r r make_alternatives s zMIMEPart.make_alternativec C s | dd| d S )Nr r r r r r r
make_mixedv s zMIMEPart.make_mixed)_dispc O sj | dks| |kr(t| d| t| | jd}|j|i | |r\d|vr\||d< | | d S )Nr Zmake_r r zContent-Disposition)r r getattrrd r; r r] )rB Z_subtyper r r r r r r _add_multiparty s
zMIMEPart._add_multipartc O s | j dg|R ddi| d S )Nr r Zinliner rB r r r r r add_related s zMIMEPart.add_relatedc O s | j dg|R i | d S )Nr r r r r r add_alternative s zMIMEPart.add_alternativec O s | j dg|R ddi| d S )Nr r r r r r r r add_attachment s zMIMEPart.add_attachmentc C s g | _ d | _d S rD r< r> rF r r r clear s zMIMEPart.clearc C s dd | j D | _ d | _d S )Nc S s& g | ] \}}| d s||fqS )r )r. r )r nr r r r r s z*MIMEPart.clear_content..r rF r r r
clear_content s zMIMEPart.clear_content)N)FNN)r )N)N)N)r r r rC rE rG r r r r r r r r r r r r r r r r r r
__classcell__r r r r r s*
7
r c s e Zd Z fddZ ZS )r c s& t j|i | d| vr"d| d< d S )Nrv rw )r r r r r r r s zEmailMessage.set_content)r r r r r r r r r r s )NT) __all__rerm rj ior r Zemailr r Zemail._policybaser r r r? Zemail._encoded_wordsr rr r compiler# r r' r7 r9 r r r r r r r s4
" N `