mi 1dZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZmZddlmZmZmZmZmZdd lmZmZmZmZmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNdd lOmPZPddlQmRZRmSZSmTZTmUZUmVZVddlWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^ddl_m`Z`ddlambZbmcZcedZeeebecfZfGddeXZgGdde1eZhdedee&de'deidzfdZj dUdede'dee&deefdZkdede'delfdZmGdde Znded ee&de'deefd!Zo dVdede'dee&d"eeedeef d#Zp dUded"eeedefd$Zq dVd%edede'dee&d"eeedeef d&Zrd'ed%ede'delfd(Zsd)ed*ed+eddfd,Ztd-euebd.ed/ebdelfd0Zvd-euebd.ed/ebd1endef d2Zwd3e]de]fd4Zxdededdzfd5Zyd6ebdee&delfd7Zzd8eld9eddedefd:Z{eehZ|exdedee&de'deefd;Z}exdedee&de'deefd<Z~d=ecd>eeide'dee&ddf d?Zd@ecdAefdBedddfdCZd=ecde'dee&ddfdDZd=ecde'dee&ddfdEZd=ecd@ecde'dee&ddf dFZd=efde'delfdGZd=efdelfdHZ dWd=efd@efde'dee&dJeldKeldelfdLZded/ebdelfdMZdedNeddeeeefdOZdPdQdedRe]de'dee&dSeideuef dTZdS)Xz Generating lines of code. N) CollectionIterator)replace)Enumauto)partialwraps)Unioncast)COMMA_PRIORITY DOT_PRIORITYSTRING_PRIORITY#get_leaves_inside_matching_bracketsmax_delimiter_priority_in_atom)FMT_OFFFMT_ONcontains_fmt_directivegenerate_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_tstring_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)LeafNodeceZdZdZdS) CannotSplitzBA readable split that fits the allotted line length is impossible.N)__name__ __module__ __qualname____doc__e/var/www/html/bestrading.cuttalo.com/scripts/kaggle-env/lib/python3.11/site-packages/black/linegen.pyrVrVesLLLLr\rVc eZdZdZdedeeddfdZd+dede e fd Z d e de e ffd Z d ede e fd Zd ede e fd Zd ede e fdZd edeedeede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZd ede e fdZ dede e fdZ!dede e fdZ"dede e fd Z#d ede e fd!Z$d ede e fd"Z%dede e fd#Z&dede e fd$Z'd ede e fd%Z(d ede e fd&Z)d ede e fd'Z*d ede e fd(Z+d ede e fd)Z,d,d*Z-xZ.S)- LineGeneratorzGenerates 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. modefeaturesreturnNcN||_||_||dSN)r`ra __post_init__)selfr`ras r]__init__zLineGenerator.__init__rs.     r\rindentc#K|js|jxj|z c_dSt|jjdkr!t |jjdrdS|j}t |j|j|z|_|VdS)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. Nrr`depth) current_linerllenleavesr-rr`)rfrh complete_lines r]linezLineGenerator.linexs     # #v - # # F t ' ( (A - -2J   $Q '3 3 - F)  di}7JV7STTTr\nodec#Kt|tr|jj}t ||jD]}|r|j||jtj kr5|j|| Ed{Vi| Ed{V|j|| Ed{V|rd|_ |jtvr|j|t|Ed{VdS)zCDefault `visit_*()` implementation. Recurses to children of `node`.r`N) isinstancerSrmbracket_trackerany_open_bracketsrr`appendtyperRCOMMENTrqprefixr&super visit_default)rfrrrxcomment __class__s r]r~zLineGenerator.visit_defaultsy dD ! ! / $ 1 A S S U U ,T BBB + +$ +%,,W5555\U]22%,,W555#yy{{******** $yy{{*******%,,W555#yy{{********  !  y **!((...77((...........r\c#nK|jo|jjtjk}|stt tjd}t tjd}|j}d|_||_|d|||| |Ed{VdS)zVisit an `x if y else z` testrurN) prev_siblingrzrRLPARrSRPARr| insert_child append_childr~)rfrralready_parenthesizedlparrparr|s r] visit_testzLineGenerator.visit_tests   F$"3"8EJ"F % $ B''D B''D[FDK DK   a & & &   d # # #%%d+++++++++++r\c#vK|dEd{V||Ed{VdS)z/Increase indentation level, maybe yield a line.rjNrqr~rfrrs r] visit_INDENTzLineGenerator.visit_INDENTsV99R==       %%d+++++++++++r\c#K|Ed{V||Ed{V|dEd{VdS)z/Decrease indentation level, maybe yield a line.Nrrs r] visit_DEDENTzLineGenerator.visit_DEDENTsz 99;;%%d+++++++++99R==         r\keywordsparensc#Kt|||j|j|jD]O}t |r#|j|vr|Ed{V||Ed{VPdS)aVisit 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_afterr`raN)normalize_invisible_parensr`rachildrenr4valuerqvisit)rfrrrrchilds r] visit_stmtzLineGenerator.visit_stmts # vDI     ] ) )EU## ' x(?(?99;;&&&&&&&zz%(( ( ( ( ( ( ( ( (  ) )r\c#dK||Ed{Vd|jd_dSNrurr~rr|rs r]visit_typeparamszLineGenerator.visit_typeparamsA%%d+++++++++"$ ar\c#dK||Ed{Vd|jd_dSNrurjrrs r]visit_typevartuplez LineGenerator.visit_typevartuplerr\c#dK||Ed{Vd|jd_dSrrrs r]visit_paramspeczLineGenerator.visit_paramspecrr\c#Ktj|jvrt|jD]\}}|dkr |j|dz jt jkrn|jtj krG|jdjtvr.t|st|||j|j dt||d||Ed{VdS)NrrjFparentr`raremove_brackets_around_commavisible)rwrap_long_dict_values_in_parensr` enumeraterrzrRCOLONrCatomr#rA#maybe_make_parens_invisible_in_atomrarDr~)rfrrirs r]visit_dictsetmakerz LineGenerator.visit_dictsetmakers  2di ? ?%dm44 H H566=Q', ;; di//!N1-26FFF 4U ; ;G<!#'!%%)]9> ,D%GGGG%%d+++++++++++r\c#K|Ed{Vd}|jD]}|jtjkrd}|rz|jt jkrQ|jdjtjkr1t|||j |j drt||dnt||dd}|jD]}| |Ed{VdS)zVisit function definition.NFTrrr) rqrrzrRRARROWrCrrrr`rarDr)rfrris_return_annotationrs r] visit_funcdefzLineGenerator.visit_funcdef s%99;; %] - -EzU\))'+$$% -:**u~a/@/E/S/S:#!Y!%5: H,D%GGGG'eUCCCC',$] ) )Ezz%(( ( ( ( ( ( ( ( ( ) )r\c#Kt|t|j|j|Ed{V|jD]}||Ed{VdS)z'Visit either a match or case statement.rN)rsetr`rarqrrrfrrrs r]visit_match_casezLineGenerator.visit_match_case%s" suu49t}    99;;] ) )Ezz%(( ( ( ( ( ( ( ( ( ) )r\c#Kt|r(||jdEd{VdS||Ed{VdS)zVisit a suite.N)r;rrr~rs r] visit_suitezLineGenerator.visit_suite/ss    0zz$-"233 3 3 3 3 3 3 3 3 3))$// / / / / / / / / /r\c#Kd}|jD]<}||tjkr!t|rt ||d|j}=|jr|jjtvrt|r,t|r| |Ed{VdS| dEd{V| |Ed{V| dEd{VdS|jr8t|jr$d|_ | |Ed{VdS| Ed{V| |Ed{VdS)z,Visit a statement without nested statements.NFrrjrru)rrRSEMIr,rDrzrr%r7r:r~rqr;r|)rfrr prev_typers r]visit_simple_stmtzLineGenerator.visit_simple_stmt6s $ ] # #E!Y%*%<%<-PUBVBV%<#D%???? II ; 04;+y88*400 )\$5G5G )--d3333333333399R==(((((((--d33333333399R==((((((((({ }T[99   --d333333333yy{{ " " " " " " "))$// / / / / / / / / /r\c#HK|Ed{Vt|j}|D]D}||Ed{V|jt jks|jtkrnEt|}||Ed{VdS)z-Visit `async def`, `async for`, `async with`.N) rqiterrrrzrRASYNCr$next)rfrrrr internal_stmts r]visit_async_stmtzLineGenerator.visit_async_stmtNs99;; &&  Ezz%(( ( ( ( ( ( ( (zU[((EJ:L,L,L-M X ::m,,,,,,,,,,,r\c#K|jD]7}|Ed{V||Ed{V8dS)zVisit decorators.N)rrqrrs r]visit_decoratorszLineGenerator.visit_decorators^sh] ) )Eyy{{ " " " " " " "zz%(( ( ( ( ( ( ( ( ( ) )r\c# Kt|jddD]\}}|j|dz}t|ts+|j}|jtjkr^|jtj krI|jdjtj kr)| dsd|vrt||t||j|j||Ed{VdS)Nrrjr)0x0b0ojr`ra)rrrvrSrlowerrzrRNUMBERrCtrailerDOT startswithrDremove_await_parensr`rar~)rfrridxleaf next_leafrs r] visit_powerzLineGenerator.visit_powerds"4="#566 0 0IC cAg.IdD)) J$$&&E U\))Ndl22&q).%);;(();<<<u$$#D$///Dty4=IIII%%d+++++++++++r\rc#>K|Ed{VdS)zBRemove a semicolon and put the other statement on a separate line.N)rqrfrs r] visit_SEMIzLineGenerator.visit_SEMI|s,99;;r\c#tK||Ed{V|Ed{VdS)zAEnd of file. Process outstanding comments and end with a newline.N)r~rqrs r]visit_ENDMARKERzLineGenerator.visit_ENDMARKERsR%%d+++++++++99;;r\c#@K|jj}|s|Ed{V|j}|}t |dko5t|dtot|dt}|r|j rtd|j dDrnt||jD]P}|Ed{V|j||Ed{VQd|_ |j||s|Ed{VdSdS||Ed{VdS)Nrrrc3K|]M}|do!t| VNdS)#N)striprr).0rqs r] z9LineGenerator.visit_STANDALONE_COMMENT..si## '',,=.tzz||<<<######r\ rtru)rmrwrxrqr splitlinesrnrrrr|anysplitrr`ryr~)rfrrxrlinesis_fmt_off_blockrs r]visit_STANDALONE_COMMENTz&LineGenerator.visit_STANDALONE_COMMENTs -=OOQQ  #yy{{ " " " " " " "   "" JJ!O :&uQx99 :&uRy&99   0 { !s##!K--d33###   ! 1DIFFF++G#yy{{*******%,,W555#yy{{********    $ $T * * *$ '99;;&&&&&&&&& ' '))$// / / / / / / / / /r\c #K|j\}}|jtjkrt |jdkr|jdjt jkr{tt jd}tt j d}| pd}| |ttj |||g||Ed{VdS)z_Force parentheses between a unary op and a binary power: -2 ** 8 -> -(2 ** 8) rj()rN)rrzrCpowerrnrR DOUBLESTARrSrrremoverrTrr~)rfrr _operatoroperandrrindexs r] visit_factorzLineGenerator.visit_factors "] 7 LDJ & &G$%%** #(E,<<< C((D C((DNN$$)E   eT$)dGT5J%K%K L L L%%d+++++++++++r\c#Kt|jdkrEt|jd||j|jrt ||jdd||Ed{VdS)a) 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): ... rrrr`raFrN)rnrrr`rarDr~rs r] visit_tnamezLineGenerator.visit_tnames t}   " "'J M! T DM( ( ( " dmA&6 F F F F%%d+++++++++++r\c#Kt|t|r3tjd|js|jjr$t|j}t|}n|j}t|}|t|d}|d}|d|krdnd}||| }| }d|j j z}t|rt||}n|}d}|rm|d|krd|z}|d|kr|dz }|dd kr>t|t|d z } | d zr|dz }d }n|sd}||z} |dkr|} |rt| dnd} t| dkrq| |z|jjkr^t||z|jjkr>|s<|jd|z d kr|| z|z| z|_n8|| z|zd z|z| z|_n!|| z|z| z|_n|| z|z| z|_|jjrG|jt(jkr2t|j|_t|j|_||Ed{VdS) Nz\\\s*\nrrjrz F r\rTr)rJr/researchrr`string_normalizationrHrIrGrnrmrlr3rFrrstripr line_lengthrzrRSTRINGr~) rfr docstringr| quote_char quote_lendocstring_started_emptyrhhas_trailing_backslashbackslash_countquoterlast_line_lengths r] visit_STRINGzLineGenerator.visit_STRINGs#*4000   N @bi DJ&G&GN @y- '3DJ?? 4I>>  J &y11F!#f++--0I"1J 'q\Z77QI!)YJ"67I*3m #t066F"4(( .3IvFF %OO-- %* " Q<:-- #iIR=J..$IR=D((&))nns9;K;KD;Q;Q7R7R&RO&*6"S( 15.,  *EA~~ ",,..5>#E3uRy>>>A JJNN(94ty7LLLF i/493HHH2Iz"y.1T99%+e^i%?%%G %+e^i%?$%F%ORW%W !'%)!;e!CDJJ#e^i7%? 9 ) =di5<.G.G0<.EI- - zT;;; CJJ ;;;;- - r\r)rrrrr~r rfrr string_leafs r] visit_fstringzLineGenerator.visit_fstringA/55  [!!! ;$ $ $- - - - - * * $))+66 6 6 6 6 6 6 6 F$$[11111111111r\c#Kt|}||d|jvr;td|jDr||Ed{VdS||Ed{VdS)Nrc3bK|]*}|jtjkdt|vV+dSrrrs r]rz.LineGenerator.visit_tstring..Trr\rr s r] visit_tstringzLineGenerator.visit_tstringPr#r\c#Ktj|jvrt|dh|j|j||Ed{VdS)Ninr)rwrap_comprehension_inr`rrar~rs r]visit_comp_forzLineGenerator.visit_comp_forsj  (DI 5 5 &D6 DM    %%d+++++++++++r\c#@K||Ed{VdSrd)r*rs r]visit_old_comp_forz LineGenerator.visit_old_comp_fors2&&t,,,,,,,,,,,r\cXt|j|_|j}t }t |dhddh|_t |hdddh|_t |dd hdh|_t |d d hd d h|_ t |hd ||_ t |d hd h|_ t |dhdh|_ t |dh||_ t ||t|_t |dhdh|_t ||dh|_t ||dh|_|j|_|j|_|j|_|j|_t ||dh|_dS)z,You are in a twisty little maze of passages.rtassert,)rr>ifelifelser0r1whiler2forr(>tryr2exceptfinallyr6withclassrbimportdelN)rr`rmrrrvisit_assert_stmt visit_if_stmtvisit_while_stmtvisit_for_stmtvisit_try_stmtvisit_except_clausevisit_with_stmtvisit_classdefr visit_expr_stmtvisit_return_stmtvisit_import_fromvisit_del_stmtrvisit_async_funcdefrvisit_decoratedrvisit_match_stmtvisit_case_block visit_guard)rfvØs r]rezLineGenerator.__post_init__s di000 Ouu!(hZSV!X!X!X$ ...f~   !(Wf4Ewi X X X%a5&/5RV-XXX% <<))))4)T)htn))))00$00000d0x~00000-T-htn---- )T)htn)))) ,,$,,,,0tDXd^ )0T)0htn)0)0)0)0V,,(4.,,,,",,$,,,,$V,V,(4.V,V,V,V,p,,(4.,,,,,t,,,,,$ 2$ 28D> 2 2 2 2 2$ 28D> 2 2 2 2`,4,HTN,,,,-t-----BBBBBBBBr\r_rqrar`rbc | ttt|||S#t$rYdSwxYwrd)rrrQrK)rqrar`s r]!_hugging_power_ops_line_to_stringrSsK d<h#E#EFFGGG tts *- ;;r[c #HK|jr|VdSt|}t|||p|}|j}|j}t ||}t ||}t||} t||} | sy|j sr|j skt|||s| rE|jr|s*|st"j|vr||g} ng} n|jrt)||s t*g} ndt,dt.dt0t2dt4dt6t.f d} t9d d d | i} t"j|vr#|jr||| t:t<| | g} n"||| | | g} n|jrt:t<| g} n| g} | t@| D]1} tC||||| }|Ed{VdS#tD$rY.wxYw|VdS) 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)r`line_strrfrqrar`rbc 3Kt||jD]D}tt||||}t |d|r |Ed{VdSEt|||Ed{VdS)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. omitrrtN)ra)generate_trailers_to_omitrlistright_hand_splitr)rfrqrar`rXrs r]_rhsztransform_line.._rhss2$8HII  -dD(NNNOO (at<<<$$$$$$$$FF(dXFFF F F F F F F F F Fr\rhsr[__call__rU)# is_commentrrSrrrLrMrOrN$contains_uncollapsable_type_commentsshould_split_rhsmagic_trailing_commar!contains_unsplittable_type_ignoreinside_bracketscontains_standalone_comments0contains_implicit_multiline_string_with_commentsrstring_processingis_defshould_split_funcdef_with_rhsleft_hand_splitobjectrrrrrrzdelimiter_splitstandalone_comment_splitryrQrun_transformerrK)rqr`rarUline_str_hugging_power_opsllsn string_mergestring_paren_strip string_splitstring_paren_wrap transformersr\r] transformresults r]transform_linerzs  d##H *$$??K8  B "BB''L,R44!"b))L*2r22  5 5 7 7J%%J%)J% !D;U V V V J% 5577 J%%J%+/*K*K*M*MJ%EEGGJ%  $ , ,(*<=LLLL :%:4FF:%'(  G G $ G0:70C GKO G d^ G G G G:2d5"z401133  $ , ,#  & #,%  !& %   # % /1I3O #u  %%%!    $T9dHxXXXF        EE     H  s/H HHcfg}d}|jD]G}|jtjkrd}|r|||jtjkrd}Ht |j|j}t|}|D])}||dt||v*|j duS)zIf a funcdef has a magic trailing comma in the return type, then we should first split the line with rhs to respect the comma. FTrk preformatted track_bracketN) rorzrRrryrrr`rlridrc)rqr`return_type_leavesin_return_typerryleaves_to_tracks r]rjrj1s&(N "" 9 # #"N  ,  % %d + + + 9 $ $!Nty 3 3 3F9:LMMO"   T((o5       &d 22r\cJeZdZeZeZeZdS)_BracketSplitComponentN)rWrXrYrheadbodytailr[r\r]rrPs/ 466D 466D 466DDDr\r _featuresc#KtjtjfD]5}g}g}g}|}d}d} t|jD] \} } | dkr| jtjkr| dz } n;| dkr5| jtjkr| dz } n| jtjkr| dz} ||urf| jtvrX| j|urOt|tr:|tjkr| dks$t| t||r|n|}| | ||ur%| j|kr|tjkr| dks| }|} |r|rn7|r|stdt|||tj} t|||tj} t|||tj}t'| | || | |fD]}|r|V dS)a/Split 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. NrrrjNo brackets found component)rRrrrrorzrr"opening_bracketrvrSr(ryrVbracket_split_build_linerrrr bracket_split_succeeded_or_raise)rqrr` leaf_type tail_leaves body_leaves head_leavescurrent_leavesmatching_bracketrlrrrrrrys r]rkrkVshj%*-&& "$ "$ "$ $(,$T[11 1 1KE4zzdi5:55 9 **QJEEY%*,,QJE+--I!111(,<<</66= #ej00UQYYt$$$/0000;!L  ! !$ ' ' ',,9 ))"ej00UQYY'+$%0N     E /;/-... #T+7M7R   D $T+7M7R   D $T+7M7R   D%T4666t$  LLLr\rXc#`Kt||}t|||||Ed{VdS)aXSplit 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`. rWrarXN)_first_right_hand_split%_maybe_split_omitting_optional_parens)rqr`rarX rhs_results r]r[r[sd)D999J4D$r\cg}g}g}|}d}d}t|jD]S}||ur ||ur|r|n|}||||ur*|jtvrt ||vr |j}|}|}T|r|r|std|||d} tj |j vr |dj r|dj|durt|} g} g} |djtjtjfv} | rdnd}t#| d|zkr| djtvr| dj| |ur|r*| | dd}| | d| d| t#| d|zkr)| djtvr| dj| |u| r| rt)| || dt*j}|j jr| djtjkrd}nS|j jt5d | | zDz }t7|t9|j |  rd }nd}|r| }|| | |z}|} t)|||t*j}| t)|||t*j} t)|||t*j}tA|| |tC|| |||S) a#Split 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. NrrrrjrrTc3NK|] }tt|V!dSrd)rnrrrs r]rz*_first_right_hand_split..sB::D NN::::::r\rrtF)"reversedroryrzr"rrrVreverser*hug_parens_with_braces_and_square_bracketsr`rrZrRSTARrrnpopinsertrrrrcCOMMArsumrrextendrrrr)rqrXrrrrrclosing_bracketrrinner_body_leaveshugged_opening_leaveshugged_closing_leaves is_unpackingunpacking_offset inner_body should_hugrrrs r]rrsB!K K K N#'O#'O%% - - [ ( (&&0;!Ld### [ ( (y,,,D1E1E"&"6"&!, //K/-...D:diGG N  H N *k"o = = --,.,."1~*uz5;K.LL %1 8q ! " "a*:&: : :!"%*.>>>!"%5 !1233  %%,,->-B-B1-E-EFFF#$ ! ( ():)>)>q)A)A B B B ! ( (,=,A,A,C,C D D D ! " "a*:&: : :!"%*.>>>!"%5 !1233 ! "%6 "1!%b)05 J . &%b).%+==! "i3c:: 58M M:::77 (WTYK%P%P%P&"'JJ!%J "/ ""#89993kA ! #T?6L6Q   D |' :P:U    $T?6L6Q   D%T4666 T4 H HHr\r]c# Ktj|vr|jjtjkr|jjs|jjtjkrj|jjs]|j sUt||j r?t|jh|} t||}t|||rt|||||Ed{VdSn#t $r}|jrnt%|jsst)|j|s]t*jrA|jjr5|jjjr$|jjjjt0jkst!d||js|jrt!d|Yd}~nd}~wwxYwt;|jt;|j|j|j|jfD]}|r|V dS)NrWrrtz>>G-gsDAA @T4(      ) !** ,SXDAAA ;   +2   +29  +29>$BSSS"R 3355 86688 "  / <3&'''3&'''8SXsx0  LLLs.os+LL49 +LLLLLLr\rc32K|]}|jtvVdSrd)rzr!rs r]rz1_prefer_split_rhs_oop_over_rhs..vs)FFtyH$FFFFFFr\Nrrjrrtcg|] }|j Sr[rzrs r] z2_prefer_split_rhs_oop_over_rhs..sBBB$DIBBBr\cg|] }|j Sr[rrs r]rz2_prefer_split_rhs_oop_over_rhs..sJJJd JJJr\Fc3@K|]}|jtjkVdSrd)rzrREQUALrs r]rz1_prefer_split_rhs_oop_over_rhs..s+IIT U[(IIIIIIr\)rrdrrrrrrrzrCrrw delimitersrrornrRrrrrrccountrr")rr]r`rhs_head_equal_countrhs_oop_head_equal_count has_closing_bracket_after_assignrs r]rrWs  6688 < 9 9 ; ; < 9 9 ; ; t /M   &M   & -M   & - 2d6G G G H $ / HLL 8KLLLLLL  A % %#(/"*=*Bek*Q*Qt FF"1EFFF F Ft  wt1AA1EFFF   t x$0tCB#(/BBBHHUUJJgl6IJJJPP   a$8;S$S$Su',$,-- 9 # # E 9( ( (/3 , E ) )  IIW\5HIII I I >$W\=== r\rrrctt|}|s.|dkrtd|dkrtd|ddSdS)aRaise :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)rnrrrV)rrrtail_lens r]rrs3t99??$$%%H  q==IJJ J \\ x     \r\rooriginalrcP|sdS|jrdS|jsdS|jdkrdStd|DrdSt d|Dd}|dSt |dkrdS|jr-|jjr!|jjjtj krdSdS)NFTrc3`K|])}|jtjkot| V*dSrd)rzrRrr8rs r]rz)_ensure_trailing_comma..sM  IM U[ D)>t)D)D%D      r\c3(K|] }|j |VdSrdrrs r]rz)_ensure_trailing_comma..s)DDd DTDDDDDDr\rb) rrirrrr*r next_siblingrzrRVBAR)rorrleaf_with_parents r]_ensure_trailing_commars ut ?u##u   QW   uDDfDDDdKKt+,,88u   # 0  # 0 5 C Cu 4r\rct|j|j}|tjurd|_|xjdz c_t |||rtt|dz ddD]g}||j tkr||j tj kr3ttj d}||dz|t}|tjurt#|}|D]X}||dt'||v||D]} || dY|tjurt+||rd|_|S)aiReturn 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. rkTrjrr/r|r})rr`rlrrrerrangernrzr$rRrrSrrrrryrcomments_aftershould_split_linerb) rorrrryr new_commarr comment_afters r]rrs x}HN ; ; ;F*///!%  !&(O D D 3v;;?B33  !9>%777!9>U[00 $U[# 6 6IMM!a%333#&55O*///=fEE<< T((o5    &44T:: < .split_wrappersO%*T8T::  J*,J a '      r\)r rrrrr)rrs` r]dont_increase_indentationrsg  :(19= $ r\ctt|jdz ddD]!}|j|jtkr|cS"dS)Nrjrr)rrnrorzr$)rqleaf_idxs r]_get_last_non_comment_leafr"sU#dk**Q.266 ;x %); ; ;OOO < 4r\rct|tjhrtj|vSt|tjtjhrtj|vSdS)N)withinT)r@rC typedargslistrTRAILING_COMMA_IN_DEFarglistargumentTRAILING_COMMA_IN_CALL)rras r]_can_add_trailing_commar)s\t123339,88t|T];<<<:-99 4r\safedelimiter_priorityc|ru|tkrj|jdjtjkrJ|jdjt kr/t tjd}|||S)Nrr/)r rorzrRrr$rSry)rrrqrs r]_safe_add_trailing_commar1sh  . 0 0 KO EK / / KO $6 6 6c**  I Kr\c#Ktjdkrtddjd}j} |t |h}n#t $rtddwxYw|tkr(||dkrtdtj j j tj}d }d td t tffd d td t tffd }t#} t%jD]`\} } | Ed{V| dko4|jt j| dz } | |ks |t*vr|| Ed{Vt-|| j}|r| j|krt1| |}|jt4kr| | krt7|||jt | } | |krh| dztjkr(|t*vr|j| dzEd{VVtj j j brt7||VdSdS)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 foundrjz7Splitting a single attribute from its owner looks wrongr`rlreTrrbc3K |ddS#t$r>Vtjjj|YdSwxYw)EAppend `leaf` to current line or to new line if appending impossible.TrrN append_safe ValueErrorrr`rlreryrrmrqs r]append_to_linez'delimiter_split..append_to_line_s &  $ $T $ = = = = = & & &   Ydj$BVL    % % % % % %  &AA&%A&c3^K|D]}|Ed{VdSrd)r)rrrrqs r]append_commentsz(delimiter_split..append_commentslsS!0066 5 5M%~m44 4 4 4 4 4 4 4 4 5 5r\)rnrorVrwmax_delimiter_priorityrrr delimiter_count_with_priorityrr`rlresysmaxsizerSrrrrgetMIGRATE_COMMENT_DELIMITERSmin bracket_depthrrzr$r)rqrar` last_leafbtr lowest_depthtrailing_comma_saferlast_non_comment_leafrrprevious_priority leaf_priorityrrms` @@r]rmrm@s| 4;1,''T1 BI B;669 6OO ;;;/00d:; l**  , ,-? @ @A E ESTTT Ydj$:NL;L &T &htn & & & & & & &5d5x~55555557t<<#DK00  $!>$'''''''''$qL R]->-> t{8a<( ) ). .  !3 3 3!%???&t,, , , , , , , ,<);<<  J4#5#E#E"9$"I"I  >/ / /H@U4U4U3#%7L ))"T((33 . . .1 s4;////&.HHH*?4;x!|+DEEEEEEEEE   Ydj$BVL/ !3\   s $A((Bc#nKstdtjjjdt dttffd }jD]<}||Ed{V |D]}||Ed{V=rVdSdS)z4Split standalone comments from the rest of the line.z*Line does not have any standalone commentsrrrbc3K |ddS#t$r>Vtjjj|YdSwxYw)rTr)rlreNrrs r]rz0standalone_comment_split..append_to_lines &  $ $T $ = = = = = & & &    T=QL    % % % % % %  &rN) rfrVrr`rlrerSrror)rqrar`rrrrms` @r]rnrns'  , , . .HFGGG Ydj$:NL &T &htn & & & & & & & 55!>$'''''''''!0066 5 5M%~m44 4 4 4 4 4 4 4 4 5r\rrrc 6 t|jd|D]}t|jtrdS |jt jkrt|||d}tt|j D]\}}t|tr(|jt jkrt||||t|tr)|jt jkrt|dh||t|tr)|jt jkrt|dh|||dkr,t|tr|jt jkrd}|dkrt|trp|jt jkr[|jt jkrFt+||s6t-|s't/||||dd rt1||d |rb|jt jkr|jt jkrlt|jt6rR|jjt8jkr8|jjd kr(t/||||d rt1||d nt|tr*|jt jkrt=|||| n|jt jkrd|vrlt?|j dkrTtA|j dr:tC|j dr |j djt j"ks&t/||||rt1||d ntG|rt1||d n|jt j$krtK|||dS|dkr,|jt8j&kr|jt j'kr|t|t6r/|j((|j(jt8j)kr|jdkrdStU|st1||d |jt8j+k}t|t6o |j|vp|}dS)aEMake 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_endmarkerr`Nrcaser0rT)rr`rarallow_star_exprrr4rrr(rrrjr),rr|rrrrzrC with_stmt_maybe_wrap_cms_in_parensrrZrrvrT annassignr case_blockguardtestlist_star_exprr expr_stmt_atom_has_magic_trailing_comma_is_atom_multilinerrDfor_stmtrrSrRNAMEremove_with_parensrnr2r9testr6 import_from_normalize_import_fromr except_clauserrr3r) rrrr`rapc check_lparrr comma_checks r]rrsDKe$GGG !"(G 4 4  FF  yDN""!$h777J!$t}"5"566l l u eT " " uzT^'C'C &Lth     eT " " uzT_'D'D &VH4(     eT " " uzTZ'?'? &TF     QJJ5$''  d555J QJJ5$''  di'' T^++25$??,&u--,3!-1 $  @$D%???? 7 @ di''I..u1488/&+uz99&,556%15 D(eUCCCCE4((' @TY$.-H-H"5$THMMMMMty(( $$''1,,!%."344-!%."455-N1%*di776$THD(eUCCCCe$$ @#D%>>>>>d...&tUE::: J%*,,I!333 5$'' @&2&+u{::K6))(// @#D%????jEK/ t,, K< ' 6;  Ul l r\rrrcvt|r7t|jdsJd|_d|jd_dS|jt jkr]||tt j d| tt j ddSdS)Nrru) r2r9rrrzrRrrrSrrr)rrrs r]r r CsU2V_R011111 $&!!! uz ! !E4 B#7#7888DR0011111 " !r\c|jdjtjkrt |jdkr|jdjt jkrb|jdjdjtjkr8t|jd|||drt||jddtt|jdjd}tt|jdjd}|jdjd}t|tr{|jt jks>|jdjtjkstd|jDr*t!|t!|dSdSdSdSdSdSdS) NrrjTrFrrc3jK|].}t|to|jtjkV/dSrd)rvrSrzrRrrs r]rz&remove_await_parens..msOud++N e>N0Nr\)rrzrRAWAITrnrCrrrrDr rSrvrTrrr()rrr`rarrbracket_contentss r]rrQs }Q ++DM0B0BQ0F0F M!  !TY . . a )!,1UZ??2 a !-1  K$D$-*:EJJJJ#4q)9)B1)EFFO"4q)9)B2)FGGO#}Q/8; *D11 0 %33#,Q/4 CC!1!:D ////////C,+0F0F . .??* 0 0CCr\ctj|vs8t|jdks |jdjt jkrdSd}tdt|jD]&}|j|jtj kr|}n'|ttj d}ttj d}|jd|}|D]}| tt j|tt j||g} |d| dSdS)zWhen enabled and safe, wrap the multiple context managers in invisible parens. It is only safe when `features` contain Feature.PARENTHESIZED_CONTEXT_MANAGERS. rrjNru)rPARENTHESIZED_CONTEXT_MANAGERSrnrrzrCrrrRrrSrrrrT testlist_gexpr) rrr`ra colon_indexrrrcontext_managersr new_childs r]rrvs= .h>> t}   " " =  DI - -"K 1c$-(( ) ) =  EK / /K E 0EJ##EJ##=;7%  E LLNNNN Id4#57GHH$O   !Y''''')r\c|jtjkrht||||drt ||dt |jdtr t|jd|||d Sd S|jtj kr4|jD]*}t |trt||||+d S|jtj kritd| Ds@t|jd|||dr#t ||jddd Sd Sd Sd S) z6Recursively hide optional parens in `with` statements.TrFrrjrc3@K|]}|jtjkVdSrd)rzrR COLONEQUALrs r]rz%remove_with_parens..s=33*. U%%333333r\rN) rzrCrrrDrvrrTrr, asexpr_testrro)rrrr`rars r]rrs( yDI . )-     = e < < < < dmA& - - U t}Q/D8 T T T T T T U U d( ( (] N NE%&& N"5$THMMMM N N d& & &s3326++--33300 & / M! )-     G dmA&6 F F F F F F ' & & & G Gr\c2|jsdSt|S)zCheck if an atom node has a magic trailing comma. Returns True for single-element tuples with trailing commas like (a,), which should be preserved to maintain their tuple type. F)rcr6)rrr`s r]rrs"  $u   r\ct|trt|jdkrdS|jd}|D]#}t|t r d|jvrdS$dS)zGCheck if an atom node is multiline (indicating intentional formatting).rFrjrT)rvrTrnr pre_orderrSr|)rrmiddlers r]rrs dD ! !S%7%7!%;%;u]1 F!!## eT " " tu|';';44 5r\Frrc |jtjtjfvsYt |sIt |s9t |r|jtjkst |r/|jtjkrt|tj st|r|jtj ks|st|tkrit j|vrt |r{|jt|jjtjkrZ|j$t+|jr|jjdks/t/|s |st1|st3|rdSt5|r|jtjtj tjtjtjtjtjtjtj tj!tj"tj!f vrdS|j#d}|j#d}tI|r4tK|r$|j#d}tM|j'(|sd|_d|_tS||||| tU|r|+|j#d|j#dj'(r5|j#dj'|j#dj'z|j#d_'|j#dj'(r|j#dj'|j'z|_'dSd S) a If 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. NasFrrrjrtrurT),rzrCrexprr0r6r<r3rr+rRrrBrrr rUNPARENTHESIZED_EXCEPT_TYPESrr!rr4rr>r=r1rAr assert_stmt return_stmtfuncdefr,tnamerdel_stmtrr2r9r?r|rrr.r) rrrr`rarrrrr7s r]rrsF $)TY/// $   0    0 TNN 0 &{d.>>> TNN ? t~--%dEK88. TNN. &{dn<< - = /t44FF4@@TNNAK+K$(:::%1%d&7882)/477 &d + + 8  8 %=T$B$B 8    8uD!! ; N N       L N   J M M M   5 M! E = DU! d 3 3!q! .fm.A.A.C.C$OOO  EKDJ+ )E      ) 0 0 G NN6?1- . . .q!(..00 OA&-0B0II")r")//11 G$ob184;F u 4r\c|jr |jdvsdSt}d} |jd}|jt jkr$d}|t||j |}n#ttf$rYdSwxYw|tko1|jjr|p#|jjt jt jhvS)zFShould `line` be immediately split with `delimiter_split()` after RHS?z[{(FrTr)rrrrorzrRraddrrwr IndexErrorrr r`rcrCrr)rqrrtrailing_commar max_prioritys r]rr\s  "'<'E'Eu eeGNKO >U[ ( (!N KK9 & & &+BB7BSS  #uu > )   ' :N H  ! &49d6F*G GsA!BBBrc#\Kt}|js|Vd|jz}d}d}t}|dD]\}}} || z }||krdS| t |jt |jzk} |jtks| rdS|r||urd}`|jtvrv|dkr|j |dz nd} | r:| jtj kr%|j t|j ||j sdS|t!||jtvr|dkr|j |dz nd} | r2| jt"vr$|t!|?|rO|t!|||||V| r:| jtj kr%|j t|j ||j sdS|jr |j }|}dS)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_reversedrrj)rrcrlenumerate_with_lengthrnrr|rzr$r"rorRrrr5rBrr#updateclear) rqrrXlengthrrinner_bracketsrr leaf_lengthhas_inline_commentprevs r]rYrYwsD  $ ^F#'O#'O"%%%N$($>$>4$>$P$P6'6' t[+ K   EE(3tz??S=M=M+MM 9* * *.@ * EE - '&&"&...16t{519--  U[00,83,dDK9EE""2d88,,, Y* * *-2QYY4;uqy))DD  %555""2d88,,, O,,--- N+++$$&&&  I,,(4/0DdDKXX5 z '"&"6"&m6'6'r\rur_rxrUc|st|}g}|||D]\}t|d|krtd|t ||]t |}tj|vs|j j dks|j j rtd|j j Ds}|si|dsO|ds5t#|dstd|jDr|S|}t)|||j|tjhz} t+||| | } t-fd | Dr| }|S) Nrz-Line transformer returned an unchanged resultrr]c3$K|] }|jV dSrd)r)rbrackets r]rz"run_transformer..s$KKw}KKKKKKr\rrtc3(K|] }|jduVdSrdrrs r]rz"run_transformer..s);;tt{d";;;;;;r\r_c3:K|]}t|VdS)rtN)r)rlnr`s r]rz"run_transformer..s1 H H2  . . . H H H H H Hr\)rrrrKrrzrrrrrWrw invisiblerrrardrroclonerroall) rqrxr`rarUrytransformed_line features_set line_copy features_fopsecond_opinions ` r]roros (!$''F%IdHd;;VV  & &t , , 8 8!"QRR R n%5D8TTTUUUUx==L*l::   '5 0 0#- 1 KKD,@,JKKK K K 1  * * , , 1 !9 9 9 ; ; 1 !9 6 6 8 8 1 q  5 5 5 1 ;;t{;;; ; ; 1  I)T4;///7#E"FFL$9dL8N H H H H H H HHH  Mr\)r[)r[r[)FF)rZrrcollections.abcrr dataclassesrenumrr functoolsrr typingr r black.bracketsr r rrrblack.commentsrrrrr black.linesrrrrrrr black.moderrr black.nodesr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCrDblack.numericsrE black.stringsrFrGrHrIrJ black.transrKrLrMrNrOrPrQblib2to3.pgen2rRblib2to3.pytreerSrTrOLeafIDrPrVr_rrSrzboolrjrrkr[rrrrrZrrrrrrrrmrnrrr rrrrrrrrYror[r\r]rps  00000000$$$$$$$$.---------&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&&N544444! &&&&&&&&  4:MMMMM/MMM |B|B|B|B|BGDM|B|B|B~ !  4Z =?yy yy&0&9y d^yyyyx33D3T3333>T ? ?%g.?6:? d^????J%'!   ! V   d^ , "dIdI dI V dIdIdIdIdIV%'! AA A A A! A V  A d^ AAAAHH H&H.2H HHHHV4t4D4! J!"&!9=! !!!!H. J... & .  ....b++"TcDj$*W2E$ 4 S  QU    .~>V V$W-V59V d^VVVVr $W-59 d^DC C !#hC 15C AKGATC  C C C C L 24 2 23 24 2 2 2 2"0d"0$"0*W:M"0RV"0"0"0"0J'( '('(&0&9'( '('('('(T-G -G-G$(-G4>w4G-G -G-G-G-G`  4 D     R D    (*/! nn n n n! n #' n  n nnnnbD4D6I'DI'sI'xF ?TI'I'I'I'd +++ ++ +! +  + $Z++++++r\