o ^iGf@sNUddlZddlmZmZddlmZddlmZddlm Z m Z m Z ddl m Z mZddlmZmZmZmZmZmZmZmZmZmZddlmZdd lmZmZe eefZhd Z e e!d <d d hZ"e e!d<hdZ#e e!d<eej$ej%hZ&e e!d<dZ'dZ(dZ)eGdddZ*dede deefddZ+eddde,de-de de.e*fd d!Z/ded"e0ddfd#d$Z1d%e,de de,fd&d'Z2d(ede d)ee3e0e0fddfd*d+Z4d,e*dede3e-e-e-ffd-d.Z5d,e*ded/e-d0e-de-f d1d2Z6ded,e*d3e0de de-f d4d5Z7d(ede d)ee3e0e0fde-fd6d7Z8d8e.ed,e*d3e0d0e-d)ee3e0e0fdeddfd9d:Z9ded,e*de deefd;d<Z:d=ede efd>d?Z;d@edAede-fdBdCZdHede de-fdIdJZ?dHede de-fdKdLZ@dMe.ede-fdNdOZAe e#Be"BfdPe,dQeBe,de-fdRdSZCdS)TN) CollectionIterator) dataclass) lru_cache)FinalOptionalUnion)ModePreview) CLOSING_BRACKETSSTANDALONE_COMMENT STATEMENT WHITESPACE container_of first_leaf_ofis_type_comment_stringmake_simple_prefixpreceding_leafsyms)token)LeafNode> # fmt:off # fmt: off# yapf: disableFMT_OFFz # fmt: skipz # fmt:skipFMT_SKIP># fmt:on # fmt: on# yapf: enableFMT_ON_COMPOUND_STATEMENTSz !:#'# ;c@sBeZdZUdZeed<eed<eed<eed<eed<eed<dS) ProtoCommentaDescribes a piece of syntax that is a comment. It's not a :class:`blib2to3.pytree.Leaf` so that: * it can be cached (`Leaf` objects should not be reused more than once as they store their lineno, column, prefix, and parent information); * `newlines` and `consumed` fields are kept separate from the `value`. This simplifies handling of special marker comments like ``# fmt: off/on``. typevaluenewlinesconsumed form_feedleading_whitespaceN)__name__ __module__ __qualname____doc__int__annotations__strboolr3r3S/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/black/comments.pyr$'s   r$leafmodereturnccsZd}t|j|jtjk|dD]}|j}t|j|j}t |j|j |dVqt ||dS)aClean the prefix of the `leaf` and generate comments from it, if any. Comments in lib2to3 are shoved into the whitespace prefix. This happens in `pgen2/driver.py:Driver.parse_tokens()`. This was a brilliant implementation move because it does away with modifying the grammar to include all the possible places in which comments can be placed. The sad consequence for us though is that comments don't "belong" anywhere. This is why this function generates simple parentless Leaf objects for comments. We simply don't know what the correct parent should be. No matter though, we can live without this. We really only need to differentiate between inline and standalone comments. The latter don't share the line with any code. Inline comments are emitted as regular token.COMMENT leaves. Standalone are emitted with a fake STANDALONE_COMMENT token identifier. r is_endmarkerr6)prefixN) list_commentsr:r%r ENDMARKERr(rr'r)rr&normalize_trailing_prefix)r5r6total_consumedpcr:r3r3r4generate_comments;s r@i)maxsizer:r9c Csg}|rd|vr |Sd}d}d}d}ttd|D]Z\}} |t| d7}td| } | s0J| \} } | sB|d7}d| vrBd}| dsQ| d rP|d7}q||kr[|s[tj } nt } t | |d }| t | ||||| d d}d}q|S) zNReturn a list of :class:`ProtoComment` objects parsed from the given `prefix`.#rFz ? | z^(\s*)(\S.*|)$ T\r6)r%r&r'r(r)r*) enumerateresplitlenmatchgroups startswithendswithrCOMMENTr make_commentappendr$)r:r9r6resultr(nlines ignored_linesr)index full_linerK whitespaceline comment_typecommentr3r3r4r;XsJ        r;r>cCsL|j|d}d|vr!|d}d|vo|d}t|||_dSd|_dS)zNormalize the prefix that's left over after generating comments. Note: don't use backslashes for formatting or you'll lose your voting rights. NrE rD)r:countrNr)r5r> remaindernl_countr)r3r3r4r=s   r=contentcCs|}|sdS|drt|r|S|ddkr|dd}|r9|ddkr9td||ds9d|dd}tj|vr^|r^d|vr^td||dr^|d d\}}|d |}|rj|dt vrjd|}d|S) aReturn a consistently formatted comment from the given `content` string. All comments (except for "##", "#!", "#:", '#'") should have a single space between the hash sign and the content. If `content` didn't start with a hash sign, one is provided. Comments containing fmt directives are preserved exactly as-is to respect user intent (e.g., `#no space # fmt: skip` stays as-is). rBrrCN r"rF :z: ) rstriprM_contains_fmt_directiverlstripr standardize_type_commentsrIstripCOMMENT_EXCEPTIONS)r`r6 type_part value_partr3r3r4rPs6    rPnodelinescCs d}|rt|||}|sdSdS)zJConvert content between `# fmt: off`/`# fmt: on` into standalone comments.TN)convert_one_fmt_off_pair)rlr6rm try_againr3r3r4normalize_fmt_offs  rprZcCs@t|jt}t|jt}|s|sdS|r|jtvrdSd||fS)zsCheck if comment should be processed for fmt handling. Returns (should_process, is_fmt_off, is_fmt_skip). )FFFT)rer&rrr%r )rZr5 is_fmt_off is_fmt_skipr3r3r4_should_process_fmt_comments   rsrqrrcCsP|jtkrdSt|}|sdS|r|jtvr|jtkrdS|r&|jtvr&dSdS)zCheck if comment is a valid standalone fmt directive. We only want standalone comments. If there's no previous leaf or if the previous leaf is indentation, it's a standalone comment in disguise. TF)r%r rr)rZr5rqrrprevr3r3r4 _is_valid_standalone_fmt_comments ruprevious_consumedc Cst|jd|d}d}d}t|D] \}}|dur|jtvr|}|dur0||kr0|jtvr0|}nq|dus9|dur;dS||}||} |j} |j} | j} | | | } |jd| }|drd|dd}| d|d|j}| |jd}d|vr| dd}||7}| | jd|_|j }|dusJdd}t|j D] \}}||ur|}nq|dusJd| |t t||dd d S) zvHandle fmt:off/on blocks that contain only comments. Returns True if a block was converted, False otherwise. Fr8Nr[rz2INTERNAL ERROR: fmt: on/off handling (prefix only)z1INTERNAL ERROR: fmt: on/off handling (leaf index)r:fmt_pass_converted_first_leafT)r;r:rGr&rr r(rNr'rIparentchildren insert_childrr )r5rZrvr6 all_comments fmt_off_idx fmt_on_idxidxcfmt_on_commentoriginal_prefix start_posend_poscontent_between_and_fmt_on hidden_valuestandalone_comment_prefixfmt_off_prefixrzleaf_idxchildr3r3r4_handle_comment_only_fmt_blocks^     rc Cs|D]O}d}t|jd|dD]B}t||\}}}|s |j}qt||||s+|j}qtt|||} | sC|rCt||||rBdSq| sFqt | |||||dSqdS)zConvert content of a single `# fmt: off`/`# fmt: on` into a standalone comment. Returns True if a pair was converted. rFr8T) leavesr;r:rsr(rulistgenerate_ignored_nodesr_handle_regular_fmt_block) rlr6rmr5rvrZshould_processrqrr ignored_nodesr3r3r4rnAsD  (rnrc sl|d}|j}|j}|jtvr||jd|_|rd|_|} n |d|d|j} ddd|D} |j|j|jtvrpd} t|dkret fdd|Dse| |jd} d| vre| dd} | | 7} |jd| } |rz| |j |j7} | dr| dd} d} |D] } | }| dur|} q|dusJd | dusJd || tt| | t|d dS) z(Handle fmt blocks with actual AST nodes.rNr\r[css|]}t|VqdSN)r1).0nr3r3r4 z,_handle_regular_fmt_block..c3s0|]}|dko|dknVqdS)rrCNr3)rrXcomment_linenor3r4rs  rwz(INTERNAL ERROR: fmt: on/off handling (1)z(INTERNAL ERROR: fmt: on/off handling (2)rx)rzr:r&rr(r'joinlinenorJanyrIr*rNremover|rr r)rrZrvrrrmr5firstrzr:rrr first_idxignoredrUr3rr4rusT      rccs:t|jtrt|||EdHdSt|}|dur|jtjkrt||dr(dSt ||drxt |j D]C\}}t |t rMt||drM|jtvrJ|VdS|jtjkrj|t|j dkrjt |j |d|drjdSt ||drsdS|Vq3n|jtjkr|jdurdS|V|j}|dur|jtjks dSdSdSdS)zStarting from the container of `leaf`, generate all leaves until `# fmt: on`. If comment is skip, returns leaf only. Stops at the end of the block. NrFrC)rer&r%_generate_ignored_nodes_from_fmt_skiprr%rr< is_fmt_onchildren_contains_fmt_onrGr{ isinstancerr INDENTrJDEDENT next_sibling)r5rZr6 containerrUrr3r3r4rs@      $rrzcCsd|jtjkrdSt|jtsdS|jjtjkr(t|jjtr(|jjjtvr(|jS|jjtvr0|SdS)aReturn the body node of a compound statement if we should respect fmt: skip. This handles one-line compound statements like: if condition: body # fmt: skip When Black expands such statements, they temporarily look like: if condition: body # fmt: skip In both cases, we want to return the body node (either the simple_stmt directly or the suite containing it). N)r%r simple_stmtrrzrsuiter!)rzr3r3r4 _find_compound_statement_contexts      r body_nodesimple_stmt_parentcCsb|D]*}|jtjkr.|jtjkr(dd|jD}t|dko%|d|uS||uSqdS)zCheck if a compound statement should be kept on one line. Returns True only for compound statements with semicolon-separated bodies, like: if True: print("a"); print("b") # fmt: skip cSsg|] }|jtjkr|qSr3)r%rr)rrr3r3r4 s  z:_should_keep_compound_statement_inline..rCrF)rr%rSEMIrrr{rJ)rrr5 simple_stmtsr3r3r4&_should_keep_compound_statement_inlines    rcCst||sgS|j}|dus|jtvrgSg}|jD]$}||ur#|St|tr7|jtjtj fvr6| |q| | q|S)zJGet header nodes for a compound statement that should be preserved inline.N) rrzr%r!r{rrrNEWLINErrQextendr)rr compound_stmt header_leavesrr3r3r4_get_compound_statement_header)s    rccsh|j}|j}g}t|jd|d}|r|j|djkrdS|dur|j|jd|_tj|vrV|g}d|jvrO|jdurO|j}|d|d|jvrO|jdus<|EdHdS|}|g}|jdurh|jdurh|j}d|jvr|jdurt |j } | r| dn|}|j t j t jfvrd|_n|d||jdur|jdur|j}d|jvr|jdusrtj|vrt|trt|} | durt| |} | r| |}|EdHdS|dur.|j tjkr0|j t j kr2d|_|j} | dur| j tjkr|d| | j} | dur| j tjks|j} | dur%| jdur%| jj t jkr%|d| jt|EdHdSdSdSdS)zLGenerate all leaves that should be ignored by the `# fmt: skip` from `leaf`.Fr8rNr[rwr\) prev_siblingrzr;r:r&r(r fix_fmt_skip_in_one_linersinsertrrr%rrrrrrrrrASYNCiter)r5rZr6rrzrcommentssiblings current_node leaf_nodesr header_nodesparent_sibling grandparentr3r3r4rBs           rrcCs<d}t|jd|dD]}|jtvrd}q |jtvrd}q |S)zDetermine whether formatting is switched on within a container. Determined by whether the last `# fmt:` comment is `on` or `off`. Fr8T)r;r:r&r r)rr6fmt_onrZr3r3r4rs  rcCs2|jD]}t|}|durt||drdSqdS)z2Determine if children have formatting switched on.NrFTF)r{rr)rr6rr5r3r3r4rs r comment_listcCs |D] }|jdr dSqdS)z Returns: True iff one of the comments in @comment_list is a pragma used by one of the more common static analysis tools for python (e.g. mypy, flake8, pylint). )z# type:z# noqaz # pylint:TF)r&rM)rrZr3r3r4contains_pragma_comments  r comment_line directivescsR|gdd|tddDdd|ttD}tfdd|DS)a Checks if the given comment contains format directives, alone or paired with other comments. Defaults to checking all directives (skip, off, on, yapf), but can be narrowed to specific ones. Matching styles: # foobar <-- single comment # foobar # foobar # foobar <-- multiple comments # foobar; foobar <-- list of comments (; separated) cSg|]}t|qSr3_COMMENT_PREFIXrhrrZr3r3r4r z+_contains_fmt_directive..rCNcSrr3rrr3r3r4rrc3s|]}|vVqdSrr3rrr3r4rrz*_contains_fmt_directive..)rIrrh_COMMENT_LIST_SEPARATORr)rrsemantic_comment_blocksr3rr4res re)DrHcollections.abcrr dataclassesr functoolsrtypingrrr black.moder r black.nodesr r r rrrrrrrblib2to3.pgen2rblib2to3.pytreerrLNrr0rr except_clause case_blockr!rirrr$r@r1r2rr;r/r=rPtuplerprsrurrnrrrrrrrrrsetrer3r3r3r4s   0   /)      L 4 @ 1)   ]