o e"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ÚsixÚ text_typerr r rr.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 rr;s   zUnparser.dispatchcCó|jD]}| |¡qdS©N©Úbodyr©rrÚstmtr r rÚ_ModuleLó  ÿzUnparser._ModulecCr/r0r1r3r r rÚ _InteractivePr6zUnparser._InteractivecCó| |j¡dSr0)rr2©rrr r rÚ _ExpressionTózUnparser._ExpressioncCs| ¡| |j¡dSr0)r rÚvaluer9r r rÚ_ExprXszUnparser._ExprcCs:| d¡| |j¡| d¡| |j¡| d¡dS)Nú(z := ú))rrÚtargetr<r9r r rÚ _NamedExpr\ó    zUnparser._NamedExprcó&ˆ d¡t‡fdd„ˆj|jƒdS)Nzimport có ˆ d¡S©Nú, ©rr r$r rÚeó z"Unparser._Import..)r rrÚ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|]}|jVqdSr0)Úname©Ú.0Únr r rÚ js€z'Unparser._ImportFrom..zfrom Ú.z import crDrErGr r$r rrHqrIz&Unparser._ImportFrom..) ÚmodulerÚextendrJr rÚlevelrrrKr r$rÚ _ImportFromgs   zUnparser._ImportFromcCs8| ¡|jD] }| |¡| d¡q| |j¡dS)Nú = )r Útargetsrrr<)rr-r@r r rÚ_Assignss    zUnparser._AssigncCsB| ¡| |j¡| d|j|jjjd¡| |j¡dS)Nú z= ) r rr@rÚbinopÚopr+r,r<rKr r rÚ _AugAssignzs zUnparser._AugAssigncCsŠ| ¡|jst|jtjƒr| d¡| |j¡|js(t|jtjƒr(| d¡| d¡| |j¡|j rC| d¡| |j ¡dSdS)Nr>r?ú: rY) r Úsimpler(r@ÚastÚNamerrÚ annotationr<rKr r rÚ _AnnAssign€s      þzUnparser._AnnAssigncCs.| d¡|jr| d¡| |j¡dSdS)NÚreturnr\)r r<rrrKr r rÚ_Returns  þzUnparser._ReturncCó| d¡dS)NÚpass©r rKr r rÚ_Pass“ózUnparser._PasscCrh)NÚbreakrjrKr r rÚ_Break–rlzUnparser._BreakcCrh)NÚcontinuerjrKr r rÚ _Continue™rlzUnparser._ContinuecrC)Nzdel crDrErGr r$r rrHžrIz"Unparser._Delete..)r rrrZrKr r$rÚ_DeleteœrMzUnparser._DeletecCs:| d¡| |j¡|jr| d¡| |j¡dSdS)Nzassert rF)r rÚtestÚmsgrrKr r rÚ_Assert s   þzUnparser._AssertcCsV| d¡| |j¡|jr| d¡| |j¡|jr)| d¡| |j¡dSdS)Nzexec ú in rF)r rr2ÚglobalsrÚlocalsrKr r rÚ_Exec§s     þzUnparser._ExeccCsp| d¡d}|jr| d¡| |j¡d}|jD]}|r$| d¡nd}| |¡q|js6| d¡dSdS)Nzprint Fú>>TrFú,)r ÚdestrrÚvaluesÚnl)rr-Zdo_commaÚer r rÚ_Print±s     ÿzUnparser._PrintcrC)Nzglobal crDrErGr r$r rrHÁrIz"Unparser._Global..©r rrrJrKr r$rÚ_Global¿rMzUnparser._GlobalcrC)Nz nonlocal crDrErGr r$r rrHÅrIz$Unparser._Nonlocal..r€rKr r$rÚ _NonlocalÃrMzUnparser._NonlocalcCó>| d¡| d¡|jr| d¡| |j¡| d¡dS)Nr>Úawaitr\r?©rr<rrKr r rÚ_AwaitÇó    zUnparser._AwaitcCrƒ)Nr>Úyieldr\r?r…rKr r rÚ_YieldÏr‡zUnparser._YieldcCrƒ)Nr>z yield fromr\r?r…rKr 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Úraiser\z from rF) r r!ÚPY3ÚexcÚcauserrÚtypeÚinstZtbackrKr r rÚ_Raiseßs*     þ     þzUnparser._RaisecCsœ| d¡| ¡| |j¡| ¡|jD]}| |¡q|jr4| d¡| ¡| |j¡| ¡|jrL| d¡| ¡| |j¡| ¡dSdS)NÚtryÚelseÚfinally)r r%rr2r&ÚhandlersÚorelseÚ finalbody©rr-Úexr r rÚ_Tryõs"         üz Unparser._TrycCsp| d¡| ¡| |j¡| ¡|jD]}| |¡q|jr6| d¡| ¡| |j¡| ¡dSdS)Nr’r“)r r%rr2r&r•r–r˜r r rÚ _TryExcepts       üzUnparser._TryExceptcCs~t|jƒdkrt|jdtjƒr| |j¡n| d¡| ¡| |j¡| ¡| d¡| ¡| |j ¡| ¡dS)Nrrr’r”) Úlenr2r(rbZ TryExceptrr r%r&r—rKr r rÚ _TryFinallys      zUnparser._TryFinallycCsv| d¡|jr| d¡| |j¡|jr+| d¡tjr%| |j¡n| |j¡| ¡| |j¡|  ¡dS)NÚexceptr\ú as ) r rrrrOr!rŒr%r2r&rKr 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>FrFTé©ééÚ*ú**r?)rÚdecorator_listr rrOr!rŒÚbasesÚkeywordsÚsysÚ version_infoÚstarargsÚkwargsr%r2r&)rr-ÚdecoÚcommar~Úar r rÚ _ClassDef3sJ                     zUnparser._ClassDefcCó| |d¡dS)NÚdef©Ú_Unparser__FunctionDef_helperrKr r rÚ _FunctionDefZr;zUnparser._FunctionDefcCr³)Nz async defrµrKr r rÚ_AsyncFunctionDef]r;zUnparser._AsyncFunctionDefcCsž| d¡|jD] }| d¡| |¡q|d|jd}| |¡| |j¡| d¡t|ddƒr?| d¡| |j¡| ¡| |j ¡|  ¡dS) Nrr¡r\r>r?ÚreturnsFz -> ) rr¨r rrOÚargsr*r¹r%r2r&)rr-Ú fill_suffixr¯Údef_strr r rZ__FunctionDef_helper`s            zUnparser.__FunctionDef_helpercCó| d|¡dS)Nzfor ©Ú_Unparser__For_helperrKr r rÚ_Forpr;z Unparser._ForcCr½)Nz async for r¾rKr r rÚ _AsyncForsr;zUnparser._AsyncForcCs|| |¡| |j¡| d¡| |j¡| ¡| |j¡| ¡|jr<| d¡| ¡| |j¡| ¡dSdS)Nrur“) r rr@rrr%r2r&r–)rr r-r r rZ __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“) r rrrr%r2r&r–rœr(rbÚIfrKr r rÚ_If„s.   ÿ    ùÿ   üz Unparser._IfcCsf| d¡| |j¡| ¡| |j¡| ¡|jr1| d¡| ¡| |j¡| ¡dSdS)Nzwhile r“)r rrrr%r2r&r–rKr r rÚ_Whilešs      üzUnparser._WhileFcs~ˆ |rdnd¡t|dƒrt‡fdd„ˆj|jƒnˆ |j¡|jr/ˆ d¡ˆ |j¡ˆ ¡ˆ |j ¡ˆ  ¡dS)Nz async with zwith ÚitemscrDrErGr r$r rrH©rIz(Unparser._generic_With..rŸ) r ÚhasattrrrrÅÚ context_exprÚ optional_varsrr%r2r&)rr-Úasync_r r$rÚ _generic_With¦s      zUnparser._generic_WithcCs| |¡dSr0©rÊrKr r rÚ_With³rlzUnparser._WithcCs|j|dddS)NT)rÉrËrKr r rÚ _AsyncWith¶szUnparser._AsyncWithcCó| t|jƒ¡dSr0)rÚreprÚsrKr 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(ÚstrÚunicodeÚlstripr9r 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ßÚ 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_)r|r*rr,)rr-rr<r.r r rràès  þzUnparser._fstring_JoinedStrcCs"|j dd¡ dd¡}||ƒdS©NÚ{z{{Ú}z}})rÐÚreplace©rr-rr<r r rÚ _fstring_Strís zUnparser._fstring_StrcCs2t|jtƒsJ‚|j dd¡ dd¡}||ƒdSrè)r(r<rÕ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érr\éÿÿÿÿÚsraz !{conversion})Ú conversionr#rçrê) rrr<ráÚrstripÚ startswithrñÚchrrâÚ format_specr*rr,)rr-rÚexprrñr.r r rÚ_fstring_FormattedValueös        z Unparser._fstring_FormattedValuecCr8r0)rÚidrKr r rÚ_Namer;zUnparser._NamecCrÎr0)rrÏr<rKr r rÚ _NameConstant rÒzUnparser._NameConstantcCs$| d¡| |j¡| d¡dS)Nú`©rrr<rKr r rÚ_Reprs  zUnparser._ReprcCs:t|ttfƒr| t|ƒ dt¡¡dS| t|ƒ¡dS)NÚinf)r(ÚfloatÚcomplexrrÏrëÚINFSTR)rr<r 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>rrrzcrDrErGr r$r rrH"rIz$Unparser._Constant..r?ú...rÔ) r<r(ÚtuplerrœrrÚEllipsisÚkind)rr-r<r 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ÏrRr!rŒrrërróÚendswith)rr-Zrepr_nr r rÚ_Num+s     ÿz Unparser._Numcs0ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)Nú[crDrErGr r$r rrH<rIz Unparser._List..ú])rrrÚeltsrKr r$rÚ_List:s zUnparser._ListcCó:| d¡| |j¡|jD]}| |¡q| d¡dS©Nr r ©rrÚeltÚ generators©rr-Úgenr r rÚ _ListComp?rBzUnparser._ListCompcCr)Nr>r?rrr r rÚ _GeneratorExpFrBzUnparser._GeneratorExpcCr)Nrérêrrr r rÚ_SetCompMrBzUnparser._SetCompcCsP| d¡| |j¡| d¡| |j¡|jD]}| |¡q| d¡dS)Nrér`rê)rrÚkeyr<rrr r rÚ _DictCompTó      zUnparser._DictCompcCsht|ddƒr | d¡n| d¡| |j¡| d¡| |j¡|jD] }| d¡| |¡q%dS)NÚis_asyncFz async for z for ruú if )r*rrr@rÚifs)rr-Ú if_clauser r rÚ_comprehension]s         þzUnparser._comprehensioncCsP| d¡| |j¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr>rz else r?)rrr2rrr–rKr r rÚ_IfExpirzUnparser._IfExpcs:|jsJ‚ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)NrécrDrErGr r$r rrHurIzUnparser._Set..rê)rrrrrKr 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)Úitemr%rß©rr&r rÚ write_items  z"Unparser._Dict..write_itemcrDrErGr r$r rrHˆrIz Unparser._Dict..rê)rrÚzipÚkeysr|)rr-r)r r(rÚ_Dictxs   zUnparser._Dictcs^ˆ d¡t|jƒdkr|jd}ˆ |¡ˆ d¡n t‡fdd„ˆj|jƒˆ d¡dS)Nr>rrrzcrDrErGr r$r rrH’rIz!Unparser._Tuple..r?)rrœrrr)rr-rr r$rÚ_Tuple‹s    zUnparser._Tupleú~Únotú+r)ÚInvertÚNotÚUAddÚUSubcCsˆ| d¡| |j|jjj¡| d¡tjr7t|jtj ƒr7t|j tj ƒr7| d¡|  |j ¡| d¡n|  |j ¡| d¡dS©Nr>r\r?) rÚunopr^r+r,r!ÚPY2r(rbr4ÚoperandÚNumrrKr r rÚ_UnaryOp–s  "    zUnparser._UnaryOpr¦r¡ú/ú%z<z>=Úiszis notÚinznot in) ÚEqÚNotEqÚLtÚLtEÚGtÚGtEÚIsÚIsNotÚInÚNotIncCsb| d¡| |j¡t|j|jƒD]\}}| d|j|jjd¡| |¡q| d¡dSr5) rrrMr*Ú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 ˆ ˆ¡Sr0rGr ©rÐrr rrH¿rIz"Unparser._BoolOp..r?)rÚboolopsr^r+rrr|rKr rerÚ_BoolOp¼s zUnparser._BoolOpc CsZ| |j¡t|jttdttddƒƒƒr t|jjtƒr | d¡| d¡| |j¡dS)NÚConstantr9r\rT) rr<r(r*rbrRÚintrÚattrrKr 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>FrFTr¢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)rr<rÚslicerKr r rÚ _Subscriptås   zUnparser._SubscriptcCs| d¡| |j¡dS)Nr¦rürKr r rÚ_Starredës zUnparser._StarredcCrh)NrrGrKr r rÚ _EllipsisðrlzUnparser._EllipsiscCr8r0)rr<rKr r rÚ_Indexór;zUnparser._IndexcCsR|jr | |j¡| d¡|jr| |j¡|jr'| d¡| |j¡dSdS)Nr#)ÚlowerrrÚupperÚsteprKr r rÚ_Sliceös    þzUnparser._Slicecst‡fdd„ˆj|jƒdS)NcrDrErGr r$r rrHrIz$Unparser._ExtSlice..)rrÚdimsrKr r$rÚ _ExtSliceszUnparser._ExtSlicecCs0| |j¡|jr| d¡| |j¡dSdSr$)rÚargrdrrKr 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Ú posonlyargsrFrFú=r z, /Ú kwonlyargsr¦ryr`Úvarargannotationr§Úkwargannotation)r*rºrœÚdefaultsÚ enumerater*rrÚvarargrÆryrdrr~Ú kw_defaultsÚkwargr€) rr-ÚfirstÚall_argsrÚindexÚelementsr±Údr r rÚ _arguments sj      €     €          €    þ ô zUnparser._argumentscCs<|jdur | d¡n | |j¡| d¡| |j¡dS)Nr§r})ryrrr<rKr r rÚ_keywordEs    zUnparser._keywordcCsD| d¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr>zlambda r`r?)rrrºr2rKr r rÚ_LambdaNs     zUnparser._LambdacCs*| |j¡|jr| d|j¡dSdS©NrŸ)rrOÚasnamerKr r rÚ_aliasVs ÿzUnparser._aliascCs0| |j¡|jr| d¡| |j¡dSdSrŽ)rrÇrÈrrKr r rÚ _withitem[r{zUnparser._withitemN)r)F)er,Ú __module__Ú __qualname__Ú__doc__r«Ústdoutrr rr%r&rr5r7r:r=rArLrXr[r_rergrkrnrprqrtrxrrr‚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ýrrr rrrrrr!r"r#r,r-r6r:r]rOr`rbrbÚAndÚOrrfrgrkrmrorprqrrrvrxrzr‹rŒrrr‘r r r rrsÆ    '        þ ÿ  :  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±rJrRÚfullnamer¥r~r r rr²os( ÿ  €ÿ €ör²cCs@|ddkr|dd…D]}t|ƒq dS|D]}t|ƒqdS)Nrz --testdirr)r²r§)rºr±r r rÚmains  ÿ ÿr´Ú__main__)r”rNrrr!r«rbr©ržrrÏÚ float_infoÚ max_10_exprrrr•r§r²r´r,Úargvr r r rÚs.  Lÿ