a ,=ic@sdZddlmZddlmZddlmZddlZddlZddlZddlm Z ddlm Z ddlm Z Gd d d e ZGd d d e jZd dZdS)z,Generate code from an annotated syntax tree.)absolute_import)division)print_functionN)annotate) formatting) fstring_utilsc@seZdZdZdS) PrintErrorz2An exception for when we failed to print the tree.N)__name__ __module__ __qualname____doc__r r c/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/pasta/base/codegen.pyrsrcseZdZdZfddZfddZddZdd Zd d Zd d Z ddZ dddZ dddZ ddZ ddZddZZS)PrinteraTraverses an AST and generates formatted python source code. This uses the same base visitor as annotating the AST, but instead of eating a token it spits one out. For special formatting information which was stored on the node, this is output exactly as it was read in unless one or more of the dependency attributes used to generate it has changed, in which case its default formatting is used. cstt|d|_dSN)superr__init__code)self __class__r rr,szPrinter.__init__c sbtdd|_ztt||Wn4tttt fyX}zt |WYd}~n d}~00|`dS)NcSsdS)NFr r r r r1zPrinter.visit..) collections defaultdict _printer_inforrvisit TypeError ValueError IndexErrorKeyErrorr)rnodeerr rr0s z Printer.visitcCsD||t|d}|j|dur(|nt|j7_||dSNcontent)prefixfmtgetrreprnsuffixrr"r%r r r visit_Num8s   zPrinter.visit_NumcCsD||t|d}|j|dur(|nt|j7_||dSr$r&r'r(rr)sr+r,r r r visit_Str>s   zPrinter.visit_StrcCs||t|d}|durjg}|jD]2}t|tjrF||jq(|t t |q(t d |}ddt |D}|jt ||7_||dS)Nr%rcSsg|] }t|qSr )to_str).0vr r r Qrz+Printer.visit_JoinedStr..)r&r'r(values isinstanceastStrappendr/r placeholderlenr)joinZget_formatted_valuesrZperform_replacementsr+)rr"r%partsvalr5r r rvisit_JoinedStrDs    zPrinter.visit_JoinedStrcCsD||t|d}|j|dur(|nt|j7_||dSr$r.r,r r r visit_BytesUs   zPrinter.visit_BytescCs|j|7_dS)N)r)rvaluer r rtoken[sz Printer.tokenFcCs4~t||}|dur|r|}|j|p*d7_dSr)r'r(r)rr" attr_nameZ token_valZallow_whitespace_prefixdefaultrAr r roptional_token^s   zPrinter.optional_tokenNcst~tdrj|rdSdj|<t|}|dusR|rVtfdd|DrV|}|j|durh|nd7_dS)aAdd the formatted data stored for a given attribute on this node. If any of the dependent attributes of the node have changed since it was annotated, then the stored formatted data for this attr_name is no longer valid, and we must use the default instead. Arguments: node: (ast.AST) An AST node to retrieve formatting information from. attr_name: (string) Name to load the formatting information from. attr_vals: (list of functions/strings) Unused here. deps: (optional, set of strings) Attributes of the node which the stored formatting data depends on. default: (string) Default formatted data for this attribute. rNTc3s*|]"}t|dt|dkVqdS)NZ__src)getattrr'r()r2depr"r r {szPrinter.attr..r)hasattrrr'r(anyr)rr"rCZ attr_valsdepsrDr>r rHrattrfs    z Printer.attrcCs(zt|dWSty"YdS0dS)NZis_elifF)r'r(AttributeErrorrr"r r r check_is_elifs zPrinter.check_is_elifcCs t|ddSNZ is_continuedFrFrOr r rcheck_is_continued_tryszPrinter.check_is_continued_trycCs t|ddSrQrRrOr r rcheck_is_continued_withszPrinter.check_is_continued_with)FF)NN)r r r r rrr-r0r?r@rBrErMrPrSrT __classcell__r r rrr"s    rcCs~t}tdd}t|D]&}t|dd}|r||d7<q|rntt |ddd\}}| || ||j S)z8Convenient function to get the python source for an AST.cSsdS)Nrr r r r rrrzto_str.. indent_diffrcSs|dr|dSdS)NrrWr )tupr r rrr)key) rrrr7walkr'r(maxsix iteritemsZset_default_indent_diffrr)treepZseen_indent_diffsr"rV_r r rr1s    r1)r __future__rrrr7rr]Z pasta.baserrr'r ExceptionrZ BaseVisitorrr1r r r rs      m