]β @ s6 d g Z d d l Z d d l Z d d l Z d d l Z d d l m Z m Z d d l m Z d d l m
Z
d d l m Z d d l m
Z d d l m Z e j Z d Z e j d
Z d d Z d d
d d Z d d Z d d Z Gd d d Z Gd d d e Z Gd d d e Z d S)Message N)BytesIOStringIO)utils)errors)compat32)charset)decode_bz; z[ \(\)<>@,;:\\"/\[\]\?=]c C sJ t | j d \ } } } | s4 | j d f S| j | j f S)N;)str partitionstrip)paramasepb r ,/opt/alt/python35/lib64/python3.5/message.py _splitparam s r Tc C s | d k r t | d k r t | t rf | d 7} t j | d | d | d } d | | f Sy | j d Wn? t k
r | d 7} t j | d d } d | | f SYn X| s t j | r d | t j | f Sd | | f Sn | Sd S)
Nr * z%s=%sasciizutf-8 z%s="%s")
len
isinstancetupler Zencode_rfc2231encodeUnicodeEncodeError tspecialssearchquote)r valuer! r r r _formatparam( s
!
r# c C sI d t | } g } x,| d d d k rD| d d } | j d } xR | d k r | j d d | | j d d | d r | j d | d } qQ W| d k r t | } | d | } d | k r | j d } | d | j j d | | d d j } | j | j | | d } q W| S)Nr
r r "z\"r =)r findcountr indexr
lowerappend)sZplistendfir r r _parseparamJ s ; 8r/ c C sB t | t r1 | d | d t j | d f St j | Sd S)Nr r r )r r r unquote)r" r r r
_unquotevalue^ s "r1 c @ s e Z d Z e d d Z d d Z d d d d d Z d
d Z d d d d
Z d d Z d d Z
d d Z d d Z d d d d Z
d d d Z d d Z d d Z d d Z d d! Z d" d# Z d$ d% Z d& d' Z d( d) Z d* d+ Z d, d- Z d. d/ Z d d0 d1 Z d2 d3 Z d4 d5 Z d d6 d7 Z d8 d9 Z d: d; Z d< d= Z d> d? Z! d@ dA Z" dB dC Z# dD dE Z$ dF dG Z% d dH dI dJ dK Z& d dH dI dL dM Z' dN dI d dO d dP dQ Z( dH dI dR dS Z) dN dI dT dU Z* d dV dW Z+ d dX dY Z, dZ d[ Z- d d\ d] Z. d d^ d_ Z/ d` da Z0 d db l1 m2 Z2 d S)cr c C sS | | _ g | _ d | _ d | _ d | _ d | _ | _ g | _ d | _ d S)Nz
text/plain) policy_headers _unixfrom_payload_charsetZpreambleZepiloguedefects
_default_type)selfr2 r r r __init__y s zMessage.__init__c C s
| j S)N) as_string)r9 r r r __str__ s zMessage.__str__Fr Nc C so d d l m } | d k r% | j n | } t } | | d d d | d | } | j | d | | j S)Nr ) Generator mangle_from_F maxheaderlenr2 unixfrom)email.generatorr= r2 r flattengetvalue)r9 r@ r? r2 r= fpgr r r r; s zMessage.as_stringc C s
| j S)N)as_bytes)r9 r r r __bytes__ s zMessage.__bytes__c C si d d l m } | d k r% | j n | } t } | | d d d | } | j | d | | j S)Nr )BytesGeneratorr> Fr2 r@ )rA rH r2 r rB rC )r9 r@ r2 rH rD rE r r r rF s zMessage.as_bytesc C s t | j t S)N)r r5 list)r9 r r r is_multipart s zMessage.is_multipartc C s
| | _ d S)N)r4 )r9 r@ r r r set_unixfrom s zMessage.set_unixfromc C s | j S)N)r4 )r9 r r r get_unixfrom s zMessage.get_unixfromc C sW | j d k r | g | _ n5 y | j j | Wn t k
rR t d Yn Xd S)Nz=Attach is not valid on a message with a non-multipart payload)r5 r* AttributeError TypeError)r9 payloadr r r attach s
zMessage.attachc C sB | j r4 | r d S| d k r) | j S| j | S| d k rl t | j t rl t d t | j | j } t | j d d j } t | t rRt j
| r| j d d } | sRy" | j | j
d d d } WqRt k
r| j d d } YqRXn= | rRy | j d } Wn! t k
rQ| j d } Yn X| s\| S| d k rut j | S| d
k rt d j | j \ } } x! | D] } | j j | | qW| S| d k r+t | } t }
y$ t j | |
d d |
j SWn t j k
r*| SYn Xt | t r>| S| S)NzExpected list, got %szcontent-transfer-encodingr r surrogateescaper replacezraw-unicode-escapezquoted-printablebase64
x-uuencodeuuencodeuuex-uuequietT)rU rV rW rX )rJ r5 r rI rN typer getr) r Z_has_surrogatesr decode get_param LookupError UnicodeErrorquopriZ decodestringr join
splitlinesr2 Z
handle_defectr uurC Error) r9 r. r\ rO cteZbpayloadr" r7 ZdefectZin_fileZout_filer r r get_payload sV " "
!
zMessage.get_payloadc C s t | d rU | d k r( | | _ d St | t sC t | } | j | j } t | d r| | j d d | _ n | | _ | d k r | j | d S)Nr r\ r rQ )hasattrr5 r Charsetr output_charsetr\ set_charset)r9 rO r r r r set_payload0 s zMessage.set_payloadc C s{ | d k r& | j d d | _ d St | t sA t | } | | _ d | k rf | j d d d | k r | j d d d | j n | j d | j | | j k r | j | j | _ d | k rw| j } y | | Wn t
k
rv| j } | rPy | j d d } Wn$ t k
rO| j | j
} Yn X| j | | _ | j d | Yn Xd S) Nr z MIME-Versionz1.0z Content-Typez
text/plainzContent-Transfer-Encodingr rQ ) del_paramr6 r rh
add_headerZget_output_charset set_paramZ body_encoder5 Zget_body_encodingrN r r_ ri )r9 r re rO r r r rj D s8
zMessage.set_charsetc C s | j S)N)r6 )r9 r r r get_charsets s zMessage.get_charsetc C s
t | j S)N)r r3 )r9 r r r __len__{ s zMessage.__len__c C s
| j | S)N)r[ )r9 namer r r __getitem__ s zMessage.__getitem__c C s | j j | } | r | j } d } xW | j D]L \ } } | j | k r4 | d 7} | | k r4 t d j | | q4 W| j j | j j | | d S)Nr r z/There may be at most {} {} headers in a message)r2 Zheader_max_countr) r3
ValueErrorformatr* header_store_parse)r9 rq valZ max_countZlnamefoundkvr r r __setitem__ s
zMessage.__setitem__c C s^ | j } g } x< | j D]1 \ } } | j | k r | j | | f q W| | _ d S)N)r) r3 r* )r9 rq
newheadersrx ry r r r __delitem__ s zMessage.__delitem__c C s | j d d | j D k S)Nc S s" g | ] \ } } | j q Sr )r) ).0rx ry r r r
s z(Message.__contains__..)r) r3 )r9 rq r r r __contains__ s zMessage.__contains__c c s# x | j D] \ } } | Vq
Wd S)N)r3 )r9 Zfieldr" r r r __iter__ s zMessage.__iter__c C s d d | j D S)Nc S s g | ] \ } } | q Sr r )r} rx ry r r r r~ s z Message.keys..)r3 )r9 r r r keys s z Message.keysc s f d d j D S)Nc s+ g | ]! \ } } j j | | q Sr )r2 header_fetch_parse)r} rx ry )r9 r r r~ s z"Message.values..)r3 )r9 r )r9 r values s zMessage.valuesc s f d d j D S)Nc s1 g | ]' \ } } | j j | | f q Sr )r2 r )r} rx ry )r9 r r r~ s z!Message.items..)r3 )r9 r )r9 r items s z
Message.itemsc C sO | j } x< | j D]1 \ } } | j | k r | j j | | Sq W| S)N)r) r3 r2 r )r9 rq failobjrx ry r r r r[ s
z Message.getc C s | j j | | f d S)N)r3 r* )r9 rq r" r r r set_raw s zMessage.set_rawc C s t | j j S)N)iterr3 copy)r9 r r r raw_items s zMessage.raw_itemsc C sh g } | j } xE | j D]: \ } } | j | k r | j | j j | | q W| sd | S| S)N)r) r3 r* r2 r )r9 rq r r rx ry r r r get_all s zMessage.get_allc K s g } xd | j D]V \ } } | d k rG | j | j d d q | j t | j d d | q W| d k r | j d | t j | | | Wt j | } | S)Nr% r r ) r r[ r/ r r
rs r* r Z
decode_params) r9 r headerr r" paramsprq rv r r r _get_params_preserveq s
z Message._get_params_preservez content-typeTc C sJ t } | j | | } | | k r+ | S| rB d d | D S| Sd S)Nc S s( g | ] \ } } | t | f q Sr )r1 )r} rx ry r r r r~ s z&Message.get_params..)r r )r9 r r r0 r r r r r
get_params s
zMessage.get_paramsc C sc | | k r | SxL | j | | D]8 \ } } | j | j k r# | rW t | S| Sq# W| S)N)r r) r1 )r9 r r r r0 rx ry r r r r] s
zMessage.get_paramz Content-Typer c C s t | t r% | r% | | | f } | | k rL | j d k rL d } n | j | } | j | d | s | s t | | | } qGt j | t | | | g } n d } x | j d | d | D]u \ } }
d } | j | j k r
t | | | } n t | |
| } | s.| } q t j | | g } q W| | j | k r| ru| j | | n | | =| | | | d =d | d <| | k rX | | | t } | j d | } | | k r+ | St j | j S)Nboundary)r r] r r rstrip)r9 r r r r r r get_boundary7 s
zMessage.get_boundaryc C s t } | j | d } | | k r6 t j d g } d } xY | D]Q \ } } | j d k r | j d d | f d } qI | j | | f qI W| s | j d d | f g } x | j D] \ } }
| j d k rgg } xG | D]? \ } }
|
d k r | j | q | j d | |
f q Wt j | }
| j | j j
| |
q | j | |
f q W| | _ d S) Nz content-typez No Content-Type header foundFr z"%s"Tr z%s=%s) r r r ZHeaderParseErrorr) r* r3 r ra r2 ru )r9 r r r Z newparamsZfoundpZpkZpvr{ hry r rx rv r r r set_boundaryD s2
zMessage.set_boundaryc C s t } | j d | } | | k r+ | St | t r | d pG d } y& | d j d } t | | } Wn" t t f k
r | d } Yn Xy | j d Wn t k
r | SYn X| j S)Nr r zus-asciir zraw-unicode-escape) r r] r r r r r^ r_ r) )r9 r r r ZpcharsetrF r r r get_content_charsetr s
zMessage.get_content_charsetc s f d d | j D S)Nc s g | ] } | j q Sr )r )r} part)r r r r~ s z(Message.get_charsets..)walk)r9 r r )r r get_charsets s zMessage.get_charsetsc C s9 | j d } | d k r d St | d j } | S)Nzcontent-dispositionr )r[ r r) )r9 r" c_dr r r get_content_disposition s
z Message.get_content_disposition)r )3__name__
__module__ __qualname__r r: r< r; rG rF rJ rK rL rP rf rk rj ro rp rr rz r| r r r r r r[ r r r rm r r r r r r r r r] rn rl r r r r r r r Zemail.iteratorsr r r r r r j s` Z /
!2
.
c @ s e Z d Z d d d Z d d Z d d Z d. d d Z d/ d0 d1 d2 h Z d d Z d d Z d d d d Z
d d d d Z d d Z d d d Z
d d d Z d d d Z d! d d" d# Z d$ d% Z d& d' Z d( d) Z d* d+ Z d, d- Z d S)3MIMEPartNc C s6 | d k r" d d l m } | } t j | | d S)Nr )default)Z email.policyr r r: )r9 r2 r r r r r: s zMIMEPart.__init__c C s, | j d } | d k r d S| j d k S)Nzcontent-dispositionF
attachment)r[ Zcontent_disposition)r9 r r r r
is_attachment s zMIMEPart.is_attachmentc c sc | j r d S| j j d \ } } | d k r[ | | k rW | j | | f Vd S| d k rk d S| d k r x) | j D] } | j | | Ed Hq Wd Sd | k r | j d | f Vd } | j d } | rx+ | j D] } | d | k r | } Pq W| d k r>| j } | r8| d n d } | d k r_| j | | Ed Hd S)Nr text multipartrelatedstartz
content-idr )r r r r(
iter_parts
_find_bodyr] rf ) r9 r preferencelistmaintypesubtypeZsubpart candidater Zsubpartsr r r r s6 zMIMEPart._find_bodyr htmlplainc C s^ t | } d } xE | j | | D]1 \ } } | | k r% | } | } | d k r% Pq% W| S)Nr )r r )r9 r Z best_prioZbodyZprior r r r get_body s
zMIMEPart.get_bodyr r alternativec c sd | j j d \ } } | d k s3 | d k r7 d S| j j } | d k r | d k r | j d } | r d } g } x9 | D]1 } | j d | k r d } q | j | q W| r | Ed Hd S| j d | Ed Hd Sg } xl | D]d } | j j d \ } } | | f | j k rW| j rW| | k rW| j | q | Vq Wd S)
Nr r r r r Fz
content-idTr )
r r rf r r] r[ r* pop _body_typesr ) r9 r r r r rw Z attachmentsr seenr r r iter_attachments s6
zMIMEPart.iter_attachmentsc c s% | j d k r! | j Ed Hd S)Nr )r rf )r9 r r r r ! s zMIMEPart.iter_partscontent_managerc O s+ | d k r | j j } | j | | | S)N)r2 r get_content)r9 r argskwr r r r ) s zMIMEPart.get_contentc O s/ | d k r | j j } | j | | | d S)N)r2 r set_content)r9 r r r r r r r . s zMIMEPart.set_contentc
C s5 | j d k rO | j } | | f } | | k rO t d j | | g } g } xU | j D]J \ } } | j j d r | j | | f qe | j | | f qe W| r t | d | j } | | _ | j
| _
| g | _
n g | _
| | _ d | | d <| d k r1| j d | d S)Nr zCannot convert {} to {}zcontent-r2 z
multipart/z Content-Typer ) r r rs rt r3 r)
startswithr* rZ r2 r5 rn )
r9 r Zdisallowed_subtypesr Zexisting_subtypeZ keep_headersZ part_headersrq r" r r r r _make_multipart3 s,
zMIMEPart._make_multipartc C s | j d d | d S)Nr r mixed)r r )r )r9 r r r r make_relatedN s zMIMEPart.make_relatedc C s | j d d | d S)Nr r )r )r )r9 r r r r make_alternativeQ s zMIMEPart.make_alternativec C s | j d f | d S)Nr )r )r9 r r r r
make_mixedT s zMIMEPart.make_mixed_dispc O s | j d k s$ | j | k r8 t | d | t | d | j } | j | | | r| d | k r| | | d <| j | d S)Nr Zmake_r2 zcontent-dispositionzContent-Disposition)r r getattrrZ r2 r rP )r9 Z_subtyper r r r r r r _add_multipartW s
zMIMEPart._add_multipartc O s | j d | d d | d S)Nr r Zinline)r )r9 r r r r r add_relateda s zMIMEPart.add_relatedc O s | j d | | d S)Nr )r )r9 r r r r r add_alternatived s zMIMEPart.add_alternativec O s | j d | d d | d S)Nr r r )r )r9 r r r r r add_attachmentg s zMIMEPart.add_attachmentc C s g | _ d | _ d S)N)r3 r5 )r9 r r r clearj s zMIMEPart.clearc C s& d d | j D | _ d | _ d S)Nc S s7 g | ]- \ } } | j j d s | | f q S)zcontent-)r) r )r} nry r r r r~ o s z*MIMEPart.clear_content..)r3 r5 )r9 r r r
clear_contentn s zMIMEPart.clear_content)r r r )r r )r r )r r )r r )r r r r: r r r r r r r r r r r r r r r r r r r r r r r s, .
r c s" e Z d Z f d d Z S) EmailMessagec s- t j | | d | k r) d | d s.
" K