B Khh/@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__r6r61/tmp/pip-install-vv81h98y/black/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:|jtkr:|j|jf|jkr:dS|||||jtkr|jd8_y|j|j|jf}Wn2t k r}zt d||Wdd}~XYnX||_ |j s|j ||j|_|jdkr.t||j}|r|jdk r||jt|j<n t||j}|r.||jt|<|jtkrr||j|jt|jf<|jd7_|j sr|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)selfrArIedelimr6r6r7markLsF              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|]\}}|kr|VqdS)Nr6).0kv)rWr6r7 sz8BracketTracker.max_delimiter_priority..)maxr<items)rQrWr6)rWr7max_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)rXp)r_r6r7r[sz?BracketTracker.delimiter_count_with_priority..)r<r^sumvalues)rQr_r6)r_r7delimiter_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>rK)rQrAr6r6r7rPs 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>r9rCrrerJrG)rQrAr6r6r7rEs   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)rCrrerJr9r?rK)rQrAr6r6r7rOs z/BracketTracker.maybe_increment_lambda_argumentscCsB|jr>|jd|jkr>|jtjkr>|jd8_|jdSdS)z=See `maybe_increment_lambda_arguments` above for explanation.rfr/TF)r?r9rCrCOLONrG)rQrAr6r6r7rFs  z5BracketTracker.maybe_decrement_after_lambda_argumentscCs|j|jdtjfS)z7Return the most recent opening square bracket (if any).r/)r;getr9rRSQB)rQr6r6r7 get_open_lsqbszBracketTracker.get_open_lsqb)r6)r)$r2r3r4r5r9int__annotations__rdictr;rr DepthNodeTyperr<LeafIDPriorityr=rlistr>rr?r@rTrUrVrr^rcrPrErOrFrlr6r6r6r7r8@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)rCrCOMMAr)rAr=r6r6r7rNs rNcCs8t|ttBdrdS|jtjkrP|jrP|jjtjtj hkrP|dksL|jt krPt S|jt kr~|jr~|jjtj tjhkr~t|jS|jtkrtS|jtjkr|dk r|jtjkrtS|jtjtjhkrdS|jdkr|jr|jjtjtjhks|jtjkrt|jtr|jjdkrtS|jdkrF|jrF|jjtjtjhkrFtS|jdkrn|jrn|jjtjkrnt S|jdkr~tS|jd kr|jr|jjtj!tj"hkr|dk r|jtjkr|jd kstS|jd kr|jr|jjtj!kr|dk r|jtjkr|jdkstS|jt#kr4|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. )ZwithinrNrdasyncif>rwelseisrgnot)%rrrrCrDOTparentrZ import_fromZ dotted_namerr0rZfactorZ star_exprr.rr'STRINGr%reZASYNCrJZcomp_forZ old_comp_for isinstanceZ prev_siblingrrZcomp_ifZ old_comp_iftestr!Zcomp_opZ comparisonrr#)rAr=r6r6r7rLsf        $       rL)noderBcCs|jtjkrdS|jd}|jd}|jtjkr<|jtjks@dSt}xH|jddD]6}t|t rp| |qVx| D]}| |qzWqVWy| St k rdSXdS)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. rrfr/N)rCrZatomchildrenrLPARRPARr8r~rrTleavesr^ ValueError)rfirstlastZbtcrAr6r6r7max_delimiter_priority_in_atomGs      r)rrBc Csytddt|D}Wntk r0tSXg}t}xt|t|D]}||}|jtkrv|t |j|f|jt krL|r|j|ddkr| \}}x,t||dD]}| t ||qWqLPqLW|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 |]\}}|jtkr|VqdS)N)rCr)rXilr6r6r7r[lsz6get_leaves_inside_matching_brackets..rfrr/)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.pytreerrLNrmrprrrqrsrrnrr!r#r%r'VBAR CIRCUMFLEXAMPER LEFTSHIFT RIGHTSHIFTPLUSMINUSSTARSLASH DOUBLESLASHPERCENTATTILDE DOUBLESTARr.r0 Exceptionr1r8rNrLrrr6r6r6r7sT(  0         `