o ÐF`"kã@s²dZddlmZmZddlZddlZddlZddlZddlZddlm Z de ej j dƒZ dd„ZGd d „d ƒZejfd d „Zd d„Zdd„ZedkrWeejdd…ƒdSdS)z'Usage: unparse.py é)Úprint_functionÚunicode_literalsN)ÚStringIOZ1eécCsJt|ƒ}z|t|ƒƒWn tyYdSw|D] }|ƒ||ƒqdS)z„Z#d?d@„Z$dAdB„Z%dCdD„Z&dEdF„Z'dGdH„Z(dIdJ„Z)dKdL„Z*dMdN„Z+dOdP„Z,dQdR„Z-dSdT„Z.dUdV„Z/dWdX„Z0dÔdZd[„Z1d\d]„Z2d^d_„Z3d`da„Z4dbdc„Z5ddde„Z6dfdg„Z7dhdi„Z8djdk„Z9dldm„Z:dndo„Z;dpdq„Zdvdw„Z?dxdy„Z@dzd{„ZAd|d}„ZBd~d„ZCd€d„ZDd‚dƒ„ZEd„d…„ZFd†d‡„ZGdˆd‰„ZHdŠd‹„ZIdŒd„ZJdŽd„ZKdd‘d’d“d”œZLd•d–„ZMd’d“d—d˜d™dšd›dœddždŸd d¡d¢œ ZNd£d¤„ZOd¥d¦d§d¨d©dªd«d¬d­d®d¯œ ZPd°d±„ZQeRjSd²eRjTd³iZUd´dµ„ZVd¶d·„ZWd¸d¹„ZXdºd»„ZYd¼d½„ZZd¾d¿„Z[dÀdÁ„Z\dÂdÄZ]dÄdÅ„Z^dÆdÇ„Z_dÈdÉ„Z`dÊdË„ZadÌdÍ„ZbdÎdÏ„ZcdÐdÑ„ZddÒS)ÕÚUnparserzŽMethods in this class recursively traverse an AST and output source code for the abstract syntax; original formatting is disregarded. cCs8||_g|_d|_| |¡td|jd|j ¡dS)zTUnparser(tree, file=sys.stdout) -> None. Print the source for tree to file.rÚ©ÚfileN)r Úfuture_importsÚ_indentÚdispatchÚprintÚflush)ÚselfÚtreerr r r Ú__init__ s  zUnparser.__init__rcCs|j dd|j|¡dS)zBIndent a piece of text, according to the current indentation levelÚ z N)r Úwriter©rÚtextr r r Úfill*sz Unparser.fillcCs|j t |¡¡dS)z+Append a piece of text to the current line.N)r rÚsixZ text_typerr r r r.szUnparser.writecCs| d¡|jd7_dS)z(Print ':', and increase the indentation.ú:rN)rr©rr r r Úenter2s zUnparser.entercCs|jd8_dS)zDecrease the indentation level.rN)rr"r r r Úleave7szUnparser.leavecCs@t|tƒr|D]}| |¡qdSt|d|jjƒ}||ƒdS)z:Dispatcher function, dispatching tree type T to method _T.NÚ_)Ú isinstanceÚlistrÚgetattrÚ __class__Ú__name__)rrÚtÚmethr r r r;s   zUnparser.dispatchcCó|jD]}| |¡qdS©N©Úbodyr©rrZstmtr r r Ú_ModuleLó  ÿzUnparser._ModulecCr-r.r/r1r r r Ú _InteractivePr3zUnparser._InteractivecCó| |j¡dSr.)rr0©rrr r r Ú _ExpressionTózUnparser._ExpressioncCs| ¡| |j¡dSr.)rrÚvaluer6r r r Ú_ExprXszUnparser._ExprcCs:| d¡| |j¡| d¡| |j¡| d¡dS)Nú(z := ú))rrÚtargetr9r6r r r Ú _NamedExpr\ó    zUnparser._NamedExprcó&ˆ d¡t‡fdd„ˆj|jƒdS)Nzimport có ˆ d¡S©Nú, ©rr r"r r Úeó z"Unparser._Import..)rrrÚnames©rr+r r"r Ú_Importcó zUnparser._Importcsz|jr|jdkrˆj dd„|jDƒ¡ˆ d¡ˆ d|j¡|jr*ˆ |j¡ˆ d¡t‡fdd„ˆj|jƒdS) NÚ __future__css|]}|jVqdSr.)Úname©Ú.0Únr r r Ú js€z'Unparser._ImportFrom..zfrom Ú.z import crArBrDr r"r r rEqrFz&Unparser._ImportFrom..) ÚmodulerÚextendrGrrÚlevelrrrHr r"r Ú _ImportFromgs   zUnparser._ImportFromcCs8| ¡|jD] }| |¡| d¡q| |j¡dS)Nú = )rÚtargetsrrr9)rr+r=r r r Ú_Assignss    zUnparser._AssigncCsB| ¡| |j¡| d|j|jjjd¡| |j¡dS)Nú z= ) rrr=rÚbinopÚopr)r*r9rHr r r Ú _AugAssignzs zUnparser._AugAssigncCsŠ| ¡|jst|jtjƒr| d¡| |j¡|js(t|jtjƒr(| d¡| d¡| |j¡|j rC| d¡| |j ¡dSdS)Nr;r<ú: rV) rÚsimpler&r=ÚastÚNamerrÚ annotationr9rHr r r Ú _AnnAssign€s      þzUnparser._AnnAssigncCs.| d¡|jr| d¡| |j¡dSdS)NÚreturnrY)rr9rrrHr r r Ú_Returns  þzUnparser._ReturncCó| d¡dS)NÚpass©rrHr r r Ú_Pass“ózUnparser._PasscCre)NÚbreakrgrHr r r Ú_Break–rizUnparser._BreakcCre)NÚcontinuergrHr r r Ú _Continue™rizUnparser._Continuecr@)Nzdel crArBrDr r"r r rEžrFz"Unparser._Delete..)rrrrWrHr r"r Ú_DeleteœrJzUnparser._DeletecCs:| d¡| |j¡|jr| d¡| |j¡dSdS)Nzassert rC)rrÚtestÚmsgrrHr r r Ú_Assert s   þzUnparser._AssertcCsV| d¡| |j¡|jr| d¡| |j¡|jr)| d¡| |j¡dSdS)Nzexec ú in rC)rrr0ÚglobalsrÚlocalsrHr r r Ú_Exec§s     þzUnparser._ExeccCsp| d¡d}|jr| d¡| |j¡d}|jD]}|r$| d¡nd}| |¡q|js6| d¡dSdS)Nzprint Fú>>TrCú,)rÚdestrrÚvaluesÚnl)rr+Zdo_commaÚer r r Ú_Print±s     ÿzUnparser._Printcr@)Nzglobal crArBrDr r"r r rEÁrFz"Unparser._Global..©rrrrGrHr r"r Ú_Global¿rJzUnparser._Globalcr@)Nz nonlocal crArBrDr r"r r rEÅrFz$Unparser._Nonlocal..r}rHr r"r Ú _NonlocalÃrJzUnparser._NonlocalcCó>| d¡| d¡|jr| d¡| |j¡| d¡dS)Nr;ÚawaitrYr<©rr9rrHr r r Ú_AwaitÇó    zUnparser._AwaitcCr€)Nr;ÚyieldrYr<r‚rHr r r Ú_YieldÏr„zUnparser._YieldcCr€)Nr;z yield fromrYr<r‚rHr r r Ú _YieldFrom×r„zUnparser._YieldFromcCsº| d¡tjr/|js|jrJ‚dS| d¡| |j¡|jr-| d¡| |j¡dSdS| d¡|jr=| |j¡|jrK| d¡| |j¡|j r[| d¡| |j ¡dSdS)NÚraiserYz from rC) rr ÚPY3ÚexcÚcauserrÚtypeÚinstZtbackrHr r r Ú_Raiseßs*     þ     þzUnparser._RaisecCsœ| d¡| ¡| |j¡| ¡|jD]}| |¡q|jr4| d¡| ¡| |j¡| ¡|jrL| d¡| ¡| |j¡| ¡dSdS)NÚtryÚelseÚfinally)rr#rr0r$ÚhandlersÚorelseÚ finalbody©rr+Úexr r r Ú_Tryõs"         üz Unparser._TrycCsp| d¡| ¡| |j¡| ¡|jD]}| |¡q|jr6| d¡| ¡| |j¡| ¡dSdS)Nrr)rr#rr0r$r’r“r•r r r Ú _TryExcepts       üzUnparser._TryExceptcCs~t|jƒdkrt|jdtjƒr| |j¡n| d¡| ¡| |j¡| ¡| d¡| ¡| |j ¡| ¡dS)Nrrrr‘) Úlenr0r&r_Z TryExceptrrr#r$r”rHr r r Ú _TryFinallys      zUnparser._TryFinallycCsv| d¡|jr| d¡| |j¡|jr+| d¡tjr%| |j¡n| |j¡| ¡| |j¡|  ¡dS)NÚexceptrYú as ) rrŒrrrLr r‰r#r0r$rHr r r Ú_ExceptHandler$s       zUnparser._ExceptHandlercCsz| d¡|jD] }| d¡| |¡q| d|j¡tjr| d¡d}|jD]}|r4| d¡nd}| |¡q*|jD]}|rI| d¡nd}| |¡q?t j dd…d krŠ|j rr|re| d¡nd}| d ¡| |j ¡|j rŠ|r}| d¡nd}| d ¡| |j ¡| d ¡n|jr­| d¡|jD] }| |¡| d¡q›| d ¡|  ¡| |j¡| ¡dS) Nrú@zclass r;FrCTé©ééÚ*ú**r<)rÚdecorator_listrrrLr r‰ÚbasesÚkeywordsÚsysÚ version_infoÚstarargsÚkwargsr#r0r$)rr+ÚdecoÚcommar{Úar r r Ú _ClassDef3sJ                     zUnparser._ClassDefcCó| |d¡dS)NÚdef©Ú_Unparser__FunctionDef_helperrHr r r Ú _FunctionDefZr8zUnparser._FunctionDefcCr°)Nz async defr²rHr r r Ú_AsyncFunctionDef]r8zUnparser._AsyncFunctionDefcCsž| d¡|jD] }| d¡| |¡q|d|jd}| |¡| |j¡| d¡t|ddƒr?| d¡| |j¡| ¡| |j ¡|  ¡dS) NrržrYr;r<ÚreturnsFz -> ) rr¥rrrLÚargsr(r¶r#r0r$)rr+Z fill_suffixr¬Zdef_strr r r Z__FunctionDef_helper`s            zUnparser.__FunctionDef_helpercCó| d|¡dS)Nzfor ©Ú_Unparser__For_helperrHr r r Ú_Forpr8z Unparser._ForcCr¸)Nz async for r¹rHr r r Ú _AsyncForsr8zUnparser._AsyncForcCs|| |¡| |j¡| d¡| |j¡| ¡| |j¡| ¡|jr<| d¡| ¡| |j¡| ¡dSdS)Nrrr) rrr=rrr#r0r$r“)rrr+r r r Z __For_helpervs        üzUnparser.__For_helpercCsî| d¡| |j¡| ¡| |j¡| ¡|jr]t|jƒdkr]t|jdt j ƒr]|jd}| d¡| |j¡| ¡| |j¡| ¡|jr]t|jƒdkr]t|jdt j ƒs,|jru| d¡| ¡| |j¡| ¡dSdS)Nzif rrzelif r) rrror#r0r$r“r™r&r_ZIfrHr r r Ú_If„s.   ÿ    ùÿ   üz Unparser._IfcCsf| d¡| |j¡| ¡| |j¡| ¡|jr1| d¡| ¡| |j¡| ¡dSdS)Nzwhile r)rrror#r0r$r“rHr r r Ú_Whilešs      üzUnparser._WhileFcs~ˆ |rdnd¡t|dƒrt‡fdd„ˆj|jƒnˆ |j¡|jr/ˆ d¡ˆ |j¡ˆ ¡ˆ |j ¡ˆ  ¡dS)Nz async with zwith ÚitemscrArBrDr r"r r rE©rFz(Unparser._generic_With..rœ) rÚhasattrrrr¿Ú context_exprÚ optional_varsrr#r0r$)rr+Úasync_r r"r Ú _generic_With¦s      zUnparser._generic_WithcCs| |¡dSr.©rÄrHr r r Ú_With³rizUnparser._WithcCs|j|dddS)NT)rÃrÅrHr r r Ú _AsyncWith¶szUnparser._AsyncWithcCó| t|jƒ¡dSr.)rÚreprÚsrHr r r Ú_BytesºózUnparser._BytescCsŠtjr | t|jƒ¡dSd|jvr| t|jƒ¡dSt|jtƒr.| dt|jƒ¡dSt|jtƒrA| t|jƒ  d¡¡dSJdƒ‚)NrÚbÚuFzshouldn't get here) r r‰rrÉrÊrr&ÚstrZunicodeÚlstripr6r r r Ú_Str½s   z Unparser._StrcCs‚| d¡tƒ}| ||j¡| ¡}d|vsd|vr ddg}ngd¢}|D]}||vr5dj||d}nq&t|ƒ}| |¡dS) Nr rú ú'''ú""")ú'ú"rÔrÓz{quote_type}{v}{quote_type})Ú quote_typeÚv)rrÚ_fstring_JoinedStrÚgetvalueÚformatrÉ)rr+ÚstringrØZ quote_typesr×r r r Ú _JoinedStrÍs  þzUnparser._JoinedStrcCs4| d¡tƒ}| ||j¡| t| ¡ƒ¡dS)Nr )rrrÙrÉrÚ)rr+rÜr r r Ú_FormattedValueás zUnparser._FormattedValuecCs.|jD]}t|dt|ƒjƒ}|||ƒqdS)NÚ _fstring_)ryr(rŒr*)rr+rr9r,r r r rÙès  þzUnparser._fstring_JoinedStrcCs"|j dd¡ dd¡}||ƒdS©NÚ{z{{Ú}z}})rÊÚreplace©rr+rr9r r r Ú _fstring_Strís zUnparser._fstring_StrcCs2t|jtƒsJ‚|j dd¡ dd¡}||ƒdSrà)r&r9rÏrãrär r r Ú_fstring_Constantñs zUnparser._fstring_ConstantcCs®|dƒtƒ}t|j|ƒ| ¡ d¡}| d¡r|dƒ||ƒ|jdkr9t|jƒ}|dvs1J‚|dj|dƒ|j rQ|dƒt |d t |j ƒj ƒ}||j |ƒ|d ƒdS) NrárrYéÿÿÿÿZsraz !{conversion})Ú conversionr!rßrâ) rrr9rÚÚrstripÚ startswithrèÚchrrÛÚ format_specr(rŒr*)rr+rÚexprrèr,r r r Ú_fstring_FormattedValueös        z Unparser._fstring_FormattedValuecCr5r.)rÚidrHr r r Ú_Namer8zUnparser._NamecCrÈr.)rrÉr9rHr r r Ú _NameConstant rÌzUnparser._NameConstantcCs$| d¡| |j¡| d¡dS)Nú`©rrr9rHr r r Ú_Reprs  zUnparser._ReprcCs:t|ttfƒr| t|ƒ dt¡¡dS| t|ƒ¡dS)NÚinf)r&ÚfloatÚcomplexrrÉrãÚINFSTR)rr9r r r Ú_write_constantszUnparser._write_constantcsž|j}t|tƒr2ˆ d¡t|ƒdkr ˆ |d¡ˆ d¡n t‡fdd„ˆj|ƒˆ d¡dS|tur=ˆ d¡dS|jd krGˆ d ¡ˆ |j¡dS) Nr;rrrwcrArBrDr r"r r rE"rFz$Unparser._Constant..r<ú...rÎ) r9r&Útuplerr™rùrÚEllipsisÚkind)rr+r9r r"r Ú _Constants      zUnparser._ConstantcCs†t|jƒ}tjr| | dt¡¡dS| d¡r| d¡d|vr,| d¡r,| dd¡}| | dt¡¡| d¡rA| d¡dSdS)Nrõú-r;z*jÚjr<) rÉrOr r‰rrãrørêÚendswith)rr+Zrepr_nr r r Ú_Num+s     ÿz Unparser._Numcs0ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)Nú[crArBrDr r"r r rE<rFz Unparser._List..ú])rrrÚeltsrHr r"r Ú_List:s zUnparser._ListcCó:| d¡| |j¡|jD]}| |¡q| d¡dS©Nrr©rrÚeltÚ generators©rr+Úgenr r r Ú _ListComp?r?zUnparser._ListCompcCr)Nr;r<r r r r r Ú _GeneratorExpFr?zUnparser._GeneratorExpcCr)Nrárâr r r r r Ú_SetCompMr?zUnparser._SetCompcCsP| d¡| |j¡| d¡| |j¡|jD]}| |¡q| d¡dS)Nrár]râ)rrÚkeyr9r r r r r Ú _DictCompTó      zUnparser._DictCompcCsht|ddƒr | d¡n| d¡| |j¡| d¡| |j¡|jD] }| d¡| |¡q%dS)NÚis_asyncFz async for z for rrú if )r(rrr=rÚifs)rr+Z if_clauser r r Ú_comprehension]s         þzUnparser._comprehensioncCsP| d¡| |j¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr;rz else r<)rrr0ror“rHr r r Ú_IfExpirzUnparser._IfExpcs:|jsJ‚ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)NrácrArBrDr r"r r rEurFzUnparser._Set..râ)rrrrrHr r"r Ú_Setrs  z Unparser._SetcsPˆ d¡‡fdd„‰‡‡fdd„}t‡fdd„|t|j|jƒƒˆ d¡dS) Nrács"ˆ |¡ˆ d¡ˆ |¡dS©Nr])rr)ÚkrØr"r r Úwrite_key_value_pairzs  z,Unparser._Dict..write_key_value_paircs6|\}}|durˆ d¡ˆ |¡dSˆ||ƒdS)Nr¤)rr)ÚitemrrØ©rrr r Ú write_items  z"Unparser._Dict..write_itemcrArBrDr r"r r rEˆrFz Unparser._Dict..râ)rrÚzipÚkeysry)rr+rr rr Ú_Dictxs   zUnparser._Dictcs^ˆ d¡t|jƒdkr|jd}ˆ |¡ˆ d¡n t‡fdd„ˆj|jƒˆ d¡dS)Nr;rrrwcrArBrDr r"r r rE’rFz!Unparser._Tuple..r<)rr™rrr)rr+r r r"r Ú_Tuple‹s    zUnparser._Tupleú~Únotú+rÿ)ZInvertZNotZUAddÚUSubcCsˆ| d¡| |j|jjj¡| d¡tjr7t|jtj ƒr7t|j tj ƒr7| d¡|  |j ¡| d¡n|  |j ¡| d¡dS©Nr;rYr<) rÚunopr[r)r*r ZPY2r&r_r'ÚoperandÚNumrrHr r r Ú_UnaryOp–s  "    zUnparser._UnaryOpr£ržú/ú%z<z>=Úiszis notÚinznot in) ZEqZNotEqZLtZLtEZGtZGtEZIsZIsNotZInZNotIncCsb| d¡| |j¡t|j|jƒD]\}}| d|j|jjd¡| |¡q| d¡dSr() rrr2r ÚopsÚ comparatorsÚcmpopsr)r*)rr+Úor{r r r Ú_Compare³s   zUnparser._CompareÚandÚorcsDˆ d¡dˆj|jj‰t‡‡fdd„ˆj|jƒˆ d¡dS)Nr;z %s cs ˆ ˆ¡Sr.rDr ©rÊrr r rE¿rFz"Unparser._BoolOp..r<)rÚboolopsr[r)rrryrHr r@r Ú_BoolOp¼s zUnparser._BoolOpc CsZ| |j¡t|jttdttddƒƒƒr t|jjtƒr | d¡| d¡| |j¡dS)NZConstantr+rYrQ) rr9r&r(r_rOÚintrÚattrrHr r r Ú _AttributeÂs *  zUnparser._AttributecCsî| |j¡| d¡d}|jD]}|r| d¡nd}| |¡q|jD]}|r/| d¡nd}| |¡q%tjdd…dkrp|jrX|rK| d¡nd}| d¡| |j¡|jrp|rc| d¡nd}| d¡| |j¡| d ¡dS) Nr;FrCTrŸr r£r¤r<) rÚfuncrr·r§r¨r©rªr«)rr+r­r{r r r Ú_CallÌs.          zUnparser._CallcCs0| |j¡| d¡| |j¡| d¡dSr)rr9rÚslicerHr r r Ú _Subscriptås   zUnparser._SubscriptcCs| d¡| |j¡dS)Nr£rórHr r r Ú_Starredës zUnparser._StarredcCre)NrúrDrHr r r Ú _EllipsisðrizUnparser._EllipsiscCr5r.)rr9rHr r r Ú_Indexór8zUnparser._IndexcCsR|jr | |j¡| d¡|jr| |j¡|jr'| d¡| |j¡dSdS)Nr!)ÚlowerrrÚupperÚsteprHr r r Ú_Sliceös    þzUnparser._Slicecst‡fdd„ˆj|jƒdS)NcrArBrDr r"r r rErFz$Unparser._ExtSlice..)rrZdimsrHr r"r Ú _ExtSliceszUnparser._ExtSlicecCs0| |j¡|jr| d¡| |j¡dSdSr)rÚargrarrHr r r Ú_argó  þz Unparser._argc CsLd}t|dgƒ|j}dgt|ƒt|jƒ|j}tt||ƒdƒD]2\}}|\}}|r/d}n| d¡| |¡|rE| d¡| |¡|tt|ddƒƒkrT| d¡q"|js^t|d dƒr¥|rcd}n| d¡| d ¡|jr¥t |jd ƒrŽ| |jj ¡|jj r| d ¡| |jj ¡n| |j¡t|d dƒr¥| d ¡| |j ¡t|d dƒrÓt|j |jƒD] \}}|r»d}n| d¡| |¡f|rÒ| d¡| |¡q²|jr"|rÜd}n| d¡t |jd ƒr| d|jj ¡|jj r| d ¡| |jj ¡dSdS| d|j¡t|ddƒr$| d ¡| |j¡dSdSdS)NTÚ posonlyargsrFrCú=r z, /Ú kwonlyargsr£rRr]Úvarargannotationr¤Úkwargannotation)r(r·r™ÚdefaultsÚ enumerater rrÚvarargrÀrRrarXrWÚ kw_defaultsÚkwargrY) rr+ÚfirstZall_argsrZÚindexÚelementsr®Údr r r Ú _arguments sj      €     €          €    þ ô zUnparser._argumentscCs<|jdur | d¡n | |j¡| d¡| |j¡dS)Nr¤rV)rRrrr9rHr r r Ú_keywordEs    zUnparser._keywordcCsD| d¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr;zlambda r]r<)rrr·r0rHr r r Ú_LambdaNs     zUnparser._LambdacCs*| |j¡|jr| d|j¡dSdS©Nrœ)rrLÚasnamerHr r r Ú_aliasVs ÿzUnparser._aliascCs0| |j¡|jr| d¡| |j¡dSdSrf)rrÁrÂrrHr r r Ú _withitem[rTzUnparser._withitemN)r)F)er*Ú __module__Ú __qualname__Ú__doc__r¨Ústdoutrrrr#r$rr2r4r7r:r>rIrUrXr\rbrdrhrkrmrnrqrur|r~rrƒr†r‡rŽr—r˜ršrr¯r´rµr³r»r¼rºr½r¾rÄrÆrÇrËrÑrÝrÞrÙrårærîrðrñrôrùrþrrrrrrrrrr"r#r)r,rZr4r;r=r_ZAndZOrrArBrErGrIrJrKrLrPrQrSrcrdrerhrir r r r rsÆ    '        þ ÿ  :  rcCsÌtjrqsztestdir..zDirectory not readable: %srz Testing %sz$ Failed to compile, exception is %s)ÚosÚlistdirÚOSErrorrr¨ÚstderrÚpathÚjoinÚisfilerr|Ú ExceptionrÉÚisdirÚtestdir)r®rGrOÚfullnamerzr{r r r r‡os( ÿ  €ÿ €ör‡cCs@|ddkr|dd…D]}t|ƒq dS|D]}t|ƒqdS)Nrz --testdirr)r‡r|)r·r®r r r Úmains  ÿ ÿr‰Ú__main__)rlrKrrr r¨r_r~rtrrÉÚ float_infoÚ max_10_exprørrrmr|r‡r‰r*Úargvr r r r Ús.  Lÿ