o ^i)@sndZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z m Z ddlmZmZmZddlmZmZmZmZmZdd lmZmZmZmZdd lmZmZm Z m!Z!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'm(Z(dd l)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDmEZEmFZFmGZGmHZHmIZImJZJmKZKmLZLmMZMmNZNdd lOmPZPddlQmRZRmSZSmTZTmUZUmVZVddlWmXZXmYZYmZZZm[Z[m\Z\m]Z]m^Z^ddl_m`Z`ddlambZbmcZcedZeeebecfZfGdddeXZgGddde1eZhdedee&de'deeifddZj dldede'dee&deefddZkdede'delfdd ZmGd!d"d"e Znded#ee&de'deefd$d%Zo  dmdede'dee&d&eeedeef d'd(Zp dlded&eeedefd)d*Zq  dmd+edede'dee&d&eeedeef d,d-Zrd.ed+ede'delfd/d0Zsd1ed2ed3eddfd4d5Ztd6euebd7ed8ebdelfd9d:Zvd6euebd7ed8ebd;endef de]de]fd?d@ZxdedeedfdAdBZydCebdee&delfdDdEZzdFeldGeddedefdHdIZ{eehZ|exdedee&de'deefdJdKZ}exdedee&de'deefdLdMZ~dNecdOeeide'dee&ddf dPdQZdRecdSefdTedddfdUdVZdNecde'dee&ddfdWdXZdNecde'dee&ddfdYdZZdNecdRecde'dee&ddf d[d\Z ]dndNefdRefde'dee&d^eldelf d_d`Zded8ebdelfdadbZdedceddeeeefdddeZdfdgdedhe]de'dee&dieideuef djdkZdS)oz Generating lines of code. N) CollectionIterator)replace)Enumauto)partialwraps)OptionalUnioncast)COMMA_PRIORITY DOT_PRIORITYSTRING_PRIORITY#get_leaves_inside_matching_bracketsmax_delimiter_priority_in_atom)FMT_OFFFMT_ONgenerate_comments list_comments)Line RHSResult append_leaves can_be_splitcan_omit_invisible_parensis_line_short_enoughline_to_string)FeatureModePreview)% ASSIGNMENTSBRACKETSCLOSING_BRACKETSOPENING_BRACKETSSTANDALONE_COMMENT STATEMENT WHITESPACEVisitorensure_visiblefstring_tstring_to_stringget_annotation_typehas_sibling_with_type is_arith_likeis_async_stmt_or_funcdefis_atom_with_invisible_parens is_docstringis_empty_tuple is_generator is_lpar_tokenis_multiline_string is_name_tokenis_one_sequence_between is_one_tupleis_parent_function_or_classis_part_of_annotation is_rpar_token is_stub_body is_stub_suiteis_tupleis_tuple_containing_staris_tuple_containing_walrusis_type_ignore_comment_string is_varargis_walrus_assignmentis_yieldsymswrap_in_parentheses)normalize_numeric_literal)fix_multiline_docstringget_string_prefixnormalize_string_prefixnormalize_string_quotes"normalize_unicode_escape_sequences)CannotTransform StringMergerStringParenStripperStringParenWrapperStringSplitter Transformer hug_power_op)token)LeafNodec@seZdZdZdS) CannotSplitzBA readable split that fits the allotted line length is impossible.N)__name__ __module__ __qualname____doc__rYrYR/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/black/linegen.pyrT_srTc seZdZdZdedeeddfddZdJd ede e fd d Z d e de e ffd d Z d ede e fddZd ede e fddZd ede e fddZd edeedeede e fddZd ede e fddZd ede e fddZd ede e fddZd ede e fdd Zd ede e fd!d"Zd ede e fd#d$Zd ede e fd%d&Zd ede e fd'd(Zd ede e fd)d*Zd ede e fd+d,Zd ede e fd-d.Z d/ede e fd0d1Z!d/ede e fd2d3Z"d/ede e fd4d5Z#d ede e fd6d7Z$d ede e fd8d9Z%d/ede e fd:d;Z&d/ede e fdd?Z(d ede e fd@dAZ)d ede e fdBdCZ*d ede e fdDdEZ+d ede e fdFdGZ,dKdHdIZ-Z.S)L LineGeneratorzGenerates reformatted Line objects. Empty lines are not emitted. Note: destroys the tree it's visiting by mutating prefixes of its leaves in ways that will no longer stringify to valid Python code on the tree. modefeaturesreturnNcCs||_||_||dSN)r\r] __post_init__)selfr\r]rYrYrZ__init__ls zLineGenerator.__init__rindentccsf|js|jj|7_dSt|jjdkr t|jjdr dS|j}t|j|j|d|_|VdS)zGenerate a line. If the line is empty, only emit if it makes sense. If the line is too long, split it first and then generate. If any lines were generated, set up a new current_line. Nrr\depth) current_linerflenleavesr,rr\)rarc complete_linerYrYrZliners  zLineGenerator.linenodec#st|trW|jj}t||jdD]3}|r|j|q|jt j kr2|j|| EdHq| EdH|j|| EdHq|rLd|_ |jt vrW|j|t|EdHdS)zCDefault `visit_*()` implementation. Recurses to children of `node`.r\N) isinstancerRrgbracket_trackerany_open_bracketsrr\appendtyperQCOMMENTrkprefixr%super visit_default)rarlrqcomment __class__rYrZrws"       zLineGenerator.visit_defaultccsn|jo |jjtjk}|s-ttjd}ttjd}|j}d|_||_|d|||| |EdHdS)zVisit an `x if y else z` testrnrN) prev_siblingrsrQLPARrRRPARru insert_child append_childrw)rarlalready_parenthesizedlparrparrurYrYrZ visit_tests    zLineGenerator.visit_testccs&|dEdH||EdHdS)z/Increase indentation level, maybe yield a line.rdNrkrwrarlrYrYrZ visit_INDENTszLineGenerator.visit_INDENTccs4|EdH||EdH|dEdHdS)z/Decrease indentation level, maybe yield a line.NrrrYrYrZ visit_DEDENTszLineGenerator.visit_DEDENTkeywordsparensccsVt|||j|jd|jD]}t|r |j|vr |EdH||EdHqdS)aVisit a statement. This implementation is shared for `if`, `while`, `for`, `try`, `except`, `def`, `with`, `class`, `assert`, and assignments. The relevant Python language `keywords` for a given statement will be NAME leaves within it. This methods puts those on a separate line. `parens` holds a set of string leaf values immediately after which invisible parens should be put.  parens_afterr\r]N)normalize_invisible_parensr\r]childrenr3valuerkvisit)rarlrrchildrYrYrZ visit_stmts  zLineGenerator.visit_stmtcc"||EdHd|jd_dSNrnrrwrrurrYrYrZvisit_typeparamszLineGenerator.visit_typeparamsccrNrnrdrrrYrYrZvisit_typevartuplerz LineGenerator.visit_typevartupleccrrrrrYrYrZvisit_paramspecrzLineGenerator.visit_paramspecccstj|jvrFt|jD]9\}}|dkrq |j|djtjkrE|jtj kr>|jdjt vr>t |s>t |||j|j ddq t||ddq ||EdHdS)NrrdFparentr\r]remove_brackets_around_commavisible)rwrap_long_dict_values_in_parensr\ enumeraterrsrQCOLONrBatomr"r@#maybe_make_parens_invisible_in_atomr]rCrw)rarlirrYrYrZvisit_dictsetmakers,  z LineGenerator.visit_dictsetmakerccs|EdHd}|jD]8}|jtjkrd}q |rE|jtjkr<|jdjtjkr -(2 ** 8) rd()rN)rrsrBpowerrhrQ DOUBLESTARrRr|r}remover~rSrrw)rarl _operatoroperandrrindexrYrYrZ visit_factors      zLineGenerator.visit_factorccsVt|jdks Jt|jd||j|jdr!t||jddd||EdHdS)a) Add potential parentheses around types in function parameter lists to be made into real parentheses in case the type hint is too long to fit on a line Examples: def foo(a: int, b: float = 7): ... -> def foo(a: (int), b: (float) = 7): ... rrrr\r]FrN)rhrrr\r]rCrwrrYrYrZ visit_tnames zLineGenerator.visit_tnamec cs.t|t|rtd|js|jjrt|j}t|}n|j}t |}|t |d}|d}|d|kr9dnd}||| }| }d|j j }t |rUt||}n|}d}|r|d|krgd|}|d|krq|d7}|dd krt |t |d } | d r|d7}d }n|sd}||} |dkr|} |rt | dnd} t | dkr| ||jjkrt |||jjkr|s|jd|d kr|| || |_n!|| |d || |_n|| || |_n || || |_|jjr |jtjkr t|j|_t|j|_||EdHdS) Nz\\\s*\nrrdrz F r\rT )rIr.researchrr\string_normalizationrGrHrFrhrgrfr2rEstriprstripr line_lengthrsrQSTRINGrw) rar docstringru quote_char quote_lendocstring_started_emptyrchas_trailing_backslashbackslash_countquoterlast_line_lengthrYrYrZ visit_STRINGsh          zLineGenerator.visit_STRINGccst|||EdHdSr_)rDrwrrYrYrZ visit_NUMBERszLineGenerator.visit_NUMBERccst|jdkr7|jd}|jd}|jtjkr|jtjks*|jtjkr7|jtjkr7t|jd||j |j d| |EdHdS)zVisit any atomrrrrdrN) rhrrsrQLSQBRSQBLBRACERBRACErr\r]rw)rarlfirstlastrYrYrZ visit_atoms     zLineGenerator.visit_atomccZt|}||d|jvr#tdd|jDr#||EdHdS||EdHdS)Nrcs(|]}|jtjkrdt|vVqdSrNrsrBfstring_replacement_fieldstr.0rrYrYrZ 5   z.LineGenerator.visit_fstring..r(rranyrrwrrarl string_leafrYrYrZ visit_fstring1 zLineGenerator.visit_fstringccr)NrcsrrrrrYrYrZrDrz.LineGenerator.visit_tstring..rrrYrYrZ visit_tstring@r zLineGenerator.visit_tstringccs8tj|jvrt|dh|j|jd||EdHdS)Ninr)rwrap_comprehension_inr\rr]rwrrYrYrZvisit_comp_forps  zLineGenerator.visit_comp_forccs||EdHdSr_)r rrYrYrZvisit_old_comp_forwsz LineGenerator.visit_old_comp_forcCsBt|jd|_|j}t}t|dhddhd|_t|hdddhd|_t|dd hdhd|_t|d d hd d hd|_ t|hd |d|_ t|d hd hd|_ t|dhdhd|_ t|dh|d|_ t||td|_t|dhdhd|_t||dhd|_t||dhd|_|j|_|j|_|j|_|j|_t||dhd|_dS)z,You are in a twisty little maze of passages.rmassert,)rr>ifelifelserrwhilerforr >tryrexceptfinallyrwithclassr^importdelN)rr\rgrrrvisit_assert_stmt visit_if_stmtvisit_while_stmtvisit_for_stmtvisit_try_stmtvisit_except_clausevisit_with_stmtvisit_classdefrvisit_expr_stmtvisit_return_stmtvisit_import_fromvisit_del_stmtrvisit_async_funcdefrvisit_decoratedrvisit_match_stmtvisit_case_block visit_guard)ravØrYrYrZr`zs0 zLineGenerator.__post_init__)r)r^N)/rUrVrWrXrrrrbintrrrkLNrwrSrrRrrrrrrrrrrrrrrrrrrrrrrrrrr r rr` __classcell__rYrYryrZr[esR   X0r[rkr]r\r^cCs,z ttt|||WStyYdSwr_)rrrPrJ)rkr]r\rYrYrZ!_hugging_power_ops_line_to_strings  r3rYc cs|jr |VdSt|}t|||p|}|j}|j}t||}t||}t||} t||} | s\|j s\|j s\t |||dsD| r\|jrK|s\|s\tj|vrY||g} nQg} nN|jrht||shtg} nBdtdtdttdtdttf dd } td d d | i} tj|vr|jr||| tt| | g} n||| | | g} n |jrtt| g} n| g} | t | D]}z t!|||||d }Wn t"yYqw|EdHdS|VdS)zTransform a `line`, potentially splitting it into many lines. They should fit in the allotted `line_length` but might not be able to. `features` are syntactical features that may be used in the output. N)r\line_strrarkr]r\r^css`t||jD]}tt||||d}t|d|dr#|EdHdSqt|||dEdHdS)zWraps calls to `right_hand_split`. The calls increasingly `omit` right-hand trailers (bracket pairs with content), meaning the trailers get glued together to split on another bracket pair instead. omitrrmN)r])generate_trailers_to_omitrlistright_hand_splitr)rarkr]r\r6rrYrYrZ_rhss  ztransform_line.._rhsrhsrY__call__r4)# is_commentrr3rrrKrLrNrM$contains_uncollapsable_type_commentsshould_split_rhsmagic_trailing_commar!contains_unsplittable_type_ignoreinside_bracketscontains_standalone_comments0contains_implicit_multiline_string_with_commentsrstring_processingis_defshould_split_funcdef_with_rhsleft_hand_splitobjectrrrrrrsdelimiter_splitstandalone_comment_splitrrrPrun_transformerrJ)rkr\r]r4line_str_hugging_power_opsllsn string_mergestring_paren_strip string_splitstring_paren_wrap transformersr:r; transformresultrYrYrZtransform_lines                rXcCsg}d}|jD]}|jtjkrd}|r|||jtjkr d}qt|j|jd}t |}|D]}|j|dt ||vdq/|j duS)zIf a funcdef has a magic trailing comma in the return type, then we should first split the line with rhs to respect the comma. FTre preformatted track_bracketN) rirsrQrrrrrr\rfridrA)rkr\return_type_leavesin_return_typerrWleaves_to_trackrYrYrZrH!s&      rHc@seZdZeZeZeZdS)_BracketSplitComponentN)rUrVrWrheadbodytailrYrYrYrZr`@s r` _featuresccstjtjfD]}g}g}g}|}d}d} t|jD]v\} } | dkr-| jtjkr-| d7} n| dkrF| jtjkr<| d7} n | jtjkrF| d8} ||urp| jtvrp| j|urpt |t rp|tjkrb| dkspt | t ||rn|n|}| | ||ur| j|krt j|vs|tjkr| dks| }|}q|r|rnq|r|stdt|||tjd} t|||tjd} t|||tjd}t| | || | |fD]}|r|VqdS)a/Split line into many lines, starting with the first matching bracket pair. Note: this usually looks weird, only use this for function definitions. Prefer RHS otherwise. This is why this function is not symmetrical with :func:`right_hand_split` which also handles optional parentheses. NrrrdNo brackets found component)rQr|rrrirsrr!opening_bracketrorRr'rrrfix_type_expansion_splitrTbracket_split_build_liner`rarbrc bracket_split_succeeded_or_raise)rkrdr\ leaf_type tail_leaves body_leaves head_leavescurrent_leavesmatching_bracketrfrrrarbrcrWrYrYrZrIFs~             rIr6ccs*t||d}t|||||dEdHdS)aXSplit line into many lines, starting with the last matching bracket pair. If the split was by optional parentheses, attempt splitting without them, too. `omit` is a collection of closing bracket IDs that shouldn't be considered for this split. Note: running this function modifies `bracket_depth` on the leaves of `line`. r5r]r6N)_first_right_hand_split%_maybe_split_omitting_optional_parens)rkr\r]r6 rhs_resultrYrYrZr9s   r9cCsg}g}g}|}d}d}t|jD]+}||ur!||ur!|r|n|}||||ur<|jtvr.rrmF)"reversedrirrrsr!r\rhrTreverser*hug_parens_with_braces_and_square_bracketsr\rr8rQSTARrrhpopinsertrjr`rbrACOMMArsumrrextendrarcrkr)rkr6rmrnrorprhclosing_bracketrrbinner_body_leaveshugged_opening_leaveshugged_closing_leaves is_unpackingunpacking_offset inner_body should_hugrrarcrYrYrZrss                rsr;c csdtj|vr|jjtjkr|jjs|jjtjkr|jjs|j st ||j rt |jh|}zt ||d}t|||rIt|||||dEdHWdSWnKty}z?|jrVn5t|js|t|j|ds|tjrw|jjrw|jjjrw|jjjjtjks|td||js|jrtd|WYd}~nd}~wwt|jt|j|j|j|jfD]}|r|VqdS)Nr5rrrmz.rcss|]}|jtvVqdSr_)rsr rvrYrYrZrgNrrdrwrmcSg|]}|jqSrYrsrvrYrYrZ wz2_prefer_split_rhs_oop_over_rhs..cSrrYrrvrYrYrZrxrFcsrr_)rsrQEQUALrvrYrYrZrr)rarBrbrcrrrhrrsrBrrp delimitersrrirhrQrrrrrAcountrxr!)rr;r\rhs_head_equal_countrhs_oop_head_equal_count has_closing_bracket_after_assignrrYrYrZrHs\ $    rrarbrccCsDtt|}|s|dkrtd|dkr td|ddSdS)aRaise :exc:`CannotSplit` if the last left- or right-hand split failed. Do nothing otherwise. A left- or right-hand split is based on a pair of brackets. Content before (and including) the opening bracket is left on one line, content inside the brackets is put on a separate line, and finally content starting with and following the closing bracket is put on a separate line. Those are called `head`, `body`, and `tail`, respectively. If the split produced the same line (all content in `head`) or ended up with an empty `body` and the `tail` is just the closing bracket, then it's considered failed. rz)Splitting brackets produced the same linerz,Splitting brackets on an empty body to save z characters is not worth itN)rhrrrT)rarbrctail_lenrYrYrZrks rkrioriginalrhcCs|sdS|jr dS|jsdS|jdkrdStdd|Dr dStdd|Dd}|dur0dSt|dkr8dS|jrI|jjrI|jjjt j krIdSdS)NFTrcss&|]}|jtjkot| VqdSr_)rsrQr~r7rvrYrYrZrs z)_ensure_trailing_comma..css|]}|jr|VqdSr_rrvrYrYrZrrr^) rrGrrrr)r next_siblingrsrQVBAR)rirrhleaf_with_parentrYrYrZ_ensure_trailing_commas2  rrgc Cst|j|jd}|tjurHd|_|jd7_t|||rHtt|dddD] }||j t kr1q'||j t j krGt t j d}||d|t}|tjurTt|}|D]}|j|dt||vd||D] } |j| ddqiqV|tjurt||rd|_|S)aiReturn a new line with given `leaves` and respective comments from `original`. If it's the head component, brackets will be tracked so trailing commas are respected. If it's the body component, the result line is one-indented inside brackets and as such has its first leaf's prefix normalized and a trailing comma added when expected. reTrdrrrYrZ)rr\rfr`rbrCrrangerhrsr#rQr~rRr}rrarrrr\comments_aftershould_split_liner@) rirrhrgrWr new_commar_r comment_afterrYrYrZrjs:      rj split_funcc s2tdtdttdtdttffdd }|S)zNormalize prefix of the first leaf in every line returned by `split_func`. This is a decorator over relevant split functions. rkr]r\r^c3s*|||D] }d|jd_|VqdSr)riru)rkr]r\ split_linerrYrZ split_wrappers  z0dont_increase_indentation..split_wrapper)rrrrrr)rrrYrrZdont_increase_indentationsrcCs8tt|jdddD]}|j|jtkr|Sq dS)Nrdrr)rrhrirsr#)rkleaf_idxrYrYrZ_get_last_non_comment_leafs rrcCs<t|tjhdr tj|vSt|tjtjhdrtj|vSdS)N)withinT)r?rB typedargslistrTRAILING_COMMA_IN_DEFarglistargumentTRAILING_COMMA_IN_CALL)rr]rYrYrZ_can_add_trailing_commas   rsafedelimiter_prioritycCsH|r"|tkr"|jdjtjkr"|jdjtkr"ttjd}|||S)Nrr)r rirsrQr~r#rRrr)rrrkrrYrYrZ_safe_add_trailing_comma"s  rc#stjdkr tddjd}j}z |jt|hd}Wn ty,tddw|tkr<||dkr.append_to_linec3s$|D] }|EdHqdSr_)r)rr)rrkrYrZappend_comments]sz(delimiter_split..append_comments)rhrirTrpmax_delimiter_priorityr\rr delimiter_count_with_priorityrr\rfrCsysmaxsizerRrrrrgetMIGRATE_COMMENT_DELIMITERSmin bracket_depthrrsr#r)rkr]r\ last_leafbtr lowest_depthtrailing_comma_saferlast_non_comment_leafrrprevious_priority leaf_priorityrY)rrgrkrZrK1st           rKc#ss tdtjjjddtdttffdd }jD]}||EdH |D] }||EdHq2q$rDVdSdS)z4Split standalone comments from the rest of the line.z*Line does not have any standalone commentsrrr^c3r)rTr)rfrCNrrrrYrZrrz0standalone_comment_split..append_to_lineN) rDrTrr\rfrCrRrrir)rkr]r\rrrrYrrZrLs   rLrlrc Cst|jd|dD] }|jtvrdSq|jtjkrt|||d}tt |j D]=\}}t |t r@|jtj kr@t||||dt |t rT|jtjkrTt|dh||dt |t rh|jtjkrht|dh||d|dkryt |t ry|jtjkryd}|rR|jtjkr|jtjkrt |jtr|jjtjkr|jjd krt||||dd rt||dd nt |t r|jtjkrt||||d n|jtjkrd |vrt|j dkrt|j drt|j dr|j djtjkst||||drt||dd nZt|rt||dd nM|jtj krt!|||dS|dkr)|jtj"kr)|jtj#kr)q(t |trF|j$durF|j$jtj%krF|jdkrFdSt&|sRt||dd |jtj'k}t |toe|j|vpe|}q(dS)aEMake existing optional parentheses invisible or create new ones. `parens_after` is a set of string leaf values immediately after which parens should be put. Standardizes on visible parentheses for single-element tuples, and keeps existing visible parentheses for other tuples and generator expressions. F) is_endmarkerr\NrcaserrTrrrrr rrrdr)(rrurrrsrB with_stmt_maybe_wrap_cms_in_parensrr8rrorS annassignr case_blockguardtestlist_star_exprrfor_stmtr{rRrQNAMErrCremove_with_parensrhr1r8testr5 import_from_normalize_import_fromr{ except_clauserrr2r~) rlrr\r]pc check_lparrr comma_checkrYrYrZrs                       rrrrcCsjt|rt|jds Jd|_d|jd_dS|jtjkr3||ttj d| ttj ddSdS)Nrrn) r1r8rrrsrQr{r~rRr|rr})rrrrYrYrZr"s rcCs|jdjtjkrt|jdkr|jdjtjkr|jdjdjtjkrt|jd|||ddr;t ||jdddt t |jdjd}t t |jdjd}|jdjd}t |t r|jtjksw|jdjtjkswtdd |jDrt|t|dSdSdSdSdSdSdS) NrrdTrFrrcss&|]}t|to|jtjkVqdSr_)rorRrsrQrrrYrYrZrLs  z&remove_await_parens..)rrsrQAWAITrhrBrr|rrCr rRrorSrrr')rlr\r]rhrbracket_contentsrYrYrZr0s<     rc Cstj|vst|jdks|jdjtjkrdSd}tdt|jD]}|j|jtj kr0|}nq!|durft tj d}t tj d}|jd|}|D]}| qJttj|ttj||g} |d| dSdS)zWhen enabled and safe, wrap the multiple context managers in invisible parens. It is only safe when `features` contain Feature.PARENTHESIZED_CONTEXT_MANAGERS. rrdNrn)rPARENTHESIZED_CONTEXT_MANAGERSrhrrsrBrrrQrrRr|r}rrS testlist_gexpr~) rlr\r] colon_indexrrrcontext_managersr new_childrYrYrZrUs0     rcCs|jtjkr-t||||ddrt||ddt|jdtr+t|jd|||dd Sd S|jtj krH|jD]}t|trEt||||dq6d S|jtj krqt dd| Dsst|jd |||ddrut||jd ddd Sd Sd Sd S) z6Recursively hide optional parens in `with` statements.TrFrrdrcsrr_)rsrQ COLONEQUALrvrYrYrZrs  z%remove_with_parens..rN) rsrBrrrCrorrSrr asexpr_testrri)rlrr\r]rrYrYrZrs>    rFrc Cs|jtjtjfvsst|sst|sst|r|jtjksst|r+|jtjkr+t |t j sst |r5|jtj kss|sgt|tkrgtj|vrstj|vrst|rs|jdurs|jjtjkrs|jdurgt|jrg|jjdksst|sst|sst|rudSt|r|jtjtj tjtjtjtj tjtj!tj"tj#tj$tj#f vrdS|j%d}|j%d}t&|rt'|r|j%d}t(|j)*|dsd|_d|_t+|||||d t,|r|-|j%d|j%dj)*r|j%dj)|j%dj)|j%d_)|j%dj)*r|j%dj)|j)|_)dSd S) a If it's safe, make the parens in the atom `node` invisible, recursively. Additionally, remove repeated, adjacent invisible parens from the atom `node` as they are redundant. Returns whether the node should itself be wrapped in invisible parentheses. NasFrrrdrmrnrT).rsrBrexprr/r5r;rrr*rQr~rA expr_stmtrr r!remove_parens_around_except_typesrUNPARENTHESIZED_EXCEPT_TYPESrrrr3rr=r<r0r@r assert_stmt return_stmtfuncdefrtnamerdel_stmtrr1r8r>rurrr-r)rlrr\r]rrrmiddlerYrYrZrs         !%&')    rc Cs|jr|jdvs dSt}d}z|jd}|jtjkr$d}|t||j j |d}Wn t t fy8YdSw|t koL|jjrC|pL|jjtjtjhvS)zFShould `line` be immediately split with `delimiter_split()` after RHS?z[{(FrTr)rrrrirsrQr~addr\rpr IndexErrorrr r\rArBrr)rkrhrtrailing_commar max_priorityrYrYrZr!s"   rrc cst}|js |Vd|j}d}d}t}|jddD]\}}} || 7}||kr,dS| t|jt|jk} |jtks?| rBdS|r}||urKd}q|jt vr||dkr[|j |dnd} | ru| jt j kru|j durut|j ||j sudS|t|q|jt vr|dkr|j |dnd} | r| jtvr|t|q|r|t|||||V| r| jt j kr|j durt|j ||j sdS|jr|j }|}qdS)aGenerate sets of closing bracket IDs that should be omitted in a RHS. Brackets can be omitted if the entire trailer up to and including a preceding closing bracket fits in one line. Yielded sets are cumulative (contain results of previous yields, too). First set is empty, unless the line should explode, in which case bracket pairs until the one that needs to explode are omitted. NT) is_reversedrrd)rrArfenumerate_with_lengthrhrrursr#r!rirQr~rhr4rr\r"updateclear) rkrr6lengthrhrinner_bracketsrr leaf_lengthhas_inline_commentprevrYrYrZr7<sr         r7rnr=rVr4c s&|st|}g}|||D]}t|d|krtd|t||dqt|}tj|vsh|j j dksh|j j rht dd|j j Dsh|sh|dsh|dsht|ddsht d d|jDrj|S|}t|||j|tjhB} t||| |d } tfd d| Dr| }|S) Nrz-Line transformer returned an unchanged resultrr;css|]}|jVqdSr_)r)rbracketrYrYrZrsz"run_transformer..rrmcss|]}|jduVqdSr_rrvrYrYrZrrr=c3s|] }t|dVqdS)rmN)r)rlnrmrYrZrr)rrrrJrrXrrrrzrUrp invisiblerrr?rBrriclonerrMall) rkrVr\r]r4rWtransformed_line features_set line_copy features_fopsecond_opinionrYrmrZrMsH      rM)rY)rYrY)F)rXrrcollections.abcrr dataclassesrenumrr functoolsrrtypingr r r black.bracketsr r rrrblack.commentsrrrr black.linesrrrrrrr black.moderrr black.nodesrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6r7r8r9r:r;r<r=r>r?r@rArBrCblack.numericsrD black.stringsrErFrGrHrI black.transrJrKrLrMrNrOrPblib2to3.pgen2rQblib2to3.pytreerRrSr0LeafIDr1rTr[rr3rXboolrHr`rIr9rsrtrrkr8rrjrrrrrrKrLrrrrrrrrr7rMrYrYrYrZs $  '$  9   | F  k D K $ 1  Y " t% * 5 rR