!.d ddlZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlmZmZmZddlmZmZddlmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*dd l+m,Z,m-Z-ed Z.e/Z0e/Z1ee,e-fZ2eGd d Z3eGd dZ4eGddZ5eGddZ6de e.de e e0e.ffdZ7 d+de3de3de e,de8ddf dZ9ddde3ded e:de8fd!Z;de3de8fd"Zde3d(e,d$e/de8fd)Z?de3de:fd*Z@dS),N) dataclassfield) CallableDictIteratorListOptionalSequenceTupleTypeVarUnioncast)COMMA_PRIORITY DOT_PRIORITYBracketTracker)ModePreview)BRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENTTEST_DESCENDANTS child_towards is_importis_multiline_stringis_one_sequence_betweenis_type_commentis_with_or_async_with_stmt replace_childsyms whitespace) str_width)token)LeafNodeTc ^eZdZUdZeed<dZeed<ee Z e e ed<ee Zeee e fed<eeZeed<d Zeed <d Zeed <d Zee ed < d3de dededd fdZd4de dedd fdZedefdZedefdZedefdZedefdZedefdZedefdZ edefdZ!edefdZ"edefdZ#edefdZ$d ddee%e gefdefd Z&e'j(fd!edefd"Z)defd#Z*defd$Z+defd%Z, d4d&e d'edefd(Z-d)e defd*Z.de de e fd+Z/d5d,Z0de defd-Z1 d4d.ede2e3e4e effd/Z5d6d0Z6de7fd1Z8defd2Z9d S)7Linez;Holds leaves and comments. Can be printed with `str(line)`.moderdepthdefault_factoryleavescommentsbracket_trackerFinside_bracketsshould_split_rhsNmagic_trailing_commaleaf preformatted track_bracketreturnc|jtvp%t|j}|sdSt j|jkr|jr |jdd=|jr4|s2|xj t|| |z c_ |j s|r|rn|j ||jjr||r||_n+||dr|||s|j|dSdS)agAdd a new `leaf` to the end of the line. Unless `preformatted` is True, the `leaf` will receive a new consistent whitespace prefix and metadata applied by :class:`BracketTracker`. Trailing commas are maybe removed, unpacked for loop variables are demoted from being delimiters. Inline comments are put aside. N)complex_subscriptT)ensure_removable)typerboolvaluestripr#COLONis_class_paren_emptyr-prefixr!is_complex_subscriptr0r/markr)r2has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr3r4r5 has_values +lib/python3.11/site-packages/black/lines.pyrGz Line.append=skI)ET$*2B2B2D2D-E-E   F ;$) # #(A # BCC ; |  KK:(A(A$(G(G KK   -| -} -  % %d + + +y- -0066504D-..td.KK -**,,,""4(( % K  t $ $ $ $ $ % %c|jjdkr<|jrtd|jr|jt krtd|||dS)zLike :func:`append()` but disallow invalid standalone comment structure. Raises ValueError when any `leaf` is appended after a standalone comment or when a standalone comment is not the first leaf on the line. rz$cannot append to standalone commentsz5cannot append standalone comments to a populated liner4N)r/r* is_comment ValueErrorr-r;rrG)rHr3r4s rJ append_safezLine.append_safe_sy   % * * I !GHHH{ ty,>>> K D| 44444rKcht|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr-r;rrHs rJrNzLine.is_commentps-4;1$RQ)<@R)RRrKc`t|o|jdjtjkS)zIs this line a decorator?r)r<r-r;r#ATrTs rJ is_decoratorzLine.is_decoratorus%Dzz=dk!n1UX==rKcTt|ot|jdS)zIs this an import line?r)r<rr-rTs rJrzLine.is_importzs#Dzz7i A777rKcTt|ot|jdS)zIs this a with_stmt line?r)r<rr-rTs rJrzLine.is_with_or_async_with_stmts$DzzH8QHHHrKct|o5|jdjtjko|jdjdkS)z Is this line a class definition?rclass)r<r-r;r#NAMEr=rTs rJis_classz Line.is_classs@ JJ 0 A#uz1 0 A$/ rKcd|jo)|jdddtdDkS)zEIs this line a class definition with a body consisting only of "..."?NcBg|]}ttjdS).)r$r#DOT).0_s rJ z&Line.is_stub_class..s16 6 6 %&DC 6 6 6 rK)r]r-rangerTs rJ is_stub_classzLine.is_stub_classsG} RSS!16 6 */((6 6 6 "  rKc6 |jd}n#t$rYdSwxYw |jd}n#t$rd}YnwxYw|jtjkr |jdkp8|jtjko#|duo|jtjko |jdkS)zBIs this a function definition? (Also returns True for async defs.)rFrRNdef)r- IndexErrorr;r#r\r=ASYNC)rH first_leaf second_leafs rJis_defz Line.is_defs QJJ   55  *.+a.KK   KKK 5:-K*2Be2K Ou{ * +4' + EJ. +!U*  s   0 ??c6t|ot|jdkor|jok|jdjt jkoK|jdjdko5|jdjt jko|jdjdkS)zzIs this a class with no base classes but using parentheses? Those are unnecessary and should be removed. (rf)) r<rSr-r]r;r#LPARr=RPARrTs rJr@zLine.is_class_paren_emptys JJ ,DK  A% ,  , A#uz1 , A$+  ,  A#uz1  ,  A$+ rKct|oD|jdjtjko$|jdjdS)z#Is the line a triple quoted string?r)z"""z''')r<r-r;r#STRINGr= startswithrTs rJis_triple_quoted_stringzLine.is_triple_quoted_stringsK JJ @ A#u|3 @ A$//?? rKcvt|jdkrdS|jdjtjkS)z/Does this line open a new level of indentation.rF)rSr-r;r#r?rTs rJ opens_blockzLine.opens_blocks4 t{  q 5{2#u{22rKfirst_leaf_matchesrct|jdkrdS|jd}|jtks|jdS|dup||jS)zIs this line converted from fmt off/skip code? If first_leaf_matches is not None, it only returns True if the first leaf of converted code matches. rRFrN)rSr-r;rfmt_pass_converted_first_leaf)rHrr3s rJis_fmt_pass_convertedzLine.is_fmt_pass_convertedsm t{  q 5{1~ I+ + +195!T) -?-?  .. .  rK depth_limitcV|jD] }|jtkr|j|krdS!dS)z)If so, needs to be split before emitting.TF)r-r;r bracket_depth)rHrr3s rJcontains_standalone_commentsz!Line.contains_standalone_commentss;K  Dy...43E3T3TtturKct} |jd}|t||jt jks|jt jkr6|js/|jd}|t|n#t$rYdSwxYwd}|j D]5\}}|D]-}t|r|st|ds||vrdSd}.6dS)Nr|r8F ignoreT) setr-addidr;r#COMMArvr=rkr.itemsr)rH ignored_ids last_leaf comment_seenleaf_idr.comments rJ$contains_uncollapsable_type_commentsz)Line.contains_uncollapsable_type_commentss)ee   BI OOByMM * * *~,,%*,,Y_, !KO 9 ...   55  !%!4!4!6!6 $ $ GX# $ $"7++$#$+GY??$#;66#ttt#  $usBB B.-B.c\|jsdStd|jDd}tdt|jDd}||krQ|jddD]A}|jt |gD]}t |drdSBdS)NFc3:K|]}|jdk |jVdSrNlinenorcr3s rJ z9Line.contains_unsplittable_type_ignore..s1SS4$+QRBRBR4;BRBRBRBRSSrKrc3:K|]}|jdk |jVdSrrrs rJrz9Line.contains_unsplittable_type_ignore..s0 O OTdkQ>N>NT[>N>N>N>N O OrKr8rT)r-nextreversedr.getrr)rH first_line last_linenoders rJ!contains_unsplittable_type_ignorez&Line.contains_unsplittable_type_ignores{ 5SS4;SSSUVWW  O OXdk%:%: O O OQR    " " BCC( $ $#}00D2>>$$G&w ::$#ttt$$urKc>td|jDS)Nc34K|]}t|VdSN)rrs rJrz2Line.contains_multiline_strings..)s+EE&t,,EEEEEErK)anyr-rTs rJcontains_multiline_stringszLine.contains_multiline_strings(s!EEEEEEEErKclosingr:c|jtvr'|jr |jdjtjksdS|jtjkrdS|jtjkr|jrV|jjtj kr<|j r5t|j ||jtj tjfrdS|sdS|jd}|jdS|jjtj kp<|j dup3t|j ||jtj tjf S|jrdS|j t|j ||jsdSdS)aTReturn True if we have a magic trailing comma, that is when: - there's a trailing comma here - it's not a one-tuple - it's not a single-element subscript Additionally, if ensure_removable: - it's not from square bracket indexing (specifically, single-element square bracket indexing) r|FT)bracketsN)r;rr-r#rRBRACERSQBparentr traileropening_bracketrLSQB subscriptlistr)rHrr:commas rJrDzLine.has_magic_trailing_comma+s{ L, , ,  - B$ 335 <5< ' '4 <5: % % N'4<77+8++K#j%*5 8u# tKOE|#u !T%77*d2.+K#j%*5   > 4  " .7N  #Wdk8 8 .4urKrc|jtkr"|jr d|_dS|jt jkrdS|jst|_d|_dS|jd}|jt jkr|j s|j rtt|j dkrIt|s:t|jdkrt|_d|_dS|jd}|jt!|g|dS)z0Add an inline or standalone comment to the line.Fr|rfrrr8T)r;rr/any_open_bracketsrAr#COMMENTr-rvr=rrSlistrr. setdefaultrrG)rHrrs rJrFzLine.append_commentis9 L. . .$6688 / GN5 <5= ( (5{ -GLGN5KO Nej ( (O )  )D)0022334499#G,,: 4;!##1 !#u BI   I33::7CCCtrKcR|jt|gS)z;Generate comments that should appear directly after `leaf`.)r.rr)rHr3s rJcomments_afterzLine.comments_afters }  D2...rKc|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.r|N)r-popr.rrextend)rHtrailing_commatrailing_comma_commentss rJrEzLine.remove_trailing_commasx**"&-"3"3B~4F4F"K"K   DKO!4!4b99@@ #     rKcJ|j}|dS|j}t|tr<|jt jkrdS|jt jkrt||}|duo*td| DS)zAReturn True iff `leaf` is part of a slice with non-trivial exprs.NFc32K|]}|jtvVdSr)r;r)rcns rJrz,Line.is_complex_subscript..s;3 3 +,AF& &3 3 3 3 3 3 rK) r/ get_open_lsqb next_sibling isinstancer%r;r listmakerrrr pre_order)rHr3 open_lsqbsubscript_starts rJrBzLine.is_complex_subscripts(6688  5#0 ot , , G#t~55u#t'999"/"F"Fd* s3 3 0?0I0I0K0K3 3 3 0 0  rKrc#Kttttgtt t tff|rtnt}||j D]p\}}t|j t|j z}d|j vrdS| |D]}|t|j z }|||fVqdS)zReturn an enumeration of leaves with their length. Stops prematurely on multiline strings and standalone comments.  N)rrr r$rr Indexenumerate_reversed enumerater-rSrAr=r)rHropindexr3lengthrs rJenumerate_with_lengthzLine.enumerate_with_lengths htn%xeTk0B'CC D"* 9     2dk?? & &KE4%%DJ7Ftz!!..t44 - -#gm,,,v% % % % % & &rKc\t|j|j|j|j|jS)N)r)r*r0r1r2)r(r)r*r0r1r2rTs rJclonez Line.clones5* 0!2!%!:     rKcV|sdSd|jz}t|j}t|}|j||j}|D]}|t |z }tj |j D]}|t |z }|dzS)zRender the line.rz ) r*iterr-rrAr=str itertoolschain from_iterabler.values)rHindentr-firstresr3rs rJ__str__z Line.__str__s 4$*$dk""V 4v4u{44  D 3t99 CC 44T]5I5I5K5KLL  G 3w<< CCTzrKc8t|jp|jS)z/Return True if the line has leaves or comments.)r<r-r.rTs rJ__bool__z Line.__bool__sDK04=111rK)FFF)r6N)r6r():__name__ __module__ __qualname____doc__r__annotations__r*intrrr-rr$dictr.rLeafIDrr/r0r<r1r2r rGrPpropertyrNrWrrr]rhror@rzr}rrsysmaxsizerrrrrDrFrrErBrr rrrrrrrKrJr(r(/sEE JJJE3NNNt444FDJ444).t)D)D)DHd64:%&DDD&+eN&K&K&KO^KKK!OT!!!"d"""+/(4.///MR % % %(, %EI %  % % % %D555D5T5555"SDSSSXS>d>>>X>84888X8IDIIIXI $   X  t   X     X $  d    X      X 3T333X3IM   %-hvt|.D%E     (?Bkd"d""""H4BFDFFFF7<<<</3< <<<<|#d#t####J/4/DJ////      $    & %&&& %tS() *&&&&*     2$222222rKr(cFeZdZUdZeed<eed<eed<eed<eed<dS) RHSResultz2Intermediate split result from a right hand split.headbodytailrclosing_bracketN)rrrrr(rr$rrKrJrrsK<< JJJ JJJ JJJrKrceZdZUdZeed<eded<eed<dZe ed<e e Z e eed<dZe ed <d e efd Zd S) LinesBlockzClass that holds information about a block of formatted lines. This is introduced so that the EmptyLineTracker can look behind the standalone comments and adjust their empty lines for class or def lines. r)previous_block original_linerbeforer+ content_linesafterr6ctt|j}||jzg|jz||jzgzS)N)r))rr(r)rrr)rH empty_lines rJ all_lineszLinesBlock.all_linessE49---.. $+ % &); ;zDJ?V>W W rKN)rrrrrrr r(rrrrrrrrrrrKrJrrs JJJ\****FCOOO$uT:::M49:::E3NNN 49      rKrceZdZUdZeed<dZeeed<dZ ee ed<e e Z eeed<dZee ed<d ed e fd Zd ed eeeffd Zd ed ed eeeffdZdS)EmptyLineTrackera{Provides a stateful method that returns the number of potential extra empty lines needed before and after the currently processed line. Note: this tracker works on lines that haven't been split yet. It assumes the prefix of the first leaf consists of optional newlines. Those newlines are consumed by `maybe_empty_lines()` and included in the computation. r)N previous_linerr+ previous_defssemantic_leading_comment current_liner6cf||\}}|jr |jjnd}|jdn||z }t |j|j|||}|jr2|j#|jjs|jjr|r|j|r||_n|jr|rd|_||_||_|S)zReturn the number of extra empty lines before and after the `current_line`. This is for separating `def`, `async def` and `class` with extra empty lines (two on module-level). rN)r)rrrr) _maybe_empty_linesrrrrr)rNrWr)rHrrrprevious_afterblocks rJmaybe_empty_linesz"EmptyLineTracker.maybe_empty_liness // == 6:6IP,22q !) A.( .&      " 1!)&3*+6*;A*2:f:05-* 1f 1,0D ))# rKc~d}|jdkr|jjrdnd}|jr?|jd}|jd}t ||}d|_nd}|j}|jr|jdj|kr|jjrq|jJ|r$|j s|jj rt d|}ntj |jvr'|jdj r|jdj sd}nZ|rd}nUd}nR|rd}nM|sI|jdjr7|jdjtjkr|jdjdvrd}nd}|j|jr|jdj|k|js|j s|j r|||S|jrD|jjr8|js1|t*s||jjkr|pddfS|jr|jj r |jr|dfS|jr|jjrd S|dfS) NrRrrrrrr|)withtryforwhileifmatchr~rr)r*r)is_pyir-rAcountminrrror"blank_line_after_nested_stub_classr]rhr;r#r?r=rrW#_maybe_empty_lines_for_class_or_defrrrzr})rHr max_allowedrmrr*s rJrz#EmptyLineTracker._maybe_empty_lines8s   " "#y/6!!QK   %,Q/J&,,T22F--F "J  F" & %T%7%;%AU%J%Jy$ )555 !4 9K9R  F^^FF>$)KK*2.7L .r2@LFFFFFFFF*2.4%+B/4 CC$+A.4MNNFFF   " " $ $ $M & %T%7%;%AU%J%JN  $ R (; R|?T R;;L&QQ Q   $", $!* $!66)6TT  $ +111Ka!# #   "+ 4  19    $"4"@ 4qyrKrc|js|j||jdS|jjr|jjr |jrdSdS|jj|jkr|jjs |jj rdSd}|jj r]|jj|jkrH|dkrB|j }|7|j 0|j j js|j j js|jdkr|}ndS|jjr|js |jjrI|jj|jkrd}n|jj|jkrd}nt|jr|jjrd}n^d}n[|j s|jr'|jj s|jrt!d|}n)d}n&|jj|jkrd}nd}n |jrdnd}|-|j }|$t#|j||jz |_d}|dfS)Nr)rrRrrRrr)rWrrGrr)rrhr*r]rorNrrrr}rrmaxr)rHrrcomment_to_add_newlinesslcnewlinesrs rJrz4EmptyLineTracker._maybe_empty_lines_for_class_or_defsc( 4   % %l 3 3 3   %4   * y L$> t4   #l&8 8 8   ' 9+/+=+D 948<   ) "(L,>>>! /C&2*8A3*8D3J!OO*-''t 9  6$ (:(C %+l.@@@ HH'- 0BBB HH!/!D4F4T! HH HH# '3'@ (/  %! #1f~~HH !HH#)L,>>>(.5qqAH " .4CN)/6AANDXX(.{rK)rrrrrrrr r(rrrrrrrr r rrrrrKrJrrs JJJ$(M8D>(((+/NHZ(/// %d ; ; ;M4:;;;59hz2999&d&z&&&&PItIc3hIIIIVJ J*-J sCxJJJJJJrKrsequencer6c#hKt|dz }t|D] }||fV|dz}dS)z;Like `reversed(enumerate(sequence))` if that were possible.rRN)rSr)rrelements rJrrsS MMA EH%%g  rKFnew_lineold_liner-r4c|D]r}t|j|j}t|||||||D]}||dsdS)a Append leaves (taken from @old_line) to @new_line, making sure to fix the underlying Node structure where appropriate. All of the leaves in @leaves are duplicated. The duplicates are then appended to @new_line and used to replace their originals in the underlying Node structure. Any comments attached to the old leaves are reattached to the new leaves. Pre-conditions: set(@leaves) is a subset of set(@old_line.leaves). rMTN)r$r;r=rrGr)r r!r-r4old_leafnew_leaf comment_leafs rJ append_leavesr&s== x~66h)))|<<<$33H== = =L OOLtO < < < < = ==rKr)line_strliner)r'cz|st|}|jrtnt}tj|vr-|||jkod|vo| S|rdSd|vr|||jkS|d^}}}|||jks|||jkrdSg}d}g} tj } t|j D]\} } | tj krd} | j dzt|kr|dn/| j dzt|kr|} | %|#|j | j dzkr| j } | dkrdS| j | krV| jt"jkrAdg}|| z }| j|vr| t|j dz ks|| j xxdz cc<| tj krt)| | j } t+| r`t| dkrdS| }| }t-||vr5| ||jn|j}t-||v5t| dkrdSt1d|DS)aFor non-multiline strings, return True if `line` is no longer than `line_length`. For multiline strings, looks at the context around `line` to determine if it should be inlined or split up. Uses the provided `line_str` rendering, if any, otherwise computes a new one. rFNrRrTc3"K|] }|dkV dSrr)rcvals rJrz'is_line_short_enough..Hs&**Csax******rK)line_to_stringpreviewr"rSrmultiline_string_handling line_lengthrsplitmathinfrr-rrGrr;r#r prev_siblingrrrrall)r(r)r'widthrrdlastcommasmultiline_stringmultiline_string_contextsmax_level_to_updateir3 had_comma ignore_ctxsctxs rJis_line_short_enoughr?s# (!$'' .II3E(44 E(OOt/ / 8H$ 855777  ((**u 8uX$"222nnT**OEAt uU||d&&&%%++8H*H*HuF'+*,(T[))'!'!4 $( * *'+I!A%F 33 a    #a'#f++55"JJLL %$0$2d6H16LLL'+&8#q== 55  !4 4 4ek9Q9Q04fK 4 4K%44c$+>N>NQR>R9R9Rt)***a/*** $( * *"%&94;M"N"N  t $ $ !,--11uu# Cc((h&&)00555:%j c((h&& $%%**t **6*** * **rKc|j}t|dkrdS|djtjkr|djtjkrd}d}|d}|dddD]}|jt vr|jtvrdS|dz }nh|jtjkr|dz }nM|jtjkr'|jtjks|jt vsdSn|jtvrdS|dkr |dkrdSdS) zReturn False if the line cannot be split *for sure*. This is not an exhaustive search but a cheap heuristic that we can use to avoid some unfortunate formattings (mostly around wrapping unsplittable code in unnecessary parentheses). rrFrrRr|r8NT) r-rSr;r#rxrbrrr\)r(r- call_count dot_countrr3s rJ can_be_splitrCKs[F 6{{Qu ay~%%&).EI*E*E  bz26r6N  Dy,,,9$444 55a ei''Q ej(( UY..$)?O2O2O 55"222uu1}}auu 4rKrhsr/cJ|j}|j}|jsdS|}||}|dkrdS|dkr,t j|jvr|tkr|j j rdS|tkrdSt|j dks Jd|j d}|j d}|jtvr"|jt vrt#|||rdS|j d}|j d } | jt$jksK| jt$jks6| jt$jkrV| jrO| jjt.jkr5|jtvrdSt3|rdSt5|| | rdSdS) aDoes `rhs.body` have a shape safe to reformat without optional parens around it? Returns True for only a subset of potentially nice looking formattings but the point is to not return false positives that end up producing lines that are too long. TrRFrrzStranded delimiterr)rr/r8r|)r6r/)rr/ delimitersmax_delimiter_prioritydelimiter_count_with_priorityr(wrap_multiple_context_managers_in_parensr)rrrrrSr-r;rr_can_omit_opening_parenr#rvrrrr rr_can_omit_closing_paren) rDr/r(bt max_prioritydelimiter_countrsecond penultimater6s rJcan_omit_invisible_parensrQos 8D B =t,,..L66|DDOu!  < I I..3/5 |##t t{  q "6  KNE [^F z%%%&+=M*M*M "4u+ N N N 4 +b/K ;r?D UZ 9 $ $ I # #  $  DL00  / / /5 u % % 4 "4d L L L 4 5rKrcd}d|jz}d}|D]>\}}}|jtvr |j|urd}|r||z }||krn.|jt vrd}?t |j|dzkrdSdS) See `can_omit_invisible_parens`.Frqr|TrR)r*rr;rrrrSr-)r(rr/ remainderr_indexr3 leaf_lengths rJrJrJsI ^F F%)%?%?%A%A!k 9( ( (T-AU-J-JI  " k !F ##y,,,!  t{  vz ) )4 5rKr6cd|jz}d}|D]0\}}}||z }||jur |s||krdS |jtvrd}1dS)rSrqFT)r*rrr;r)r(r6r/rseen_other_bracketsrUr3rVs rJrKrKs ^F%)%?%?%A%A''!k+ 4' ' '" f &;&;tt'<Y* * *"&  5rKcFt|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr>)r(s rJr,r,s t99??4  rKr)Arr1r dataclassesrrtypingrrrrr r r r r rblack.bracketsrrr black.moderr black.nodesrrrrrrrrrrrrr r! black.stringsr"blib2to3.pgen2r#blib2to3.pytreer$r%r&rrrLNr(rrrrr<r&rr?rCrQrJrKr,rrKrJrcs ((((((((                        HGGGGGGGGG$$$$$$$$ $##### &&&&&&&& GCLL   4: l2l2l2l2l2l2l2 l2^             * LLLLLLL L^!%q/1JNS==="=,0J=FJ= ====202X+X+X+ X+X+),X+ X+X+X+X+v!t!!!!!HQ QQ QQQQh$$ST2$CD"!!#!!!!!!rK