yIf.tV UdZddlZddlmZmZmZmZmZmZm Z m Z m Z m Z ej dkrddlmZnddlmZddlmZddlmZddlmZmZdd lmZmZdd lmZdd lmZdd lmZm Z m!Z!m"Z"ej#eej$Z%ee&d <e dZ'e e e!fZ(e)Z*e)Z+ej,ej-ej.hZ/ee&d<e%j0e%j1e%j2e%j3e%j4e%j5e%j6e%j7e%j8e%j9h Z:ee&d<dZ;ee&d<dej<e;<ddhZ=ee&d<ej>ej?ej@ejAejBejChZDee&d<ejEejFejGejHejIejJejKejLejMejNejOejPejQejRhZSee&d<ejLejRhZTee&d<eTejMhzZUee&d<e%jVe%jWe%jXe%jYe%jZhZ[ee&d<e%j\e%j]e%j^e%j_e%j`e%jae%jbhZcee&d<e%jde%jee%jfe%jge%jhe%jie%jje%jke%jle%jme%jne%joe%jXe%jpe%jqe%jrhZsee&d<e%jte%juhZvee&d<hdZwee&d<e%jxe%j`e%jyhZzee&d <ej{ej|ej}ej~ejejiZee&d!<eeZee&d"<eeZee&d#<eezZee&d$<eeje;ejejejhzZee&d%<d&Zed'(Gd)d*ee'Zd+e d,ed-ed.efd/Zdxd1e)d2ed3ed.efd4Zd5ee(d.ee fd6Zd5ee(d7eee+d.efd8Zd5ee(d.ee+fd9Zd:e!d;e(d.ee(fd<Zd=e(d>e(d.dfd?Zd+e d.e(fd@Zd5e(d.ee fdAZd5e(d.efdBZd5ed-ed.efdCZd5e(d.efdDZd5e(d.efdEZd5e(d.efdFZej{ej|ffdGe dHe dIee dJe e)e)fd.ef dKZd5e(d.efdLZdyd5e(dNed.efdOZd5e(d.efdPZd5e(d.efdQZd+e dRe e+d.efdSZd5e!d.efdTZd5e!d.e fdUZd5e(d.efdVZd5e!d.efdWZd5e!d.efdXZd5e!d.efdYZd5e(d.efdZZd5e(d.efd[Zd+e d.efd\Zd+e d.efd]Zd+e d.efd^Zd+e d.efd_Zd+e d.efd`Zd+e d.efdaZd+e d.efdbZd+e d.efdcZdded.efdeZd'dfdge!dhe(died.dfdjZd5e(d.ee(fdkZd+e d.dfdlZdmed.ee fdnZdmed.ee fdoZdmed.ee fdpZdmed.ee fdqZd+e d.edrfdsZd+e d.efdtZd5e(d.ee fduZd5e(d.ee fdvZd+e d.e(fdwZdS)zz> blib2to3 Node/Leaf transformation-related utility functions. N) FinalGenericIteratorListLiteralOptionalSetTupleTypeVarUnion) ) 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)selfrGnamevisitfs ?/var/www/piapp/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) isinstancerchildrenrU)rQrGchilds rTrPzVisitor.visit_defaultsd dD ! ! - - -::e,,,,,,,,,, - - - -rVN) __name__ __module__ __qualname____doc__LNrrrUrPrVrTrFrFsfLL0"0!00000-"-!------rVrFleafcomplex_subscriptmoderHcd}d}d}|j}|j}|j}|tvr|S|tjkr|S| Jd||tjkr,|jtjtj tj hvr|S|tj kr|jtj kr|S|j } | st|} | r| jtvr|S|tjkr2| jtjkr|S| jtjkr|s|S|S| jtjkrf| jr]| jjtjtjtjtjhvr|S| jjtjkr| jSn]| jtjkrAt3| tjkr$t3| jtj kr|S| jt6vr!t9| t:t<zr|Sn| jtjkr2| jr*| jjtjtj hvr|r|n|Sn| jr*| jjtjkr| jt@vr|S| jtj!kr#|jr|jjtj"kr|Sn'| jtvr|S| jtj#kr|S|jtjtjhvr| r| jtjkr|Sn|jtjkr| r| jtjkr|Sn|jtjkr[| s|S|tjkr| jtHvr|Snh| jtjkr| jS| jtjkr|Sn3|jtHvr,| 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| jtjht6zvr|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|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|tj0tj1tj2hvr|Sn|jtj3kr\|tj(kr| r| jtj(kr|Snp|tj0kr!|dkr|S| r| jtj(kr|Sn>|jtj kr|S|jtj4kr|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)5rLparentvaluerBrCOMMENTCOLONr subscript subscriptlistsliceopLBRACEfstring_replacement_field prev_siblingpreceding_leafr?COMMAEQUALarglistargument parameters varargslist typedargslistprefixSTAR parent_type star_exprr( is_varargr)r*factorr&AT decoratorBANGr,trailerLPARRPARDOTLSQB dotted_nameclassdef COLONEQUALatom dictsetmaker DOUBLESTARNAMENUMBERSTRING import_from except_clause) rarbrcNOSPACE DOUBLESPACEtpvprevprevp prevp_parents rT whitespacers BE"K A A AO EM ==TDTT===EKAF   +  ELQVt'EEE  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 & & & 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 LrV nl_count form_feed empty_linec,|r||dz zdz|zS||zS)z$Generate a normalized prefix string. r`)rrrs rTmake_simple_prefixrs/Ahl+t3j@@   rVrGc|rb|j}|rPt|tr|S t|dS#t $rYdSwxYw|j}|bdS)z3Return the first leaf that precedes `node`, if any.N)rrrXrlistleaves IndexErrorri)rGress rTrsrss    #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)rLprev_siblings_arerr)rGrs rTrrsc t bzt| u yF2Ju T.ss < <  ?1  $ 3 3  ;$/ ) )    *v/B/G8/S/S    rVcvt|tr|S|jrt|jdSdS)z(Returns the first leaf of the node tree.rN)rXrrY first_leaf_ofrs rTrrs=$  }T]1-...trVcl|jtjtjtjtjhvS)z?Whether node is an arithmetic or a binary arithmetic expression)rLr arith_expr shift_exprxor_exprand_exprrs rT is_arith_liker$s* 9     rVct|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)rXrrLrrrrjset intersectionrunify_docstring_detectionrir simple_stmtrrrrNEWLINEINDENTrxrl)rGrcr{s rT 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 5rVc|jtjkoWt|jdko?|jdjt jko|jdjt jkS)z+Return True if `node` holds an empty tuple.rr)rLrrlenrYrrrrs rTis_empty_tuplerOs` TY 0   ! # 0 M!  !UZ / 0 M!  !UZ / rVcx|jtjkr`t|}||jtjkrdSt |jdko|jdjtjkS|jtvo7t |jdko|jdjtjkS)zMReturn True if `node` holds a tuple with one element, with or without parens.NFrr) rLrrunwrap_singleton_parenthesis testlist_gexprrYrrtr=rGgexps rT is_one_tuplerYs yDI+D11 <49(:::54=!!Q&O4=+;+@EK+OO ^# 1   ! # 1 M!  !U[ 0rVc|jtjkrdSt|}||jtjkrdSt d|jDS)zDReturn True if `node` holds a tuple that contains a walrus operator.FNc3@K|]}|jtjkVdSN)rLrnamedexpr_test).0rZs rT z-is_tuple_containing_walrus..qs,LLUuzT00LLLLLLrV)rLrrrranyrYrs rTis_tuple_containing_walrusris[ yDIu ' - -D |tyD$666u LLdmLLL L LLrVopeningclosingrbracketsc|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) rL bracket_depth enumerate LookupErrorrrtrirrvrz) rrrrdepth_opening_indexracommasrs rTis_one_sequence_betweenrts  gl#x//u  !A %E )& 1 1AA 7?? E ?@@@ FaN~'   7?? E* E ! !di5;&>&> aKF{ t{/ "4  !  A:rVcRt|}|duo|jtjkS)z7Return True iff `node` is of the shape ( test := test )N)rrLrr)rGinners rTis_walrus_assignmentrs) ( . .E   Bt/B!BBrVFlastcF|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 ) rLrrrrYrrrrr)rGrs rTis_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'rVcr|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) rLrrrpowerrYallmaprrrs rTis_simple_decorator_expressionrs yEJt yDJ =  a %37qt9LMMNN &&*Q24=3D4PPP   5rVcV|jtjkrdSt|r |jdkrdS|jtjkrdSt |jdkrdS|j\}}}|jtj kr$|jtj krt|SdS)zAReturn True if `node` holds a `yield` or `yield from` expression.TyieldFr ) rLr yield_expr is_name_tokenrjrrrYrrris_yield)rGlparexprrpars rTrrs yDO##tTtzW44t yDIu 4=Qu}D$ yEJ49 #:#:~~ 5rVrgc|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)rLr(rirr~)rargrs rTrrsY y((( (u Avx 5 H 6V rVc,|jtjkS)z&Return True if the node is an f-string)rLrfstringrs rT is_fstringrs 9 $$rVct|t|jd}ttj||j}|pd|_|S)z/Converts an fstring node back to a string node.N)r{r)rNrr{rrr get_linenolineno)rGstring_without_prefix string_leafs rTfstring_to_stringr sYIIc$+&6&6&8&89u|%:4;OOOK**/aK rVct|trt|rt|}nt|tr|}ndSt |jod|jvS)zKReturn True if `leaf` is a multiline string that actually spans many lines.Fr)rXrrr rrrj)rGras rTis_multiline_stringr  sl$*T"2"2 && D$  u TZ ( ( ?TTZ-??rVc|jtjtjhvsJ|jJ|jjtjtjhvSr)rLrsuiterrifuncdefrrs rTis_parent_function_or_classrsF 9T%56 6 6 6 6 ; " " " ;  dm< <.Es3HHUY,,,HHHHHHrV) rXrrLrrrrYr{rrr)rGrZs rTrr8s dD ! !TY$2B%B%Bu 4=Qu M! E L     I J$) # I   1 $ I HHHHH H H rVcbt|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. Frrre) rXrrLrrrYrrrjr)rGfirstrs rTis_atom_with_invisible_parensr Is$di!7!7u-"DM"$54E5$  J%* $  K2   tT " "  I #   J"  rVc>t|pt|Sr) is_empty_lpar is_empty_rparras rT is_empty_parr%[s    5-"5"55rVcB|jtjko |jdkSNre)rLrrrjr$s rTr"r"_ 9 " 7tzR'77rVcB|jtjko |jdkSr')rLrrrjr$s rTr#r#cr(rVc|j}|j}|j}t|tjko9|dkr|r|jt jkp|dko|o|jt jkS)z9Return True if the given leaf starts an import statement.rhfrom) rirLrjboolrrr import_namer)rarrrs rT is_importr.gss A A A  UZ (] ?q ?QVt/?%? BV @@af0@&@   rVc$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,rLrrrjrir with_stmtASYNC next_siblingr$s rTis_with_or_async_with_stmtr4us  UZ / J&  / K / K  .      U[  5   5   "dn 4   rVct|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,rLrr2rir async_stmtrr$s rTis_async_stmt_or_funcdefr7sJ  U[  F K F K $2D E E  rVcr|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:)rLrjrrkr! startswithrarrs rTis_type_commentr;s6 A A  23 3 O Y8O8OOrVcf|j}|j}|tjthvot |S)zGReturn True if the given leaf is a type comment with ignore annotation.)rLrjrrkr!is_type_ignore_comment_stringr:s rTis_type_ignore_commentr>s2 A A  23 3 X8UVW8X8XXrVrjc,|dS)zSReturn True if the given string match with type comment with ignore annotation.z# type: ignore)r9)rjs rTr=r=s   , - --rV)visiblerirZr@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). (re)rN) rrrrr{rrrrr)rirZr@rrr{indexrs rTwrap_in_parenthesesrEs  72CC 3 3D  72CC 3 3D \FEL LLNN aETYud 344II y)))))rVct|jdkrdS|j\}}}|jtjkr|jtjksdS|S)zqReturns `wrapped` if `node` is of the shape ( wrapped ). Parenthesis can be optional. Returns None otherwiser N)rrYrLrrr)rGrwrappedrs rTrrsS 4=Qt-D'4 I # # UZ(?(?t NrVc~|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`). rBrCN)rLrrrjrr$s rTensure_visiblerIs@  yEJ ej   ! rVnlc,|jtjkSr)rLrrrJs rTrr 7ej  rVc,|jtjkSr)rLrrrLs rT is_lpar_tokenrOrMrVc,|jtjkSr)rLrrrLs rT is_rpar_tokenrQrMrVc,|jtjkSr)rLrrrLs rTis_number_tokenrSs 7el ""rV)rHparamNc|j}|O|jr|jjtjkrdS|jr|jjt jkrdS|j}|OdS)zrkeysr?valuesr@rArtFSTRING_MIDDLE FSTRING_ENDrrBrVrFr,rNrrrsrr}rrrrrrrrrrrrrrrrr r rrrrr r%r"r#r.r4r7r;r>r=rErrIrrOrQrSrXrZr\r^r`r`rVrTrs7                          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) K   J -  ...-----gaj--/.-DmTmmTmcmmmm`!!!!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>,,,,4BrV