"Te'UddlZddlZddlZddlZddlZddlmZmZddl m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZddlmZmZddlmZddlm Z dd l!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+dd l,m-Z-dd l.m/Z/dd l0m1Z1m2Z2d dl3m4Z4d dl5m6Z6d dl7m8Z8m9Z9d dl:m;Z;mZ>d dl?m@Z@d dlAmBZBd dlCmDZDmEZEd dlFmZmGZGd dlHmIZIeeJdfZKejdk(ZMdZNie*ee+ede"ede"jede%ede%j*ede(jed e&jede&jed!e&jedd"e&jed!d"e&jede&jed d#e&jed$e&jed$e&jede&jed%e)ed&e'ede$jed'e$jed!e$jed(e$jed d#e$jed(e$jFed'e#ed$d"i Z`e eKefead)<ie*ee+ed*e"ede"jed+e%ed%e%j*ed+e(jed,e&jed+e&jed-e&jed+d"e&jed-d"e&jed+e&jed,d#e&jed'e&jed'e&jed+e&jed%e)ed&e'ed%e$jed'e$jed-e$jed(e$jed,d#e$jed(e$jFed'e#ed$d"i Zbe eKefead.<e`ebd/Zcd0ZdGd1d2eZeGd3d4eeZfGd5d6eeZgeehehfZiGd7d8eZjGd9d:e@Zkd;eehdZlemd?k(rddlnZnddlZenjd@AZpepjdBdCdDEepjdFdGdHdIddJKepjdLdMdNdIdOdPKepjdQdRdSdIdTUepjdVdWehdXddYZepjd[d\d]dIdOd^Kepjd_d`dIdadOdbcepjdddedfddghepjdidjdkddlhepjdmdnddodpqepjdrdsehddtduvepjdwehddxdyvepjZsddzltm;Z;e;esjesj{Z:esjd|k(rejjZzekezesjesjesjesjesjesjesjesjh} Znyekj esjesjesjesjesjesjesjesjesjh~ Ze:j eesj yy)N)ABCabstractmethod)Path) AnyDictIterableList NamedTupleOptionalSequenceSetTupleTypeUnion)Lexer)get_lexer_by_nameguess_lexer_for_filename)Style)get_style_by_name) CommentErrorGenericKeywordNameNumberOperatorStringToken Whitespace) ClassNotFound)Lines)PaddingPaddingDimensions) loop_firstcell_len)Color blend_rgb)ConsoleConsoleOptions JustifyMethod RenderResult) JupyterMixin) Measurement)SegmentSegments)r StyleType)Text.WindowsmonokaiwhitecolorTdimcyanbluemagentagreen)r8 underline)r8boldred bright_blueyellow bright_red)r@ ANSI_LIGHT bright_black bright_cyanbright_magenta bright_green ANSI_DARK) ansi_light ansi_darkc@eZdZdZededefdZedefdZy) SyntaxThemezBase class for a syntax theme. token_typereturnct)z'Get a style for a given Pygments token.NotImplementedError)selfrPs -/usr/lib/python3/dist-packages/rich/syntax.pyget_style_for_tokenzSyntaxTheme.get_style_for_token~ "!ct)zGet the background color.rSrUs rVget_background_stylez SyntaxTheme.get_background_stylerXrYN) __name__ __module__ __qualname____doc__r TokenTyperrWr\rYrVrOrO{s@("i"E"""e""rYrOcLeZdZdZdeeeefddfdZde de fdZ de fdZ y) PygmentsSyntaxThemez.Syntax theme that delegates to Pygments theme.themerQNci|_t|tr t||_n||_|jj |_t|j|_ y#t $rtd|_YRwxYw)Ndefaultbgcolor) _style_cache isinstancestrr_pygments_style_classr background_color_background_colorr_background_style)rUres rV__init__zPygmentsSyntaxTheme.__init__sx46 eS ! J->u-E**/D &!%!;!;!L!L!&t/E/E!F ! J->y-I* JsA((BBrPc V |j|S#t$r |jj|}|d}|d}t |rd|znd|rd|zn |j |d|d|d}n##t$rt j }YnwxYw||j|<Y|SwxYw) z"Get a style from a Pygments class.r8ri#z#000000r@italicr?)r8rir@rtr?)rjKeyErrorrmstyle_for_tokenrronull)rUrPpygments_styler8ristyles rVrWz'PygmentsSyntaxTheme.get_style_for_tokens 2$$Z0 0 2 !%!;!;!K!KJ!W'w/(3).#+I-4C'M$:P:P'/)(3,[9   %  %-2D  j ) ! 2s1 B(A2.line_tokenizes\ L5-2-=-=d-CD) E#:?//$:O7J%",j8.C"CC$DsA AAc3<Kt }d}rdz nd}||kr1 t|\}}|df|jdr|dz }||kr1|D]2\}}||f|jds$|dz }s,|k\s2yy#t$rYCwxYww)zConvert tokens to spans.rr$Nr)iternext StopIterationendswith) tokensline_no _line_start _token_typerrP_get_theme_styleline_end line_startrs rVtokens_to_spansz)Syntax.highlight..tokens_to_spanss!-/2FG4>*q.AK"K/"15f.K %dm+ >>$/#qLG"K/.4&) E$&6z&BCC >>$/#qLG'Gx,? % & -"!"s?!BB !B%B;BB B BBBBc38K|]\}}||fywr{rb).0rPrrs rV z#Syntax.highlight.. s(#) E,Z89#szon )rtransparent_backgroundr3rrrrWrrappendrrrrlr r append_tokensrrnstylizer_apply_stylized_ranges) rUrr base_stylerrrrrrrrs ` @@@@@rV highlightzSyntax.highlightsL))+ #::I ]]&    ;;:: 0d00 = KK (2$ HDxc3h'@D&%Xe_8L2M)N&&.""?#45""#-2-=-=d-C#$$0 s4#8#8"9:;   ' ' - rYryrrcP|jjt|||y)a Adds a custom style on a part of the code, that will be applied to the syntax display when it's rendered. Line numbers are 1-based, while column indexes are 0-based. Args: style (StyleType): The style to apply. start (Tuple[int, int]): The start of the range, in the form `[line number, column index]`. end (Tuple[int, int]): The end of the range, in the form `[line number, column index]`. N)rrr)rUryrrs rV stylize_rangezSyntax.stylize_ranges" $$%:5%%MNrYblendc|jj|jz}|j}| |jrt j S|jtj}| |jr|xst j St|j|j|}t j|S)N) cross_fade) rr\rriis_system_definedr(rgrrr3r) get_truecolor from_triplet)rUrrrnforeground_color new_colors rV_get_line_numbers_colorzSyntax._get_line_numbers_color's;;;;=@U@UU+33  #'7'I'I==? "00<  #'7'I'I#6u}} 6  * * ,  * * , !!),,rYcd}|jrAtt|j|jj dzt z}|S)z?Get the number of characters used to render the numbers column.rr)rlenrlrrcountNUMBERS_COLUMN_DEFAULT_PADDING)rU column_widths rV_numbers_column_widthzSyntax._numbers_column_width6sJ   C$))//$*??@A01 rYconsolec |j}|jr3tjtdtjfS|jdvrtj ||j jtjt|j|j}tj ||j jtjtd|jd|j}n|tdz}|tdz}|||fS)z>Get background, number, and highlight styles for line numbers.Tr9)256 truecolorr7g?)r@r8F) rrrrw color_systemchainrrWrr3rr)rUrr number_stylehighlight_number_styles rV_get_number_styleszSyntax._get_number_stylesAs//1  2 2::<4%**,> >   #7 7 ;;  // ;D88:;%% L &+[[  // ;4t'C'CC'HI%% & ",eo=L%5%8H%H "/EEErYoptionsctj|j\}}}}||z}|j5|j|jz|zdz}t |j|S|j j}|j|z|rtd|Dndz}|jr|dz }t |j|S)Nr$c32K|]}t|ywr{r&)rlines rVrz*Syntax.__rich_measure__..ds4d8D>4sr) r"unpackrrrr/r splitlinesmaxr) rUrrrrightrrwidthliness rV__rich_measure__zSyntax.__rich_measure__Xs$NN4<<85!T, ?? &OOd&@&@@7JQNEt995A A $$&  & & 8=s4e441 F     QJE455u==rYc#Kt|j||}|jr3t||jj |jy|yw)Nrypad)r1 _get_syntaxrr"rr\)rUrrsegmentss rV__rich_console__zSyntax.__rich_console__jsRD,,Wg>? << @ @ B  NsA A"c#| K|jj}|j4|jr|j|j z dz n|jn |j}|j |j\}}|j||j}|js/|js"|js|s|jd|j|jjtztdz|j z}|j"r0|j$s$|j'|j(|}d|_|jr,|j-||j/| Ed{y|j1||j/|dd |j dd } | D] } | Ed{y|jxsd \} } d} | rt3d| dz } |j5d|}|jr| t7|kDry|| | }|j"r|j$s|j|jjtztdz|j z}t9dj;|j'|j(|tdzj5dd}|j }|j/|}|j<j>}t@}|d}|jBrdnd}|jE|\}}}tG||jH| zD]@\}}|jr(|j1||j/dd || }nMtK|j-|d}|jLr|g}n!|jO||j|| g}|jr|d|zdz|}tQ|D]y\}}|r`tS|jU|dz dz}||r!||td|||n|d||||n||Ed{|{,|D]}|Ed{|Cy777+7w)zc Get the Segments for the Syntax object, excluding any vertical/horizontal padding Nr$rTr9)rycrop)r)rr)rheightr)ryr new_lines)NNr) allow_blankF)rtz> u❱ )rrr)r rMrAr7z )+rrrr max_widthr _process_coderrrr remove_suffixrrWrrrr ascii_onlywith_indent_guidesroverflowrenderupdate render_linesr splitrr3joinr __contains__r0legacy_windowsr enumeraterlistradjust_line_lengthr%rlrjust) rUrrrr ends_on_nlprocessed_coderry syntax_lines syntax_linerend_line line_offsetrnumbers_column_widthrender_optionshighlight_line_Segmentr line_pointerrrrrr  wrapped_linesrwrapped_line_left_padfirst wrapped_line line_columns rVrzSyntax._get_syntaxus"&!5!5!7!N!N&$$""T%?%??!C&& &*%7%7 %B" N~~ndoo>  ""4($$&++11':;D/"''(  !!'*<*<..t}}E.J & ++">>'..z."B*  '33NND&NQ//" 4  $0+K***+ #>, H aa0K*.**Tz**R ??SZ'+h/E   g&8&8$$&++11':;D/"''( T e##DMMeAT9T#Ut.  $99 j9--::D>&55t6  # #G ,    "'udoo .KL* #MGT~~ ' 4 4"))v)F*22 !5!   G <=??%-JM!33$*44"2$: : 4%M  (0..46F)%,6m+D #'E<&)'l&8&89MPQ9Q&RUX&X )'2"*<U9K"LL"*;8N"OO"*41G"HH"*; "EE33+++"N #%2#L+++"N#Q* #m+^,,sJFR<R2AR<R5J4R< R8 R<#R:$R<5R<8R<:R<rc|j}dgtjd|tjDcgc]}|j dzc}t |dz}|j D]Q}t||j}t||j}|2|5|j|j||Sycc}w)z Apply stylized ranges to a text instance, using the given code to determine the right portion to apply the style to. Args: text (Text): Text instance to apply the style to. rr)flagsr$N) plainrefinditer MULTILINErrr#_get_code_index_for_syntax_positionrrry)rUrrmatchnewlines_offsetsstylized_rangerrs rVrzSyntax._apply_stylized_rangesszz    [[t2<<H !   IM  #33 ?N7 ."6"6E6 ."4"4C S_ ^115#> ?sCc|jd}|r|n|dz}|jrtj|n|}|j|j}||fS)a Applies various processing to a raw code string (normalises it so it always ends with a line return, dedents it if necessary, etc.) Args: code (str): The raw code string to process Returns: Tuple[bool, str]: the boolean indicates whether the raw code ends with a line return, while the string is the processed code. r)rrtextwrap expandtabsr)rUrr/r0s rVrzSyntax._process_codesY]]4( !+/3{{HOON + (224==A>))rYr{)g333333?)rr*rr+rQr/)2r]r^r_r`rr~rrO classmethodrrlr DEFAULT_THEMErboolintr rr r#rqrrrrrar(rpropertyrrr3rr2rrfloatrrr*rrr+r-rrr0rrrrbrYrVrrsA( ..  U3 #34    *7"DH.2$(*.#%&!$@$@UCZ $@ S+%& $@  $@$@$@U8C=(3-#?@A$@"#c(+$@SM$@$@$@#3-$@$@ #!$@" #$@L -1)6"04.2$(*.#%&!< < < eSj)* < S+%& <  < < U38_-< < "#c(+< SM< < < #3-< < #!< " #< < |'s'(3-'3''R 9 % x$ u  EISSU8C=(3-#?@AS  Sj O O'5 O >+;> >$  )7   D#D# D# '  D#L?4?D?<*#*%c *:*rYrrGpositionrQct|}|\}}||kDst||dzkry|dz }||dz||z dz }t||}|||zS)a Returns the index of the code string for the given positions. Args: newlines_offsets (Sequence[int]): The offset of each newline character found in the code snippet. position (SyntaxPosition): The position to search for. Returns: Optional[int]: The index of the code string for this position, or `None` if the given position's line number is out of range (if it's the column that is out of range we silently clamp its value so that it reaches the end of the line) r$N)rmin)rGrR lines_count line_number column_index line_index line_lengths rVrErE.sy&'K (K[ C(8$9[1_$MqJ":>25Ej5QQTUUK{L1L J ', 66rY__main__z&Render syntax to the console with Rich) descriptionrPATHzpath to file, or - for stdin)metavarhelpz-cz --force-color force_color store_truezforce color for non-terminals)destactionrgr^z-iz--indent-guidesrFzdisplay indent guidesz-lz--line-numbersrzrender line numbers)rarbr^z-wz--widthrz*width of output (default will auto-detect))typerargr^z-rz--wraprzword wrap long linesz-sz --soft-wrap soft_wrapzenable soft wrapping mode)rbrargr^z-tz--themerezpygments theme)rargr^z-bz--background-colorrnzOverride background colorz-xz--lexerrz Lexer name)rgrar^z-pz --paddingrr")rcrgrar^z--highlight-liner7z)The line number (not index!) to highlight)r*)force_terminalr-) rrrrrernrrr)rrrrernrrr)rd)os.pathrplatformrBsysrJabcrrpathlibrtypingrrrr r r r r rrrpygments.lexerrpygments.lexersrrpygments.stylerr~pygments.stylesrpygments.tokenrrrrrrrrrr pygments.utilr rich.containersr! rich.paddingr"r#_loopr%cellsr'r8r(r)rr*r+r,r-jupyterr.measurer/segmentr0r1ryr2rr3rlrasystemWINDOWSrMPreprocWordBuiltinFunction NamespaceClass Exception DecoratorVariableConstant AttributeTagDeletedInsertedHeading SubheadingPromptrErrJrrrOrdrrOrrrrEr]argparseArgumentParserparser add_argument parse_argsargs rich.consoler_rrstdinreadrrrrrernrrr7syntaxrprintrdrbrYrVrs #    !G1-   (!3#II! &# #s(O (// y ( & 57&G$& Ut_& OOU( &  U &  LL%f% & MM5y)& LL%f%& MM5w'& NNE$7& JJGt4& NNE'& NNE 5& MM5u%& MM5u%&  NNE'!&" HHe-(#&$ E ! E  OOU. e'* OOU% ID9 NNEt$ MM5|, 5u -5& DE! ":% 57%N+% Ut_% OOU/ %  U ' %  LL%m, % MM5/0% LL%m,% MM5~.% NNE >% JJNd;% NNE .% NNE 0t<% MM5|,% MM5|,%  NNE .!%" HHe-(#%$ E ! E & OOU. e.1 OOU% $44@ NNEt$ MM5|, 5u -5% 4 5 !:%/YG!" "# "'&+'&T&k&BsCx J H *\H *V7sm7/=7 c]76 z $X $ $<F  +     ,     $     "      9     #     (   igy?O    (        kQYY    8     D$T%5%5TZZHG yyCyy~~//**nn**!22,,LL!001  !! II//**nn**!22,,LL!001"   MM&DNNM3krY