o FDi @s"ddlZddlZddlmZmZmZddlmZmZddl m Z m Z m Z m Z ddlmZmZmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(dd l)m*Z*dd l+m,Z,m-Z-e d Z.e/Z0e/Z1e e,e-fZ2eGd d d Z3eGdddZ4eGdddZ5eGdddZ6dee.dee7e0e.ffddZ8 d4de3de3de9e,de:ddf ddZ;dd d!e3d"ed#ed(e4d)e/de:fd*d+Z?d!e3d,e,d)e/de:fd-d.Z@d!e3d/e,d)e/de:fd0d1ZAd!e3deZ-d?e defd@dAZ.dBe defdCdDZ/de de e fdEdFZ0dVdGdHZ1de defdIdJZ2 dUdKede3e4e5e e ffdLdMZ6dWdNdOZ7de8fdPdQZ9defdRdSZ:d S)XLinez;Holds leaves and comments. Can be printed with `str(line)`.F)reprmoderdepthdefault_factoryleavescommentsbracket_trackerinside_bracketsshould_split_rhsNmagic_trailing_commaleaf preformatted track_bracketreturncCs|jtvp|jtjkpt|j}|sdS|jtjkr%|jr%|j dd=|j r:|s:|j t || ||j d7_ |jsA|rA|r]|j||j jrT||rS||_n ||r]|||sj|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_subscriptr()typerr"FSTRING_MIDDLEboolvaluestripCOLONis_class_paren_emptyr,prefixr is_complex_subscriptr(r/r.markr1has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr2r3r4 has_valuerHD/var/www/Datamplify/venv/lib/python3.10/site-packages/black/lines.pyrE5s4         z Line.appendcCsN|jjdks |jr|jrtd|jr|jtkrtd|j||ddS)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 liner3N) r.r)any_open_for_or_lambda is_comment ValueErrorr,r8rrE)rFr2r3rHrHrI append_safe^s zLine.append_safecCst|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr,r8rrFrHrHrIrLrszLine.is_commentcCst|o |jdjtjkS)zIs this line a decorator?r)r:r,r8r"ATrQrHrHrI is_decoratorwszLine.is_decoratorcCt|o t|jdS)zIs this an import line?r)r:rr,rQrHrHrIr|zLine.is_importcCrT)zIs this a with_stmt line?r)r:rr,rQrHrHrIrrUzLine.is_with_or_async_with_stmtcCs*t|o|jdjtjko|jdjdkS)z Is this line a class definition?rclass)r:r,r8r"NAMEr;rQrHrHrIis_classs z Line.is_classcCs&|jo|jddddtdDkS)zEIs this line a class definition with a body consisting only of "..."?NcSg|]}ttjdqS.r#r"DOT.0_rHrHrI  z&Line.is_stub_class..)rXr,rangerQrHrHrI is_stub_classszLine.is_stub_classcCsz|jd}Wn tyYdSwz|jd}Wn ty$d}Ynw|jtjkr0|jdkpD|jtjkoD|duoD|jtjkoD|jdkS)zBIs this a function definition? (Also returns True for async defs.)rFrONdef)r, IndexErrorr8r"rWr;ASYNC)rF first_leaf second_leafrHrHrIis_defs$    z Line.is_defcCs4|jo|jddttjdgddtdDkS)zHIs this line a function definition with a body consisting only of "..."?N:cSrZr[r]r_rHrHrIrbrcz$Line.is_stub_def..rd)rlr,r#r"r=rerQrHrHrI is_stub_defs$ zLine.is_stub_defcCs`t|o/t|jdko/|jo/|jdjtjko/|jdjdko/|jdjtjko/|jdjdkS)zzIs this a class with no base classes but using parentheses? Those are unnecessary and should be removed. (rd)) r:rPr,rXr8r"LPARr;RPARrQrHrHrIr>s zLine.is_class_paren_emptycCsF|r |jdjtjkr dS|jdj}|drdS|dr!dSdS)z#Is the line a triple quoted string?rF)z"""z'''T)zr'''zr"""zR'''zR""")r,r8r"STRINGr; startswith)rFr;rHrHrI_is_triple_quoted_strings   zLine._is_triple_quoted_stringcCrT)zIs the line a docstring?r)r:rr,rQrHrHrIrrUzLine.is_docstringcCsdd|jDtjdkS)z Is the line a chained assignmentcSsg|]}|jqSrHr8r`r2rHrHrIrbsz.Line.is_chained_assignment..rO)r,countr"EQUALrQrHrHrIis_chained_assignmentszLine.is_chained_assignmentcCs$t|jdkr dS|jdjtjkS)z/Does this line open a new level of indentation.rF)rPr,r8r"r=rQrHrHrI opens_blockszLine.opens_blockfirst_leaf_matchesrcCsFt|jdkr dS|jd}|jtks|jdurdS|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. rOFrN)rPr,r8rfmt_pass_converted_first_leaf)rFrr2rHrHrIis_fmt_pass_converteds    zLine.is_fmt_pass_convertedcCs |jD] }|jtkr dSqdS)z)If so, needs to be split before emitting.TF)r,r8rrFr2rHrHrIcontains_standalone_commentss  z!Line.contains_standalone_commentscCs^t|jddD]#\}}|tjkrq t|}t|dkrq |D] }||r+dSq q dS)zFChck if we have an implicit multiline string with comments on the linecSs|jSNry)r2rHrHrIszGLine.contains_implicit_multiline_string_with_comments..rOTF) itertoolsgroupbyr,r"rvlistrPcomments_after)rF leaf_typeleaf_group_iterator leaf_listr2rHrHrI0contains_implicit_multiline_string_with_commentss    z5Line.contains_implicit_multiline_string_with_commentscCst}z)|jd}|t||jtjks|jtjkr+|js+|jd}|t|Wn t y6YdSwd}|j D]\}}|D]}t |rX|sTt |sX||vrXdSd}qDq>dS)Nr~r6FT)setr,addidr8r"COMMArur;rhr-itemsrr)rF ignored_ids last_leaf comment_seenleaf_idr-commentrHrHrI$contains_uncollapsable_type_commentss0     z)Line.contains_uncollapsable_type_commentscCs|jsdStdd|jDd}tddt|jDd}||kr?|jddD]}|jt|gD] }t|r=dSq3q(dS)NFcs |] }|jdkr|jVqdSrNlinenorzrHrHrI =z9Line.contains_unsplittable_type_ignore..rcsrrrrzrHrHrIr?rr6T)r,nextreversedr-getrr)rF first_line last_linenoderrHrHrI!contains_unsplittable_type_ignore,sz&Line.contains_unsplittable_type_ignorecCstdd|jDS)Ncs|]}t|VqdSr)rrzrHrHrIrNz2Line.contains_multiline_strings..)anyr,rQrHrHrIcontains_multiline_stringsMszLine.contains_multiline_stringsclosingcCs|jtvr|jr|jdjtjksdS|jtjkrdS|jtjkrT|jdurR|jjtj krR|j durRt |j ||jtj tjfdrR|j dusGJ|j jtjksPJdSdS|jrYdS|j durht |j ||jshdSdS)zReturn True if we have a magic trailing comma, that is when: - there's a trailing comma here - it's not from single-element square bracket indexing - it's not a one-tuple r~FTN)brackets)r8rr,r"rRBRACERSQBparentrtraileropening_bracketrLSQB prev_sibling subscriptlistr)rFrrHrHrIrBPs:        zLine.has_magic_trailing_commarcCs|jtkr|jrd|_dS|jtjkrdS|js"t|_d|_dS|jd}|jtjkrV|j sV|j rVt t |j dkrVt |sVt |jdkrQt|_d|_dS|jd}|jt|g|dS)z0Add an inline or standalone comment to the line.Fr~rdrqr6T)r8rr.any_open_bracketsr?r"COMMENTr,rur;rrPrrr- setdefaultrrE)rFrrrHrHrIrD|s6     zLine.append_commentcCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)r-rrrrHrHrIrszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.r~N)r,popr-rrextend)rFtrailing_commatrailing_comma_commentsrHrHrIrCs zLine.remove_trailing_commacCsj|j}|dur dS|j}t|tr&|jtjkrdS|jtjkr&t ||}|duo4t dd| DS)zAReturn True iff `leaf` is part of a slice with non-trivial exprs.NFcss|]}|jtvVqdSr)r8r)r`nrHrHrIrs  z,Line.is_complex_subscript..) r. get_open_lsqb next_sibling isinstancer$r8r listmakerrrr pre_order)rFr2 open_lsqbsubscript_startrHrHrIr@s     zLine.is_complex_subscript is_reversedccsttttgttttff|rtnt}||j D]+\}}t |j t |j }d|j vr2dS| |D] }|t |j 7}q7|||fVqdS)zReturn an enumeration of leaves with their length. Stops prematurely on multiline strings and standalone comments.  N)r rrr#rtupleIndexenumerate_reversed enumerater,rPr?r;r)rFropindexr2lengthrrHrHrIenumerate_with_lengths  zLine.enumerate_with_lengthcCst|j|j|j|j|jdS)N)r(r)r/r0r1)r&r(r)r/r0r1rQrHrHrIclonesz Line.clonecCs|sdSd|j}t|j}t|}|j||j}|ddd|D7}tj |j }dd|D}|d|7}|dS)zRender the line.rz rcsrrstrrzrHrHrIrrzLine.__str__..cSsg|]}t|qSrHr)r`rrHrHrIrbsz Line.__str__..) r)iterr,rr?r;joinrchain from_iterabler-values)rFindentr,firstres comments_iterr-rHrHrI__str__s  z Line.__str__cCst|jp|jS)z/Return True if the line has leaves or comments.)r:r,r-rQrHrHrI__bool__sz Line.__bool__)FFF)r5N)r5r&);__name__ __module__ __qualname____doc__rr(r__annotations__r)intrr,r#dictr-LeafIDr r.r/r:r0r1rrErNpropertyrLrSrrrXrfrlror>rxrr}rrrrrrrrrBrDrrCr@rrrrrrrrrHrHrHrIr&'s     )  $!,%   r&c@s:eZdZUdZeed<eed<eed<eed<eed<dS) RHSResultz2Intermediate split result from a right hand split.headbodytailrclosing_bracketN)rrrrr&rr#rHrHrHrIrs  rc@szeZdZUdZeed<eded<eed<dZe ed<e e dZ e e ed<dZe ed <d Zeed <d e e fd dZdS) 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_linesafterF form_feedr5cCs8tt|jd}t|j|j|}|g|j||jgS)N)r()rr&r(rrrrr)rF empty_liner?rHrHrI all_linesszLinesBlock.all_linesN)rrrrrrrr&rrrrrrrrr:rrHrHrHrIrs     rc @seZdZUdZeed<dZeeed<dZ ee ed<e e dZ e eed<dZee ed<d ed e fd d Zd ed eeeffd dZd ededed eeeffddZdS)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_liner5cCs|jdkot|jod|jdjv}||\}}|jr |jjnd}td||}|jrH|jjdurHt|jj jdkrH|jj j rH|j sH|j sHd}t |j|j||||d}|jrq|jdusm|jjsp|jjrf|rp|jdusm|rp||_n|jrv|ryd|_||_||_|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). rz NrO)r(rrrrr)r)r:r,r?_maybe_empty_linesrrmaxrPrrrXrlrr(rLrrSr)rFrrrrprevious_afterblockrHrHrImaybe_empty_lines$sV    z"EmptyLineTracker.maybe_empty_linescCsd}|jdkr|jjr dnd}|jr&|jd}|jd}t||}d|_nd}t|}|j}d}|jrL|jdj|krL|j }|jrL|jdj|ks<|j sR|j rX|j ||j dur_dS|jrr|j j rhdS|j jrr|j j rrdS|dur|j dus}J|jjr|j r|jsd}n8|r|j s|j j r|rdnd}n(|rd}n#d}n |rd}n|s|jr|jdjtjkr|jdjd vrd}nd}|js|j s|j r||||S|j jr|j jdkr|jdkr|jstj|jvrd S|j jr |js |jtd s ||j jkr |pddfS|dfS) NrOrrqrrr~rrrrO)withtryforwhileifmatch)rOrr)r)r(is_pyir,r?r{minr:rrrlrXrErrrrfr8r"r=r;rS#_maybe_empty_lines_for_class_or_defrralways_one_newline_after_importr)rFr max_allowedrjruser_had_newliner) previous_defrHrHrIrVs               z#EmptyLineTracker._maybe_empty_linesrr cCs|jdusJ|jjr|jjr|jrdSdS|jj|jkr5|jjs%|jjr5|jjr+dS|r1ddfSddfSd}|jjrf|jj|jkrf|dkrf|j }|durd|j durd|j j jsd|j j j sd|j dkrd|}ndS|jjr|jsq|jjr|jj|jkr{d}nI|jj|jkrd}n?|jr|jjrd}n5d}n2|jj|jkrd}n(|js|jr|jjs|jrtd|}nd}nd}n|jrdnd}|jjr|sd}|dur|j }|durt|j ||j|_ d}|dfS)NrrrOrrq)rrSr(rrfr)rXrlrLrrrrrr rorr)rFrrr comment_to_add_newlinesslcnewlinesrrHrHrIr sv    z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)rrrrrrrrr&rrrrrrrrrrr:r rHrHrHrIrs$ 2] rsequencer5ccs2t|d}t|D] }||fV|d8}q dS)z;Like `reversed(enumerate(sequence))` if that were possible.rON)rPr)rrelementrHrHrIrs     rFnew_lineold_liner,r3cCsR|D]$}t|j|j}t|||j||d||D] }|j|ddqqdS)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). rJTN)r#r8r;rrEr)rrr,r3old_leafnew_leaf comment_leafrHrHrI append_leaves s rr)line_strliner(rcCsV|st|}tj|vrt||jkod|vo| S|r!dSd|vr,t||jkS|d^}}}t||jksCt||jkrEdSg}d}g}tj} t |j D]\} } | tjkrd} | j dt |krm| dn | j dt |krz|} | dur|dur|j | j dkr| j } | dkrdS| j | kr| jtjkrdg} | |7} |js| j dkr| t |j dks| j| vr|| j d7<| tjkrt| | j } t| r| jr| jjtjks| jjr| jjjtjkrdSt |dkrdS| }| }t||vr| ||jdurn |j}t||vsqSt |dkr"dStdd|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. rFNrOrTcss|]}|dkVqdSrrH)r`valrHrHrIrrz'is_line_short_enough..)line_to_stringrmultiline_string_handlingr! line_lengthrsplitmathinfrr, bracket_depthrPrErr8r"rr/rr rrrtest dictsetmakerrall)rr(rrralastcommasmultiline_stringmultiline_string_contextsmax_level_to_updateir2 had_comma ignore_ctxsctxrHrHrIis_line_short_enough$sz         r0cCs|j}t|dkr dS|djtjkrr|djtjkrrd}d}|d}|dddD]G}|jtvr>|jtvr9dS|d7}n(|jtjkrI|d7}n|jtjkr^|jtjks]|jtvs]dSn|jtvrfdS|dkrq|dkrqdSq*dS) 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). rqFrrOr~r6NT) r,rPr8r"rvr^rrrW)rr, call_count dot_countrr2rHrHrI can_be_splits0         r3rhsrc Cs|j}d}t|jD]'}|r||jurd}|jtkr|sdS|s1|jtvr1|j|jvr1|jr1|}q |j}|j s:dS| }| |}|dkrIdS|dkrW|t krW|j jrWdS|tkr]dSt|jdkshJd|jd}|jd} |jtvr| jtvrt|||drdS|jd } |jd } | jtjks| jtjks| jtjkr| jr| jjtjkr| jtvrdSt|rdSt|| |d 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. NFTrOrqzStranded delimiterr)rrr6r~)r'r)rrr,rr8rrr;r. delimitersmax_delimiter_prioritydelimiter_count_with_priorityr rrr rPr_can_omit_opening_parenr"rurrrrrr_can_omit_closing_paren) r4rrrr2bt max_prioritydelimiter_countrsecond penultimater'rHrHrIcan_omit_invisible_parenss`            r?rcCsd}d|j}d}|D]%\}}}|jtvr|j|urd}|r2||7}||kr+dS|jtvr2d}q t|j|dkr>dSdS) See `can_omit_invisible_parens`.Frpr~TrO)r)rr8rrrrPr,)rrr remainderr_indexr2 leaf_lengthrHrHrIr8 s"  r8r'cCsZd|j}d}|D]\}}}||7}||jur#|s||kr"dSq |jtvr*d}q dS)r@rpFT)r)rrr8r)rr'rrseen_other_bracketsrBr2rCrHrHrIr9#s    r9cCst|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr<)rrHrHrIr4srr)Crr!collections.abcrrr dataclassesrrtypingrrr r black.bracketsr r r black.moderr black.nodesrrrrrrrrrrrrrrrrr black.stringsr!blib2to3.pgen2r"blib2to3.pytreer#r$r%rrrLNr&rrrrrrr:rrr0r3r?r8r9rrHrHrHrIszL   N "p   a$ a