o !QçdäNã@sÊdZgd¢ZddlZddlZddlZddlZddlmZddlm Z m Z ddl m Z dZ dZe d ¡Ze d ej¡ZGd d „d ƒZGd d„deƒZdZGdd„deƒZeeejdƒƒZdeZejZdS)z:Classes to generate plain text from a message object tree.)Ú GeneratorÚDecodedGeneratorÚBytesGeneratoréN)Údeepcopy)ÚStringIOÚBytesIO)Ú_has_surrogatesÚ_Ú z \r\n|\r|\nz^From c@s°eZdZdZd'ddœdd„Zdd„Zd(d d „Zd d „Zd d„Zdd„Z dd„Z dd„Z dd„Z dd„Z dd„ZeZdd„Zdd„Zdd „Zd!d"„Zed)d#d$„ƒZed%d&„ƒZdS)*rzŠGenerates output from a Message object tree. This basic generator writes the message to the given file object as plain text. N©ÚpolicycCs6|dur |dur dn|j}||_||_||_||_dS)aõCreate the generator for message flattening. outfp is the output file-like object for writing the message to. It must have a write() method. Optional mangle_from_ is a flag that, when True (the default if policy is not set), escapes From_ lines in the body of the message by putting a `>' in front of them. Optional maxheaderlen specifies the longest length for a non-continued header. When a header line is longer (in characters, with tabs expanded to 8 spaces) than maxheaderlen, the header will split as defined in the Header class. Set maxheaderlen to zero to disable header wrapping. The default is 78, as recommended (but not required) by RFC 2822. The policy keyword specifies a policy object that controls a number of aspects of the generator's operation. If no policy is specified, the policy associated with the Message object passed to the flatten method is used. NT)Ú mangle_from_Ú_fpÚ _mangle_from_Ú maxheaderlenr )ÚselfÚoutfpr rr ©rú6/opt/alt/python310/lib64/python3.10/email/generator.pyÚ__init__$s  zGenerator.__init__cCs|j |¡dS©N)rÚwrite©rÚsrrrrDszGenerator.writeFcCsâ|jdur|jn|j}|dur|j|d}|jdur!|j|jd}|j|_| |j¡|_d|_| |j¡|_|j}|j}z-||_||_|r\|  ¡}|sTdt   t   ¡¡}|  ||j¡|  |¡W||_||_dS||_||_w)aÿPrint the message object tree rooted at msg to the output file specified when the Generator instance was created. unixfrom is a flag that forces the printing of a Unix From_ delimiter before the first object in the message tree. If the original message has no From_ delimiter, a `standard' one is crafted. By default, this is False to inhibit the printing of any From_ delimiter. Note that for subobjects, no From_ line is printed. linesep specifies the characters used to indicate a new line in the output. The default value is determined by the policy specified when the Generator instance was created or, if none was specified, from the policy associated with the msg. N)Úlinesep©Zmax_line_lengthÚz From nobody )r ÚclonerrÚ_NLÚ_encodeÚ _encoded_NLZ_EMPTYÚ_encoded_EMPTYZ get_unixfromÚtimeÚctimerÚ_write)rÚmsgÚunixfromrr Zold_gen_policyZold_msg_policyZufromrrrÚflattenHs0    ÿzGenerator.flattencCs|j||jd|jdS)z1Clone this generator with the exact same options.Nr )Ú __class__rr )rÚfprrrrys ýzGenerator.clonecCótƒSr)r©rrrrÚ _new_bufferszGenerator._new_buffercCs|Srrrrrrr‘szGenerator._encodecCsX|sdSt |¡}|dd…D] }| |¡| |j¡q|dr*| |d¡dSdS)Néÿÿÿÿ)ÚNLCREÚsplitrr)rÚlinesÚlinerrrÚ _write_lines•s  ÿzGenerator._write_linescCsÐ|j}zd|_| ¡|_}| |¡W||_|j}|`n ||_|j}|`w|rJt|ƒ}| d¡dur:|d|d<n| d|d¡| d|d¡t|ddƒ}|durZ| |¡n||ƒ|j  |  ¡¡dS)Núcontent-transfer-encodingrúContent-Transfer-Encodingú content-typeéÚ_write_headers) rÚ _munge_cter,Ú _dispatchrÚgetZreplace_headerÚgetattrr7rÚgetvalue)rr%ZoldfpZsfpZ munge_cteÚmethrrrr$¥s,  þ  zGenerator._writecCst| ¡}| ¡}t ||f¡ dd¡}t|d|dƒ}|dur4| dd¡}t|d|dƒ}|dur4|j}||ƒdS)Nú-r Z_handle_)Úget_content_maintypeÚget_content_subtypeÚ UNDERSCOREÚjoinÚreplacer;Ú _writeBody)rr%ÚmainÚsubZspecificr=Zgenericrrrr9Ìs  zGenerator._dispatchcCs6| ¡D]\}}| |j ||¡¡q| |j¡dSr)Ú raw_itemsrr Zfoldr©rr%ÚhÚvrrrr7àszGenerator._write_headerscCsž| ¡}|dur dSt|tƒstdt|ƒƒ‚t|jƒr?| d¡}|dur?t|ƒ}|d=|  ||¡| ¡}|d|df|_ |j rHt   d|¡}| |¡dS)Nzstring payload expected: %sÚcharsetr3r5ú>From )Ú get_payloadÚ isinstanceÚstrÚ TypeErrorÚtyperÚ_payloadZ get_paramrZ set_payloadr8rÚfcrerFr2)rr%ÚpayloadrKrrrÚ _handle_textês$    ÿ zGenerator._handle_textc Cs–g}| ¡}|dur g}nt|tƒr| |¡dSt|tƒs!|g}|D]}| ¡}| |¡}|j|d|jd|  |  ¡¡q#|  ¡}|sU|j   |¡}| |¡}| |¡|jdurs|jret d|j¡} n|j} | | ¡| |j¡| d||j¡|rˆ|j | d¡¡|D]} | |jd||j¡|j | ¡qŠ| |jd|d|j¡|jdurÉ|jr¿t d|j¡} n|j} | | ¡dSdS)NF©r&rrLz--r)rMrNrOrÚlistr,rr'rÚappendr<Z get_boundaryr rBÚ_make_boundaryZ set_boundaryÚpreamblerrSrFr2rÚpopÚepilogue) rr%ZmsgtextsZsubpartsÚpartrÚgÚboundaryZalltextrZZ body_partr\rrrÚ_handle_multipartsL           ûzGenerator._handle_multipartcCs4|j}|jdd|_z | |¡W||_dS||_w)Nrr)r rr`)rr%ÚprrrÚ_handle_multipart_signed<s  z"Generator._handle_multipart_signedcCsšg}| ¡D]:}| ¡}| |¡}|j|d|jd| ¡}| |j¡}|r;|d|jkr;|  |j  |dd…¡¡q|  |¡q|j   |j  |¡¡dS)NFrVr-) rMr,rr'rr<r/r r!rXrBrr)rr%Zblocksr]rr^Útextr0rrrÚ_handle_message_delivery_statusGs    z)Generator._handle_message_delivery_statuscCs^| ¡}| |¡}|j}t|tƒr"|j| d¡d|jd| ¡}n|  |¡}|j   |¡dS)NrFrV) r,rrRrNrWr'rMrr<rrr)rr%rr^rTrrrÚ_handle_message\s    zGenerator._handle_messagecCsxt tj¡}dt|d}|dur|S|}d} | dt |¡dtj¡}|  |¡s/ |S|dt |ƒ}|d7}q) Nz===============z==rTz^--z(--)?$Ú.r6) ÚrandomZ randrangeÚsysÚmaxsizeÚ_fmtÚ _compile_reÚreÚescapeÚ MULTILINEÚsearchrO)ÚclsrcÚtokenr_ÚbZcounterZcrerrrrYus  þûzGenerator._make_boundarycCs t ||¡Sr)rlÚcompile©rprÚflagsrrrrk‡s zGenerator._compile_re)NN)FNr)Ú__name__Ú __module__Ú __qualname__Ú__doc__rrr'rr,rr2r$r9r7rUrDr`rbrdreÚ classmethodrYrkrrrrrs0  ÿ 1' :  rcsPeZdZdZdd„Zdd„Zdd„Zdd „Z‡fd d „ZeZ e d d „ƒZ ‡Z S)raþGenerates a bytes version of a Message object tree. Functionally identical to the base Generator except that the output is bytes and not string. When surrogates were used in the input to encode bytes, these are decoded back to bytes for output. If the policy has cte_type set to 7bit, then the message is transformed such that the non-ASCII bytes are properly content transfer encoded, using the charset unknown-8bit. The outfp object must accept bytes in its write method. cCs|j | dd¡¡dS)NÚasciiÚsurrogateescape)rrÚencoderrrrr™szBytesGenerator.writecCr*r)rr+rrrr,œszBytesGenerator._new_buffercCs | d¡S©Nr{)r}rrrrrŸs zBytesGenerator._encodecCs8| ¡D]\}}|j |j ||¡¡q| |j¡dSr)rGrrr Z fold_binaryrrHrrrr7¢szBytesGenerator._write_headerscs^|jdurdSt|jƒr%|jjdks%|jrt d|j¡|_| |j¡dStt |ƒ  |¡dS)NZ7bitrL) rRrr Zcte_typerrSrFr2ÚsuperrrU)rr%©r(rrrUªs zBytesGenerator._handle_textcCst | d¡|¡Sr~)rlrsr}rtrrrrk¹szBytesGenerator._compile_re) rvrwrxryrr,rr7rUrDrzrkÚ __classcell__rrr€rrŒs   rzD[Non-text (%(type)s) part of message omitted, filename %(filename)s]c@s(eZdZdZdddœdd„Zdd„ZdS) rz±Generates a text representation of a message. Like the Generator base class, except that non-text parts are substituted with a format string representing the part. Nr cCs0tj|||||d|durt|_dS||_dS)aŒLike Generator.__init__() except that an additional optional argument is allowed. Walks through all subparts of a message. If the subpart is of main type `text', then it prints the decoded payload of the subpart. Otherwise, fmt is a format string that is used instead of the message payload. fmt is expanded with the following keywords (in %(keyword)s format): type : Full MIME type of the non-text part maintype : Main MIME type of the non-text part subtype : Sub-MIME type of the non-text part filename : Filename of the non-text part description: Description associated with the non-text part encoding : Content transfer encoding of the non-text part The default value for fmt is None, meaning [Non-text (%(type)s) part of message omitted, filename %(filename)s] r N)rrÚ_FMTrj)rrr rZfmtr rrrrÇs ÿ  zDecodedGenerator.__init__c Cs„| ¡D];}| ¡}|dkrt|jdd|dq|dkrqt|j| ¡| ¡| ¡| d¡| dd¡| d d ¡d œ|dqdS) NrcF)Údecode)ÚfileZ multipartz [no filename]zContent-Descriptionz[no description]r4z [no encoding])rQÚmaintypeZsubtypeÚfilenameÚ descriptionÚencoding) Úwalkr?ÚprintrMrjZget_content_typer@Ú get_filenamer:)rr%r]r…rrrr9ås* ÿÿù ÷øzDecodedGenerator._dispatch)NNN)rvrwrxryrr9rrrrrÁs  ÿ rr6z%%0%dd)ryÚ__all__rlrhr"rgÚcopyrÚiorrZ email.utilsrrAÚNLrsr.rnrSrrr‚rÚlenÚreprriZ_widthrjrYrrrrÚs,   t3;