U \Sh&@sUdZddlmZddlZddlmZejdejdejdej dej d ej d ej d ej d ejd ejdejdejdejdejdejdejdejdejdejdiZded<ed%ddddddZed&ddddddZd'd dd!dd"dZGd#d$d$ejZdS)(z'Helpers for AST (Abstract Syntax Tree).) annotationsN)overload+and&|^/z//~z<<@*%not**orz>>-zdict[type[ast.AST], str] OPERATORSNonestr)nodecodereturncCsdSNrrrrD/root/rtd-docs/venv/lib/python3.8/site-packages/sphinx/pycode/ast.pyunparsesrast.ASTcCsdSrrrrrrr$sast.AST | Nonez str | NonecCs(|dkr dSt|tr|St||S)zUnparse an AST to string.N) isinstancer_UnparseVisitorvisitrrrrr)s  c@sdeZdZdCdddddZdddd d ZeD]Zeed ej<q.d ddd dZd dddddZ dddddZ dddddZ dddddZ dddddZ dddd d!Zd"ddd#d$Zd%ddd&d'Zd(ddd)d*Zd+ddd,d-Zd.ddd/d0Zd1ddd2d3Zd4ddd5d6Zd7ddd8d9Zd:ddd;d<Zd=ddd>d?Zd@dAZdBS)Dr!rrr)rrcCs ||_dSr)r)selfrrrr__init__4sz_UnparseVisitor.__init__r)rrcCs t|jSr)r __class__r#rrrr _visit_op7sz_UnparseVisitor._visit_opvisit_zast.argcCs(|jr|jd||jS|jSdS)N: ) annotationargr"r&rrr visit_arg<sz_UnparseVisitor.visit_argr)r+defaultrcCs>||}|r:|jr(|d||7}n|d||7}|S)z&Unparse a single argument to a string.z = %sz=%s)r"r*)r#r+r-namerrr_visit_arg_with_defaultBs  z'_UnparseVisitor._visit_arg_with_defaultz ast.argumentsc Cslt|j}t|j}t|j}||7}tt||D]}|ddq4t|j}tt|t|jD]}|ddqdg}t |jD]\}} | | | ||q|jr| dt |jD]"\}} | | | |||q|j r| d| |j |jr|j s| dt |jD] \}} | | | ||q"|jrb| d| |jd|S)Nrr r r, )listdefaultslenargs posonlyargsrangeinsert kw_defaults kwonlyargs enumerateappendr/varargr"kwargjoin) r#rr2 positionalsr5_r8r4ir+rrrvisit_argumentsLs2      z_UnparseVisitor.visit_argumentsz ast.AttributecCs||jd|jS)N.)r"valueattrr&rrrvisit_Attributeosz_UnparseVisitor.visit_Attributez ast.BinOpcsPt|jtjr,dtj|j|j|jfSdfdd|j|j|jfDS)Nr c3s|]}|VqdSrr".0er#rr vsz._UnparseVisitor.visit_BinOp..) r opastPowr>mapr"leftrightr&rrLr visit_BinOprsz_UnparseVisitor.visit_BinOpz ast.BoolOpcs*d|j}|fdd|jDS)Nz %s c3s|]}|VqdSrrHrIrLrrrMzsz/_UnparseVisitor.visit_BoolOp..)r"rNr>values)r#rrNrrLr visit_BoolOpxsz_UnparseVisitor.visit_BoolOpzast.CallcsFdfdd|jDfdd|jD}|jd|dS)Nr0csg|]}|qSrrHrIrLrr }sz._UnparseVisitor.visit_Call..cs$g|]}|jd|jqS)=)r+r"rDrJkrLrrrW~s())r>r4keywordsr"func)r#rr4rrLr visit_Call|sz_UnparseVisitor.visit_Callz ast.ConstantcCsX|jtkrdSt|jtttfrJ|jr>t|j|p.c3s|]}|VqdSrrH)rJvrLrrrMscss|]\}}|d|VqdS)r)Nr)rJrZrgrrrrMs{r0})keysrUzipr>)r#rrjrUitemsrrLr visit_Dictsz_UnparseVisitor.visit_Dictz ast.IndexcCs ||jSr)r"rDr&rrr visit_Indexsz_UnparseVisitor.visit_Indexz ast.LambdacCsd||jS)Nzlambda %s: ...)r"r4r&rrr visit_Lambdasz_UnparseVisitor.visit_Lambdazast.Listcs"ddfdd|jDdS)N[r0c3s|]}|VqdSrrHrIrLrrrMsz-_UnparseVisitor.visit_List..]r>eltsr&rrLr visit_Listsz_UnparseVisitor.visit_Listzast.NamecCs|jSr)idr&rrr visit_Namesz_UnparseVisitor.visit_Namezast.Setcs"ddfdd|jDdS)Nrhr0c3s|]}|VqdSrrHrIrLrrrMsz,_UnparseVisitor.visit_Set..rirrr&rrLr visit_Setsz_UnparseVisitor.visit_Setz ast.Subscriptcsddddd}||jrNdfdd|jjD}|jd |d St|jtjr||jjrdfd d|jjjD}|jd |d S|jd |jd SdS) Nrbool)rDrcSs,t|tjo*t|jo*tdd|jD S)Ncss|]}t|tjVqdSr)r rOStarred)rJeltrrrrMszK_UnparseVisitor.visit_Subscript..is_simple_tuple..)r rOTuplerxrsany)rDrrris_simple_tuples  z8_UnparseVisitor.visit_Subscript..is_simple_tupler0c3s|]}|VqdSrrHrIrLrrrMsz2_UnparseVisitor.visit_Subscript..rprqc3s|]}|VqdSrrHrIrLrrrMs)slicer>rsr"rDr rOIndex)r#rr}rsrrLrvisit_Subscripts z_UnparseVisitor.visit_Subscriptz ast.UnaryOpcCsHt|jtjr,||jd||jS||j||jS)NrG)r rNrONotr"operandr&rrr visit_UnaryOpsz_UnparseVisitor.visit_UnaryOpz ast.TuplecsZt|jdkrdSt|jdkr4d|jdSddfdd|jDd SdS) Nrz()z(%s,)r[r0c3s|]}|VqdSrrHrIrLrrrMsz._UnparseVisitor.visit_Tuple..r\)r3rsr"r>r&rrLr visit_Tuples z_UnparseVisitor.visit_TuplecCstdt|jdS)NzUnable to parse %s object)NotImplementedErrortype__name__r&rrr generic_visitsz_UnparseVisitor.generic_visitN)r)r __module__ __qualname__r$r'rZ_oplocalsr,r/rBrFrTrVr_rfrmrnrortrvrwrrrrrrrrr!3s, # r!)r)r)r)__doc__ __future__rrOtypingrAddAndBitAndBitOrBitXorDivFloorDivInvertLShiftMatMultMultModrrPOrRShiftSubUAddUSubr__annotations__r NodeVisitorr!rrrrs`