o ^iΟ @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|jtjtjfvpt|j}|sdS|jtjkr(|j r(|j dd=|j r=|s=|j t || ||jd7_ |jsD|rD|r`|j||jjrW||rV||_n ||r`|||sm|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_MIDDLETSTRING_MIDDLEboolvaluestripCOLONis_class_paren_emptyr,prefixr is_complex_subscriptr(r/r.markr1has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr2r3r4 has_valuerIP/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/black/lines.pyrF5s4         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,r8rrF)rGr2r3rIrIrJ append_safe^s zLine.append_safecCst|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr,r8rrGrIrIrJrMrszLine.is_commentcCst|o |jdjtjkS)zIs this line a decorator?r)r;r,r8r"ATrRrIrIrJ is_decoratorwszLine.is_decoratorcCt|o t|jdS)zIs this an import line?r)r;rr,rRrIrIrJr|zLine.is_importcCrU)zIs this a with_stmt line?r)r;rr,rRrIrIrJrrVzLine.is_with_or_async_with_stmtcCs*t|o|jdjtjko|jdjdkS)z Is this line a class definition?rclass)r;r,r8r"NAMEr<rRrIrIrJis_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_rIrIrJ  z&Line.is_stub_class..)rYr,rangerRrIrIrJ 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.)rFrPNdef)r, IndexErrorr8r"rXr<ASYNC)rG first_leaf second_leafrIrIrJis_defs$    z Line.is_defcCs4|jo|jddttjdgddtdDkS)zHIs this line a function definition with a body consisting only of "..."?N:cSr[r\r^r`rIrIrJrcrdz$Line.is_stub_def..re)rmr,r#r"r>rfrRrIrIrJ 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. (re)) r;rQr,rYr8r"LPARr<RPARrRrIrIrJr?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)rGr<rIrIrJ_is_triple_quoted_strings   zLine._is_triple_quoted_stringcCrU)zIs the line a docstring?r)r;rr,rRrIrIrJrrVzLine.is_docstringcCsdd|jDtjdkS)z Is the line a chained assignmentcSsg|]}|jqSrIr8rar2rIrIrJrcsz.Line.is_chained_assignment..rP)r,countr"EQUALrRrIrIrJis_chained_assignmentszLine.is_chained_assignmentcCs$t|jdkr dS|jdjtjkS)z/Does this line open a new level of indentation.rF)rQr,r8r"r>rRrIrIrJ 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. rPFrN)rQr,r8rfmt_pass_converted_first_leaf)rGrr2rIrIrJis_fmt_pass_converteds  zLine.is_fmt_pass_convertedcCs |jD] }|jtkr dSqdS)z)If so, needs to be split before emitting.TF)r,r8rrGr2rIrIrJcontains_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|jSNrz)r2rIrIrJszGLine.contains_implicit_multiline_string_with_comments..rPTF) itertoolsgroupbyr,r"rwlistrQcomments_after)rG leaf_typeleaf_group_iterator leaf_listr2rIrIrJ0contains_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 ||j dr^|sZt||j ds^||vr^dSd}qDq>dS)Nrr6Fr(T)setr,addidr8r"COMMArvr<rir-itemsrr(r)rG ignored_ids last_leaf comment_seenleaf_idr-commentrIrIrJ$contains_uncollapsable_type_commentss8       z)Line.contains_uncollapsable_type_commentscCs|jsdStdd|jDd}tddt|jDd}||krB|jddD]}|jt|gD] }t||jdr@dSq3q(dS) NFcs |] }|jdkr|jVqdSrNlinenor{rIrIrJ =z9Line.contains_unsplittable_type_ignore..rcsrrrr{rIrIrJr?rr6rT)r,nextreversedr-getrrr()rG first_line last_linenoderrIrIrJ!contains_unsplittable_type_ignore,sz&Line.contains_unsplittable_type_ignorecCstdd|jDS)Ncs|]}t|VqdSr)rr{rIrIrJrNz2Line.contains_multiline_strings..)anyr,rRrIrIrJcontains_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 rFTN)brackets)r8rr,r"rRBRACERSQBparentrtraileropening_bracketrLSQB prev_sibling subscriptlistr)rGrrIrIrJrCPsD       zLine.has_magic_trailing_commarcCs|jtkr|jrd|_dS|jtjkrdS|js"t|_d|_dS|jd}|jtjkrY|j sY|j rYt t |j dkrYt ||jdsYt |jdkrTt|_d|_dS|jd}|jt|g|dS) z0Add an inline or standalone comment to the line.Frrerrrr6T)r8rr.any_open_bracketsr@r"COMMENTr,rvr<rrQrrr(r- setdefaultrrF)rGrrrIrIrJrE|s<      zLine.append_commentcCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)r-rrrrIrIrJrszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.rN)r,popr-rrextend)rGtrailing_commatrailing_comma_commentsrIrIrJrDs 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)ranrIrIrJrs  z,Line.is_complex_subscript..) r. get_open_lsqb next_sibling isinstancer$r8r listmakerrrr pre_order)rGr2 open_lsqbsubscript_startrIrIrJrAs     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,rQr@r<r)rGropindexr2lengthrrIrIrJenumerate_with_lengths  zLine.enumerate_with_lengthcCst|j|j|j|j|jdS)N)r(r)r/r0r1)r&r(r)r/r0r1rRrIrIrJclonesz Line.clonecCs|sdSd|j}t|j}t|}|j||j}|ddd|D7}tj |j }dd|D}|d|7}|dS)zRender the line.rz rcsrrstrr{rIrIrJrrzLine.__str__..cSsg|]}t|qSrIr)rarrIrIrJrcsz Line.__str__..) r)iterr,rr@r<joinrchain from_iterabler-values)rGindentr,firstres comments_iterr-rIrIrJ__str__s  z Line.__str__cCst|jp|jS)z/Return True if the line has leaves or comments.)r;r,r-rRrIrIrJ__bool__sz Line.__bool__)FFF)r5N)r5r&);__name__ __module__ __qualname____doc__rr(r__annotations__r)intrr,r#dictr-LeafIDr r.r/r;r0r1rrFrOpropertyrMrTrrrYrgrmrpr?ryrr~rrrrrrrrrCrErrDrArrrrrrrrrIrIrIrJr&'s     )  $!,%   r&c@s:eZdZUdZeed<eed<eed<eed<eed<dS) RHSResultz2Intermediate split result from a right hand split.headbodytailrclosing_bracketN)rrrrr&rr#rIrIrIrJrs  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)Nr)rr&r(rrrrr)rG empty_liner@rIrIrJ all_linesszLinesBlock.all_linesN)rrrrrrrr&rrrrrrrrr;rrIrIrIrJrs     rc @seZdZUdZeed<dZeeed<dZ ee ed<e e dZ e eed<dZee ed<d ed e fd d Zd ed efd dZd ed eeeffddZd 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||}tj |j vr7| |r6d}n|jrV|jjdurVt |jj jdkrV|jj jrV|jsV|jsVd}t|j |j||||d}|jr|jdus{|jjs~|jjrt|r~|jdus{|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). rz rPN)r(rrrrr)r)r;r,r@_maybe_empty_linesrrmaxrfix_module_docstring_detectionr(_line_is_module_docstringrQrrrYrmrrMrrTr)rGrrrrprevious_afterblockrIrIrJmaybe_empty_lines$sh      z"EmptyLineTracker.maybe_empty_linescCsZ|j}|sdSt|jjdks|jjr|js|jrdS|j}r+|jjs&dS|j}s dS)NFrPT)rrQrr,rrYrmrM)rGrrrIrIrJr[s"  z*EmptyLineTracker._line_is_module_docstringcCsd}|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) NrPrrrrrrrrrrP)withtryforwhileifmatch)rPrr)r)r(is_pyir,r@r|minr;rrrmrYrFrrrrgr8r"r>r<rT#_maybe_empty_lines_for_class_or_defrralways_one_newline_after_importr)rGr max_allowedrkruser_had_newliner) previous_defrIrIrJrks              z#EmptyLineTracker._maybe_empty_linesrrcCs|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)NrrrPrrr)rrTr(r rgr)rYrmrMrrrrrr rprr)rGrrrcomment_to_add_newlinesslcnewlinesrrIrIrJrs   z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)rrrrrrrrr&rrrrrrrr;rrrrrrIrIrIrJrs& 7] rsequencer5ccs2t|d}t|D] }||fV|d8}q dS)z;Like `reversed(enumerate(sequence))` if that were possible.rPN)rQr)rrelementrIrIrJrs     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). rKTN)r#r8r<rrFr)rrr,r3old_leafnew_leaf comment_leafrIrIrJ 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. rFNrPrTcss|]}|dkVqdSrrI)ravalrIrIrJrrz'is_line_short_enough..)line_to_stringrmultiline_string_handlingr! line_lengthrsplitmathinfrr, bracket_depthrQrFrr8r"rr/rr rrrtest dictsetmakerrall)rr(rrrblastcommasmultiline_stringmultiline_string_contextsmax_level_to_updateir2 had_comma ignore_ctxsctxrIrIrJis_line_short_enough9s          r4cCs|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). rrFrrPrr6NT) r,rQr8r"rwr_rrrX)rr, call_count dot_countrr2rIrIrJ can_be_splits0         r7rhsr#c 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. NFTrPrrzStranded delimiterr)rr#r6r)r+r#)rrr,rr8rrr<r. delimitersmax_delimiter_prioritydelimiter_count_with_priorityr rrr rQr_can_omit_opening_parenr"rvrrrrrr_can_omit_closing_paren) r8r#rrr2bt max_prioritydelimiter_countrsecond penultimater+rIrIrJcan_omit_invisible_parenssl            rCrcCsd}d|j}d}|D]%\}}}|jtvr|j|urd}|r2||7}||kr+dS|jtvr2d}q t|j|dkr>dSdS) See `can_omit_invisible_parens`.FrqrTrP)r)rr8rrrrQr,)rrr# remainderr_indexr2 leaf_lengthrIrIrJr<s"  r<r+cCsZd|j}d}|D]\}}}||7}||jur#|s||kr"dSq |jtvr*d}q dS)rDrqFT)r)rrr8r)rr+r#rseen_other_bracketsrFr2rGrIrIrJr=8s    r=cCst|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr=)rrIrIrJr!Isr!r)Crr%collections.abcrrr dataclassesrrtypingrrr r black.bracketsr r r black.moderr black.nodesrrrrrrrrrrrrrrrrr black.stringsr!blib2to3.pgen2r"blib2to3.pytreer#r$r%rrrLNr&rrrrrrr;rrr4r7rCr<r=r!rIrIrIrJs|L   N "   a$ a