o #iJ @s*ddlZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z m Z mZmZddlmZmZmZddlmZmZddlmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)ddl*m+Z+ddl,m-Z-dd l.m/Z/m0Z0e d Z1e2Z3e2Z4ee/e0fZ5eGd d d Z6eGd ddZ7eGdddZ8eGdddZ9de e1dee e3e1ffddZ: d3de6de6de e/de;ddf ddZd e6de;fd%d&Z?d'e7d(e2de;fd)d*Z@d e6d+e/d(e2de;fd,d-ZAd e6d.e/d(e2de;fd/d0ZBd e6de=fd1d2ZCdS)4N) dataclassfield) CallableDictIteratorListOptionalSequenceTupleTypeVarUnioncast)COMMA_PRIORITY DOT_PRIORITYBracketTracker)ModePreview)BRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENTTEST_DESCENDANTS child_towards is_docstring is_funcdef is_importis_multiline_stringis_one_sequence_betweenis_type_commentis_type_ignore_commentis_with_or_async_with_stmtmake_simple_prefix replace_childsyms whitespace) str_width)token)LeafNodeTc @seZdZUdZeddZeed<dZe ed<ee dZ e e ed<eedZeee e fed <eedZeed <dZeed <dZeed <d Zee ed< dSde dededd fddZdTde dedd fddZedefddZedefddZedefddZedefddZedefdd Z edefd!d"Z!edefd#d$Z"edefd%d&Z#edefd'd(Z$edefd)d*Z%edefd+d,Z&edefd-d.Z'd d/d0ee(e gefdefd1d2Z)defd3d4Z*defd5d6Z+defd7d8Z,defd9d:Z-defd;d<Z. dTd=e d>edefd?d@Z/dAe defdBdCZ0de de e fdDdEZ1dUdFdGZ2de defdHdIZ3 dTdJede4e5e6e e ffdKdLZ7dVdMdNZ8de9fdOdPZ:defdQdRZ;d S)WLinez;Holds leaves and comments. Can be printed with `str(line)`.F)reprmoderdepthdefault_factoryleavescommentsbracket_trackerinside_bracketsshould_split_rhsNmagic_trailing_commaleaf preformatted track_bracketreturncCs|jtvp t|j}|sdStj|jkr|jr|jdd=|jr4|s4|j t || ||j d7_ |j s;|r;|rY|j||j jrN||rM||_n |j|ddrY|||sf|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,T)ensure_removable)typerboolvaluestripr&COLONis_class_paren_emptyr0prefixr$is_complex_subscriptr,r3r2markr5has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr6r7r8 has_valuerLR/var/www/GraceOrthoHospital/venv_grace/lib/python3.10/site-packages/black/lines.pyrI@s,       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 liner7N) r2r-any_open_for_or_lambda is_comment ValueErrorr0r=rrI)rJr6r7rLrLrM append_safeds zLine.append_safecCst|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr0r=rrJrLrLrMrPxszLine.is_commentcCst|o |jdjtjkS)zIs this line a decorator?r)r>r0r=r&ATrUrLrLrM is_decorator}szLine.is_decoratorcCt|o t|jdS)zIs this an import line?r)r>rr0rUrLrLrMrzLine.is_importcCrX)zIs this a with_stmt line?r)r>r r0rUrLrLrMr rYzLine.is_with_or_async_with_stmtcCs*t|o|jdjtjko|jdjdkS)z Is this line a class definition?rclass)r>r0r=r&NAMEr?rUrLrLrMis_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_rLrLrM  z&Line.is_stub_class..)r\r0rangerUrLrLrM 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.)rFrSNdef)r0 IndexErrorr=r&r[r?ASYNC)rJ first_leaf second_leafrLrLrMis_defs$    z Line.is_defcCs4|jo|jddttjdgddtdDkS)zHIs this line a function definition with a body consisting only of "..."?N:cSr^r_rarcrLrLrMrfrgz$Line.is_stub_def..rh)rpr0r'r&rArirUrLrLrM 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. (rh)) r>rTr0r\r=r&LPARr?RPARrUrLrLrMrBs zLine.is_class_paren_emptycCsR|r |jdjtjkr dS|jdj}|drdStj|jvr'|dr'dSdS)z#Is the line a triple quoted string?rF)z"""z'''T)zr'''zr"""zR'''zR""") r0r=r&STRINGr? startswithraccept_raw_docstringsr,)rJr?rLrLrMis_triple_quoted_strings  zLine.is_triple_quoted_stringcCsdd|jDtjdkS)z Is the line a chained assignmentcSsg|]}|jqSrLr=rdr6rLrLrMrfsz.Line.is_chained_assignment..rS)r0countr&EQUALrUrLrLrMis_chained_assignmentszLine.is_chained_assignmentcCs$t|jdkr dS|jdjtjkS)z/Does this line open a new level of indentation.rF)rTr0r=r&rArUrLrLrM 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. rSFrN)rTr0r=rfmt_pass_converted_first_leaf)rJrr6rLrLrMis_fmt_pass_converteds    zLine.is_fmt_pass_convertedcCs |jD] }|jtkr dSqdS)z)If so, needs to be split before emitting.TF)r0r=rrJr6rLrLrMcontains_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|jSNr~)r6rLrLrMszGLine.contains_implicit_multiline_string_with_comments..rSTF) itertoolsgroupbyr0r&rzlistrTcomments_after)rJ leaf_typeleaf_group_iterator leaf_listr6rLrLrM0contains_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)Nrr:FT)setr0addidr=r&COMMAryr?rlr1itemsrr)rJ ignored_ids last_leaf comment_seenleaf_idr1commentrLrLrM$contains_uncollapsable_type_comments s0     z)Line.contains_uncollapsable_type_commentscCs|jsdStdd|jDd}tddt|jDd}||kr?|jddD]}|jt|gD] }t|r=dSq3q(dS)NFcs |] }|jdkr|jVqdSrNlinenorrLrLrM @z9Line.contains_unsplittable_type_ignore..rcsrrrrrLrLrMrBrr:T)r0nextreversedr1getrr)rJ first_line last_linenoderrLrLrM!contains_unsplittable_type_ignore/sz&Line.contains_unsplittable_type_ignorecCstdd|jDS)Ncss|]}t|VqdSr)rrrLrLrMrQz2Line.contains_multiline_strings..)anyr0rUrLrLrMcontains_multiline_stringsPszLine.contains_multiline_stringsclosingr<cCs|jtvr|jr|jdjtjksdS|jtjkrdS|jtjkrD|jdurB|jjtj krB|j durBt |j ||jtj tjfdrBdSdS|j rIdS|j durXt |j ||jsXdSdS)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) rFTN)brackets)r=rr0r&rRBRACERSQBparentr#traileropening_bracketrLSQBr)rJrr<rLrLrMrFSs6        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.Frrhrur:T)r=rr2any_open_bracketsrCr&COMMENTr0ryr?rrTrrr1 setdefaultrrI)rJrrrLrLrMrHs6     zLine.append_commentcCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)r1rrrrLrLrMrszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.rN)r0popr1rrextend)rJtrailing_commatrailing_comma_commentsrLrLrMrGs 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)r=r)rdnrLrLrMrs  z,Line.is_complex_subscript..) r2 get_open_lsqb next_sibling isinstancer(r=r# listmaker subscriptlistrr pre_order)rJr6 open_lsqbsubscript_startrLrLrMrDs     zLine.is_complex_subscriptrccsttttgttttff|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 rr r'rr Indexenumerate_reversed enumerater0rTrCr?r)rJropindexr6lengthrrLrLrMenumerate_with_lengths  zLine.enumerate_with_lengthcCst|j|j|j|j|jdS)N)r,r-r3r4r5)r*r,r-r3r4r5rUrLrLrMclonesz Line.clonecCsz|sdSd|j}t|j}t|}|j||j}|D]}|t|7}qtj |j D]}|t|7}q0|dS)zRender the line.rz ) r-iterr0rrCr?strrchain from_iterabler1values)rJindentr0firstresr6rrLrLrM__str__s  z Line.__str__cCst|jp|jS)z/Return True if the line has leaves or comments.)r>r0r1rUrLrLrM__bool__sz Line.__bool__)FFF)r9N)r9r*)<__name__ __module__ __qualname____doc__rr,r__annotations__r-intrr0rr'dictr1rLeafIDrr2r3r>r4r5rrIrRpropertyrPrWrr r\rjrprsrBr}rrrrrrrrrrFrHrrGrDrr rrrrrrrLrLrLrMr*2s     $  $! /%   r*c@s:eZdZUdZeed<eed<eed<eed<eed<dS) RHSResultz2Intermediate split result from a right hand split.headbodytailrclosing_bracketN)rrrrr*rr'rLrLrLrMrs  rc@szeZdZUdZeed<eded<eed<dZe ed<e e dZ e eed<dZe ed <d Zeed <d e efd 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_feedr9cCs8tt|jd}t|j|j|}|g|j||jgS)N)r,)rr*r,r!rrrr)rJ empty_linerCrLrLrM all_linesszLinesBlock.all_linesN)rrrrrrrr*rrrrrrrrrr>rrLrLrLrMrs     rc @seZdZUdZeed<dZeeed<dZ ee ed<e e dZ eeed<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_liner9cCs"tj|jvo|jdkot|jod|jdjv}||\}}|jr&|jj nd}|j dur/dn||}tj |jvrX|jrX|jjdurXt |jj jdkrX|jj jrX|jsX|jsXd}t|j|j||||d}|jr|j dus}|j js|j jrv|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 NrS)r,rrrrr)rallow_form_feedsr,r-r>r0rC_maybe_empty_linesrrrmodule_docstring_newlinesrTrr}r\rprrPrWr)rJrrrrprevious_afterblockrLrLrMmaybe_empty_lines)sb      z"EmptyLineTracker.maybe_empty_linesc CsFd}|jdkr|jjr dnd}|jr&|jd}|jd}t||}d|_nd}t|}|j}d}|jrL|jdj|krL|j }|jrL|jdj|ks<|dur|j dusWJ|jjr|rk|j sk|j j rk|rhdnd}n7t j |jvrz|jrz|jszd}n(|rd}n#d}n |rd}n|s|jr|jdjtjkr|jdjdvrd}nd}|js|j s|jr||||S|j r|j jr|js|jtds||j jkr|pddfS|j r|j jr|jrt j|jvrd S|dfSt j|jvot|jd p|j o|j jdo|j jdjot|j jdj }|j r|j jr|sd S|dfS) NrSrrurrr)withtryforwhileifmatchrrrSrr)r-r,is_pyir0rCrminr>rrrrpr"blank_line_after_nested_stub_classr\rjr=r&rAr?rW#_maybe_empty_lines_for_class_or_defrrr}$no_blank_line_before_class_docstringallow_empty_first_line_in_blockrrrr) rJr max_allowedrnruser_had_newliner- previous_defis_empty_first_line_okrLrLrMrbs              z#EmptyLineTracker._maybe_empty_linesrrcCs|js |j||jdurdS|jjr|jjr|jrdSdS|jj|jkr0|jjs.|jj r0dSd}|jj ra|jj|jkra|dkra|j }|dur_|j dur_|j j js_|j j js_|jdkr_|}ndS|jjr|jsl|jjr|jj|jkrvd}n_|jj|jkrd}nU|jr|jjrd}nKd}nHtj|jvr|jj|jkrd}n8|j s|jr|jj s|jrtd|}n%d}n"|jj|jkrd}nd}n|jrdnd}tj|jvr|jjr|sd}|dur|j }|durt|j||j|_d}|dfS)NrrrrSru)rWrrIrr,rrjr-r\rprPrrrrrr+blank_line_between_nested_and_def_stub_filerdummy_implementationsrsmaxr)rJrrrcomment_to_add_newlinesslcnewlinesrrLrLrMr s       z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)rrrrrrrrr*rrrrrrrrr rrr>r rLrLrLrMrs$ 9i rsequencer9ccs2t|d}t|D] }||fV|d8}q dS)z;Like `reversed(enumerate(sequence))` if that were possible.rSN)rTr)rrelementrLrLrMr+s     rFnew_lineold_liner0r7cCsR|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). rNTN)r'r=r?r"rIr)rrr0r7old_leafnew_leaf comment_leafrLrLrM append_leaves3s rr)line_strliner,rcCs|st|}tj|vr tnt}tj|vr$|||jko#d|vo#| S|r*dSd|vr5|||jkS|d^}}}|||jksL|||jkrNdSg}d}g} t j } t |j D]\} } | t j krd} | j dt|krv|dn | j dt|kr|} | dur|dur|j | j dkr| j } | dkrdS| j | kr| jtjkrdg}|| 7}| j|vr| t|j dks|| j d7<| t j krt| | j } t| rt| dkrdS| }| }t||vr| ||jdurn |j}t||vsq\t| dkrdStdd|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. rFNrSrTcss|]}|dkVqdSrrL)rdvalrLrLrMrrz'is_line_short_enough..)line_to_stringrrespect_east_asian_widthr%rTmultiline_string_handling line_lengthrsplitmathinfrr0 bracket_depthrIrr=r&r prev_siblingrrrrall)r r,rwidthrrelastcommasmultiline_stringmultiline_string_contextsmax_level_to_updateir6 had_comma ignore_ctxsctxrLrLrMis_line_short_enoughKsn          r6cCs|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). ruFrrSrr:NT) r0rTr=r&rzrbrrr[)r r0 call_count dot_countrr6rLrLrM can_be_splits0         r9rhsr%c Cs|j}d}t|jD]'}|r||jurd}|jtkr|sdS|s1|jtvr1|j|jvr1|jr1|}q |j}|j s:dS| }| |}|dkrIdS|dkr]t j |jvr]|tkr]|jjr]dS|tkrcdSt|jdksnJd|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. NFTrSruzStranded delimiterr)rr%r:r)r-r%)rrr0rr=rrr?r2 delimitersmax_delimiter_prioritydelimiter_count_with_priorityr(wrap_multiple_context_managers_in_parensr,rrr rrTr_can_omit_opening_parenr&ryrrrr#rr_can_omit_closing_paren) r:r%r rr6bt max_prioritydelimiter_countrsecond penultimater-rLrLrMcan_omit_invisible_parenssf             rFrcCsd}d|j}d}|D]%\}}}|jtvr|j|urd}|r2||7}||kr+dS|jtvr2d}q t|j|dkr>dSdS) See `can_omit_invisible_parens`.FrtrTrS)r-rr=rrrrTr0)r rr% remainderr_indexr6 leaf_lengthrLrLrMr?/s"  r?r-cCsZd|j}d}|D]\}}}||7}||jur#|s||kr"dSq |jtvr*d}q dS)rGrtFT)r-rrr=r)r r-r%rseen_other_bracketsrIr6rJrLrLrMr@Hs    r@cCst|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr@)r rLrLrMr"Ysr"r)Drr' dataclassesrrtypingrrrrrr r r r r black.bracketsrrr black.moderr black.nodesrrrrrrrrrrrrrr r!r"r#r$ black.stringsr%blib2to3.pgen2r&blib2to3.pytreer'r(r)rrrLNr*rrrrr>rrr6r9rFr?r@r"rLrLrLrMsz0 P   H "   [$ e