B 18™\ªNã@s’dZddlZddlZddlZddlZddlZdeejjdƒZ dd„Z Gdd„dƒZ ej fd d „Z d d „Zd d„ZedkrŽeejdd…ƒdS)z'Usage: unparse.py éNZ1eécCsNt|ƒ}y|t|ƒƒWntk r,YnXx|D]}|ƒ||ƒq4WdS)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„Z0dYdZ„Z1d[d\„Z2d]d^„Z3d_d`„Z4dadb„Z5dcdd„Z6dedf„Z7dgdh„Z8didj„Z9dkdl„Z:dmdn„Z;dodp„Zdudv„Z?dwdx„Z@dydz„ZAd{d|„ZBd}d~dd€dœZCd‚dƒ„ZDdd€d„d…d†d‡dˆd‰dŠd‹dŒddŽdœ ZEdd‘„ZFd’d“d”d•d–d—d˜d™dšd›dœœ ZGddž„ZHeIjJdŸeIjKd iZLd¡d¢„ZMd£d¤„ZNd¥d¦„ZOd§d¨„ZPd©dª„ZQd«d¬„ZRd­d®„ZSd¯d°„ZTd±d²„ZUd³d´„ZVdµd¶„ZWd·d¸„ZXd¹dº„ZYd»d¼„ZZd½d¾„Z[d¿S)ÁÚUnparserzŽMethods in this class recursively traverse an AST and output source code for the abstract syntax; original formatting is disregarded. cCs2||_d|_| |¡td|jd|j ¡dS)zTUnparser(tree, file=sys.stdout) -> None. Print the source for tree to file.rÚ)ÚfileN)rÚ_indentÚdispatchÚprintÚflush)ÚselfÚtreerr r r Ú__init__s  zUnparser.__init__r cCs|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 |¡dS)z+Append a piece of text to the current line.N)rr)rrr r r r+szUnparser.writecCs| d¡|jd7_dS)z(Print ':', and increase the indentation.ú:rN)rr)rr r r Úenter/s zUnparser.entercCs|jd8_dS)zDecrease the indentation level.rN)r)rr r r Úleave4szUnparser.leavecCsDt|tƒr&x|D]}| |¡qWdSt|d|jjƒ}||ƒdS)z:Dispatcher function, dispatching tree type T to method _T.NÚ_)Ú isinstanceÚlistrÚgetattrÚ __class__Ú__name__)rrÚtÚmethr r r r8s   zUnparser.dispatchcCsx|jD]}| |¡qWdS)N)Úbodyr)rrÚstmtr r r Ú_ModuleIs zUnparser._ModulecCs| ¡| |j¡dS)N)rrÚvalue)rrr r r Ú_ExprNszUnparser._Exprcs&ˆ d¡t‡fdd„ˆj|jƒdS)Nzimport cs ˆ d¡S)Nz, )rr )rr r ÚTóz"Unparser._Import..)rr rÚnames)rr#r )rr Ú_ImportRs zUnparser._ImportcsRˆ d¡ˆ d|j¡|jr,ˆ |j¡ˆ d¡t‡fdd„ˆj|jƒdS)Nzfrom Ú.z import cs ˆ d¡S)Nz, )rr )rr r r*\r+z&Unparser._ImportFrom..)rrÚlevelÚmoduler rr,)rr#r )rr Ú _ImportFromVs    zUnparser._ImportFromcCs<| ¡x"|jD]}| |¡| d¡qW| |j¡dS)Nz = )rÚtargetsrrr()rr#Útargetr r r Ú_Assign^s   zUnparser._AssigncCsB| ¡| |j¡| d|j|jjjd¡| |j¡dS)Nú z= ) rrr3rÚbinopÚopr!r"r()rr#r r r Ú _AugAssignes zUnparser._AugAssigncCs†| ¡|js&t|jtjƒr&| d¡| |j¡|jsPt|jtjƒrP| d¡| d¡| |j¡|j r‚| d¡| |j ¡dS)Nú(ú)z: z = ) rZsimplerr3ÚastÚNamerrÚ annotationr()rr#r r r Ú _AnnAssignks      zUnparser._AnnAssigncCs*| d¡|jr&| d¡| |j¡dS)NÚreturnr5)rr(rr)rr#r r r Ú_Returnxs  zUnparser._ReturncCs| d¡dS)NÚpass)r)rr#r r r Ú_Pass~szUnparser._PasscCs| d¡dS)NÚbreak)r)rr#r r r Ú_BreakszUnparser._BreakcCs| d¡dS)NÚcontinue)r)rr#r r r Ú _Continue„szUnparser._Continuecs&ˆ d¡t‡fdd„ˆj|jƒdS)Nzdel cs ˆ d¡S)Nz, )rr )rr r r*‰r+z"Unparser._Delete..)rr rr2)rr#r )rr Ú_Delete‡s zUnparser._DeletecCs6| d¡| |j¡|jr2| d¡| |j¡dS)Nzassert z, )rrÚtestÚmsgr)rr#r r r Ú_Assert‹s    zUnparser._Assertcs&ˆ d¡t‡fdd„ˆj|jƒdS)Nzglobal cs ˆ d¡S)Nz, )rr )rr r r*”r+z"Unparser._Global..)rr rr,)rr#r )rr Ú_Global’s zUnparser._Globalcs&ˆ d¡t‡fdd„ˆj|jƒdS)Nz nonlocal cs ˆ d¡S)Nz, )rr )rr r r*˜r+z$Unparser._Nonlocal..)rr rr,)rr#r )rr Ú _Nonlocal–s zUnparser._NonlocalcCs>| d¡| d¡|jr0| d¡| |j¡| d¡dS)Nr9Úawaitr5r:)rr(r)rr#r r r Ú_Awaitšs     zUnparser._AwaitcCs>| d¡| d¡|jr0| d¡| |j¡| d¡dS)Nr9Úyieldr5r:)rr(r)rr#r r r Ú_Yield¢s     zUnparser._YieldcCs>| d¡| d¡|jr0| d¡| |j¡| d¡dS)Nr9z yield fromr5r:)rr(r)rr#r r r Ú _YieldFromªs     zUnparser._YieldFromcCsT| d¡|js|jrt‚dS| d¡| |j¡|jrP| d¡| |j¡dS)NÚraiser5z from )rÚexcÚcauseÚAssertionErrorrr)rr#r r r Ú_Raise²s     zUnparser._RaisecCsœ| d¡| ¡| |j¡| ¡x|jD]}| |¡q.W|jrl| d¡| ¡| |j¡| ¡|jr˜| d¡| ¡| |j¡| ¡dS)NÚtryÚelseÚfinally)rrrr%rÚhandlersÚorelseZ finalbody)rr#Úexr r r Ú_Try½s        z Unparser._TrycCsb| d¡|jr&| d¡| |j¡|jrB| d¡| |j¡| ¡| |j¡| ¡dS)NÚexceptr5z as )rÚtyperrÚnamerr%r)rr#r r r Ú_ExceptHandlerÏs      zUnparser._ExceptHandlercCsÒ| d¡x"|jD]}| d¡| |¡qW| d|j¡| d¡d}x,|jD]"}|rh| d¡nd}| |¡qTWx,|jD]"}|r–| d¡nd}| |¡q‚W| d¡| ¡| |j¡|  ¡dS) Nrú@zclass r9Fz, Tr:) rÚdecorator_listrrr`ÚbasesÚkeywordsrr%r)rr#ÚdecoÚcommaÚer r r Ú _ClassDefÛs*          zUnparser._ClassDefcCs| |d¡dS)NÚdef)Ú_Unparser__FunctionDef_helper)rr#r r r Ú _FunctionDefñszUnparser._FunctionDefcCs| |d¡dS)Nz async def)rk)rr#r r r Ú_AsyncFunctionDefôszUnparser._AsyncFunctionDefcCsœ| d¡x"|jD]}| d¡| |¡qW|d|jd}| |¡| |j¡| d¡|jr|| d¡| |j¡| ¡| |j¡|  ¡dS)Nrrbr5r9r:z -> ) rrcrrr`ÚargsZreturnsrr%r)rr#Z fill_suffixrfZdef_strr r r Z__FunctionDef_helper÷s         zUnparser.__FunctionDef_helpercCs| d|¡dS)Nzfor )Ú_Unparser__For_helper)rr#r r r Ú_Forsz Unparser._ForcCs| d|¡dS)Nz async for )ro)rr#r r r Ú _AsyncFor szUnparser._AsyncForcCsx| |¡| |j¡| d¡| |j¡| ¡| |j¡| ¡|jrt| d¡| ¡| |j¡| ¡dS)Nz in rX) rrr3rrrr%rr[)rrr#r r r Z __For_helper s       zUnparser.__For_helpercCsÊ| d¡| |j¡| ¡| |j¡| ¡xf|jr˜t|jƒdkr˜t|jdt j ƒr˜|jd}| d¡| |j¡| ¡| |j¡| ¡q4W|jrÆ| d¡| ¡| |j¡| ¡dS)Nzif rrzelif rX) rrrHrr%rr[Úlenrr;ÚIf)rr#r r r Ú_Ifs$          z Unparser._IfcCsb| d¡| |j¡| ¡| |j¡| ¡|jr^| d¡| ¡| |j¡| ¡dS)Nzwhile rX)rrrHrr%rr[)rr#r r r Ú_While1s     zUnparser._WhilecsBˆ d¡t‡fdd„ˆj|jƒˆ ¡ˆ |j¡ˆ ¡dS)Nzwith cs ˆ d¡S)Nz, )rr )rr r r*?r+z Unparser._With..)rr rÚitemsrr%r)rr#r )rr Ú_With=s   zUnparser._WithcsBˆ d¡t‡fdd„ˆj|jƒˆ ¡ˆ |j¡ˆ ¡dS)Nz async with cs ˆ d¡S)Nz, )rr )rr r r*Fr+z%Unparser._AsyncWith..)rr rrvrr%r)rr#r )rr Ú _AsyncWithDs   zUnparser._AsyncWithcCs| t|jƒ¡dS)N)rÚreprÚs)rr#r r r Ú_BytesLszUnparser._BytescCs| t|jƒ¡dS)N)rryrz)rrr r r Ú_StrOsz Unparser._StrcCs6| d¡t ¡}| ||j¡| t| ¡ƒ¡dS)Nr)rÚioÚStringIOÚ_fstring_JoinedStrryÚgetvalue)rr#Ústringr r r Ú _JoinedStrRs zUnparser._JoinedStrcCs6| d¡t ¡}| ||j¡| t| ¡ƒ¡dS)Nr)rr}r~Ú_fstring_FormattedValueryr€)rr#rr r r Ú_FormattedValueXs zUnparser._FormattedValuecCs2x,|jD]"}t|dt|ƒjƒ}|||ƒqWdS)NÚ _fstring_)Úvaluesr r_r")rr#rr(r$r r r r^s zUnparser._fstring_JoinedStrcCs"|j dd¡ dd¡}||ƒdS)Nú{z{{ú}z}})rzÚreplace)rr#rr(r r r Ú _fstring_StrcszUnparser._fstring_StrcCs2t|jtƒst‚|j dd¡ dd¡}||ƒdS)Nr‡z{{rˆz}})rr(ÚstrrUr‰)rr#rr(r r r Ú_fstring_ConstantgszUnparser._fstring_ConstantcCs®|dƒt ¡}t|j|ƒ| ¡ d¡}| d¡r<|dƒ||ƒ|jdkrrt|jƒ}|dksdt ‚|d|›ƒ|j r¢|dƒt |dt |j ƒj ƒ}||j |ƒ|d ƒdS) Nr‡rr5éÿÿÿÿZsraú!rr…rˆ)r}r~r r(r€ÚrstripÚ startswithÚ conversionÚchrrUÚ format_specr r_r")rr#rÚexprr‘r$r r r rƒls       z Unparser._fstring_FormattedValuecCs| |j¡dS)N)rÚid)rr#r r r Ú_Name~szUnparser._NamecCs8t|ttfƒr&| t|ƒ dt¡¡n| t|ƒ¡dS)NÚinf)rÚfloatÚcomplexrryr‰ÚINFSTR)rr(r r r Ú_write_constantszUnparser._write_constantcsr|j}t|tƒrbˆ d¡t|ƒdkr@ˆ |d¡ˆ d¡nt‡fdd„ˆj|ƒˆ d¡n ˆ |j¡dS)Nr9rrú,cs ˆ d¡S)Nz, )rr )rr r r*r+z$Unparser._Constant..r:)r(rÚtuplerrrr›r )rr#r(r )rr Ú _Constant‡s     zUnparser._ConstantcCs| t|jƒ¡dS)N)rryr()rr#r r r Ú _NameConstant”szUnparser._NameConstantcCs| t|jƒ dt¡¡dS)Nr—)rryÚnr‰rš)rr#r r r Ú_Num—sz Unparser._Numcs0ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)Nú[cs ˆ d¡S)Nz, )rr )rr r r*r+z Unparser._List..ú])rr rÚelts)rr#r )rr Ú_List›s zUnparser._ListcCs>| d¡| |j¡x|jD]}| |¡qW| d¡dS)Nr¢r£)rrÚeltÚ generators)rr#Úgenr r r Ú _ListComp s    zUnparser._ListCompcCs>| d¡| |j¡x|jD]}| |¡qW| d¡dS)Nr9r:)rrr¦r§)rr#r¨r r r Ú _GeneratorExp§s    zUnparser._GeneratorExpcCs>| d¡| |j¡x|jD]}| |¡qW| d¡dS)Nr‡rˆ)rrr¦r§)rr#r¨r r r Ú_SetComp®s    zUnparser._SetCompcCsT| d¡| |j¡| d¡| |j¡x|jD]}| |¡q4W| d¡dS)Nr‡z: rˆ)rrÚkeyr(r§)rr#r¨r r r Ú _DictCompµs     zUnparser._DictCompcCsf|jr| d¡n | d¡| |j¡| d¡| |j¡x"|jD]}| d¡| |¡qFWdS)Nz async for z for z in z if )Zis_asyncrrr3rZifs)rr#Z if_clauser r r Ú_comprehension¾s       zUnparser._comprehensioncCsP| d¡| |j¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr9z if z else r:)rrr%rHr[)rr#r r r Ú_IfExpÊs      zUnparser._IfExpcs:|js t‚ˆ d¡t‡fdd„ˆj|jƒˆ d¡dS)Nr‡cs ˆ d¡S)Nz, )rr )rr r r*Ör+zUnparser._Set..rˆ)r¤rUrr r)rr#r )rr Ú_SetÓs  z Unparser._SetcsPˆ d¡‡fdd„‰‡‡fdd„}t‡fdd„|t|j|jƒƒˆ d¡dS) Nr‡cs"ˆ |¡ˆ d¡ˆ |¡dS)Nz: )rr)ÚkÚv)rr r Úwrite_key_value_pairÛs  z,Unparser._Dict..write_key_value_paircs4|\}}|dkr&ˆ d¡ˆ |¡n ˆ||ƒdS)Nz**)rr)Úitemr±r²)rr³r r Ú write_itemàs   z"Unparser._Dict..write_itemcs ˆ d¡S)Nz, )rr )rr r r*ér+z Unparser._Dict..rˆ)rr ÚzipÚkeysr†)rr#rµr )rr³r Ú_DictÙs    zUnparser._Dictcs^ˆ d¡t|jƒdkr8|jd}ˆ |¡ˆ d¡nt‡fdd„ˆj|jƒˆ d¡dS)Nr9rrrœcs ˆ d¡S)Nz, )rr )rr r r*ór+z!Unparser._Tuple..r:)rrrr¤rr )rr#r¦r )rr Ú_Tupleìs    zUnparser._Tupleú~Únotú+ú-)ÚInvertÚNotÚUAddÚUSubcCsD| d¡| |j|jjj¡| d¡| |j¡| d¡dS)Nr9r5r:)rÚunopr7r!r"rÚoperand)rr#r r r Ú_UnaryOp÷s    zUnparser._UnaryOpÚ*rbú/ú%z<>ú|ú^ú&z//z**) ÚAddÚSubÚMultÚMatMultÚDivÚModÚLShiftÚRShiftÚBitOrÚBitXorÚBitAndÚFloorDivÚPowcCsN| d¡| |j¡| d|j|jjjd¡| |j¡| d¡dS)Nr9r5r:)rrÚleftr6r7r!r"Úright)rr#r r r Ú_BinOps    zUnparser._BinOpz==z!=úz>=Úiszis notÚinznot in) ÚEqÚNotEqÚLtÚLtEÚGtÚGtEÚIsÚIsNotÚInÚNotIncCsf| d¡| |j¡x@t|j|jƒD].\}}| d|j|jjd¡| |¡q&W| d¡dS)Nr9r5r:) rrrØr¶ÚopsZ comparatorsÚcmpopsr!r")rr#Úorhr r r Ú_Compare s   zUnparser._CompareÚandÚorcsDˆ d¡dˆj|jj‰t‡‡fdd„ˆj|jƒˆ d¡dS)Nr9z %s cs ˆ ˆ¡S)N)rr )rzrr r r*r+z"Unparser._BoolOp..r:)rÚboolopsr7r!r rr†)rr#r )rzrr Ú_BoolOps zUnparser._BoolOpcCsh| |j¡t|jtjƒr(t|jjtƒsDt|jtjƒrNt|jjtƒrN| d¡| d¡| |j ¡dS)Nr5r.) rr(rr;ÚNumr ÚintÚConstantrÚattr)rr#r r r Ú _Attributes    zUnparser._AttributecCs„| |j¡| d¡d}x,|jD]"}|r6| d¡nd}| |¡q"Wx,|jD]"}|rd| d¡nd}| |¡qPW| d¡dS)Nr9Fz, Tr:)rÚfuncrrnre)rr#rgrhr r r Ú_Call$s      zUnparser._CallcCs0| |j¡| d¡| |j¡| d¡dS)Nr¢r£)rr(rÚslice)rr#r r r Ú _Subscript2s   zUnparser._SubscriptcCs| d¡| |j¡dS)NrÅ)rrr()rr#r r r Ú_Starred8s zUnparser._StarredcCs| d¡dS)Nz...)r)rr#r r r Ú _Ellipsis=szUnparser._EllipsiscCs| |j¡dS)N)rr()rr#r r r Ú_Index@szUnparser._IndexcCsN|jr| |j¡| d¡|jr.| |j¡|jrJ| d¡| |j¡dS)Nr)ÚlowerrrÚupperÚstep)rr#r r r Ú_SliceCs    zUnparser._Slicecst‡fdd„ˆj|jƒdS)Ncs ˆ d¡S)Nz, )rr )rr r r*Nr+z$Unparser._ExtSlice..)r rZdims)rr#r )rr Ú _ExtSliceMszUnparser._ExtSlicecCs,| |j¡|jr(| d¡| |j¡dS)Nz: )rÚargr=r)rr#r r r Ú_argQs  z Unparser._argcCs„d}dgt|jƒt|jƒ|j}xNt|j|ƒD]>\}}|rDd}n | d¡| |¡|r2| d¡| |¡q2W|js€|jrÒ|rŠd}n | d¡| d¡|jrÒ| |jj¡|jj rÒ| d¡| |jj ¡|jr.xRt|j|j ƒD]@\}}|rüd}n | d¡| |¡f|rê| d¡| |¡qêW|j r€|rBd}n | d¡| d|j j¡|j j r€| d¡| |j j ¡dS)NTFz, ú=rÅz: z**) rrrnÚdefaultsr¶rrÚvarargÚ kwonlyargsrr=Ú kw_defaultsÚkwarg)rr#ÚfirstrÚaÚdr r r Ú _argumentsXsJ              zUnparser._argumentscCs<|jdkr| d¡n| |j¡| d¡| |j¡dS)Nz**r)rrrr()rr#r r r Ú_keyword‚s     zUnparser._keywordcCsD| d¡| d¡| |j¡| d¡| |j¡| d¡dS)Nr9zlambda z: r:)rrrnr%)rr#r r r Ú_LambdaŠs      zUnparser._LambdacCs&| |j¡|jr"| d|j¡dS)Nz as )rr`Zasname)rr#r r r Ú_alias’s zUnparser._aliascCs,| |j¡|jr(| d¡| |j¡dS)Nz as )rZ context_exprZ optional_varsr)rr#r r r Ú _withitem—s  zUnparser._withitemN)r )\r"Ú __module__Ú __qualname__Ú__doc__ÚsysÚstdoutrrrrrrr'r)r-r1r4r8r>r@rBrDrFrGrJrKrLrNrPrQrVr]rarirlrmrkrprqrortrurwrxr{r|r‚r„rrŠrŒrƒr–r›ržrŸr¡r¥r©rªr«r­r®r¯r°r¸r¹rÂrÄr6rÚrêrìr;ÚAndÚOrrïrðrõr÷rùrúrûrürrrr rrrrr r r r r s®                *r c Csft|dƒ}t |j¡d}WdQRXt|d|d}| ¡}WdQRXt||dtjƒ}t||ƒdS)NÚrbrÚr)ÚencodingÚexec) ÚopenÚtokenizeÚdetect_encodingÚreadlineÚreadÚcompiler;Ú PyCF_ONLY_ASTr )ÚfilenameÚoutputÚpyfilerÚsourcerr r r Ú roundtrips  r(c CsØydd„t |¡Dƒ}Wn&tk r>td|tjdYn–Xx’|D]Š}tj ||¡}tj |¡r¼t   ¡}td|ƒyt ||ƒWqÐt k r¸}ztdt |ƒƒWdd}~XYqÐXqFtj |¡rFt|ƒqFWdS)NcSsg|]}| d¡r|‘qS)z.py)Úendswith)Ú.0r r r r ú ©sztestdir..zDirectory not readable: %s)rz Testing %sz$ Failed to compile, exception is %s)ÚosÚlistdirÚOSErrorrrÚstderrÚpathÚjoinÚisfiler}r~r(Ú ExceptionryÚisdirÚtestdir)r r,r Úfullnamer%rhr r r r5§s   $ r5cCsF|ddkr,x4|dd…D] }t|ƒqWnx|D] }t|ƒq2WdS)Nrz --testdirr)r5r()rnr r r r Úmain¹s   r7Ú__main__)rrr;rr}r,ryÚ float_infoÚ max_10_expršr r rr(r5r7r"Úargvr r r r Ús$