qjtJ UdZddlZddlmZmZmZmZmZmZm Z ej dkrddlm Z nddl m Z ddl mZddlmZddlmZmZdd lmZmZdd lmZdd lmZdd lmZmZmZmZej eej!Z"ee#d <edZ$e eefZ%e&Z'e&Z(ej)ej*ej+hZ,ee#d<e"j-e"j.e"j/e"j0e"j1e"j2e"j3e"j4e"j5e"j6h Z7ee#d<dZ8ee#d<dej9e8<ddhZ:ee#d<ej;ej<ej=ej>ej?ej@hZAee#d<ejBejCejDejEejFejGejHejIejJejKejLejMejNejOhZPee#d<ejIejOhZQee#d<eQejJhzZRee#d<e"jSe"jTe"jUe"jVe"jWhZXee#d<e"jYe"jZe"j[e"j\e"j]e"j^e"j_hZ`ee#d<e"jae"jbe"jce"jde"jee"jfe"jge"jhe"jie"jje"jke"jle"jUe"jme"jne"johZpee#d<e"jqe"jrhZsee#d<hdZtee#d<e"jue"j]e"jvhZwee#d <ejxejyejzej{ej|ej}iZ~ee#d!<ee~Zee#d"<ee~Zee#d#<eezZee#d$<eeje8ejejejhzZee#d%<d&Zed'(Gd)d*ee$Zd+ed,ed-ed.efd/Zdxd1e&d2ed3ed.efd4Zd5ee%d.eefd6Zd5ee%d7eee(d.efd8Zd5ee%d.ee(fd9Zd:ed;e%d.ee%fd<Zd=e%d>e%d.dfd?Zd+ed.e%fd@Zd5e%d.eefdAZd5e%d.efdBZd5ed-ed.efdCZd5e%d.efdDZd5e%d.efdEZd5e%d.efdFZejxejyffdGedHedIeedJee&e&fd.ef dKZd5e%d.efdLZdyd5e%dNed.efdOZd5e%d.efdPZd5e%d.efdQZd+edRee(d.efdSZd5ed.efdTZd5ed.efdUZd5e%d.efdVZd5ed.efdWZd5ed.efdXZd5ed.efdYZd5e%d.efdZZd5e%d.efd[Zd+ed.efd\Zd+ed.efd]Zd+ed.efd^Zd+ed.efd_Zd+ed.efd`Zd+ed.efdaZd+ed.efdbZd+ed.efdcZdded.efdeZd'dfdgedhe%died.dfdjZd5e%d.ee%fdkZd+ed.dfdlZdmed.e efdnZdmed.e efdoZdmed.e efdpZdmed.e efdqZd+ed.edrfdsZd+ed.efdtZd5e%d.eefduZd5e%d.eefdvZd+ed.e%fdwZdS)zz> blib2to3 Node/Leaf transformation-related utility functions. N)FinalGenericIteratorLiteralOptionalTypeVarUnion) ) TypeGuard) mypyc_attr) CACHE_DIR)ModePreview)get_string_prefixhas_triple_quotes)pygram)token)NLLeafNode type_reprsymsT WHITESPACE STATEMENTSTANDALONE_COMMENTandorLOGIC_OPERATORS COMPARATORSMATH_OPERATORSSTARSVARARGS_SPECIALSVARARGS_PARENTSUNPACKING_PARENTSTEST_DESCENDANTS TYPED_NAMES>%=&=*=+=-=/=@=^=|=**=//=<<=>>=:= ASSIGNMENTSIMPLICIT_TUPLEBRACKETOPENING_BRACKETSCLOSING_BRACKETSBRACKETSALWAYS_NO_SPACE7T)allow_interpreted_subclassescJeZdZdZdedeefdZdedeefdZdS)VisitorzBBasic lib2to3 visitor that yields things of type `T` on `visit()`.nodereturnc#K|jdkrtj|j}n!tt |j}t |d|d}|r||Ed{VdS||Ed{VdS)azMain method to visit `node` and its children. It tries to find a `visit_*()` method for the given `node.type`, like `visit_simple_stmt` for Node objects or `visit_INDENT` for Leaf objects. If no dedicated `visit_*()` method is found, chooses `visit_default()` instead. Then yields objects of type `T` from the selected visitor. visit_N)typertok_namestrrgetattr visit_default)selfrDnamevisitfs [/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/black/nodes.pyvisitz Visitor.visits 9s??>$),DDy++,,D 55  0vd|| # # # # # # # # #))$// / / / / / / / / /c#Kt|tr%|jD]}||Ed{VdSdS)zCDefault `visit_*()` implementation. Recurses to children of `node`.N) isinstancerchildrenrR)rNrDchilds rQrMzVisitor.visit_defaultsd dD ! ! - - -::e,,,,,,,,,, - - - -rSN) __name__ __module__ __qualname____doc__LNrrrRrMrSrQrCrCsfLL0"0!00000-"-!------rSrCleafcomplex_subscriptmoderEc^d}d}d}|j}|j}|j}|tvr|S|tjkr|S| Jd||tjkr,|jtjtj tj hvr|S|tj kr|jtj kr|S|j } | sDt|} | r| jtvr|S|tjkr2| jtjkr|S| jtjkr|s|S|S| jtjkrf| jr]| jjtjtjtjtjhvr|S| jjtjkr| jSn| jtjkrqt3| tjkrTt3| jtj ks0t6j|vr$t3| jtjkr|S| jt<vr!t?| t@tBzr|Sn| jtjkr2| jr*| jjtjtj hvr|r|n|Sn| jr*| jjtj"kr| jtFvr|S| jtj$kr#|jr|jjtj%kr|Sn'| jtvr|S| jtj&kr|S|jtjtjhvr| r| jtjkr|Sn|jtjkr| r| jtjkr|Sn|jtjkr[| s|S|tjkr| jtNvr|Snh| jtjkr| jS| jtjkr|Sn3|jtNvr,| s(t|} | r| jtjkr|Sn|jtj(kra|tj)ks|tj*kr|S| s$|tj+ks|tj,kr|Sn| jtjkr|Sn|jtjkr^|tjkr|S| s*t|} | r| jtj)kr|Sn0| jtjht<zvr|Sn|jtj%kr|S|jtj-krC| r|St|} | r*| jtj$ks| jtj+kr|Sn|jtj.kr-|tj)kr|S| r| jtj)kr|Sn_|jtjtj hvr^| s/|j Jd|jjtj kr|S|S|tj/ks| jtj/kr|S|s|Sn|jtj0kr| r|tj+kr|Sn|jtj1kr| r| jtj2kr|Sn|jtj"tjhvr| st|} | r| jtvr|S| j} | J| jtjkr!| jtjtj hvr|S| jtjkr| jtjkr|Sn|tj3tj4tj5hvr|Sn|jtj6kr\|tj+kr| r| jtj+kr|Snp|tj3kr!|dkr|S| r| jtj+kr|Sn>|jtj kr|S|jtj7kr|tjkr|S|S) zReturn whitespace prefix if needed for the given `leaf`. `complex_subscript` signals whether the given leaf is part of a subscription which has non-trivial arguments, like arithmetic expressions or function calls.  z Nz/INTERNAL ERROR: hand-made leaf without parent: )withinzsubscripts are always parentedimport)8rIparentvaluer?rCOMMENTCOLONr subscript subscriptlistsliceopLBRACEfstring_replacement_field prev_siblingpreceding_leafr<COMMAEQUALarglistargument parameters varargslist typedargslistprefixSTAR parent_type star_exprr$pep646_typed_star_arg_type_var_tuple tname_starr% is_varargr&r'factorr#AT decoratorBANGr)trailerLPARRPARDOTLSQB dotted_nameclassdef COLONEQUALatom dictsetmaker DOUBLESTARNAMENUMBERSTRING import_from except_clause) r^r_r`NOSPACE DOUBLESPACEtpvprevprevp prevp_parents rQ whitespacers BE"K A A AO EM ==TDTT===EKAF   +  ELQVt'EEE  D Cq!!  &666I   zU[(( u{**3D* L : $ $| (<$LMO$ ) I\&$*<<<!<' J%* $ $E""dn44EL))T-???@DHH#EL11T_DD I Z+ + +9J'JKKK   Z5; & &| : 1dndl5S S S 19uur9 L  !T[00 n,,I Z58 # # #QX]dn5T5TI & & & ej  v$/4<000 tyEK//I0 4# # #  DI,,I 4% % % I   y ++ ,Y%+ % %;  Y%+ % %I& ;   "1%%E EJ%+55 4<    ??a5:ooI EI~~ej "1Y%+ % %I& 4=   I "1%%E EJ%*44 5Y5;-*:: : :I; 4> ! ! 4# # #  Iq!!  eh..%* 2I2II3J 4=  ??I  DI++I DNDL1 1 1 8'')I'''x} 222 I %" " "di53C&C&CL" I  49    ANNI 4$ $ $  DI!111I DK0 0 0 "1%%E EJ*:::   C.. u{**|/@DM/Q/Q 5:u|U\: : :I; 4# # #  >>  UY.. %*__H}}   UY.. 4<   4% % %  ??I LrS nl_count form_feed empty_linec,|r||dz zdz|zS||zS)z$Generate a normalized prefix string. r])rrrs rQmake_simple_prefixrs/Ahl+t3j@@   rSrDc|rb|j}|rPt|tr|S t|dS#t $rYdSwxYw|j}|bdS)z3Return the first leaf that precedes `node`, if any.N)rorUrlistleaves IndexErrorrf)rDress rQrprps    #t$$   CJJLL))"--   tt {   4s&A AAtokensc|sdS|d|duS|sdS|j|dkrdSt|j|ddS)a;Return if the `node` and its previous siblings match types against the provided list of tokens; the provided `node`has its type matched against the last element in the list. `None` can be used as the first element to declare that the start of the list is anchored at the start of its parent's children.TrNF)rIprev_siblings_arero)rDrs rQrrsc t bzt| u yF2Ju T.ss < <)r^ same_prefix containerrfs rQ container_ofrs +KI ! >  ?1  $ 3 3  ;$/ ) )    *v/B/G8/S/S    rScvt|tr|S|jrt|jdSdS)z(Returns the first leaf of the node tree.rN)rUrrV first_leaf_ofrs rQrrs=$  }T]1-...trScl|jtjtjtjtjhvS)z?Whether node is an arithmetic or a binary arithmetic expression)rIr arith_expr shift_exprxor_exprand_exprrs rQ is_arith_likers* 9     rSct|trO|jtjkrdSt |j}t|drdStj |vrZ|j rS|j jtj kr9|j js-|j j r!|j j jtjkrdSt!|j dtjtjtj grdSt!|j tjtjtj grdSdS)NFbBfFT)rUrrIrrrrgset intersectionrunify_docstring_detectionrfr simple_stmtrorrNEWLINEINDENTruri)rDr`rxs rQ is_docstringr)s$ 9 $ $5"4:.. v;; # #F + + 5 )T11 K 2 K  0 0 0 ( 1 K  1 K  #t 6 6t dEM5<9IJtt TEU&VWWt 5rSc|jtjkoWt|jdko?|jdjt jko|jdjt jkS)z+Return True if `node` holds an empty tuple.rr)rIrrlenrVrrrrs rQis_empty_tuplerJs` TY 0   ! # 0 M!  !UZ / 0 M!  !UZ / rScx|jtjkr`t|}||jtjkrdSt |jdko|jdjtjkS|jtvo7t |jdko|jdjtjkS)zMReturn True if `node` holds a tuple with one element, with or without parens.NFrr) rIrrunwrap_singleton_parenthesis testlist_gexprrVrrqr:rDgexps rQ is_one_tuplerTs yDI+D11 <49(:::54=!!Q&O4=+;+@EK+OO ^# 1   ! # 1 M!  !U[ 0rSc|jtjkrdSt|}||jtjkrdSt d|jDS)zDReturn True if `node` holds a tuple that contains a walrus operator.FNc3@K|]}|jtjkVdSN)rIrnamedexpr_test).0rWs rQ z-is_tuple_containing_walrus..ls,LLUuzT00LLLLLLrS)rIrrrranyrVrs rQis_tuple_containing_walrusrds[ yDIu ' - -D |tyD$666u LLdmLLL L LLrSopeningclosingrbracketsc|j|jf|krdS|jdz}t|D] \}}||urn tdd}|dz }||dD]a}||urnZ|j}||krL|jtjkr7|dz }|jr+|jjtjtj hvr|dz }nb|dkS)zIReturn True if content between `opening` and `closing` is a one-sequence.Frz#Opening paren not found in `leaves`rNr) rI bracket_depth enumerate LookupErrorrrqrfrrsrw) rrrrdepth_opening_indexr^commasrs rQis_one_sequence_betweenros  gl#x//u  !A %E )& 1 1AA 7?? E ?@@@ FaN~'   7?? E* E ! !di5;&>&> aKF{ t{/ "4  !  A:rScRt|}|duo|jtjkS)z7Return True iff `node` is of the shape ( test := test )N)rrIrr)rDinners rQis_walrus_assignmentrs) ( . .E   Bt/B!BBrSFlastcF|jtjko t|jdko?|jdjt jko|jdjt jkp|oWt|jdko?|jdjt jko|jdjt j kpY|oWt|jdko?|jdjt jko|jdjt j kS)z?Return True iff `node` is a trailer valid in a simple decoratorrrrr ) rIrrrrVrrrrr)rDrs rQis_simple_decorator_trailerrs 9 $    ! # 4 a %2 4 a %3   4DM""a' 4 a %3 4 a %3   4DM""a' 4 a %3 4 a %3'rScr|jtjkrdS|jtjkr|jr|jdjtjkobt tt|jddo3t|jdkpt|jddSdS)a\Return True iff `node` could be a 'dotted name' decorator This function takes the node of the 'namedexpr_test' of the new decorator grammar and test if it would be valid under the old decorator grammar. The old grammar was: decorator: @ dotted_name [arguments] NEWLINE The new grammar is : decorator: @ namedexpr_test NEWLINE Trrrr)rF) rIrrrpowerrVallmaprrrs rQis_simple_decorator_expressionrs yEJt yDJ =  a %37qt9LMMNN &&*Q24=3D4PPP   5rScV|jtjkrdSt|r |jdkrdS|jtjkrdSt |jdkrdS|j\}}}|jtj kr$|jtj krt|SdS)zAReturn True if `node` holds a `yield` or `yield from` expression.TyieldFr ) rIr yield_expr is_name_tokenrgrrrVrrris_yield)rDlparexprrpars rQrrs yDO##tTtzW44t yDIu 4=Qu}D$ yEJ49 #:#:~~ 5rSrdc|jtvs|jsdS|j}|jtjkr|jsdS|j}|j|vS)aFReturn True if `leaf` is a star or double star in a vararg or kwarg. If `within` includes VARARGS_PARENTS, this applies to function signatures. If `within` includes UNPACKING_PARENTS, it applies to right hand-side extended iterable unpacking (PEP 3132) and additional unpacking generalizations (PEP 448). F)rIr%rfrr{)r^rdrs rQr~r~sY y((( (u Avx 5 H 6V rSc,|jtjkS)z&Return True if the node is an f-string)rIrfstringrs rQ is_fstringrs 9 $$rSct|t|jd}ttj||j}|pd|_|S)z/Converts an fstring node back to a string node.N)rxr)rKrrxrrr get_linenolineno)rDstring_without_prefix string_leafs rQfstring_to_stringr sYIIc$+&6&6&8&89u|%:4;OOOK**/aK rSct|trt|rt|}nt|tr|}ndSt |jod|jvS)zKReturn True if `leaf` is a multiline string that actually spans many lines.Fr)rUrrr rrrg)rDr^s rQis_multiline_stringr sl$*T"2"2 && D$  u TZ ( ( ?TTZ-??rSc|jtjtjhvsJ|jJ|jjtjtjhvSr)rIrsuiterrffuncdefrrs rQis_parent_function_or_classrsF 9T%56 6 6 6 6 ; " " " ;  dm< <.@s3HHUY,,,HHHHHHrS) rUrrIrrrrVrxrrr)rDrWs rQrr3s dD ! !TY$2B%B%Bu 4=Qu M! E L     I J$) # I   1 $ I HHHHH H H rScbt|ts|jtjkrdS|jd|jd}}t|toT|jt jko?|jdko4t|to|jt j ko |jdkS)zGiven a `LN`, determines whether it's an atom `node` with invisible parens. Useful in dedupe-ing and normalizing parens. Frrrb) rUrrIrrrVrrrgr)rDfirstrs rQis_atom_with_invisible_parensrDs$di!7!7u-"DM"$54E5$  J%* $  K2   tT " "  I #   J"  rSc>t|pt|Sr) is_empty_lpar is_empty_rparr^s rQ is_empty_parr$Vs    5-"5"55rScB|jtjko |jdkSNrb)rIrrrgr#s rQr!r!Z 9 " 7tzR'77rScB|jtjko |jdkSr&)rIrrrgr#s rQr"r"^r'rSc|j}|j}|j}t|tjko9|dkr|r|jt jkp|dko|o|jt jkS)z9Return True if the given leaf starts an import statement.refrom) rfrIrgboolrrr import_namer)r^rrrs rQ is_importr-bss A A A  UZ (] ?q ?QVt/?%? BV @@af0@&@   rSc$t|jtjko+|jdko |jo|jjt jkpBt|jtjko |j o|j jt jkS)zDReturn True if the given leaf starts a with or async with statement.with) r+rIrrrgrfr with_stmtASYNC next_siblingr#s rQis_with_or_async_with_stmtr3ps  UZ / J&  / K / K  .      U[  5   5   "dn 4   rSct|jtjko*|jo#|jjt jt jhvS)zReturn True if the given leaf starts an async def/for/with statement. Note that `async def` can be either an `async_stmt` or `async_funcdef`, the latter is used when it has decorators. )r+rIrr1rfr async_stmtrr#s rQis_async_stmt_or_funcdefr6~sJ  U[  F K F K $2D E E  rScr|j}|j}|tjthvo|dS)aBReturn True if the given leaf is a type comment. This function should only be used for general type comments (excluding ignore annotations, which should use `is_type_ignore_comment`). Note that general type comments are no longer used in modern version of Python, this function may be deprecated in the future.z# type:)rIrgrrhr startswithr^rrs rQis_type_commentr:s6 A A  23 3 O Y8O8OOrScf|j}|j}|tjthvot |S)zGReturn True if the given leaf is a type comment with ignore annotation.)rIrgrrhris_type_ignore_comment_stringr9s rQis_type_ignore_commentr=s2 A A  23 3 X8UVW8X8XXrSrgc,|dS)zSReturn True if the given string match with type comment with ignore annotation.z# type: ignore)r8)rgs rQr<r<s   , - --rS)visiblerfrWr?c:ttj|rdnd}ttj|rdnd}|j}d|_|pd}t tj|||g}||_| ||dS)zWrap `child` in parentheses. This replaces `child` with an atom holding the parentheses and the old child. That requires moving the prefix. If `visible` is False, the leaves will be valueless (and thus invisible). (rb)rN) rrrrrxrrrrr)rfrWr?rrrxindexrs rQwrap_in_parenthesesrDs  72CC 3 3D  72CC 3 3D \FEL LLNN aETYud 344II y)))))rSct|jdkrdS|j\}}}|jtjkr|jtjksdS|S)zqReturns `wrapped` if `node` is of the shape ( wrapped ). Parenthesis can be optional. Returns None otherwiser N)rrVrIrrr)rDrwrappedrs rQrrsS 4=Qt-D'4 I # # UZ(?(?t NrSc~|jtjkr d|_dS|jtjkr d|_dSdS)zMake sure parentheses are visible. They could be invisible as part of some statements (see :func:`normalize_invisible_parens` and :func:`visit_import_from`). rArBN)rIrrrgrr#s rQensure_visiblerHs@  yEJ ej   ! rSnlc,|jtjkSr)rIrrrIs rQrr 7ej  rSc,|jtjkSr)rIrrrKs rQ is_lpar_tokenrNrLrSc,|jtjkSr)rIrrrKs rQ is_rpar_tokenrPrLrSc,|jtjkSr)rIrrrKs rQis_number_tokenrRs 7el ""rS)rEparamNc|j}|O|jr|jjtjkrdS|jr|jjt jkrdS|j}|OdS)zrqFSTRING_MIDDLE FSTRING_ENDrr?rUrCr+rKrrrprrrzrrrrrrrrrtuplerrrrrr~rr r rrrrrr$r!r"r-r3r6r:r=r<rDrrHrrNrPrRrWrYr[r]r_r]rSrQrs  NNNNNNNNNNNNNNNNNNw       ++++++&&&&&&!!!!!!$$$$$$$$>>>>>>>> 555555555555)#e### GCLL 4:  \5<? E???LOMMNLMOO  5    E%9!"&&& J M M N O   U J  K O  J K J K  M H K   E,-u---5;-/%///LML  INL5 ILLMMONIMMOOLIJ!%$j$/2 U222 U$(?OOOO J  J  L%, #gllnn--%---#gnn..//%///"%55%555) K   J -  ...-----gaj--/.-DsTssTscssssl!!!!3!RU!!!!"(4.$ =HRL =$x7I2J =t = = = = hrl x'9    DbXb\2R2B242222"t2x~tr$B r d     MRMDMMMM"'UZ 8 "" " " J"CHo "  """"JCrCdCCCC b402$*D#h-D.%T%d%%%% DT @b @T @ @ @ @=d=t====JtJJJJJ******.rd"t$6t66666888888888888 D T     T d     4 D    P$P4PPPPYY$YYYY...... EI****R*T*T****$ r hrl      $    !b!Yt_!!!!!b!Yt_!!!!!b!Yt_!!!!##y#### d w/F'G    111111 ,R,HTN,,,,,B,8D>,,,,4BrS