^h SrSSKrSSKrSSKJrJr SSKJr SSKJ r J r SSK J r J r SSKJrJrJr SSKJrJrJrJrJr SS KJrJrJr SS KJrJrJrJ r J!r!J"r"J#r# SS K$J%r%J&r&J'r' SS K(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;Jr>J?r?J@r@JArAJBrBJCrCJDrDJErEJFrFJGrGJHrHJIrIJJrJJKrKJLrLJMrM SS KNJOrO SSKPJQrQJRrRJSrSJTrTJUrU SSKVJWrWJXrXJYrYJZrZJ[r[J\r\J]r] SSK^J_r_ SSK`JaraJbrb \crd\\a\b4re"SS\W5rf"SS\0\5rgS\S\\%S\&S\\h4SjriSPS\S\&S\\%S\\4SjjrjS\S\&S\k4Sjrl"SS\ 5rmS\S\\%S\&S\\4S jrnSQS\S\&S\\%S!\\dS\\4 S"jjroSPS\S!\\dS\4S#jjrpSQS$\S\S\&S\\%S!\\dS\\4 S%jjrqS&\S$\S\&S\k4S'jrrS(\S)\S*\SS4S+jrsS,\t\aS-\S.\aS\k4S/jruS,\t\aS-\S.\aS0\mS\4 S1jrvS2\\S\\4S3jrwS\S\\c4S4jrxS5\aS\\%S\k4S6jryS7\kS8\cS\S\4S9jrz\\1r{\wS\S\\%S\&S\\4S:j5r|\wS\S\\%S\&S\\4S;j5r}S<\bS=\~\hS\&S\\%SS4 S>jrS?\bS@\eSA\cSS4SBjrS<\bS\&S\\%SS4SCjrS<\bS\&S\\%SS4SDjrS<\bS?\bS\&S\\%SS4 SEjrSRS<\eS?\eS\&S\\%SF\kS\k4 SGjjrS\S.\aS\k4SHjrS\SI\cS\\~\d4SJjrSKSL.S\SM\\S\&S\\%SN\hS\t\4 SOjjrg)Sz Generating lines of code. N) CollectionIterator)replace)Enumauto)partialwraps)OptionalUnioncast)COMMA_PRIORITY DOT_PRIORITYSTRING_PRIORITY#get_leaves_inside_matching_bracketsmax_delimiter_priority_in_atom)FMT_OFFgenerate_comments list_comments)Line RHSResult append_leaves can_be_splitcan_omit_invisible_parensis_line_short_enoughline_to_string)FeatureModePreview)% ASSIGNMENTSBRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENT STATEMENT WHITESPACEVisitorensure_visiblefstring_to_stringget_annotation_typehas_sibling_with_type is_arith_likeis_async_stmt_or_funcdefis_atom_with_invisible_parens is_docstringis_empty_tuple is_generator is_lpar_tokenis_multiline_string is_name_tokenis_one_sequence_between is_one_tupleis_parent_function_or_classis_part_of_annotation is_rpar_token is_stub_body is_stub_suiteis_tupleis_tuple_containing_staris_tuple_containing_walrusis_type_ignore_comment_string is_varargis_walrus_assignmentis_yieldsymswrap_in_parentheses)normalize_numeric_literal)fix_multiline_docstringget_string_prefixnormalize_string_prefixnormalize_string_quotes"normalize_unicode_escape_sequences)CannotTransform StringMergerStringParenStripperStringParenWrapperStringSplitter Transformer hug_power_op)token)LeafNodec\rSrSrSrSrg) CannotSplit_zBA readable split that fits the allotted line length is impossible.N)__name__ __module__ __qualname____firstlineno____doc____static_attributes__rWG/home/james-whalen/.local/lib/python3.13/site-packages/black/linegen.pyrUrU_sLr^rUc ^\rSrSrSrS\S\\SS4SjrS+S\ S\ \ 4S jjr S \ S\ \ 4U4S jjrS \S\ \ 4S jrS \S\ \ 4S jrS \S\ \ 4SjrS \S\\S\\S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4SjrS \S\ \ 4Sjr S \S\ \ 4Sjr!S\S\ \ 4Sjr"S\S\ \ 4Sjr#S\S\ \ 4S jr$S \S\ \ 4S!jr%S \S\ \ 4S"jr&S\S\ \ 4S#jr'S\S\ \ 4S$jr(S \S\ \ 4S%jr)S \S\ \ 4S&jr*S \S\ \ 4S'jr+S \S\ \ 4S(jr,S,S)jr-S*r.U=r/$)- LineGeneratorezGenerates reformatted Line objects. Empty lines are not emitted. Note: destroys the tree it's visiting by mutating prefixes of its leaves in ways that will no longer stringify to valid Python code on the tree. modefeaturesreturnNc@XlX lU UR5 gN)rcrd __post_init__)selfrcrds r___init__LineGenerator.__init__ls    r^indentc#p# UR(d UR=RU- slg[URR5S:Xa([ URRS5(agURn[ UR URU-S9UlUv g7f)zGenerate a line. If the line is empty, only emit if it makes sense. If the line is too long, split it first and then generate. If any lines were generated, set up a new current_line. Nrrcdepth) current_linerplenleavesr,rrc)rirl complete_lines r_lineLineGenerator.liners     # #v - #  t  '' (A -2J    $ $Q '3 3  ))  dii}7J7JV7STsB4B6nodec#># [U[5(Ga5URRR 5n[ U5HnU(aURR U5 M'UR[R:Xa5URR U5 UR5ShvN MzUR5ShvN URR U5 UR5ShvN M U(aSUl UR[;aURR U5 [TU]9U5ShvN gNNNaN 7f)zCDefault `visit_*()` implementation. Recurses to children of `node`.N) isinstancerRrqbracket_trackerany_open_bracketsrappendtyperQCOMMENTruprefixr%super visit_default)rirwr|comment __class__s r_rLineGenerator.visit_defaults dD ! ! $ 1 1 A A S S U ,T2$%%,,W5\\U]]2%%,,W5#yy{** $yy{**%%,,W5#yy{**3 !  yy *!!((.7(...+++ /sIB9E/<E'=E/E)2E/ E+ AE/!E-"E/)E/+E/-E/c## UR=(a' URR[R:HnU(dp[ [RS5n[ [R S5nUR nSUlXSlURSU5 URU5 URU5ShvN gN7f)zVisit an `x if y else z` testryrN) prev_siblingr~rQLPARrRRPARr insert_child append_childr)rirwalready_parenthesizedlparrparrs r_ visit_testLineGenerator.visit_tests    F$"3"3"8"8EJJ"F % B'D B'D[[FDK K   a &   d #%%d+++sCCC Cc#x# URS5ShvN URU5ShvN gNN7f)z/Increase indentation level, maybe yield a line.rnNrurrirws r_ visit_INDENTLineGenerator.visit_INDENTs499R=  %%d+++ !+s:6:8::c## UR5ShvN URU5ShvN URS5ShvN gN7N N 7f)z/Decrease indentation level, maybe yield a line.Nrrs r_ visit_DEDENTLineGenerator.visit_DEDENTsN 99;%%d+++99R=    , !s1AAAAAA AAAkeywordsparensc## [XURURS9 URHTn[ U5(a(UR U;aUR 5ShvN URU5ShvN MV gN"N 7f)a~Visit a statement. This implementation is shared for `if`, `while`, `for`, `try`, `except`, `def`, `with`, `class`, `assert`, and assignments. The relevant Python language `keywords` for a given statement will be NAME leaves within it. This methods puts those on a separate line. `parens` holds a set of string leaf values immediately after which invisible parens should be put.  parens_afterrcrdN)normalize_invisible_parensrcrdchildrenr3valueruvisit)rirwrrchilds r_ visit_stmtLineGenerator.visit_stmtsk # DII  ]]EU## x(?99;&&zz%( ( ( #& (s$A"B $B%B =B > B  B c#j# URU5ShvN SURSlgN7fNryrrrrrs r_visit_typeparamsLineGenerator.visit_typeparams/%%d+++"$ a , 313c#j# URU5ShvN SURSlgN7fNryrnrrs r_visit_typevartuple LineGenerator.visit_typevartuplerrc#j# URU5ShvN SURSlgN7frrrs r_visit_paramspecLineGenerator.visit_paramspecrrc #"# [RUR;a[UR5Hup#US:XaM URUS- R [ R:XdM=UR [R:XaTURSR [;a3[U5(d#[UUURURSS9 M[XSS9 M URU5ShvN gN7f)NrrnFparentrcrdremove_brackets_around_commavisible)rwrap_long_dict_values_in_parensrc enumeraterr~rQCOLONrBatomr"r@#maybe_make_parens_invisible_in_atomrdrCr)rirwirs r_visit_dictsetmaker LineGenerator.visit_dictsetmakers  2 2dii ?%dmm46==Q',, ; dii/!NN1-226FF 4U ; ;;!#'!%%)]]9> ,DG#5$%%d+++sA-D3BDD Dc #(# UR5ShvN SnURHnUR[R:XaSnM%U(dM.UR[ R :Xa\URSR[R:Xa1[UUURURSS9(a [XSS9 O [XSS9 SnM URHnURU5ShvN M gNN 7f)zVisit function definition.NFTrrr) rurr~rQRARROWrBrrrrcrdrCr)rirwis_return_annotationrs r_ visit_funcdefLineGenerator.visit_funcdefs99; %]]EzzU\\)'+$%%::*u~~a/@/E/E/S:#!YY!%5: ,DG'UC',$#"]]Ezz%( ( (#+ , )s(DD -(2 ** 8) rn()rN)rr~rBpowerrrrQ DOUBLESTARrRrrremoverrSrr)rirw _operatoroperandrrindexs r_ visit_factorLineGenerator.visit_factors "]] LLDJJ &G$$%*  #((E,<,<< C(D C(DNN$)E   eT$))dT5J%K L%%d+++sC>DDDc## [UR5S:Xde[URSXRURS9(a[ XRSSS9 UR U5ShvN gN7f)z Add potential parentheses around types in function parameter lists to be made into real parentheses in case the type hint is too long to fit on a line Examples: def foo(a: int, b: float = 7): ... -> def foo(a: (int), b: (float) = 7): ... rrrrcrdFrN)rrrrrcrdrCrrs r_ visit_tnameLineGenerator.visit_tnamesg4==!Q&&& . MM! T DMM  mmA&6 F%%d+++sA8B:B;Bc## [U5 [U5(Ga>[R"SUR5(GdUR R (a![UR5n[U5nO URn[U5nU[U5SnUSnUSU:waSOSnX%U*nU(+nSURR-n[U5(a [X'5nOUR5nSnU(a]USU:XaSU-nUSU:XaUS- nUSS :Xa7[U5[UR!S 55- n U S -(aUS- nS nO U(dSnXE-n US:XaUR#5n U(a[U S5OSn [U 5S:aX-UR R$:ai[U5U-UR R$::aCU(d<URSU- S :XaX:-U-U -UlO5X:-U-S -U-U -UlOX:-U-U -UlOX:-U-U -UlUR R (aRUR&[(R*:Xa4[UR5Ul[UR5UlUR-U5ShvN gN7f) Nz\\\s*\nrrnrz F r\rT )rIr.researchrrcstring_normalizationrGrHrFrrrqrpr2rEstriprstrip splitlines line_lengthr~rQSTRINGr) rir docstringr quote_char quote_lendocstring_started_emptyrlhas_trailing_backslashbackslash_countquotelineslast_line_lengths r_ visit_STRINGLineGenerator.visit_STRINGs*40   bii DJJ&G&Gyy--3DJJ? 4I>  JJ &y1F!#f+-0I"1J 'q\Z7QI!YJ7I*3m #t00666F"4((3IF %OO- %* "Q<:- #iIR=J.$IR=D(&))ns9;K;KD;Q7R&RO&*"S( 15., *EA~ ",,.5>3uRy>A JN(4tyy7L7LLF i/4993H3HH2zz"y.1T9%+^i%?%%G %+^i%?$%F%ORW%W !')!;e!CDJ#^i7%? 99 ) )dii5<<.G0.LineGenerator.visit_fstring..s4- &zzT;;; DCJ &s"==)r(rranyrrr)rirw string_leafs r_ visit_fstringLineGenerator.visit_fstrings{'-  [! ;$$ $- - * * ))+6 6 6 $$[111 71s$A"B$B%B>B?BBc## [RUR;a![US1URURS9 UR U5ShvN gN7f)Ninr)rwrap_comprehension_inrcrrdrrs r_visit_comp_forLineGenerator.visit_comp_forFsH  ( (DII 5 &D6 DMM %%d+++sAAAAc#B# URU5ShvN gN7frg)r<rs r_visit_old_comp_for LineGenerator.visit_old_comp_forMs&&t,,,s c[URS9UlURn[ 5n[ US1SS1S9Ul[ U1SkSS1S9Ul[ USS 1S1S9Ul[ US S 1S S 1S9Ul [ U1S kUS9Ul [ US 1S 1S9Ul [ US1S1S9Ul [ US1US9Ul [ X[S9Ul[ US1S1S9Ul[ XS1S9Ul[ XS1S9UlUR&UlUR*UlUR.UlUR.Ul[ XS1S9Ulg)z,You are in a twisty little maze of passages.rcassert,)rr>ifelifelserErFwhilerGforr:>tryrGexceptfinallyrKwithclassreimportdelN)rrcrqrrrvisit_assert_stmt visit_if_stmtvisit_while_stmtvisit_for_stmtvisit_try_stmtvisit_except_clausevisit_with_stmtvisit_classdefrvisit_expr_stmtvisit_return_stmtvisit_import_fromvisit_del_stmtrvisit_async_funcdefrvisit_decoratedrvisit_match_stmtvisit_case_block visit_guard)rivØs r_rhLineGenerator.__post_init__Psm dii0 OOu!(hZSV!X$ .f~ !(Wf4Ewi X%a5&/5RV-X% )4)T)htn)00$00d0x~00-T-htn- )T)htn) ,,$,0tDXd^ ,T,htn, ,,(4.,",,$,&V,V,(4.V,p,,(4.,,t,,$ 2$ 28D> 2`,4,HTN,-t--BBr^rarurdrcrec `[[[XU555$![a gf=frg)rrrPrJ)rurdrcs r_!_hugging_power_ops_line_to_stringrips0 d<#EFGG s  --c ## UR(aUv g[U5n[XU5=(d UnURnURn[ XV5n[ XV5n[XV5n [XV5n UR5(dUR(dUR(dz[XUS9(dUR5(aVUR(aUR5(d0UR!5(d["R$U;aXx/n O/n OUR&(a[)X5(d[*/n OS[,S[.S[0[2S[4S[6[.4 Sjn [9S S S U 05"5n ["R$U;a+UR(aUUU [:[<U U /n O*UUU U U /n O"UR(a[:[<U /n OU /n U R?[@5 U Hn[CXXUS 9nUShvN g Uv gN ![Da M1f=f7f) zTransform a `line`, potentially splitting it into many lines. They should fit in the allotted `line_length` but might not be able to. `features` are syntactical features that may be used in the output. N)rcline_strrirurdrcrec 3# [XR5H4n[[XX$S95n[ USUS9(dM*UShvN g [XUS9ShvN gNN7f)zWraps calls to `right_hand_split`. The calls increasingly `omit` right-hand trailers (bracket pairs with content), meaning the trailers get glued together to split on another bracket pair instead. omitrrBN)rd)generate_trailers_to_omitrlistright_hand_splitr)rirurdrcrnrs r__rhstransform_line.._rhssk2$8H8HI-d(NO (at<<$$$J(XF F F% Gs)\rSrSr\"5r\"5r\"5rSrg)_BracketSplitComponentirWN) rXrYrZr[rheadbodytailr]rWr^r_rrs 6D 6D 6Dr^r _featuresc#&# [R[R4GH`n/n/n/nUnSnSn [UR5GH#upU S:Xa$U R [R:XaU S- n OMU S:aGU R [R:XaU S- n O#U R [R :XaU S-n XuLasU R [;a_U RULaP[U[5(a;U[R:XaU S:d![U 5 [U5 U(aUOUnURU 5 XvLdGM U R U:XdGMU nUnGM& U(dGMWU(dGMa O W(aW(d [S5e[WX[R S9n [WX[R"S9n [X@U[R$S9n['XU5 XU4HnU(dM Uv M g7f)aSplit line into many lines, starting with the first matching bracket pair. Note: this usually looks weird, only use this for function definitions. Prefer RHS otherwise. This is why this function is not symmetrical with :func:`right_hand_split` which also handles optional parentheses. NrrrnNo brackets found component)rQrr%rrsr~r&r!opening_bracketrzrRr'r}rUbracket_split_build_linerrrr bracket_split_succeeded_or_raise)rurrc leaf_type tail_leaves body_leaves head_leavescurrent_leavesmatching_bracketrprrrrrrs r_rrsjj%**- "$ "$ "$ $+/$T[[1KEzdii5::5 99 *QJEYY%**,QJE-II!11((,<</66#ejj0UQYt$/00;  ! !$ ',99 )'+$%0N728   I.J ;-.. #T7M7R7R D $T7M7R7R D $+7M7R7R D%T6t$ 6L%s%E HH'H:HBH Hrnc#H# [XS9n[X@XUS9ShvN gN7f)aDSplit line into many lines, starting with the last matching bracket pair. If the split was by optional parentheses, attempt splitting without them, too. `omit` is a collection of closing bracket IDs that shouldn't be considered for this split. Note: running this function modifies `bracket_depth` on the leaves of `line`. rmrdrnN)_first_right_hand_split%_maybe_split_omitting_optional_parens)rurcrdrn rhs_results r_rqrq\s*)9J4$s " "c2/n/n/nUnSnSn[UR5HdnXSLaXLa U(aUOUnURU5 XRLdM-UR[;dMC[ U5U;dMTUR nUnUnMf U(aU(aU(d [S5eUR5 UR5 UR5 Sn [RUR;Ga5USR(Ga USR USLGa [U5n /n /n USR[R[R 4;n U (aSOSn[#U 5SU-:aU SR[;aU SR XLaU(a"U RU R%S55 SnU RU R%S55 U R'SU R%55 [#U 5SU-:a-U SR[;aU SR XLaMU (aU (a[)U UU S[*R,S9nURR.(a$U SR[R0:XaSnOQURR2[5S X-55- n[7U[9URUS 9S 9(aS nOSnU(aU nUR;U 5 X-nUn [)X@U[*R<S9nU c[)X0U[*R,S9n [)X U[*R>S9n[AUU U5 [CUU UXg5$) aSplit the line into head, body, tail starting with the last bracket pair. Note: this function should not have side effects. It's relied upon by _maybe_split_omitting_optional_parens to get an opinion whether to prefer splitting on the right side of an assignment statement. NrrrrnrrTc3J# UHn[[U55v M g7frg)rrr0r2rs r_r3*_first_right_hand_split..s!: MD NN Ms!#rrBF)"reversedrsr}r~r!rrrUreverser*hug_parens_with_braces_and_square_bracketsrcrrprQSTARrrrpopinsertrrrrzCOMMArsumrrextendrrrr)rurnrrrrrclosing_bracketrrinner_body_leaveshugged_opening_leaveshugged_closing_leaves is_unpackingunpacking_offset inner_body should_hugrrrs r_rrps]!K K K N&*O&*O%  (&0;d#  (yy,,D1E"&"6"6"&!,& K-..D::diiG N N * *k"o = -,.,."1~**uzz5;K;K.LL %1q ! "a*:&: :!"%**.>>!"%55 23 %,,->-B-B1-EF#$ ! ( ():)>)>q)A B ! ( (,=,A,A,C D ! "a*:&: :!"%**.>>!"%55 23 !%61!%b)055 J ..%b)..%++=! "ii33c: 5 M:7 (WTYYK%P"'J!%J/ ""#893A ! #?6L6Q6Q D |' :P:U:U  $?6L6Q6Q D%T46 T4 HHr^rtc#~# [RU;aURR[R :XaURR (dURR[R:XaURR (dsUR(db[XR5(aH[UR51Ukn[XS9n[XPU5(a[XQX#US9ShvN g[;UR5 [;UR5 UR4UR&UR84HnU(dM Uv M gNh![ Ga;nUR"(aSnAN[%UR&5(d[)UR&US9(d[*R,(a|URR.(aaURR.R.(a<URR.R.R[0R2:Xd [!S5UeUR4R75(dUR8R75(a [!S5UeSnAGNSnAff=f7f)NrmrrBzG-gDAA@4 BH3&&'3&&'88SXXsxx0 6L1K  ))!**,SXXDA;;++22++2299++2299>>$BSBSS"R 3355886688!  9+ sVCJ=!(E4 E2 E4AJ=) J=2E44 J:?J5J=DJ5/J=5J::J=rcURR5(d>URR5(dURR5(ag[R (aUR R(aUR RR(aUR RRR[R:XaKURRR(a&[SURR55$[URR5S:a5URRSR[ R":Xdg[SURRSS55(dg[%UR['X"R(S- S 9S 9(dgURR*bgURRVs/sHo3RPM snR-[ R"5nURRVs/sHo3RPM snR-[ R"5nUS:aXE:ag S n[/URR5H;nUR[ R":Xa OUR[0;dM9Sn O U=(dA [S URR55=(a [%URUS 9$s snfs snf) zv Returns whether we should prefer the result from a split omitting optional parens (rhs_oop) over the original (rhs). Tc3Z# UH!oR[R:Hv M# g7frg)r~rQrrs r_r31_prefer_split_rhs_oop_over_rhs..3sL8K99 +8K)+rc3F# UHoR[;v M g7frg)r~r rs r_r3r:sF1EyyH$1Es!NrrnrrBFc3Z# UH!oR[R:Hv M# g7frg)r~rQEQUALrs r_r3r_sI5HT U[[(5Hr)rr{rrrrrrr~rBrr{ delimitersr5rsrrrQrrrrrzcountrr!)rrtrcrrhs_head_equal_countrhs_oop_head_equal_count has_closing_bracket_after_assigns r_rrsd  6688 << 9 9 ; ; << 9 9 ; ; //    & &    & & - -    & & - - 2 2d6G6G G HH $ $ / /L 8K8KLLL  A %#((//"*=*B*Bekk*Q F"1EF F F  wt1A1AA1EF  xx$$036((//B/$II/BHHU6=ll6I6IJ6Id 6IJPP  a$8$S',$,,- 99 #  99( (/3 ,  . )  IW\\5H5HI I >$W\\= CJs !M.Mrrrc[[U5R55nU(d'US:Xa [S5eUS:a[SUS35egg)a[Raise :exc:`CannotSplit` if the last left- or right-hand split failed. Do nothing otherwise. A left- or right-hand split is based on a pair of brackets. Content before (and including) the opening bracket is left on one line, content inside the brackets is put on a separate line, and finally content starting with and following the closing bracket is put on a separate line. Those are called `head`, `body`, and `tail`, respectively. If the split produced the same line (all content in `head`) or ended up with an empty `body` and the `tail` is just the closing bracket, then it's considered failed. rz)Splitting brackets produced the same linerz,Splitting brackets on an empty body to save z characters is not worth itN)rrr0rrU)rrrtail_lens r_rrfs_3t9??$%H  q=IJ J \>xjI    r^rsoriginalrcU(dgUR(agUR(dgURS:wag[SU55(ag[ SU5S5nUcg[ U5S:XagUR (aNUR R(a3UR RR[R:Xagg)NFTrc3# UH8oR[R:H=(a [U5(+v M: g7frg)r~rQrr7rs r_r3)_ensure_trailing_comma..s0 QW U[[ D)>t)D%DDQWsAAc3J# UHoR(dMUv M g7frgrrs r_r3rsDfd TTfs# #re) rrrr5rr)r next_siblingr~rQVBAR)rsrrleaf_with_parents r__ensure_trailing_commars  ??#  QW DfDdK+,8   # # 0 0  # # 0 0 5 5 C r^rc[URURS9nU[RLaSUlU=RS- sl[ XU5(a[[U5S- SS5HjnXR[:XaMXR[R:wa/[[RS5nURUS-U5 O [5nU[R La [#U5nUHHnUR%US['U5U;S9 UR)U5Hn UR%U SS9 M MJ U[RLa[+XB5(aSUlU$)aQReturn a new line with given `leaves` and respective comments from `original`. If it's the head component, brackets will be tracked so trailing commas are respected. If it's the body component, the result line is one-indented inside brackets and as such has its first leaf's prefix normalized and a trailing comma added when expected. roTrnrrDrr)rrcrprrr|rrangerrr~r#rQrrRrrrrr}rcomments_aftershould_split_linery) rsrrrrr new_commarr comment_afters r_rrsT x}}HNN ;F*///!%  !&O D D3v;?B39>>%779>>U[[0 $U[[# 6IMM!a%34$'5O*///=fE T(o5  &44T:M MM-dM ;; *///4E55#' Mr^ split_funcc ^[T5S[S[[S[S[ [4U4Sjj5nU$)zNormalize prefix of the first leaf in every line returned by `split_func`. This is a decorator over relevant split functions. rurdrcrec3^># T"XU5HnSURSlUv M g7fr)rsr)rurdrc split_liners r_ split_wrapper0dont_increase_indentation..split_wrappers2%TT:J*,J  a ' ;s*-)r rrrrr)rrs` r_dont_increase_indentationrsM  :(19= $ r^c[[UR5S- SS5H(nURUR[:wdM&Us $ g)Nrnrr)rrrrsr~r#)ruleaf_idxs r__get_last_non_comment_leafrsE#dkk*Q.26 ;;x % %); ;O7 r^rc[U[R1S9(a[RU;$[U[R [R 1S9(a[RU;$g)N)withinT)r?rB typedargslistrTRAILING_COMMA_IN_DEFarglistargumentTRAILING_COMMA_IN_CALL)rrds r__can_add_trailing_commarsQt1123,,88t||T]];<--99 r^safedelimiter_prioritycU(aU[:XawURSR[R:waLURSR[ :wa+[ [RS5nURU5 U$)NrrD)r rsr~rQrr#rRr})rrrurs r__safe_add_trailing_commars_ . 0 KKO EKK / KKO $6 6c*  I Kr^c#^^^# [TR5S:Xa [S5SeTRSnTRnUR [ U51S9nU[:Xa URU5S:Xa [S5e[TRTRTRS 9m[RnS nS [S [ [4UU4S jjmS [S [ [4UU4Sjjn[#T5n [%TR5GHoupT"U 5ShvN U S:=(a4 UR&R)[ TRU S- 55n X:wd U[*;aU"U 5ShvN [-XkR.5nU(aU R.U:Xa [1X5nUR2[4:XaX:Xa [7XuT5mUR&R)[ U 55n X:XdMU S-[TR5:a*U[*;a U"TRU S-5ShvN Tv [TRTRTRS 9mGMr T(a[7XuT5mTv gg![ a [S5Sef=fGNGN:No7f)zSplit according to delimiters of the highest priority. If the appropriate Features are given, the split will add trailing commas also in function signatures and calls that contain `*` and `**`. rz Line emptyNrexcludezNo delimiters foundrnz7Splitting a single attribute from its owner looks wrongrcrpr|Trrec3># TRUSS9 g![aA Tv [TRTRTR S9mTR U5 gf=f7f)EAppend `leaf` to current line or to new line if appending impossible.Trr N append_safe ValueErrorrrcrpr|r}rrqrus r_append_to_line'delimiter_split..append_to_line#sa &  $ $T $ = & YYdjj$BVBVL    %  &&A$A$AA!A$ A!!A$c3b># TRU5HnT"U5ShvN M gN 7frg)r)rrrrus r_append_comments(delimiter_split..append_comments0s,!006M%m4 4 47 4s /- /)rrrsrUr{max_delimiter_priorityrrrdelimiter_count_with_priorityrrcrpr|sysmaxsizerRrrrrgetMIGRATE_COMMENT_DELIMITERSmin bracket_depthrr~r#r)rurdrc last_leafbtr lowest_depthtrailing_comma_saferlast_non_comment_leafrrprevious_priority leaf_priorityrrqs` @@r_rrs 4;;1,'T1 BI  B;669 6O l*  , ,-? @A ESTT YYdjj$:N:NL;;L &T &htn & &5d5x~557t<#DKK0!$'''$qL R]]->-> t{{8a<( ).   3!%??&t, , ,<););< 4#5#5#E"9$"I  >>/ /H4U3#L ))"T(3  .1 s4;;//&.HH*4;;x!|+DEEE YYdjj$BVBVL=1D/ \  G ;/0d:;D ( -"FsYAK J* CK .K/AK  KBK AK K AK *KK K  K c#^^# TR5(d [S5e[TRTRTR S9mS[ S[[4UU4SjjnTRH;nU"U5ShvN TRU5HnU"U5ShvN M M= T(aTv ggN=N7f)z4Split standalone comments from the rest of the line.z*Line does not have any standalone commentsr rrec3># TRUSS9 g![aA Tv [TRTRTR S9mTR U5 gf=f7f)r Tr)rpr|Nr rs r_r0standalone_comment_split..append_to_linejsa &  $ $T $ = &  T=Q=QL    %  &rN) r}rUrrcrpr|rRrrsr)rurdrcrrrrqs` @r_rr^s  , , . .FGG YYdjj$:N:NL &T &htn & & !$'''!006M%m4 4 47  ( 5s$BCC#C(C)CCrwrc  [URSS9HnUR[;dM g UR[ R :Xa [XU5 Sn[[UR55GH upg[U[5(a(UR[ R:Xa [XqX#S9 [U[5(a*UR[ R:Xa [US1X#S9 [U[5(a*UR[ R :Xa [US1X#S9 US:Xa5[U[5(a UR[ R":XaSnU(GaUR[ R$:XaUR[ R&:Xa[UR([*5(a`UR(R[,R.:Xa8UR(RS :Xa[1UUUUSS 9(a [3XSS 9 GO[U[5(a*UR[ R :Xa [5XpX#S 9 GOUR[ R$:XaS U;a~[7UR5S:Xae[9URS5(aH[;URS5(a+URSR[ R<:XdpX#S9(a [3XSS 9 GO[?U5(a [3XSS 9 OUR[ R@:Xa[CXU5 gUS:Xa?UR[,RD:Xa!UR[ RF:XaGMG[U[*5(aGURHb:URHR[,RJ:XaURS:Xa g[MU5(d [3XSS 9 UR[,RN:Hn[U[*5=(a URU;=(d UnGM g)a1Make existing optional parentheses invisible or create new ones. `parens_after` is a set of string leaf values immediately after which parens should be put. Standardizes on visible parentheses for single-element tuples, and keeps existing visible parentheses for other tuples and generator expressions. F) is_endmarkerNrcaserErTrIrrrr:rrrnr)(rrrrr~rB with_stmt_maybe_wrap_cms_in_parensrrprrzrS annassignr case_blockguardtestlist_star_exprrfor_stmtrrRrQNAMErrCremove_with_parensrrr1r8testr5 import_from_normalize_import_fromr except_clauserrr2r) rwrrcrdpc check_lparrr comma_checks r_rrs>DKKe< 88w  = yyDNN"!$h7J!$t}}"56  eT " "uzzT^^'C &t  eT " "uzzT__'D &VH4  eT " "uzzTZZ'? &TF  QJ5$'' d555J  dii'II.u11488&&++uzz9&&,,56%15 (UCE4((TYY$..-H"5TMtyy( $'1,!%.."344!%.."455NN1%**dii76T(UCe$$#D>d...&tE: JJ%**,II!3!33 5$''&&2&&++u{{:KK6)(//#D?jjEKK/ t, KK< ' 6; q7r^rrrc~[U5(a;[URS5(deSUlSURSlgUR[ R :waTURU[[ RS55 UR[[ RS55 gg)Nrry) r1r8rrr~rQrrrRrrr)rrrs r_r7r7sUV__R01111 $&! uzz !E4 B#78DR01 "r^cURSR[R:XGa[ UR5S:GaURSR[ R :XGaxURSRSR[R:XGa>[URSUUUSS9(a[XRSSS9 [[URSRS5n[[URSRS5nURSRSn[U[5(aUR[ R:wdLURSR[R:Xd![SUR55(a[!U5 [!U5 ggggggg) NrrnTrFrrc3# UH9n[U[5=(a UR[R:Hv M; g7frg)rzrRr~rQrr1s r_r3&remove_await_parens..s5!:ud+N e>N>N0NN!:sAA)rr~rQAWAITrrrBrrrrCr rRrzrSrr5r')rwrcrdrrbracket_contentss r_rrs }}Q +DMM0BQ0F MM!  ! !TYY . a ))!,11UZZ?2 a !-1 $D--*:EJ#4q)9)B)B1)EFO"4q)9)B)B2)FGO#}}Q/88; *D11 %%3#,,Q/44 C!1!:!: // 2+@ /1G+r^c[RU;dD[UR5S::d+URSR[ R :XagSn[S[UR55H2nURUR[R:XdM0Un O Ub[[RS5n[[RS5nURSUnUHnUR5 M [[ R U[[ RU5U/5n UR!SU 5 gg)zWhen enabled and safe, wrap the multiple context managers in invisible parens. It is only safe when `features` contain Feature.PARENTHESIZED_CONTEXT_MANAGERS. rrnNry)rPARENTHESIZED_CONTEXT_MANAGERSrrrr~rBrrrQrrRrrrrS testlist_gexpr) rwrcrd colon_indexrrrcontext_managersr new_childs r_r-r-(s ..h> t}}  " ==  DII -!%K 1c$--( ) ==  EKK /K *EJJ#EJJ#==;7%E LLN& IId4#5#57GH$O  !Y')r^cUR[R:XaX[UUUUSS9(a [ XSS9 [ UR S[5(a[UR SXUS9 g g UR[R:Xa5UR H$n[ U[5(dM[X@X#S9 M& g UR[R:Xa^[SUR555(d8[UR SUUUSS9(a[ XR SSS9 g g g g ) z6Recursively hide optional parens in `with` statements.TrFrrnrc3Z# UH!oR[R:Hv M# g7frg)r~rQ COLONEQUALrs r_r3%remove_with_parens..us 32?$ U%%%-rrN) r~rBrrrCrzrrSr4rD asexpr_testr5rs)rwrrcrdrs r_r4r4Rs( yyDII . )-   e < dmmA& - - t}}Q/8 T . d(( (]]E%&&"5TM# d&& &s326++-300 / MM! )-   mmA&6 F 0 &r^rc UR[R[R4;Gd[ U5(Gd[ U5(Gd[ U5(aUR[R:XGdt[ U5(a>UR[R:Xa [U[R5(Gd&[U5(aUR[R:wdU(d[U5[:a[ R"U;a[$R&U;a[ U5(aUR(bUR(R[R*:XaqUR,b4[/UR,5(aUR,R0S:Xd0[3U5(d [5U5(d[7U5(ag[9U5(aUR[R:[R[R<[R>[R*[R@[R[RB[RD[RF[RH[RF4 ;agURJSnURJSn[MU5(GaZ[OU5(GaIURJSn[QURRRU55(dSUlSUl[WUUUUUS9 [YU5(aUR[URJS5 URJSRRRU5(aEURJSRRURJSRR-URJSl)URJSRRRU5(a+URJSRRURR-Ul)gg) zIf it's safe, make the parens in the atom `node` invisible, recursively. Additionally, remove repeated, adjacent invisible parens from the atom `node` as they are redundant. Returns whether the node should itself be wrapped in invisible parentheses. asFrrrnryrT).r~rBrexprr/r5r;rLr,r*rQrrA expr_stmtrr r!remove_parens_around_except_typesrUNPARENTHESIZED_EXCEPT_TYPESrr8rr3rr=r<r0r@r. assert_stmt return_stmtfuncdefrDtnamer2del_stmtrr1r8r>rrrr-r)rwrrcrdrr)r*middles r_rrs $))TYY// $      TNNv{{d.>.>> TNN t~~-%dEKK88 TNNv{{dnn< -.t4F 99TA88HDTNNKK+KK$$(:(::%%1%d&7&788))//47 &d + + #D ) )   D!! ;; NN NN          LL NN    JJ MM MM MM   MM! E == DU d 3 3q! .fmm.A.A.CDDEKDJ+ )E   ) 0 0 NN6??1- .q!((..00OOA&--0B0I0II")r"))//11$oob1884;;F  r^c*UR(aURS;dg[5nSnURSnUR[ R :XaSnUR[U55 URRUS9nU[:H=(a] URR=(a U=(d7 URR[ R"[ R$1;$![[4a gf=f)zFShould `line` be immediately split with `delimiter_split()` after RHS?z[{(FrTr)rrrrsr~rQraddrr{r IndexErrorrr rcrzrBrr6)rurr trailing_commar max_prioritys r_rrs  " "'<'<'E eGNKKO >>U[[ (!N KK9 &++BB7BS  > )   ' ' :N H  ! ! & &499d6F6F*G G  #sA"C??DDrc#&# [5nUR(dUv SUR-nSnSn[5nURSS9GH=upxn X9- nX1:a gU [ UR 5[ UR 5-:n UR[:XdU (a gU(aXLaSnMlUR[;aUS:aURUS- OSn U (aRU R[R:Xa4URb'[URXR5(d gUR[!U55 GMGMUR[;dGM*US:aURUS- OSn U (a1U R[";aUR[!U55 GM|U(a?UR[!U55 UR%U5 UR'5 Uv U (aRU R[R:Xa4URb'[URXR5(d gUR (dGM/URnUnGM@ g7f)aGenerate sets of closing bracket IDs that should be omitted in a RHS. Brackets can be omitted if the entire trailer up to and including a preceding closing bracket fits in one line. Yielded sets are cumulative (contain results of previous yields, too). First set is empty, unless the line should explode, in which case bracket pairs until the one that needs to explode are omitted. NT) is_reversedrrn)rrzrpenumerate_with_lengthrrrrr~r#r!rsrQrrr4rZrr"updateclear) rurrnlengthrrinner_bracketsrr leaf_lengthhas_inline_commentprevs r_rorosD  $ $ ^F&*O&*O"%%N$($>$>4$>$P [   (3tzz?S=M+MM 99* *.@  &"&..16t{{519- U[[0,,83,,dKK ""2d8,/YY* *-2QY4;;uqy)DD %55""2d8,O,- N+$$& II,((4/0D0DdKKXXzzz"&"6"6"&m%QsE1J8DJ=Jryrvrrkc ^U(d [U5n/nU"XT5HFn[U5RS5U:Xa [S5eUR [ UTUS95 MH [ U5n[RU;dURRS:wdURR(a[SURR55(dxUR5(dcUSR5(dKUSR!5(d3[#USTS9(d![SUR$55(aU$UR'5n[)XUR$5 U[R1-n [+XTXS 9n [-U4S jU 55(aU nU$) Nr z-Line transformer returned an unchanged resultrrtc38# UHoRv M g7frg)r)r2brackets r_r3"run_transformer..qsK,J}},JsrrBc3<# UHoRSLv M g7frgrrs r_r3rlzs;{t{{d"{srvc38># UHn[UTS9v M g7f)rBN)r)r2lnrcs r_r3rls H2  .s)rr0rrJrrrrrrrXr{ invisibler5rrxr{rrsclonerrall) rurrcrdrkrtransformed_line features_set line_copy features_fopsecond_opinions ` r_rr[sv !$'F%dd;  & &t , 8!"QR R n%5D8TU < x=L**l:    ' '5 0##-- KD,@,@,J,JK K K  * * , , !9 9 9 ; ; !9 6 6 8 8 q  5 ;t{{; ; ;  I)4;;/7#E#E"FFL$dLN H HHH Mr^)rW)rWrW)F)r\rrcollections.abcrr dataclassesrenumrr functoolsrr typingr r r black.bracketsr rrrrblack.commentsrrr black.linesrrrrrrr black.moderrr black.nodesrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCblack.numericsrD black.stringsrErFrGrHrI black.transrJrKrLrMrNrOrPblib2to3.pgen2rQblib2to3.pytreerRrSreLeafIDrfrUrar0rirboolrrrrqrrrrrprrrrrrrrrrrr7rr-r4rrrorrWr^r_rs 0$((ED.-&&&&&&&&&&N5!&  4:M/M HBGDMHBV ! c] =?y yy&0&9y d^yx33D3T3>T = =%g.=6:= d^=F%'!   ! V   d^ , "dI dI V dIdIV%'! A A A A! A V  A d^ AHH H&H.2H HV4t4D4! J!"&!9=! !H. J... & .  .b++"Thsm$*W2E$ 4 S  QU .~>V V$W-V59V d^VVr $W-59 d^Dq q !#hq 15q AKGATq  q h 24 2 23 24 2"0d"0$"0*W:M"0RV"0J'( '('(&0&9'( '(T-G -G-G$(-G4>w4G-G -Gj*/ o o o o! o #' o  odD4D6I'DI'sI'xF ?TI'd + ++ +! +  + $Z+r^