o be @sddlmZmZddlZddlZddlmZmZmZm Z m Z m Z m Z m Z mZmZddlmZmZddlmZddlmZmZddlmZddlmZmZdd lmZmZmZdd lm Z m!Z!m"Z"m#Z#dd lm$Z$m%Z%m&Z&m'Z'dd lm(Z(ed Z)e*Z+e*Z,eGdddZ-eGdddZ.de e)de e e+e)ffddZ/ d2de-de-de ede0ddf ddZ1ddde-d e*d!e2de0fd"d#Z3de-de0fd$d%Z4 &d3de-d e*d'ee,de0fd(d)Z5de-d*ed e*de0fd+d,Z6de-d-ed e*de0fd.d/Z7de-de2fd0d1Z8dS)4) dataclassfieldN) Callable CollectionDictIteratorListOptionalSequenceTupleTypeVarcast)NodeLeaf)token)BracketTracker DOT_PRIORITY)Mode)STANDALONE_COMMENTTEST_DESCENDANTS)BRACKETSOPENING_BRACKETSCLOSING_BRACKETS)syms whitespace replace_child child_towards)is_multiline_string is_importis_type_commentlast_two_except)is_one_tuple_betweenTc @s>eZdZUdZeed<dZeed<ee dZ e e ed<ee dZeee e fed<eedZeed<d Zeed <d Zeed <d Zee ed <dDde dedd fddZdDde dedd fddZedefddZedefddZedefddZedefddZedefddZedefdd Z edefd!d"Z!edefd#d$Z"e#j$fd%edefd&d'Z%defd(d)Z&defd*d+Z'defd,d-Z( dDd.e d/edefd0d1Z)d2e defd3d4Z*de de e fd5d6Z+dEd7d8Z,de defd9d:Z- dDd;ede.e/e0e effdd?Z2de3fd@dAZ4defdBdCZ5d S)GLinez;Holds leaves and comments. Can be printed with `str(line)`.moderdepthdefault_factoryleavescommentsbracket_trackerFinside_bracketsshould_split_rhsNmagic_trailing_commaleaf preformattedreturncCs|jtvp t|j}|sdStj|jkr|jr|jdd=|jr2|s2|j t || |d7_ |j s7|sU|j ||jjrJ||rI||_n |j|ddrU|||sb|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)Zcomplex_subscriptT)ensure_removable)typerboolvaluestriprCOLONis_class_paren_emptyr(prefixris_complex_subscriptr+r*markr$r-has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr.r/Z has_valuerAJ/var/www/secure340b-portal/env/lib/python3.10/site-packages/black/lines.pyr?0s(        z Line.appendcCsD|jjdkr|jr td|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 liner/N)r*r% is_comment ValueErrorr(r3rr?)r@r.r/rArArB append_safePs zLine.append_safecCst|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr(r3rr@rArArBrDaszLine.is_commentcCst|o |jdjtjkS)zIs this line a decorator?r)r4r(r3rATrIrArArB is_decoratorfszLine.is_decoratorcCst|o t|jdS)zIs this an import line?r)r4rr(rIrArArBrkszLine.is_importcCs*t|o|jdjtjko|jdjdkS)z Is this line a class definition?rclass)r4r(r3rNAMEr5rIrArArBis_classps z Line.is_classcCs&|jo|jddddtdDkS)zEIs this line a class definition with a body consisting only of "..."?NcSsg|]}ttjdqS).)rrDOT).0_rArArB |s z&Line.is_stub_class..)rNr(rangerIrArArB is_stub_classyszLine.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.)rFrGNdef)r( IndexErrorr3rrMr5ASYNC)r@ first_leafZ second_leafrArArBis_defs$    z Line.is_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. (rU)) r4rHr(rNr3rLPARr5RPARrIrArArBr8s zLine.is_class_paren_emptycCs,t|o|jdjtjko|jdjdS)z#Is the line a triple quoted string?r)z"""z''')r4r(r3rSTRINGr5 startswithrIrArArBis_triple_quoted_strings zLine.is_triple_quoted_string depth_limitcCs*|jD]}|jtkr|j|krdSqdS)z)If so, needs to be split before emitting.TF)r(r3rZ bracket_depth)r@rfr.rArArBcontains_standalone_commentss z!Line.contains_standalone_commentscCst}z)|jd}|t||jtjks|jtjkr+|js+|jd}|t|Wn t y6YdSwd}|j D]\}}|D]}t |rY|sUt |dsY||vrYdSd}qDq>dS)Nr1F ignoreT) setr(addidr3rCOMMArbr5rYr)itemsr)r@Z ignored_ids last_leafZ comment_seenZleaf_idr)commentrArArB$contains_uncollapsable_type_commentss8      z)Line.contains_uncollapsable_type_commentscCs|jsdStdd|jDd}tddt|jDd}||kr@|jddD]}|jt|gD] }t|dr>dSq3q(dS) NFcs |] }|jdkr|jVqdSrNlinenorRr.rArArB z9Line.contains_unsplittable_type_ignore..rcsrrrsrtrvrArArBrwrxr1riT)r(nextreversedr)getrlr)r@ first_line last_linenoderprArArB!contains_unsplittable_type_ignores z&Line.contains_unsplittable_type_ignorecCstdd|jDS)Ncss|]}t|VqdSN)rrvrArArBrwsz2Line.contains_multiline_strings..)anyr(rIrArArBcontains_multiline_stringsszLine.contains_multiline_stringsclosingr2cCs|jtvr|jr|jdjtjksdS|jtjkrdS|jtjkr6|s%dS|jd}t|jo4|jjt j kS|j r;dSt |j ||jsEdSdS)zReturn True if we have a magic trailing comma, that is when: - there's a trailing comma here - it's not a one-tuple Additionally, if ensure_removable: - it's not from square bracket indexing rhFT)r3rr(rrmRBRACERSQBr4parentr listmakerrr!opening_bracket)r@rr2commarArArBr<s&    zLine.has_magic_trailing_commarpcCs|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.FrhrUr^r1T)r3rr*Zany_open_bracketsr9rCOMMENTr(rbr5rrHlistrr) setdefaultrlr?)r@rprorArArBr>s<     zLine.append_commentcCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)r)r{rl)r@r.rArArBcomments_afterBszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.rhN)r(popr)rlrextend)r@Ztrailing_commaZtrailing_comma_commentsrArArBr=Fs 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)r3r)rRnrArArBrw\s  z,Line.is_complex_subscript..) r*Z get_open_lsqbZ next_sibling isinstancerr3rrZ subscriptlistrrZ pre_order)r@r.Z open_lsqbZsubscript_startrArArBr:Ns     zLine.is_complex_subscriptrzccsttttgttttff|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 rrr Indexenumerate_reversed enumerater(rHr9r5r)r@rzopindexr.lengthrprArArBenumerate_with_length`s  zLine.enumerate_with_lengthcCst|j|j|j|j|jdS)N)r$r%r+r,r-)r#r$r%r+r,r-rIrArArBcloneusz Line.clonecCsz|sdSd|j}t|j}t|}|j||j}|D]}|t|7}qtj |j D]}|t|7}q0|dS)zRender the line.rz ) r%iterr(ryr9r5str itertoolschain from_iterabler)values)r@indentr(firstresr.rprArArB__str__~s  z Line.__str__cCst|jp|jS)z/Return True if the line has leaves or comments.)r4r(r)rIrArArB__bool__sz Line.__bool__F)r0N)r0r#)6__name__ __module__ __qualname____doc__r__annotations__r%intrrr(rrdictr)rLeafIDrr*r+r4r,r-r r?rFpropertyrDrKrrNrWr\r8resysmaxsizergrqrrr<r>rr=r:rr rrrrrrrArArArBr#"sj     $! !%   r#c@seZdZUdZdZeed<dZee ed<dZ e ed<e e dZee ed <d e d ee e ffd d Zd e d ee e ffddZd e de d ee e ffddZdS)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. Fis_pyiN previous_linerprevious_afterr& previous_defs current_liner0cCs:||\}}|jdurdn||j}||_||_||fS)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). Nr)_maybe_empty_linesrr)r@rbeforeafterrArArBmaybe_empty_linessz"EmptyLineTracker.maybe_empty_linescCsd}|jdkr|jr dnd}|jr%|jd}|jd}t||}d|_nd}|j}|jrS|jd|krS|j|jrC|r@dnd}n|rGdnd}|jrS|jd|ks4|js\|j s\|j rb| ||S|j rx|j j rx|j sx||j jkrx|puddfS|j r|j j r|jr|dfS|dfS)NrGrr^rrrh)r%rr(r9countminrrrKr\rN#_maybe_empty_lines_for_class_or_defrrre)r@rZ max_allowedr[rr%rArArBrsF         z#EmptyLineTracker._maybe_empty_linesrcCs|js |j|j|jdurdS|jjr|jr|jrdSdS|jj|jkr0|jjs.|jjr0dS|jj rA|jj|jkrA|dkrAdS|jrr|jj|jkrNd}n&|jsU|jjrb|jr_|jjr_d}nd}n|jsh|jro|jjsod}nd}nd}|jr}|r}|d8}|dfS)N)rr)rrGrrGr^) rKrr?r%rrrWrNr\rD)r@rrnewlinesrArArBrsP    z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)rrrrrr4rrr r#rrrrrrr rrrrArArArBrs   ' rsequencer0ccs2t|d}t|D] }||fV|d8}q dS)z;Like `reversed(enumerate(sequence))` if that were possible.rGN)rHrz)rrelementrArArBrs     rFnew_lineold_liner(r/cCsR|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). rCTN)rr3r5rr?r)rrr(r/Zold_leafZnew_leafZ comment_leafrArArB append_leavess rr)line_strline line_lengthrcCs*|st|}t||kod|vo| S)zReturn True if `line` is no longer than `line_length`. Uses the provided `line_str` rendering, if any, otherwise computes a new one. r)line_to_stringrHrg)rrrrArArBis_line_short_enough/s rcCs|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). r^FrrGrhr1NT) r(rHr3rrcrQrrrM)rr(Z call_count dot_countryr.rArArB can_be_split=s0         rrAomit_on_explodec CsJ|j}|jsdS|}||dkrdS|tkrdSt|jdks&Jd|jd}|jd}|jtvrC|jt vrCt |||drCdS|jd}|jd }|j rfz t |j|d \}}Wn t yeYdSw|jtjks|jtjks|jtjkr|jr|jjtjkr|jtvrdSt|rdS|j r|jtjkrdSt|||d rdSdS) aDoes `line` 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. TrGFr^zStranded delimiterr)rrr1rh)Zomit)lastr)r* delimitersZmax_delimiter_priorityZdelimiter_count_with_priorityrrHr(r3rr_can_omit_opening_parenr-r LookupErrorrrbrrrrtrailerrrm_can_omit_closing_paren) rrrZbtZ max_priorityrsecondZ penultimaterrArArBcan_omit_invisible_parensasR           rrcCsd}d|j}d}|D]%\}}}|jtvr|j|urd}|r2||7}||kr+dS|jtvr2d}q t|j|dkr>dSdS) See `can_omit_invisible_parens`.Fr]rhTrG)r%rr3rrrrHr()rrr remainderr_indexr. leaf_lengthrArArBrs"  rrcCsZd|j}d}|D]\}}}||7}||jur#|s||kr"dSq |jtvr*d}q dS)rr]FT)r%rrr3r)rrrrZseen_other_bracketsrr.rrArArBrs    rcCst|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr6)rrArArBrsrr)rA)9 dataclassesrrrrtypingrrrrrr r r r r Zblib2to3.pytreerrZblib2to3.pgen2rZblack.bracketsrrZ black.moderZ black.nodesrrrrrrrrrrrrr r!r"rrrr#rrr4rrrrrrrrrArArArBsd0   r"{  ' N