o ^iP @s dZddlZddlmZmZmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZmZdeed eeeeffd d Zd eeefd efd dZd eeeefded eeeeffddZd eeeefdeded eeeeffddZded eeeefd dfddZ de d efddZ!GdddedZ"dede#ed dfdd Z$de d dfd!d"Z%d#ee d$e d dfd%d&Z&d'e d efd(d)Z'd*e e ee fd e#efd+d,Z(eGd-d.d.Z)deded ee)fd/d0Z*d1ed2ee)d3ed efd4d5Z+dS)6z?Functions related to Black's formatting by line ranges feature.N) CollectionIteratorSequence) dataclass)Union) LNSTANDALONE_COMMENTLeafNodeVisitor first_leaf furthest_ancestor_with_last_leaf last_leafsyms)ASYNCNEWLINE line_rangesreturnc Csg}|D]9}|d}t|dkrtd|zt|d}t|d}Wnty5td|dw|||fq|S)N-z:Incorrect --line-ranges format, expect 'START-END', found rzrsz7_convert_nodes_to_standalone_comment..rrt)r^r rujoinrwryr r)rrfr^r{rur~r}rNr!r!r"rjhs4     rjrncCs$|jtkr|jS|jt|dS)z5Returns the line number of the leaf node's last line.r&)rPrlinenorzr')rnr!r!r"_leaf_line_ends r node_or_nodescCst|tr/|}|s tSt|d}t|d}|r,|r,|j}t|}tt||dStS|}t|trBtt|jt|dSt|}t|}|rZ|rZtt|jt|dStS)z5Returns the line range of this node or list of nodes.rrsr) rOrGrCr rrrrEr )rrr{r| line_startline_endrNr!r!r"rXs&    rXc@s:eZdZUdZeed<eed<eed<eed<eed<dS) _LinesMappingz1-based lines mapping from original source to modified source. Lines [original_start, original_end] from original source are mapped to [modified_start, modified_end]. The ranges are inclusive on both ends. r5 original_endr4r6r3N)rbrcrdrer__annotations__boolr!r!r!r"rs  rc Cstd|jdd|jdd}|}g}t|D]c\}}|dkr:|jdks+|jdkr9|td|jd|jddn ||d}|t|j|j d|j|j|j d|jdd|t |dkr||t|jd|j|j |jd|j|j ddq|S)aReturns a sequence of _LinesMapping by diffing the sources. For example, given the following diff: import re - def func(arg1, - arg2, arg3): + def func(arg1, arg2, arg3): pass It returns the following mappings: original -> modified (1, 1) -> (1, 1), is_changed_block=False (the "import re" line) (2, 3) -> (2, 2), is_changed_block=True (the diff) (4, 4) -> (3, 3), is_changed_block=False (the "pass" line) You can think of this visually as if it brings up a side-by-side diff, and tries to map the line ranges from the left side to the right side: (1, 1)->(1, 1) 1. import re 1. import re (2, 3)->(2, 2) 2. def func(arg1, 2. def func(arg1, arg2, arg3): 3. arg2, arg3): (4, 4)->(3, 3) 4. pass 3. pass Args: original_source: the original source. modified_source: the modified source. NT)keependsrrF)r5rr4r6r3) difflibSequenceMatcher splitlinesget_matching_blocks enumerateabrrsizer)r.r/matchermatching_blocksr7iblockprevious_blockr!r!r"r0sT       r0 original_liner7 start_indexcCsL|}|t|kr$||}|j|kr|jkr|S|d7}|t|ks|S)zGReturns the original index of the lines mappings for the original line.r)rr5r)rr7rr}mappingr!r!r"r2 s  r2),rercollections.abcrrr dataclassesrtypingr black.nodesrrr r r r r rrblib2to3.pgen2.tokenrrrzrGtuplerr#rr$r-rArMrRrFrCrIrZrjrrXrr0r2r!r!r!r"sf  , "  "P  665"  O