
    ziJ                        d dl Z d dlZd dlmZmZ d dlmZmZmZm	Z	m
Z
mZmZmZmZmZ d dlmZmZmZ d dlmZmZ d dlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z) d dl*m+Z+ d dl,m-Z- d d	l.m/Z/m0Z0  ed
          Z1e2Z3e2Z4ee/e0f         Z5e G d d                      Z6e G d d                      Z7e G d d                      Z8e G d d                      Z9dee1         deee3e1f                  fdZ:	 d+de6de6de	e/         de;ddf
dZ<ddde6ded e=de;fd!Z>de6de;fd"Z?d#e7d$e2de;fd%Z@de6d&e/d$e2de;fd'ZAde6d(e/d$e2de;fd)ZBde6de=fd*ZCdS ),    N)	dataclassfield)
CallableDictIteratorListOptionalSequenceTupleTypeVarUnioncast)COMMA_PRIORITYDOT_PRIORITYBracketTracker)ModePreview)BRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENTTEST_DESCENDANTSchild_towardsis_docstring
is_funcdef	is_importis_multiline_stringis_one_sequence_betweenis_type_commentis_type_ignore_commentis_with_or_async_with_stmtmake_simple_prefixreplace_childsyms
whitespace)	str_width)token)LeafNodeTc            	          e Zd ZU dZ ed          Zeed<   dZe	ed<    ee
          Zee         ed<    ee          Zeeee         f         ed	<    ee          Zeed
<   dZeed<   dZeed<   dZee         ed<   	 d6dedededdfdZd7dededdfdZedefd            Zedefd            Zedefd            Zedefd            Zedefd            Z edefd            Z!edefd            Z"edefd            Z#edefd            Z$edefd            Z%edefd            Z&edefd             Z'dd!d"ee(egef                  defd#Z)defd$Z*defd%Z+defd&Z,defd'Z-defd(Z.	 d7d)ed*edefd+Z/d,edefd-Z0dedee         fd.Z1d8d/Z2dedefd0Z3	 d7d1ede4e5e6ee	f                  fd2Z7d9d3Z8de9fd4Z:defd5Z;dS ):Linez;Holds leaves and comments. Can be printed with `str(line)`.F)reprmoder   depthdefault_factoryleavescommentsbracket_trackerinside_bracketsshould_split_rhsNmagic_trailing_commaleafpreformattedtrack_bracketreturnc                    |j         t          v p%t          |j                                                  }|sdS t
          j        |j         k    r| j        r
| j        dd= | j        r:|s8|xj	        t          ||                     |          | j                  z  c_	        | j        s|r|rn| j                            |           | j        j        r|                     |          r|| _        n+|                     |d          r|                                  |                     |          s| j                            |           dS dS )ag  Add 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_subscriptr.   T)ensure_removable)typer   boolvaluestripr'   COLONis_class_paren_emptyr2   prefixr%   is_complex_subscriptr.   r5   r4   markr7   has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr8   r9   r:   	has_values        V/var/www/tmov.alphamb/tmov_inventario/venv/lib/python3.11/site-packages/black/lines.pyrL   zLine.append@   sq    I)ET$*2B2B2D2D-E-E	 	F;$)##(A#BCC ; 	| 	 KK:"&";";D"A"AY   KK
  	-| 	-} 	- %%d+++y- -0066 504D-..td.KK -**,,,""4(( 	%Kt$$$$$	% 	%    c                     | j         j        dk    s| j                                         r<| j        rt	          d          | j        r|j        t          k    rt	          d          |                     ||           dS )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.
        r   z$cannot append to standalone commentsz5cannot append standalone comments to a populated liner9   N)	r4   r/   any_open_for_or_lambda
is_comment
ValueErrorr2   r@   r   rL   )rM   r8   r9   s      rO   append_safezLine.append_safed   s      &!++#::<< ,  I !GHHH{ ty,>>> K   	D|44444rP   c                 h    t          | j                  dk    o| j        d         j        t          k    S )z"Is this line a standalone comment?   r   )lenr2   r@   r   rM   s    rO   rT   zLine.is_commentx   s-     4;1$RQ)<@R)RRrP   c                 `    t          |           o| j        d         j        t          j        k    S )zIs this line a decorator?r   )rA   r2   r@   r'   ATrZ   s    rO   is_decoratorzLine.is_decorator}   s%     Dzz=dk!n1UX==rP   c                 T    t          |           ot          | j        d                   S )zIs this an import line?r   )rA   r   r2   rZ   s    rO   r   zLine.is_import   s#     Dzz7iA777rP   c                 T    t          |           ot          | j        d                   S )zIs this a with_stmt line?r   )rA   r!   r2   rZ   s    rO   r!   zLine.is_with_or_async_with_stmt   s$     DzzH8QHHHrP   c                     t          |           o5| j        d         j        t          j        k    o| j        d         j        dk    S )z Is this line a class definition?r   class)rA   r2   r@   r'   NAMErB   rZ   s    rO   is_classzLine.is_class   s@     JJ 0A#uz10A$/	
rP   c                 d    | j         o)| j        dd         d t          d          D             k    S )zEIs this line a class definition with a body consisting only of "..."?Nc                 B    g | ]}t          t          j        d           S .r(   r'   DOT.0_s     rO   
<listcomp>z&Line.is_stub_class.<locals>.<listcomp>   s1     6
 6
 6
%&DC  6
 6
 6
rP      )rc   r2   rangerZ   s    rO   is_stub_classzLine.is_stub_class   sG     } 
RSS!1 6
 6
*/((6
 6
 6
 "
 	
rP   c                 6   	 | j         d         }n# t          $ r Y dS w xY w	 | j         d         }n# t          $ r d}Y nw xY w|j        t          j        k    r|j        dk    p8|j        t          j        k    o#|duo|j        t          j        k    o
|j        dk    S )zBIs this a function definition? (Also returns True for async defs.)r   FrX   Ndef)r2   
IndexErrorr@   r'   rb   rB   ASYNC)rM   
first_leafsecond_leafs      rO   is_defzLine.is_def   s    	QJJ 	 	 	55		*.+a.KK 	 	 	KKK	5:-K*2Be2K 
Ou{* +4'+ EJ.+ !U*		
s    
0 ??c                     | j         oE| j        dd         t          t          j        d          gd t          d          D             z   k    S )zHIs this line a function definition with a body consisting only of "..."?N:c                 B    g | ]}t          t          j        d           S rg   ri   rk   s     rO   rn   z$Line.is_stub_def.<locals>.<listcomp>   s7     O
 O
 O
%&DC  O
 O
 O
rP   ro   )rx   r2   r(   r'   rD   rp   rZ   s    rO   is_stub_defzLine.is_stub_def   sb     { 
t{233/Dc4J4J3K O
 O
*/((O
 O
 O
 4
  
 	
rP   c                 6   t          |           ot          | j                  dk    or| j        ok| j        d         j        t
          j        k    oK| j        d         j        dk    o5| j        d         j        t
          j        k    o| j        d         j        dk    S )zzIs this a class with no base classes but using parentheses?

        Those are unnecessary and should be removed.
              (ro   ))	rA   rY   r2   rc   r@   r'   LPARrB   RPARrZ   s    rO   rE   zLine.is_class_paren_empty   s     JJ ,DK  A%,, A#uz1, A$+	,
 A#uz1, A$+	
rP   c                     | r | j         d         j        t          j        k    rdS | j         d         j        }|                    d          rdS t          j        | j        v r|                    d          rdS dS )z#Is the line a triple quoted string?r   F)z"""z'''T)zr'''zr"""zR'''zR""")	r2   r@   r'   STRINGrB   
startswithr   accept_raw_docstringsr.   )rM   rB   s     rO   is_triple_quoted_stringzLine.is_triple_quoted_string   s      	t{1~*el::5A$N++ 	4(DI55%:J:J,;
 ;
5 4urP   c                 f    d | j         D                                 t          j                  dk    S )z Is the line a chained assignmentc                     g | ]	}|j         
S  r@   rl   r8   s     rO   rn   z.Line.is_chained_assignment.<locals>.<listcomp>   s    222d	222rP   rX   )r2   countr'   EQUALrZ   s    rO   is_chained_assignmentzLine.is_chained_assignment   s/     32dk22288EEIIrP   c                 v    t          | j                  dk    rdS | j        d         j        t          j        k    S )z/Does this line open a new level of indentation.r   F)rY   r2   r@   r'   rD   rZ   s    rO   opens_blockzLine.opens_block   s4     t{q  5{2#u{22rP   first_leaf_matchesr   c                    t          | j                  dk    rdS | j        d         }|j        t          k    s|j        dS |du p ||j                  S )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.
        rX   Fr   N)rY   r2   r@   r   fmt_pass_converted_first_leaf)rM   r   r8   s      rO   is_fmt_pass_convertedzLine.is_fmt_pass_converted   sm     t{q  5{1~I+++195!T) 
-?-?..
 .
 	
rP   c                 @    | j         D ]}|j        t          k    r dS dS )z)If so, needs to be split before emitting.TF)r2   r@   r   rM   r8   s     rO   contains_standalone_commentsz!Line.contains_standalone_comments   s4    K 	 	Dy...tt / urP   c                     t          j        | j        d           D ]W\  }}|t          j        k    rt          |          }t          |          dk    r9|D ]}|                     |          r  dS XdS )zFChck if we have an implicit multiline string with comments on the linec                     | j         S Nr   )r8   s    rO   <lambda>zGLine.contains_implicit_multiline_string_with_comments.<locals>.<lambda>   s    di rP   rX   TF)	itertoolsgroupbyr2   r'   r   listrY   comments_after)rM   	leaf_typeleaf_group_iterator	leaf_listr8   s        rO   0contains_implicit_multiline_string_with_commentsz5Line.contains_implicit_multiline_string_with_comments   s    .7.?K///
 /
 
	  
	 *I* EL((011I9~~""!    &&t,,  444   urP   c                    t                      }	 | j        d         }|                    t          |                     |j        t
          j        k    s|j        t
          j        k    r6|j        s/| j        d         }|                    t          |                     n# t          $ r Y dS w xY wd}| j
                                        D ]4\  }}|D ],}t          |          r|st          |          s||vr  dS d}-5dS )Nr   r=   FT)setr2   addidr@   r'   COMMAr   rB   rt   r3   itemsr   r    )rM   ignored_ids	last_leafcomment_seenleaf_idr3   comments          rO   $contains_uncollapsable_type_commentsz)Line.contains_uncollapsable_type_comments  s'   ee	BIOOByMM***~,,%*,,Y_, !KO	9... 	 	 	55	 !%!4!4!6!6 		$ 		$GX# $ $"7++ $# $27;;$#;66#ttt#$ us   BB   
B.-B.c                 Z   | j         sdS t          d | j         D             d          }t          d t          | j                   D             d          }||k    rP| j         dd          D ]@}| j                            t          |          g           D ]}t          |          r  dS AdS )NFc              3   :   K   | ]}|j         d k    |j         V  dS r   Nlinenor   s     rO   	<genexpr>z9Line.contains_unsplittable_type_ignore.<locals>.<genexpr>@  s1      SS4$+QRBRBR4;BRBRBRBRSSrP   r   c              3   :   K   | ]}|j         d k    |j         V  dS r   r   r   s     rO   r   z9Line.contains_unsplittable_type_ignore.<locals>.<genexpr>B  s0      OOTdkQ>N>NT[>N>N>N>NOOrP   r=   T)r2   nextreversedr3   getr   r    )rM   
first_line	last_linenoder   s        rO   !contains_unsplittable_type_ignorez&Line.contains_unsplittable_type_ignore/  s    { 	5 SS4;SSSUVWW
OOXdk%:%:OOOQR
 
	 "" BCC( $ $#}00D2>> $ $G-g66 $#ttt$$ urP   c                 >    t          d | j        D                       S )Nc              3   4   K   | ]}t          |          V  d S r   )r   r   s     rO   r   z2Line.contains_multiline_strings.<locals>.<genexpr>Q  s+      EE&t,,EEEEEErP   )anyr2   rZ   s    rO   contains_multiline_stringszLine.contains_multiline_stringsP  s!    EEEEEEEErP   closingr?   c                    |j         t          v r'| j        r | j        d         j         t          j        k    sdS |j         t          j        k    rdS |j         t          j        k    r_|j        V|j        j         t          j	        k    r<|j
        5t          |j
        || j        t          j        t          j        f          rdS dS | j        rdS |j
        t          |j
        || j                  sdS dS )aT  Return 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   FTN)brackets)r@   r   r2   r'   r   RBRACERSQBparentr$   traileropening_bracketr   LSQBr   )rM   r   r?   s      rO   rI   zLine.has_magic_trailing_commaS  s     L,,, -B$335<5<''4<5:%%*N'4<77+7++K#j%*5	   8 u4> 	4".7N#Wdk8
 8
. 4urP   r   c                    |j         t          k    r"| j                                        r	d|_        dS |j         t
          j        k    rdS | j        st          |_         d|_        dS | j        d         }|j         t
          j        k    r|j	        s|j
        rt          t          |j
                                                            dk    rIt          |          s:t          | j                  dk     rt          |_         d|_        dS | j        d         }| j                            t!          |          g                               |           dS )z0Add an inline or standalone comment to the line. Fr   ro   r   r=   T)r@   r   r4   any_open_bracketsrF   r'   COMMENTr2   r   rB   r   rY   r   r   r3   
setdefaultr   rL   )rM   r   r   s      rO   rK   zLine.append_comment  s9    L...$6688 /  GN5<5=((5{ 	-GLGN5KO	Nej((O )  ) D)0022334499#G,, : 4;!##1!#uBI  I33::7CCCtrP   c                 R    | j                             t          |          g           S )z;Generate comments that should appear directly after `leaf`.)r3   r   r   r   s     rO   r   zLine.comments_after  s     }  D2...rP   c                    | j                                         }| j                            t          |          g           }| j                            t          | j         d                   g                               |           dS )z@Remove the trailing comma and moves the comments attached to it.r   N)r2   popr3   r   r   extend)rM   trailing_commatrailing_comma_commentss      rO   rJ   zLine.remove_trailing_comma  sx    **"&-"3"3B~4F4F"K"K  DKO!4!4b99@@#	
 	
 	
 	
 	
rP   c                 J   | j                                         }|dS |j        }t          |t                    r<|j        t          j        k    rdS |j        t          j        k    rt          ||          }|duo*t          d |                                D                       S )zAReturn True iff `leaf` is part of a slice with non-trivial exprs.NFc              3   2   K   | ]}|j         t          v V  d S r   )r@   r   )rl   ns     rO   r   z,Line.is_complex_subscript.<locals>.<genexpr>  s;       3
 3
+,AF&&3
 3
 3
 3
 3
 3
rP   )r4   get_open_lsqbnext_sibling
isinstancer)   r@   r$   	listmakersubscriptlistr   r   	pre_order)rM   r8   	open_lsqbsubscript_starts       rO   rG   zLine.is_complex_subscript  s    (6688	5#0ot,, 	G#t~55u#t'999"/"F"Fd* 
s 3
 3
0?0I0I0K0K3
 3
 3
 0
 0
 	
rP   r   c              #     K   t          t          t          t                   gt          t
          t          t          f                  f         |rt          nt                    } || j	                  D ]p\  }}t          |j                  t          |j                  z   }d|j        v r dS |                     |          D ]}|t          |j                  z  }|||fV  qdS )zReturn an enumeration of leaves with their length.

        Stops prematurely on multiline strings and standalone comments.
        
N)r   r   r
   r(   r   r   Indexenumerate_reversed	enumerater2   rY   rF   rB   r   )rM   r   opindexr8   lengthr   s          rO   enumerate_with_lengthzLine.enumerate_with_length  s       htn%xeTk0B'CCD"*9	
 
 2dk?? 	& 	&KE4%%DJ7Ftz!!..t44 - -#gm,,,v%%%%%	& 	&rP   c                 \    t          | j        | j        | j        | j        | j                  S )N)r.   r/   r5   r6   r7   )r,   r.   r/   r5   r6   r7   rZ   s    rO   clonez
Line.clone  s5    * 0!2!%!:
 
 
 	
rP   c                 V   | sdS d| j         z  }t          | j                  }t          |          }|j         | |j         }|D ]}|t          |          z  }t          j        	                    | j
                                                  D ]}|t          |          z  }|dz   S )zRender the line.r   z    )r/   iterr2   r   rF   rB   strr   chainfrom_iterabler3   values)rM   indentr2   firstresr8   r   s          rO   __str__zLine.__str__  s     	4$*$dk""V4v4u{44 	 	D3t99CC 44T]5I5I5K5KLL 	  	 G3w<<CCTzrP   c                 8    t          | j        p| j                  S )z/Return True if the line has leaves or comments.)rA   r2   r3   rZ   s    rO   __bool__zLine.__bool__  s    DK04=111rP   )FFF)r;   N)r;   r,   )<__name__
__module____qualname____doc__r   r.   r   __annotations__r/   intr   r2   r   r(   dictr3   r   LeafIDr   r4   r5   rA   r6   r7   r	   rL   rV   propertyrT   r]   r   r!   rc   rq   rx   r}   rE   r   r   r   r   r   r   r   r   r   r   rI   rK   r   rJ   rG   r   r   r   r   r   r   r   r   r   rP   rO   r,   r,   2   sI        EEE"""D$"""E3NNNt444FDJ444).t)D)D)DHd64:%&DDD&+eN&K&K&KO^KKK!OT!!!"d"""+/(4./// MR"% "%"%(,"%EI"%	"% "% "% "%H5 5 5D 5T 5 5 5 5( SD S S S XS >d > > > X> 84 8 8 8 X8 ID I I I XI 
$ 
 
 
 X
 
t 
 
 
 X
 
 
 
 
 X
$ 
T 
 
 
 X
 
d 
 
 
 X
     X Jt J J J XJ 3T 3 3 3 X3 IM
 
 
%-hvt|.D%E
	
 
 
 
(d    $    "d " " " "H4    BFD F F F F 7<- --/3-	- - - -^#d #t # # # #J/4 /DJ / / / /
 
 
 

 
$ 
 
 
 
&  %& &&	%tS()	*& & & &*
 
 
 
     2$ 2 2 2 2 2 2rP   r,   c                   F    e Zd ZU dZeed<   eed<   eed<   eed<   eed<   dS )	RHSResultz2Intermediate split result from a right hand split.headbodytailr   closing_bracketN)r   r   r   r  r,   r  r(   r   rP   rO   r  r    sK         <<
JJJ
JJJ
JJJrP   r  c                       e Zd ZU dZeed<   ed          ed<   eed<   dZe	ed<    e
e          Zee         ed<   dZe	ed	<   d
Zeed<   dee         fdZd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_liner   beforer0   content_linesafterF	form_feedr;   c                     t          t          | j                            }t          | j        | j        |          }|g| j        z   || j        z  gz   S )N)r.   )r   r,   r.   r"   r  r  r  r  )rM   
empty_linerF   s      rO   	all_lineszLinesBlock.all_lines  sP    49---..
#DKLLx$,,
TZ0G/HHHrP   N)r   r   r   r  r   r  r	   r,   r  r  r   r   r  r   r   r  r  rA   r  r   rP   rO   r  r    s           JJJ\****FCOOO$uT:::M49:::E3NNNItI49 I I I I I IrP   r  c            
           e Zd ZU dZeed<   dZee         ed<   dZ	ee
         ed<    ee          Zee         ed<   dZee
         ed<   d	ed
e
fdZd	ed
eeef         fdZd	ededed
eeef         fdZdS )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_liner  r0   previous_defssemantic_leading_commentcurrent_liner;   c                    t           j        | j        v o2|j        dk    o't	          |j                  od|j        d         j        v }|                     |          \  }}| j        r| j        j	        nd}| j
        dn||z
  }t           j        |j        v rV| j        rO| j        j        Ct          | j        j        j                  dk    r!| j        j        j        r|j        s	|j        sd}t#          | j        | j        ||||          }|j        r2| j
        #| j
        j        s| j
        j        r|r| j        |r|| _        n|j        r|rd| _        || _
        || _        |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).
        r   z
NrX   )r.   r  r  r  r  r  )r   allow_form_feedsr.   r/   rA   r2   rF   _maybe_empty_linesr  r  r  module_docstring_newlinesrY   r  r   rc   rx   r  rT   r]   r  )rM   r  r  r  r  previous_afterblocks          rO   maybe_empty_linesz"EmptyLineTracker.maybe_empty_lines)  s    $	1 8"a'8\())8 ,-a077	 	 //==6:6IP,22q
 !) A.( 	 -1BBB# C#2:D'5<==BB#1I C!* C.:.A C F.&
 
 
 " 
	1!)&3 * +6 * ;A * 2:f:05-* 	1f 	1,0D))#rP   c                 ~   d}|j         dk    r| j        j        rdnd}|j        r?|j        d         }|j                            d          }t          ||          }d|_        nd}t          |          }|j         }d }| j        rL| j        d         j         |k    r6| j        	                                }| j        r| j        d         j         |k    6|| j
        J | j        j        rH|r|j        s| j
        j        r|rdnd}nst          j        | j        v r|j        r
|j        sd}nO|rd}nJd}nG|rd}nB|s>|j         r7|j        d         j        t"          j        k    r|j        d         j        dvrd}nd}|j        s|j        s|j        r|                     |||          S | j
        rD| j
        j        r8|j        s1|                    t,                    s|| j
        j         k    r|pddfS | j
        r,| j
        j        r |j        rt          j        |j        v rd	S |dfS t          j        |j        v oot7          |j        d                    pT| j
        oM| j
        j        d         o;| j
        j        d         j        o$t;          | j
        j        d         j                   }| j
        r| j
        j        r|sd
S |dfS )NrX   r   r   r   r   r   )withtryforwhileifmatchr   r   rX   r   r   )r/   r.   is_pyir2   rF   r   minrA   r  r   r  rx   r   "blank_line_after_nested_stub_classrc   rq   r@   r'   rD   rB   r]   #_maybe_empty_lines_for_class_or_defr   r   r   $no_blank_line_before_class_docstringallow_empty_first_line_in_blockr   r   r   r   )	rM   r  max_allowedrv   r  user_had_newliner/   previous_defis_empty_first_line_oks	            rO   r   z#EmptyLineTracker._maybe_empty_linesb  s   ""#y/6!!QK 	%,Q/J&,,T22F--F "JF<<"  	4T%7%;%AU%J%J-1133L   	4T%7%;%AU%J%J #%111y # !4 9K9R "29QQFF>$)KK$- L(6 L FF FFFF FF$* %+B/4CC$+A.4MN N FFF$ 	(; 	|?T 	;;f&6  
 	$",	$ !*	$ !66)6TT		$
 +111Ka!## 	"+	 4	
 ;|?PPPt19
 3|7HH 	 !4Q!7888 & L*1!4L*1!4;L 't'9'@'C'JKKK 	 	".	 +	
 4qyrP   r  r5  c                 X   |j         s| j                            |           | j        dS | j        j         r| j        j        r	|j        rdS dS | j        j        |j        k     r| j        j        s| j        j	        rdS d }| j        j
        r]| j        j        |j        k    rH|dk    rB| j        }|7|j        0|j        j        j        s|j        j        j        s|j        dk    r|}ndS | j        j        r|j        s| j        j        rI| j        j        |j        k     rd}n| j        j        |j        k    rd}n|j        r| j        j        rd}nd}nt           j        |j        v r| j        j        |j        k    rd}n~|j	        s|j         r'| j        j	        s|j        rt%          d|          }nLd}nI| j        j        |j        k    rd}n1d}n.|j        rdnd}t           j        | j        v r| j        j        r|sd}|-|j        }|$t+          |j        |          |j        z
  |_        d}|dfS )Nr-  r,  r   rX   r   )r]   r  rL   r  r.   r.  rq   r/   rc   rx   rT   r  r  r  r   r  r   +blank_line_between_nested_and_def_stub_filer/  dummy_implementationsr}   maxr  )rM   r  r  r5  comment_to_add_newlinesslcnewlinesr  s           rO   r1  z4EmptyLineTracker._maybe_empty_lines_for_class_or_def  s    ( 	4%%l333%4* 	y L$> t4#l&888' 9+/+=+D 9 48<)	"(L,>>>!/C&2*8A 3*8D 3 J!OO*-''t9 .	$ #(:(C #%+l.@@@ HH'-0BBB HH!/ !D4F4T ! HH HH C|GXXX&,|/AAA#'3'@(/  % !  #1f~~HH  !HH#)L,>>>(.5qqAH -::&2 ;( ; ".4CN)/6AANDXX (. {rP   )r   r   r   r  r   r  r  r	   r,   r  r  r   r   r  r   r  r$  r   r  r   rA   r1  r   rP   rO   r  r    s%          JJJ$(M8D>(((+/NHZ(/// %d ; ; ;M4:;;;59hz29997d 7z 7 7 7 7rgt gc3h g g g gR] ]*-]AE]	sCx] ] ] ] ] ]rP   r  sequencer;   c              #   h   K   t          |           dz
  }t          |           D ]}||fV  |dz  }dS )z;Like `reversed(enumerate(sequence))` if that were possible.rX   N)rY   r   )r?  r   elements      rO   r   r   +  sS      MMAEH%%  g
 rP   Fnew_lineold_liner2   r9   c                     |D ]r}t          |j        |j                  }t          ||           |                     ||           |                    |          D ]}|                     |d           sdS )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).
    rR   TN)r(   r@   rB   r#   rL   r   )rB  rC  r2   r9   old_leafnew_leafcomment_leafs          rO   append_leavesrH  3  s      = =x~66h)))|<<<$33H== 	= 	=LOOLtO<<<<	== =rP   r   )line_strliner.   rI  c                   |st          |           }t          j        |v rt          nt          }t          j        |vr- ||          |j        k    od|vo|                                  S |                                 rdS d|vr ||          |j        k    S |                    d          ^}}} ||          |j        k    s ||          |j        k    rdS g }d}g }	t          j
        }
t          | j                  D ]\  }}|
t          j
        k    rd}|j        dz   t	          |          k    r|                    d           n/|j        dz   t	          |          k     r|                                }|%|#|j        |j        dz   k    r|j        }
|dk    r dS |j        |
k    rV|j        t"          j        k    rAdg}||	z  }|j        |v r|t	          | j                  dz
  k    s||j        xx         dz  cc<   |
t          j
        k    rt)          |
|j                  }
t+          |          r`t	          |	          dk    r dS |}|}t-          |          |v r5|	                    |           |j        n|j        }t-          |          |v 5t	          |	          dk    rdS t1          d |D                       S )a  For 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.
    r   FNrX   r   Tc              3   "   K   | ]
}|d k    V  dS r   r   )rl   vals     rO   r   z'is_line_short_enough.<locals>.<genexpr>  s&      **Csax******rP   )line_to_stringr   respect_east_asian_widthr&   rY   multiline_string_handlingline_lengthr   splitmathinfr   r2   bracket_depthrL   r   r@   r'   r   prev_siblingr/  r   r   r   all)rJ  r.   rI  widthr   rm   lastcommasmultiline_stringmultiline_string_contextsmax_level_to_updateir8   	had_commaignore_ctxsctxs                   rO   is_line_short_enoughrb  K  s)     (!$'' 9TAAIIsE(44E(OOt// 8H$855777	
 ((** u8uX$"222nnT**OEAtuU||d&&&%%++8H*H*Hu F'+*,-1XT[)) '! '!4$(**'+I!A%F33a    #a'#f++55"JJLL	%$0$2d6H16LLL '+&8#q== 55!444ek9Q9Q 04fK44K%44c$+>N>NQR>R9R9Rt)***a/***$(**"%&94;M"N"Nt$$ 	!,--11uu#Cc((h&&)00555:%j	 c((h&& $%%**t**6******rP   c                    | j         }t          |          dk     rdS |d         j        t          j        k    r|d         j        t          j        k    rd}d}|d         }|ddd         D ]}|j        t          v r|j        t          vr dS |dz  }nh|j        t          j        k    r|dz  }nM|j        t          j        k    r'|j        t          j        k    s|j        t          v s dS n|j        t          vr dS |dk    r	|dk    r dS dS )	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).
    r   Fr   rX   r   r=   NT)	r2   rY   r@   r'   r   rj   r   r   rb   )rJ  r2   
call_count	dot_countr   r8   s         rO   can_be_splitrf    s    [F
6{{Quay~%%&).EI*E*E
	bz26r6N 	 	Dy,,,9$444 55a

ei''Q		ej((	UY..$)?O2O2O 55"222uu1}}auu4rP   rhsrQ  c                    | j         }d}t          |j                  D ]K}|r||j        u rd}|j        t
          k    r|s dS |s%|j        t          v r|j        |j        v r	|j        r|}L|j        }|j	        sdS |
                                }|                    |          }|dk    rdS |dk    r,t          j        |j        v r|t          k    r| j        j        rdS |t$          k    rdS t'          |j                  dk    s
J d            |j        d         }|j        d         }	|j        t(          v r"|	j        t          vrt+          |||          rdS |j        d	         }
|j        d
         }|j        t,          j        k    sK|j        t,          j        k    s6|j        t,          j        k    rV|j        rO|j        j        t6          j        k    r5|
j        t(          v rdS t;          |          rdS t=          |||          rdS dS )a  Does `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.
    NFTrX   r   zStranded delimiterr   )r   rQ  r=   r   )rY  rQ  )r
  r   r2   r   r@   r   r   rB   r4   
delimitersmax_delimiter_prioritydelimiter_count_with_priorityr   (wrap_multiple_context_managers_in_parensr.   r   r	  r!   r   rY   r   _can_omit_opening_parenr'   r   r   r   r   r$   r   r   _can_omit_closing_paren)rg  rQ  rJ  r  r8   btmax_prioritydelimiter_countr   secondpenultimaterY  s               rO   can_omit_invisible_parensrt    sT    8D '+O%% # # 	#t'FFF"O9***?*55	#	---$33
 4 #O		B= t,,..L66|DDOu!<	II..3 / 5
 |##tt{q   "6    KNE[^Fz%%%&+=M*M*M"4u+NNN 	4 +b/K;r?D 		UZ9$$ I## $ DL00 ///5u%% 	 4"4dLLL 	45rP   r   c                   d}d| j         z  }d}|                                 D ]>\  }}}|j        t          v r|j        |u rd}|r||z  }||k    r n.|j        t
          v rd}?t          | j                  |dz   k    rdS dS ) See `can_omit_invisible_parens`.Fr   r   TrX   )r/   r   r@   r   r   r   rY   r2   )rJ  r   rQ  	remainderr   _indexr8   leaf_lengths           rO   rm  rm  /  s    I^FF%)%?%?%A%A  !k9(((T-AU-J-JI 	"k!F##y,,,!	 t{vz))45rP   rY  c                    d| j         z  }d}|                                 D ]0\  }}}||z  }||j        u r|s||k    r dS  |j        t          v rd}1dS )rv  r   FT)r/   r   r   r@   r   )rJ  rY  rQ  r   seen_other_bracketsrx  r8   ry  s           rO   rn  rn  H  s    ^F%)%?%?%A%A ' '!k+4'''" f&;&;tt '< Y***"&5rP   c                 F    t          |                               d          S )zmReturns the string representation of @line.

    WARNING: This is known to be computationally expensive.
    r   )r   rC   )rJ  s    rO   rN  rN  Y  s    
 t99??4   rP   r   )Dr   rS  dataclassesr   r   typingr   r   r   r   r	   r
   r   r   r   r   black.bracketsr   r   r   
black.moder   r   black.nodesr   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   black.stringsr&   blib2to3.pgen2r'   blib2to3.pytreer(   r)   r*   r  r   r  LNr,   r  r  r  r   rA   rH  r   rb  rf  rt  rm  rn  rN  r   rP   rO   <module>r     s        ( ( ( ( ( ( ( (                        H G G G G G G G G G $ $ $ $ $ $ $ $                                       ( $ # # # # #             & & & & & & & & GCLL	
4: B2 B2 B2 B2 B2 B2 B2 B2J         I I I I I I I I* N N N N N N N Nb! %q/1J     NS= =="=,0J=FJ=	= = = =2 02X+ X+ X+
X+X+),X+	X+ X+ X+ X+v!t ! ! ! ! !Hb	bb 
b b b bJ$ $ S T    2$  C D    "! !# ! ! ! ! ! !rP   