a g g@sddlZddlZddlZddlmZmZddlmZmZm Z m Z m Z m Z m Z mZmZmZddlmZmZmZddlmZmZddlmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*dd l+m,Z,m-Z-ed Z.e/Z0e/Z1ee,e-fZ2eGd d d Z3eGd ddZ4eGdddZ5eGdddZ6e e.e e e0e.fdddZ7d-e3e3e e,e8ddddZ9dde3ee:e8dddZ;e3e8dd d!Ze3e,e/e8d(d)d*Z?e3e:dd+d,Z@dS).N) dataclassfield) CallableDictIteratorListOptionalSequenceTupleTypeVarUnioncast)COMMA_PRIORITY DOT_PRIORITYBracketTracker)ModePreview)BRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENTTEST_DESCENDANTS child_towards is_importis_multiline_stringis_one_sequence_betweenis_type_commentis_with_or_async_with_stmt replace_childsyms whitespace) str_width)token)LeafNodeTc@sveZdZUdZeed<dZeed<ee dZ e e ed<ee dZeee e fed<eedZeed<d Zeed <d Zeed <d Zee ed <dLe eed dddZdMe ed dddZeedddZeedddZeedddZeedddZeedddZeeddd Z eedd!d"Z!eedd#d$Z"eedd%d&Z#eedd'd(Z$d d)ee%e gefed*d+d,Z&e'j(feed-d.d/Z)edd0d1Z*edd2d3Z+edd4d5Z,dNe eed6d7d8Z-e ed9d:d;Z.e e e d<d=d>Z/d dd?d@Z0e ed<dAdBZ1dOee2e3e4e efdCdDdEZ5dddFdGZ6e7ddHdIZ8eddJdKZ9d S)PLinez;Holds leaves and comments. Can be printed with `str(line)`.moderdepthdefault_factoryleavescommentsbracket_trackerFinside_bracketsshould_split_rhsNmagic_trailing_comma)leaf preformatted track_bracketreturncCs|jtvpt|j}|s dStj|jkr>|jr>|jdd=|jrd|sd|j t || |d7_ |j sr|rr|r|j ||jjr||r||_n|j|ddr|||s|j|dS)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)typerboolvaluestripr"COLONis_class_paren_emptyr+prefixr is_complex_subscriptr.r-markr'r0has_magic_trailing_commaremove_trailing_commaappend_commentappend)selfr1r2r3Z has_valuerEN/var/www/html/llm_bihealth/app/venv/lib/python3.9/site-packages/black/lines.pyrC=s$       z Line.append)r1r2r4cCsD|jjdkr2|jrtd|jr2|jtkr2td|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 liner2N)r-r( is_comment ValueErrorr+r7rrC)rDr1r2rErErF append_safe_s zLine.append_safer4cCst|jdko|jdjtkS)z"Is this line a standalone comment?r)lenr+r7rrDrErErFrHpszLine.is_commentcCst|o|jdjtjkS)zIs this line a decorator?r)r8r+r7r"ATrNrErErF is_decoratoruszLine.is_decoratorcCst|ot|jdS)zIs this an import line?r)r8rr+rNrErErFrzszLine.is_importcCst|ot|jdS)zIs this a with_stmt line?r)r8rr+rNrErErFrszLine.is_with_or_async_with_stmtcCs*t|o(|jdjtjko(|jdjdkS)z Is this line a class definition?rclass)r8r+r7r"NAMEr9rNrErErFis_classs z Line.is_classcCs&|jo$|jddddtdDkS)zEIs this line a class definition with a body consisting only of "..."?NcSsg|]}ttjdqS).)r#r"DOT).0_rErErF sz&Line.is_stub_class..)rSr+rangerNrErErF is_stub_classszLine.is_stub_classcCsz|jd}Wnty"YdS0z|jd}WntyHd}Yn0|jtjkr`|jdkp|jtjko|duo|jtjko|jdkS)zBIs this a function definition? (Also returns True for async defs.)rFrLNdef)r+ IndexErrorr7r"rRr9ASYNC)rD first_leafZ second_leafrErErFis_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. (rZ)) r8rMr+rSr7r"LPARr9RPARrNrErErFr<s zLine.is_class_paren_emptycCs,t|o*|jdjtjko*|jdjdS)z#Is the line a triple quoted string?r)z"""z''')r8r+r7r"STRINGr9 startswithrNrErErFis_triple_quoted_strings zLine.is_triple_quoted_stringcCs$t|jdkrdS|jdjtjkS)z/Does this line open a new level of indentation.rF)rMr+r7r"r;rNrErErF opens_blockszLine.opens_blockfirst_leaf_matches)rnr4cCsFt|jdkrdS|jd}|jtks0|jdur4dS|dupD||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. rLFrN)rMr+r7rZfmt_pass_converted_first_leaf)rDrnr1rErErFis_fmt_pass_converteds  zLine.is_fmt_pass_converted) depth_limitr4cCs*|jD]}|jtkr|j|krdSqdS)z)If so, needs to be split before emitting.TF)r+r7r bracket_depth)rDrpr1rErErFcontains_standalone_commentss z!Line.contains_standalone_commentscCst}zR|jd}|t||jtjks>|jtjkrV|jsV|jd}|t|Wnt ylYdS0d}|j D]<\}}|D].}t |r|st |ds||vrdSd}qq|dS)Nrkr5F ignoreT) setr+addidr7r"COMMArgr9r^r,itemsr)rDZ ignored_ids last_leafZ comment_seenZleaf_idr,commentrErErF$contains_uncollapsable_type_commentss2     z)Line.contains_uncollapsable_type_commentscCs|js dStdd|jDd}tddt|jDd}||kr|jddD].}|jt|gD]}t|drfdSqfqPdS) NFcss|]}|jdkr|jVqdSrNlinenorWr1rErErF z9Line.contains_unsplittable_type_ignore..rcss|]}|jdkr|jVqdSr|r}rrErErFrrr5rsT)r+nextreversedr,getrvr)rD first_line last_linenoderzrErErF!contains_unsplittable_type_ignores  z&Line.contains_unsplittable_type_ignorecCstdd|jDS)Ncss|]}t|VqdSN)rrrErErFr)rz2Line.contains_multiline_strings..)anyr+rNrErErFcontains_multiline_strings(szLine.contains_multiline_strings)closingr6r4cCs|jtvr"|jr"|jdjtjks&dS|jtjkr6dS|jtjkr|jr||jjtj kr||j r|t |j ||jtj tjfdr|dS|sdS|jd}|jdurdS|jjtj kp|j dupt |j ||jtj tjfd S|jrdS|j durt |j ||jsdSdS)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) rkFT)ZbracketsN)r7rr+r"rwRBRACERSQBparentrtraileropening_bracketrLSQB subscriptlistr)rDrr6commarErErFr@+s\            zLine.has_magic_trailing_comma)rzr4cCs|jtkr|jrd|_dS|jtjkr.dS|jsDt|_d|_dS|jd}|jtjkr|j s|j rt t |j dkrt |st |jdkrt|_d|_dS|jd}|jt|g|dS)z0Add an inline or standalone comment to the line.FrkrZrcr5T)r7rr-Zany_open_bracketsr=r"COMMENTr+rgr9rrMlistrr, setdefaultrvrC)rDrzryrErErFrBis<     zLine.append_comment)r1r4cCs|jt|gS)z;Generate comments that should appear directly after `leaf`.)r,rrv)rDr1rErErFcomments_afterszLine.comments_aftercCs>|j}|jt|g}|jt|jdg|dS)z@Remove the trailing comma and moves the comments attached to it.rkN)r+popr,rvrextend)rDZtrailing_commaZtrailing_comma_commentsrErErFrAs  zLine.remove_trailing_commacCsj|j}|durdS|j}t|trL|jtjkr6dS|jtjkrLt ||}|duoht dd| DS)zAReturn True iff `leaf` is part of a slice with non-trivial exprs.NFcss|]}|jtvVqdSr)r7r)rWnrErErFrsz,Line.is_complex_subscript..) r-Z get_open_lsqbZ next_sibling isinstancer$r7rZ listmakerrrrZ pre_order)rDr1Z open_lsqbZsubscript_startrErErFr>s     zLine.is_complex_subscript)rr4ccsttttgttttff|r&tnt}||j D]V\}}t |j t |j }d|j vrbdS| |D]}|t |j 7}ql|||fVq6dS)zReturn an enumeration of leaves with their length. Stops prematurely on multiline strings and standalone comments.  N)r rr r#rr Indexenumerate_reversed enumerater+rMr=r9r)rDropindexr1lengthrzrErErFenumerate_with_lengths  zLine.enumerate_with_lengthcCst|j|j|j|j|jdS)N)r'r(r.r/r0)r&r'r(r.r/r0rNrErErFclonesz Line.clonecCsz|sdSd|j}t|j}t|}|j||j}|D]}|t|7}qrr rrrrrrrErErErFr&/sz     " $! >%  r&c@s:eZdZUdZeed<eed<eed<eed<eed<dS) RHSResultz2Intermediate split result from a right hand split.headbodytailrZclosing_bracketN)rrrrr&rr#rErErErFrs rc@sneZdZUdZeed<eded<eed<dZe ed<e e dZ e eed<dZe ed <e ed d d Zd 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_linerbeforer) content_linesafterrKcCs.tt|jd}||jg|j||jgS)N)r')rr&r'rrr)rDZ empty_linerErErF all_linesszLinesBlock.all_linesN)rrrrrrrr&rrrrrrrrrrErErErFrs    rc@seZdZUdZeed<dZeeed<dZ ee ed<e e dZ eeed<dZee ed<ee d d d Zeeeefd d d ZeeeeefdddZdS)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_liner4cCs||\}}|jr|jjnd}|jdur.dn||}t|j|j|||d}|jr|jdus~|jjs|jjrp|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). rN)r'rrrr) _maybe_empty_linesrrrrr'rHrPr)rDrrrZprevious_afterblockrErErFmaybe_empty_liness>   z"EmptyLineTracker.maybe_empty_linescCsd}|jdkr|jjrdnd}|jrL|jd}|jd}t||}d|_nd}|j}|jr8|jdj|kr8|jjr|jdusJ|r|j s|jj rtd|}n8t j |jvr|jdj r|jdj sd}n|rd}nd}nN|rd}nD|s(|jdjr(|jdjtjkr(|jdjdvr(d}nd}|jqV|jsP|j sP|j r\|||S|jr|jjr|js|jtds||jjkr|pddfS|jr|jj r|jr|dfS|jr|jjrd S|dfS) NrLrrcrrrk)withtryforwhileifmatchrmrr)r(r'is_pyir+r=countminrrrarZ"blank_line_after_nested_stub_classrSr\r7r"r;r9rrP#_maybe_empty_lines_for_class_or_defrrorjrl)rDrZ max_allowedr`rr(rErErFr8s~              z#EmptyLineTracker._maybe_empty_lines)rrr4cCs|js|j||jdur dS|jjr>|jjr:|jr:dSdS|jj|jkr`|jjs\|jj r`dSd}|jj r|jj|jkr|dkr|j }|dur|j dur|j j js|j j js|jdkr|}ndS|jjrt|js|jjr$|jj|jkrd}n2|jj|jkrd}n|jr|jjrd}nd}nN|j s4|jrX|jj sX|jrRtd|}nd}n|jj|jkrnd}nd}n|jrdnd}|dur|j }|durt|j||j|_d}|dfS)Nr)rrLrrLrc)rPrrCrr'rr\r(rSrarHrrrrlrrmaxr)rDrrZcomment_to_add_newlinesZslcnewlinesrrErErFrs~       z4EmptyLineTracker._maybe_empty_lines_for_class_or_def)rrrrrrrrr&rrrrrrrrr rrrrErErErFrs (L r)sequencer4ccs0t|d}t|D]}||fV|d8}qdS)z;Like `reversed(enumerate(sequence))` if that were possible.rLN)rMr)rrelementrErErFrs   rF)new_lineold_liner+r2r4cCsR|D]H}t|j|j}t|||j||d||D]}|j|ddq8qdS)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). rGTN)r#r7r9rrCr)rrr+r2Zold_leafZnew_leafZ comment_leafrErErF append_leavess  rr)line_str)liner'rr4cCs.|s t|}|jrtnt}tj|vrD|||jkoBd|voB| S|rPdSd|vrf|||jkS|d^}}}|||jks|||jkrdSg}d}g} t j } t |j D]R\} } | t j krDd} | j dt|kr|dn| j dt|kr|} | durD|durD|j | j dkrD| j } | dkrDdS| j | kr| jtjkrdg}|| 7}| j|vr| t|j dks|| j d7<| t j krt| | j } t| rt| dkrdS| }| }t||vr| ||jdurq|j}qqt| 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. rFNrLrTcss|]}|dkVqdSr|rE)rWvalrErErFrHrz'is_line_short_enough..)line_to_stringpreviewr!rMrZmultiline_string_handling line_lengthrrsplitmathinfrr+rqrCrr7r"rwZ prev_siblingrrrrall)rr'rwidthrrXlastZcommasZmultiline_stringZmultiline_string_contextsZmax_level_to_updateir1Z had_commaZ ignore_ctxsctxrErErFis_line_short_enoughsp           r)rr4cCs|j}t|dkrdS|djtjkr|djtjkrd}d}|d}|dddD]}|jtvr||jtvrrdS|d7}nP|jtjkr|d7}n:|jtjkr|jtjks|jtvsdSn|jtvrdS|dkrT|dkrTdSqTdS) 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). rcFrrLrkr5NT) r+rMr7r"rhrVrrrR)rr+Z call_count dot_countrr1rErErF can_be_splitKs,         r)rhsrr4c CsD|j}|j}|jsdS|}||}|dkr4dS|dkr\tj|jvr\|tkr\|j j r\dS|t krhdSt |j dks~Jd|j d}|j d}|jtvr|jtvrt|||drdS|j d}|j d } | jtjks| jtjks| jtjkr@| jr@| jjtjkr@|jtvrdSt|r,dSt|| |d r@dSdS) 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. TrLFrczStranded delimiterr)rrr5rk)rr)rr- delimitersZmax_delimiter_priorityZdelimiter_count_with_priorityrZ(wrap_multiple_context_managers_in_parensr'rrrrrMr+r7rr_can_omit_opening_parenr"rgrrrrrr_can_omit_closing_paren) rrrZbtZ max_priorityZdelimiter_countrsecondZ penultimaterrErErFcan_omit_invisible_parensosV              r)rrrr4cCs~d}d|j}d}|D]H\}}}|jtvr<|j|ur\}}}||7}||jurF|s>||krTdSq|jtvrd}qdS)rrbFT)r(rrr7r)rrrrZseen_other_bracketsrr1rrErErFrs    rcCst|dS)zmReturns the string representation of @line. WARNING: This is known to be computationally expensive. r)rr:)rrErErFrsr)F)ArrrZ dataclassesrrtypingrrrrrr r r r r Zblack.bracketsrrrZ black.moderrZ black.nodesrrrrrrrrrrrrrr Z black.stringsr!Zblib2to3.pgen2r"Zblib2to3.pytreer#r$r%rrrLNr&rrrrr8rrrrrrrrrErErErFsZ0 @   2 P    [% T