!.dbdv UdZddlZddlmZmZmZmZmZmZm Z m Z ej dkrddlm Z nddl m Z ej dkrddlmZnddl mZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZmZmZmZejeejZ e e!d<e dZ"e eefZ#e$Z%e$Z&ej'ej(ej)hZ*e e!d<e j+e j,e j-e j.e j/e j0e j1e j2e j3e j4h Z5e e!d<dZ6e e!d<dej7e6<ddhZ8e e!d<ej9ej:ej;ej<ej=ej>hZ?e e!d<ej@ejAejBejCejDejEejFejGejHejIejJejKejLejMhZNe e!d<ejGejMhZOe e!d<eOejHhzZPe e!d<e jQe jRe jSe jTe jUhZVe e!d<e jWe jXe jYe jZe j[e j\e j]hZ^e e!d<e j_e j`e jae jbe jce jde jee jfe jge jhe jie jje jSe jke jlhZme e!d<e jne johZpe e!d<hdZqe e!d <e jre j[e jshZte e!d!<ejuejvejwejxejyejziZ{e e!d"<e|e{}Z~e e!d#<e|e{Ze e!d$<e~ezZe e!d%<eeje6hzZe e!d&<d'Zed()Gd*d+ee"Zd,ed-ed.efd/Zd0ee#d.eefd1Zd0ee#d2eee&d.efd3Zd0ee#d.ee&fd4Zd5ed6e#d.ee#fd7Zd8e#d9e#d.dfd:Zd,ed.e#fd;Zd0e#d.eefd<Zd0e#d.efd=Zd,ed.efd>Zd0e#d.efd?Zd0e#d.efd@Zd0e#d.efdAZejuejvffdBedCedDeedEee$e$fd.ef dFZd0e#d.efdGZdjd0e#dIed.efdJZd0e#d.efdKZd0e#d.efdLZd,edMee&d.efdNZd,ed.efdOZd0ed.efdPZd0e#d.efdQZd0e#d.efdRZd,ed.efdSZd,ed.efdTZd,ed.efdUZd,ed.efdVZd,ed.efdWZd,ed.efdXZdkd,edZed.efd[Zd(d\d]ed^e#d_ed.dfd`Zd0e#d.ee#fdaZd,ed.dfdbZdced.eefddZdced.eefdeZdced.eefdfZdced.eefdgZdced.eefdhZd,ed.efdiZdS)lz> blib2to3 Node/Leaf transformation-related utility functions. N)GenericIteratorListOptionalSetTupleTypeVarUnion))Final)r ) TypeGuard) mypyc_attr) CACHE_DIR)has_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)selfrCnamevisitfs +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) isinstancerchildrenrQ)rMrCchilds rPrLzVisitor.visit_defaultsd dD ! ! - - -::e,,,,,,,,,, - - - -rRN) __name__ __module__ __qualname____doc__LNrrrQrLrRrPrBrBsfLL0"0!00000-"-!------rRrBleafcomplex_subscriptrDc4d}d}d}|j}|j}|j}|tvr|S|tjkr|S| Jd||tjkr,|jtjtj tj hvr|S|j }|st|} | r| jtvr|S|tjkr2| jtjkr|S| jtjkr|s|S|S| jtjkrf| jr]| jjtjtjtjtjhvr|S| jjtjkr| jSnF| jtjkrAt/| tjkr$t/| jtj kr|S| jt2vr!t5| t6t8zr|Sn| jtjkr2| jr*| jjtjtj hvr|r|n|Snz| jr*| jjtjkr| jt<vr|S| jtjkr#|jr|jjtj kr|Sn|jtvr|S|jtjtjhvr|r|jtjkr|Sn|jtjkr|r|jtjkr|Sn||jtjkr[|s|S|tjkr|jtBvr|SnA|jtjkr|jS|jtjkr|Sn |jtBvr,|s(t|} | r| jtjkr|Sn|jtj"kra|tj#ks|tj$kr|S|s$|tj%ks|tj&kr|Snu|jtjkr|Sn\|jtjkr^|tjkr|S|s*t|} | r| jtj#kr|Sn |jtjht2zvr|Sn|jtj kr|S|jtj'krC|r|St|} | r*| jtjks| jtj%kr|Snz|jtj(kr-|tj#kr|S|r|jtj#kr|Sn8|jtjtj hvr7|s/|j Jd|jjtj kr|S|S|s|Sn|jtj)kr|r|tj%kr|Sn|jtj*kr|r|jtj+kr|Sn|jtjtjhvr|st|} | r| jtvr|S| j} | J| jtjkr!| jtjtj hvr|S| jtjkr| jtjkr|Sn|tj,tj-tj.hvr|Sn|jtj/kr\|tj%kr|r|jtj%kr|Snp|tj,kr!|dkr|S|r|jtj%kr|Sn>|jtj kr|S|jtj0kr|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)1rHparentvaluer>rCOMMENTCOLONr subscript subscriptlistsliceop prev_siblingpreceding_leafr;COMMAEQUALarglistargument parameters varargslist typedargslistprefixSTAR parent_type star_exprr% is_varargr&r'factorr#AT decoratorr)trailerLPARRPARDOTLSQB dotted_nameclassdefatom dictsetmaker DOUBLESTARNAMENUMBERSTRING import_from except_clause) r]r^NOSPACE DOUBLESPACEtpvprevprevp prevp_parents rP whitespacers BEK A A AO EM ==TDTT===EKAF   +   D :q!!  &666I   zU[(( u{**3D* L : $ $| (<$LMO$ ) I\&$*<<<!<' J%* $ $E""dn44EL))T-???I Z+ + +9J'JKKK   Z5; & &| : 1dndl5S S S 19uur9 L  !T[00 n,,I Z58 # # #QX]dn5T5TI & & & 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" 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 LrRrCc|rb|j}|rPt|tr|S t|dS#t $rYdSwxYw|j}|bdS)z3Return the first leaf that precedes `node`, if any.N)rkrTrlistleaves IndexErrorrd)rCress rPrlrls    #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)rHprev_siblings_arerk)rCrs rPrrsc t bzt| u yF2Ju T.ss < <  ?1  $ 3 3  ;$/ ) )    *v/B/G8/S/S    rRcvt|tr|S|jrt|jdSdS)z(Returns the first leaf of the node tree.rN)rTrrU first_leaf_ofrs rPrrs=$  }T]1-...trRcl|jtjtjtjtjhvS)z?Whether node is an arithmetic or a binary arithmetic expression)rHr arith_expr shift_exprxor_exprand_exprrs rP is_arith_likers* 9     rRct|jdtjtjt jgrdSt|jt jtjt jgrdSdS)NTF) rrdrNEWLINEINDENTr simple_stmtrqrgr]s rP is_docstringr sb dEM5<9IJtt TEU&VWWt 5rRc|jtjkoWt|jdko?|jdjt jko|jdjt jkS)z+Return True if `node` holds an empty tuple.r)rHrrlenrUrr}r~rs rPis_empty_tuplers` TY 0   ! # 0 M!  !UZ / 0 M!  !UZ / rRcx|jtjkr`t|}||jtjkrdSt |jdko|jdjtjkS|jtvo7t |jdko|jdjtjkS)zMReturn True if `node` holds a tuple with one element, with or without parens.NFrr) rHrrunwrap_singleton_parenthesis testlist_gexprrUrrmr9rCgexps rP is_one_tupler&s yDI+D11 <49(:::54=!!Q&O4=+;+@EK+OO ^# 1   ! # 1 M!  !U[ 0rRc|jtjkrdSt|}||jtjkrdSt d|jDS)zDReturn True if `node` holds a tuple that contains a walrus operator.FNc3@K|]}|jtjkVdSN)rHrnamedexpr_test).0rVs rP z-is_tuple_containing_walrus..>s,LLUuzT00LLLLLLrR)rHrrrranyrUrs rPis_tuple_containing_walrusr6s[ yDIu ' - -D |tyD$666u LLdmLLL L LLrRopeningclosingrbracketsc|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) rH bracket_depth enumerate LookupErrorrrmrdrrors) rrrrdepth_opening_indexr]commasrs rPis_one_sequence_betweenrAs  gl#x//u  !A %E )& 1 1AA 7?? E ?@@@ FaN~'   7?? E* E ! !di5;&>&> aKF{ t{/ "4  !  A:rRcRt|}|duo|jtjkS)z7Return True iff `node` is of the shape ( test := test )N)rrHrr)rCinners rPis_walrus_assignmentrfs) ( . .E   Bt/B!BBrRFlastcF|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 ) rHrr|rrUrrrr}r~)rCrs rPis_simple_decorator_trailerrls 9 $    ! # 4 a %2 4 a %3   4DM""a' 4 a %3 4 a %3   4DM""a' 4 a %3 4 a %3'rRcr|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) rHrrrpowerrUallmaprrrs rPis_simple_decorator_expressionrs yEJt yDJ =  a %37qt9LMMNN &&*Q24=3D4PPP   5rRcV|jtjkrdSt|r |jdkrdS|jtjkrdSt |jdkrdS|j\}}}|jtj kr$|jtj krt|SdS)zAReturn True if `node` holds a `yield` or `yield from` expression.TyieldFr ) rHr yield_expr is_name_tokenrerrrUrr}r~is_yield)rClparexprrpars rPrrs yDO##tTtzW44t yDIu 4=Qu}D$ yEJ49 #:#:~~ 5rRrbc|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)rHr%rdrrw)r]rbrs rPrxrxsY y((( (u Avx 5 H 6V rRc<t|jod|jvS)zKReturn True if `leaf` is a multiline string that actually spans many lines. )rrers rPis_multiline_stringrs TZ ( ( ?TTZ-??rRc*t|jdks`|jdjtjks@|jdjtjks |jdjtjkrdSt|jdS)z2Return True if `node` is a suite with a stub body.rrr Fr)rrUrHrrrDEDENT is_stub_bodyrs rP is_stub_suitersx DMa =  EM 1 1 =  EL 0 0 =  EL 0 0u  a( ) ))rRc>t|tr|jtjkrdSt |jdkrdS|jd}|jtjko5t |jdkotd|jDS)zCReturn True if `node` is a simple statement containing an ellipsis.Frrr c3RK|]"}|ttjdkV#dS).N)rrr)rr]s rPrzis_stub_body..s3HHUY,,,HHHHHHrR) rTrrHrrrrUrr)rCrVs rPrrs dD ! !TY$2B%B%Bu 4=Qu M! E di I   1 $ I HHHHH H HrRcbt|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. Frrr`) rTrrHrrrUrr}rer~)rCfirstrs rPis_atom_with_invisible_parensrs$di!7!7u-"DM"$54E5$  J%* $  K2   tT " "  I #   J"  rRc>t|pt|Sr) is_empty_lpar is_empty_rparrs rP is_empty_parrs    5-"5"55rRcB|jtjko |jdkSNr`)rHrr}rers rPrr 9 " 7tzR'77rRcB|jtjko |jdkSr)rHrr~rers rPrrrrRc|j}|j}|j}t|tjko9|dkr|r|jt jkp|dko|o|jt jkS)z9Return True if the given leaf starts an import statement.rcfrom) rdrHreboolrrr import_namer)r]rrrs rP is_importr  ss A A A  UZ (] ?q ?QVt/?%? BV @@af0@&@   rRc$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) rrHrrrerdr with_stmtASYNC next_siblingrs rPis_with_or_async_with_stmtrs  UZ / J&  / K / K  .      U[  5   5   "dn 4   rRct|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. )rrHrrrdr async_stmt async_funcdefrs rPis_async_stmt_or_funcdefr&sJ  U[  F K F K $2D E E  rRr`suffixcx|j}|j}|tjthvo|d|zS)zdReturn True if the given leaf is a special comment. Only returns true for type comments for now.z# type:)rHrerrfr startswith)r]rrrs rPis_type_commentr3s< A A  23 3 X YQWEW8X8XXrR)visiblerdrVrc: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). (r`)rN) rrr}r~rtrrrrr)rdrVrrrrtindexrs rPwrap_in_parenthesesr;s  72CC 3 3D  72CC 3 3D \FEL LLNN aETYud 344II y)))))rRct|jdkrdS|j\}}}|jtjkr|jtjksdS|S)zqReturns `wrapped` if `node` is of the shape ( wrapped ). Parenthesis can be optional. Returns None otherwiser N)rrUrHrr}r~)rCrwrappedrs rPrrMsS 4=Qt-D'4 I # # UZ(?(?t NrRc~|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`). rrN)rHrr}rer~rs rPensure_visibler"[s@  yEJ ej   ! rRnlc,|jtjkSr)rHrrr#s rPrrg 7ej  rRc,|jtjkSr)rHrr}r%s rP is_lpar_tokenr(kr&rRc,|jtjkSr)rHrr~r%s rP is_rpar_tokenr*or&rRc,|jtjkSr)rHrrr%s rPis_string_tokenr,s 7el ""rRc,|jtjkSr)rHrrr%s rPis_number_tokenr/wr-rRc|j}|O|jr|jjtjkrdS|jr|jjt jkrdS|j}|OdS)z6Returns whether this leaf is part of type annotations.NTF)rdrkrHrRARROWrtname)r]rs rPis_part_of_annotationr3{sg{H     X%:%?5<%O%O4 ? x3tzAA4?   5rR)F)r`)rZsystypingrrrrrrr r version_infor typing_extensionsrmypy_extensionsr black.cacher black.stringsrblib2to3rblib2to3.pgen2rblib2to3.pytreerrrr initializepython_symbolsr__annotations__rr[intLeafIDNodeTyperrrrif_stmt while_stmtfor_stmttry_stmtrr funcdefr match_stmt case_blockrrrIr!LESSGREATEREQEQUALNOTEQUAL LESSEQUAL GREATEREQUALr"VBAR CIRCUMFLEXAMPER LEFTSHIFT RIGHTSHIFTPLUSMINUSruSLASH DOUBLESLASHPERCENTrzTILDErr#r$r%rorpr|rsrrr&rr listmakerrtestlist_star_expr subject_exprpatternr'testlambdefor_testand_testnot_test comparisonrwrrrrrtermrr(r2 tname_starr)r8testlistexprlistr9r}r~rRSQBLBRACERBRACEr:setkeysr;valuesr<r=rmr>r1rBrrJrrlrrvrrrrrrrrrrrrrrrxrrrrrrrr rrrrrr"rr(r*r,r/r3r\rRrPrpsl  PPPPPPPPPPPPPPPPPPPPv''''''w       ++++++&&&&&&!!!!!!++++++ 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)U[:L,MMMMM  ...-----gaj--/.-DcTcc#ccccL"(4.$ =HRL =$x7I2J =t = = = = hrl x'9    DbXb\2R2B242222"t2x~t t      r d     MRMDMMMM"'UZ 8 "" " " J"CHo "  """"JCrCdCCCC b402$*D#h-D.@d@t@@@@ * * * * * * r d     t$6t66666888888888888 D T     T d     4 D    YY$YYTYYYYEI****R*T*T****$ r hrl      $    !b!Yt_!!!!!b!Yt_!!!!!b!Yt_!!!!##y######y####        rR