a g g/@sUdZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z m Z ejdkr^ddlmZn ddlmZddlmZmZmZmZmZmZmZmZmZmZddlmZdd lmZm Z e ee fZ!e"Z#e"Z$e"Z%e"Z&d Z'ee(d <d Z)ee(d <dZ*ee(d<dZ+ee(d<dZ,ee(d<dZ-ee(d<ej.dej/dej0dej1dej2dej3dej4dej5dej6dej7dej8dej9dej:dej;diZZ?eGd#d$d$Z@d0ee ee&d%d&d'ZAd1ee ee&d%d(d)ZBe!e&d*d+d,ZCe ee e$d-d.d/ZDdS)2z,Builds on top of nodes.py to track brackets.N) dataclassfield)DictIterableListOptionalSequenceSetTupleUnion))Final) BRACKETCLOSING_BRACKETS COMPARATORSLOGIC_OPERATORSMATH_OPERATORSOPENING_BRACKETSUNPACKING_PARENTSVARARGS_PARENTS is_varargsyms)token)LeafNodeCOMPREHENSION_PRIORITYCOMMA_PRIORITYTERNARY_PRIORITYLOGIC_PRIORITY STRING_PRIORITY COMPARATOR_PRIORITY r r MATH_PRIORITIES DOT_PRIORITYc@seZdZdZdS)BracketMatchErrorzLRaised when an opening bracket is unable to be matched to a closing bracket.N)__name__ __module__ __qualname____doc__r6r6Q/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/black/brackets.pyr1<sr1c@sDeZdZUdZdZeed<eedZ e e e e fefed<eedZe eefed<dZeeed<eedZeeed <eedZeeed <eedZeeed <edd d dZedddZd#eeedddZd$eedddZeed ddZeed ddZ eed ddZ!eed dd Z"eedd!d"Z#dS)%BracketTrackerz"Keeps track of brackets on a line.rdepth)default_factory bracket_match delimitersNprevious_for_loop_depths_lambda_argument_depths invisible)leafreturnc Cs|jtjkrdS|jdkr:|jtvr:|j|jf|jvr:dS|||||jtvr|jd8_z|j|j|jf}Wn4t y}zt d||WYd}~n d}~00||_ |j s|j ||j|_|jdkr0t||j}|r|jdur||jt|j<n t||j}|r0||jt|<|jtvrt||j|jt|jf<|jd7_|j st|j |||_||||dS)aMark `leaf` with bracket-related metadata. Keep track of delimiters. All leaves receive an int `bracket_depth` field that stores how deep within brackets a given leaf is. 0 means there are no enclosing brackets that started on this line. If a leaf is itself a closing bracket and there is a matching opening bracket earlier, it receives an `opening_bracket` field with which it forms a pair. This is a one-directional link to avoid reference cycles. Closing bracket without opening happens on lines continued from previous breaks, e.g. `) -> "ReturnType":` as part of a funcdef where we place the return type annotation on its own line of the previous closing RPAR. If a leaf is a delimiter (a token on which Black can split the line if needed) and it's on depth 0, its `id()` is stored in the tracker's `delimiters` field. Nrr/zDUnable to match a closing bracket to the following opening bracket: )typerCOMMENTr9rr;'maybe_decrement_after_for_loop_variable&maybe_decrement_after_lambda_argumentspopKeyErrorr1opening_bracketvaluer@appendZ bracket_depthis_split_before_delimiterr=r<idis_split_after_delimiterrr maybe_increment_lambda_arguments!maybe_increment_for_loop_variable)selfrArIedelimr6r6r7markLsT           zBracketTracker.mark)rBcCs t|jS)zBReturn True if there is an yet unmatched open bracket on the line.)boolr;rQr6r6r7any_open_bracketssz BracketTracker.any_open_bracketsr6)excluderBcstfdd|jDS)zReturn the highest priority of a delimiter found on the line. Values are consistent with what `is_split_*_delimiter()` return. Raises ValueError on no delimiters. c3s|]\}}|vr|VqdSNr6).0kvrXr6r7 z8BracketTracker.max_delimiter_priority..)maxr<items)rQrXr6r]r7max_delimiter_prioritysz%BracketTracker.max_delimiter_priority)priorityrBcs2|js dSp|tfdd|jDS)zReturn the number of delimiters with the given `priority`. If no `priority` is passed, defaults to max priority on the line. rc3s|]}|krdVqdS)r/Nr6)rZprcr6r7r^r_z?BracketTracker.delimiter_count_with_priority..)r<rbsumvalues)rQrcr6rer7delimiter_count_with_prioritys z,BracketTracker.delimiter_count_with_prioritycCs:|jtjkr6|jdkr6|jd7_|j|jdSdS)zIn a for loop, or comprehension, the variables are often unpacks. To avoid splitting on the comma in this situation, increase the depth of tokens between `for` and `in`. forr/TF)rCrNAMErJr9r>rKrQrAr6r6r7rPs z0BracketTracker.maybe_increment_for_loop_variablecCsL|jrH|jd|jkrH|jtjkrH|jdkrH|jd8_|jdSdS)z>See `maybe_increment_for_loop_variable` above for explanation.inr/TF)r>r9rCrrjrJrGrkr6r6r7rEs  z6BracketTracker.maybe_decrement_after_for_loop_variablecCs:|jtjkr6|jdkr6|jd7_|j|jdSdS)zIn a lambda expression, there might be more than one argument. To avoid splitting on the comma in this situation, increase the depth of tokens between `lambda` and `:`. lambdar/TF)rCrrjrJr9r?rKrkr6r6r7rOs z/BracketTracker.maybe_increment_lambda_argumentscCsB|jr>|jd|jkr>|jtjkr>|jd8_|jdSdS)z=See `maybe_increment_lambda_arguments` above for explanation.rlr/TF)r?r9rCrCOLONrGrkr6r6r7rFs  z5BracketTracker.maybe_decrement_after_lambda_argumentscCs|j|jdtjfS)z7Return the most recent opening square bracket (if any).r/)r;getr9rRSQBrVr6r6r7 get_open_lsqbszBracketTracker.get_open_lsqb)r6)r)$r2r3r4r5r9int__annotations__rdictr;rr DepthNodeTyperr<LeafIDPriorityr=rlistr>rr?r@rTrUrWrrbrhrPrErOrFrrr6r6r6r7r8@s"  "<    r8)rAr=rBcCs|jtjkrtSdS)zReturn the priority of the `leaf` delimiter, given a line break after it. The delimiter priorities returned here are from those delimiters that would cause a line break after themselves. Higher numbers are higher priority. r)rCrCOMMArrAr=r6r6r7rNs rNcCs8t|ttBdrdS|jtjkrP|jrP|jjtjtj hvrP|dusL|jt vrPt S|jt vr~|jr~|jjtj tjhvr~t|jS|jtvrtS|jtjkr|dur|jtjkrtS|jtjtjhvrdS|jdkr|jr|jjtjtjhvs|jtjkrt|jtr|jjdkrtS|jdkrF|jrF|jjtjtjhvrFtS|jdvrn|jrn|jjtjkrnt S|jdkr~tS|jd kr|jr|jjtj!tj"hvr|dur|jtjkr|jd kstS|jd kr|jr|jjtj!kr|dur|jtjkr|jdkstS|jt#vr4|jr4t$SdS) zReturn the priority of the `leaf` delimiter, given a line break before it. The delimiter priorities returned here are from those delimiters that would cause a line break before themselves. Higher numbers are higher priority. )ZwithinrNriasyncif>r~elseisrmnot)%rrrrCrDOTparentrZ import_fromZ dotted_namerr0rZfactorZ star_exprr.rr'STRINGr%rjASYNCrJZcomp_forZ old_comp_for isinstanceZ prev_siblingrrZcomp_ifZ old_comp_iftestr!Zcomp_opZ comparisonrr#r|r6r6r7rLs        $      rL)noderBcCs|jtjkrdS|jd}|jd}|jtjkr<|jtjks@dSt}|jddD]2}t|t rn| |qT| D]}| |qvqTz | WSt yYdS0dS)zReturn maximum delimiter priority inside `node`. This is specific to atoms with contents contained in a pair of parentheses. If `node` isn't an atom or there are no enclosing parentheses, returns 0. rrlr/N)rCrZatomchildrenrLPARRPARr8rrrTleavesrb ValueError)rfirstlastZbtcrAr6r6r7max_delimiter_priority_in_atomGs         r)rrBc Csztddt|D}Wnty2tYS0g}t}t|t|D]}||}|jtvrv|t |j|f|jt vrL|r|j|ddkr| \}}t||dD]}| t ||qqLqqL|S)zReturn leaves that are inside matching brackets. The input `leaves` can have non-matching brackets at the head or tail parts. Matching brackets are included. css |]\}}|jtvr|VqdSrY)rCr)rZilr6r6r7r^ksz6get_leaves_inside_matching_brackets..rlrr/)next enumerate StopIterationsetrangelenrCrrKrrrGaddrM) r start_indexZ bracket_stackidsrrA_startjr6r6r7#get_leaves_inside_matching_bracketscs&      r)N)N)Er5sysZ dataclassesrrtypingrrrrrr r r version_infoZtyping_extensionsrZ black.nodesrrrrrrrrrrZblib2to3.pgen2rZblib2to3.pytreerrLNrsrvrxrwryrrtrr!r#r%r'VBAR CIRCUMFLEXAMPER LEFTSHIFT RIGHTSHIFTPLUSMINUSSTARSLASH DOUBLESLASHPERCENTATTILDE DOUBLESTARr.r0 Exceptionr1r8rNrLrrr6r6r6r7sX(  0          `