o MEg_g@sUddlZddlZddlmZddlZddlmZmZddl m Z m Z m Z m Z mZmZmZmZmZddlmZmZddlmZddlmZddlmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%dd l&m'Z'd d l(m)Z)d d l*m+Z+m,Z,d d l-m.Z.m/Z/m0Z0m1Z1d dl2m3Z3d dl4m5Z5d dl6m7Z7d dl8mZd dl9m:Z:ee;dfZdZ?ie$ee%eddeeddej@eddeeddejedde"jAedde jBedde jCedde jDeddde jEeddde jFedde jGeddde jHedde jIedde jJedde jKed de#ed!de!eddejLed"dejMeddejNedd#ejOedddejPedd#ejed"deedddi ZQe ed?e]j^d@dAdBd=dCdDd?e]j^dEdFdGd=dHdIe]j^dJdKe_dLddMdNe]j^dOdPdQd=dCdRd?e]j^dSdTd=dUdCdVdWe]j^dXdYdZdd[d\e]j^d]d^d_dd`d\e]j^dadbdcdddedfe]`Zaddglbm.Z.e.eajceajddhZ-eajedikre[jfgZheXeheajieajjeajkeajleajmeajndjZoneXjpeajeeajjeajkeajleajmeajndkZoe-jqeoeajrdldSdS)mN)Lines)ABCabstractmethod) AnyDictIterableListOptionalSetTupleTypeUnion)get_lexer_by_nameguess_lexer_for_filename)Style)get_style_by_name) CommentErrorGenericKeywordNameNumberOperatorStringToken Whitespace) ClassNotFound) loop_first)Color blend_rgb)ConsoleConsoleOptions JustifyMethod RenderResult) JupyterMixin) Measurement)Segment)Text.WindowsmonokaiwhitecolorTdimcyanbluemagentagreen)r- underline)r-boldred bright_blueyellow bright_red)r5 ANSI_LIGHT bright_black bright_cyanbright_magenta bright_green ANSI_DARK) ansi_light ansi_darkc@s8eZdZdZededefddZedefddZdS) SyntaxThemezBase class for a syntax theme. token_typereturncCt)z'Get a style for a given Pygments token.NotImplementedError)selfrCrIe/var/www/eduai.edurigo.com/doc_train/edurigo_ai/Puru/venv/lib/python3.10/site-packages/rich/syntax.pyget_style_for_tokenjzSyntaxTheme.get_style_for_tokencCrE)zGet the background color.rFrHrIrIrJget_background_styleorLz SyntaxTheme.get_background_styleN) __name__ __module__ __qualname____doc__r TokenTyperrKrNrIrIrIrJrBgs rBc@sNeZdZdZdeeeefddfddZde de fdd Z de fd d Z dS) PygmentsSyntaxThemez.Syntax theme that delagates to Pygments theme.themerDNcCs^i|_t|trzt||_Wntytd|_Ynw||_|jj|_t|jd|_ dS)Ndefaultbgcolor) _style_cache isinstancestrr_pygments_style_classrbackground_color_background_colorr_background_style)rHrUrIrIrJ__init__xs   zPygmentsSyntaxTheme.__init__rCc Csz|j|WStyOz|j|}Wn ty"t}Yn%w|d}|d}t|r2d|nd|r9d|n|j|d|d|dd}||j|<Y|Sw) z"Get a style from a Pygments class.r-rX#z#000000r5italicr4)r-rXr5rbr4)rYKeyErrorr\style_for_tokenrnullr^)rHrCpygments_stylestyler-rXrIrIrJrKs(     z'PygmentsSyntaxTheme.get_style_for_tokencC|jSNr_rMrIrIrJrNz(PygmentsSyntaxTheme.get_background_style) rOrPrQrRr r[r PygmentsStyler`rSrrKrNrIrIrIrJrTus  rTc@sJeZdZdZdeeefddfddZdedefdd Zdefd d Z dS) ANSISyntaxThemez$Syntax theme to use standard colors. style_maprDNcCs$||_t|_t|_i|_dSri)rnrre_missing_styler_rY)rHrnrIrIrJr`s   zANSISyntaxTheme.__init__rCcCsnz|j|WSty6|jj}t|}|j}|r-||}|dur%|}n|dd}|s||j|<|YSw)zLook up style in the style map.N)rYrcrngettuplero)rHrC get_styletokenrg_stylerIrIrJrKs     z#ANSISyntaxTheme.get_style_for_tokencCrhrirjrMrIrIrJrNrkz$ANSISyntaxTheme.get_background_style) rOrPrQrRrrSrr`rKrNrIrIrIrJrms rmc@seZdZUdZeeed<eed<ede e efdefddZ e ddd d d d d dd dd d e de de e efde de dedeeeefdeeedeedede dee de dd fddZede ddd d d d d dd df de de de e efde de deeeefdedeeedeedede dee de ddfdd Zdefd!d"Zd#edeefd$d%Z d9d e deeeefdefd&d'Zd:d)edefd*d+Zedefd,d-Zd.edeeeeffd/d0Z . 1 2 3  4d;d5d6Z!d.ed2e"de#fd7d8Z$d S)<SyntaxaConstruct a Syntax object to render syntax highlighted code. Args: code (str): Code to highlight. lexer_name (str): Lexer to use (see https://pygments.org/docs/lexers/) theme (str, optional): Color theme, aka Pygments style (see https://pygments.org/docs/styles/#getting-a-list-of-available-styles). Defaults to "monokai". dedent (bool, optional): Enable stripping of initial whitespace. Defaults to False. line_numbers (bool, optional): Enable rendering of line numbers. Defaults to False. start_line (int, optional): Starting number for line numbers. Defaults to 1. line_range (Tuple[int, int], optional): If given should be a tuple of the start and end line to render. highlight_lines (Set[int]): A set of line numbers to highlight. code_width: Width of code to render (not including line numbers), or ``None`` to use all available width. tab_size (int, optional): Size of tabs. Defaults to 4. word_wrap (bool, optional): Enable word wrapping. background_color (str, optional): Optional background color, or None to use theme color. Defaults to None. indent_guides (bool, optional): Show indent guides. Defaults to False. r\_themenamerDcCs2t|tr|S|tvrtt|}|St|}|S)zGet a syntax theme instance.)rZrBRICH_SYNTAX_THEMESrmrT)clsrxrUrIrIrJ get_themes  zSyntax.get_themeFrN) rUdedent line_numbers start_line line_rangehighlight_lines code_widthtab_size word_wrapr] indent_guidescode lexer_namerUr}r~rrrrrrr]rc Cst||_||_||_||_||_||_|pt|_| |_| |_ | |_ | |_ | r+t | dnt |_ | |_|||_dS)NrW)rrr}r~rrsetrrrrr]rbackground_stylerr{rw)rHrrrUr}r~rrrrrrr]rrIrIrJr`s zSyntax.__init__zutf-8pathencodingcCst|d|d }|}Wdn1swYd}d}ztj|\}}|r8|d}t|}|j}Wn t yBYnw|durYzt ||j}Wn t yXYnw|||||||||| | | | | d S)a?Construct a Syntax object from a file. Args: path (str): Path to file to highlight. encoding (str): Encoding of file. theme (str, optional): Color theme, aka Pygments style (see https://pygments.org/docs/styles/#getting-a-list-of-available-styles). Defaults to "emacs". dedent (bool, optional): Enable stripping of initial whitespace. Defaults to True. line_numbers (bool, optional): Enable rendering of line numbers. Defaults to False. start_line (int, optional): Starting number for line numbers. Defaults to 1. line_range (Tuple[int, int], optional): If given should be a tuple of the start and end line to render. highlight_lines (Set[int]): A set of line numbers to highlight. code_width: Width of code to render (not including line numbers), or ``None`` to use all available width. tab_size (int, optional): Size of tabs. Defaults to 4. word_wrap (bool, optional): Enable word wrapping of code. background_color (str, optional): Optional background color, or None to use theme color. Defaults to None. indent_guides (bool, optional): Show indent guides. Defaults to False. Returns: [Syntax]: A Syntax object that may be printed to the console rt)rNrV.) rUr}r~rrrrrrr]r) openreadosrsplitextlstriplowerrrxrr)rzrrrUr}r~rrrrrrr]r code_filerlexerr_ext extensionrIrIrJ from_pathsH%   zSyntax.from_pathcCs|j|j}|S)zGet the base style.)rwrNr)rH default_stylerIrIrJ_get_base_styleNszSyntax._get_base_stylerCcCs|j|}|jS)zGet a color (if any) for the given token. Args: token_type (TokenType): A token type tuple from Pygments. Returns: Optional[Color]: Color from theme, or None for no color. )rwrKr-)rHrCrgrIrIrJ_get_token_colorSs zSyntax._get_token_colorcs |}|jr dnd}t|||j|j d}|jjz t|jdd|jdWnt y5| Y|Sw|rg|\dt t t tfffdd dt t tttfffd d }||n|fd d D|jdur|d|j|S)aHighlight code and return a Text instance. Args: code (str): Code to highlight. line_range(Tuple[int, int], optional): Optional line range to highlight. Returns: Text: A text instance containing highlighted syntax. rVleft)justifyrgrno_wrapFT)stripnlensurenltabsizerDc3s@D]\}}|r|d\}}}|||fV|s qdS)zSplit tokens to one per line. N) get_tokens partition)rCrt line_tokennew_line)rrrIrJ line_tokenizesz'Syntax.highlight..line_tokenizec3st}d}d}||kr(t|\}}|dfV|dr$|d7}||ks|D]\}}||fV|drE|d7}|krEdSq*dS)zConvert tokens to spans.rrNr)iternextendswith)tokensline_no _line_start _token_typertrC)_get_theme_styleline_end line_startrrIrJtokens_to_spanss$      z)Syntax.highlight..tokens_to_spansc3s |] \}}||fVqdSrirI).0rCrt)rrIrJ s  z#Syntax.highlight..Nzon )rtransparent_backgroundr(rrrwrKrrrappendrr rr[r r append_tokensrr]stylize)rHrr base_stylertextrrI)rrrrrrrJ highlight_s@   + ( zSyntax.highlight333333?blendcCsr|j|j}|j}|dus|jrtS|tj }|dus#|jr)|p(tSt | | |d}t |S)N) cross_fade) rwrNrrXis_system_definedrrVrrr(r get_truecolor from_triplet)rHrrr]foreground_color new_colorrIrIrJ_get_line_numbers_colors   zSyntax._get_line_numbers_colorcCs,d}|jrtt|j|jdd}|S)z?Get the number of characters used to render the numbers column.rr)r~lenr[rrcount)rH column_widthrIrIrJ_numbers_column_widthszSyntax._numbers_column_widthconsolec Cs|}|jrttddtfS|jdvrAt||jtj t| d|j }t||jtj td| dd|j }n|tdd}|tdd}|||fS)z>Get background, number, and highlight styles for line numbers.Tr.)256 truecolorr,g?)r5r-F) rrrre color_systemchainrwrKrr(rr)rHrr number_stylehighlight_number_stylerIrIrJ_get_number_styless&     zSyntax._get_number_stylesr!optionsr"r&cCs0|jdur|j|j}t|j|St|j|jSri)rrr& max_width)rHrrwidthrIrIrJ__rich_measure__s   zSyntax.__rich_measure__c csj|j}|jdur|jr|j|jdn|jn|j}d}|jr-|j\}}td|d}|j d}|r8|jn|jd} |j rEt | n| } | |j } || |j} ||\} } } |js|js|js|sl| d||jttdd|j}|jr|js| j|j |d} d| _|jr|j| |j|dd EdHdS|j| |j|dd |jddd }|D]}|EdHqdS| jd|d }|jr|||}|jr|js||jttdd|j}td |j|j |djddd }|j}|j|d}|j!j"}t#}|d |d | }|d}|j$rdnd}t%||j&|D]\}}|jr;|j||jdd| | d}nt'|j|dd}|j(rL|g}n |j)||j| | dg}|jrt*|D]@\}}|rt+|,|dd }||r||tddV||| Vn|d| V||| Vn|V|EdH|Vq`q$|D] }|EdH|Vqq$dS)NrrrTr.)rgcrop)r)r)rheight)rgpad new_lines) allow_blank z> u❱ )r)rgr)endrr6r,z )-rrrr~rrrmaxrrr}textwrap expandtabsrrrr remove_suffixrwrKrrrr ascii_onlywith_indent_guidesoverflowrenderupdate render_linessplitr(joinr __contains__r'legacy_windows enumeraterlistradjust_line_lengthrr[rjust) rHrrrr line_offsetrend_line ends_on_nlrrrrrrg syntax_lines syntax_linelinesnumbers_column_widthrender_optionshighlight_line_Segmentpaddingr line_pointerrline wrapped_linessegmentsfirst wrapped_line line_columnrIrIrJ__rich_console__s                         zSyntax.__rich_console__ri)r)rr!rr"rDr&)%rOrPrQrRr rl__annotations__rB classmethodr r[r{ DEFAULT_THEMEboolintr r r r`rrrrSrrr(rfloatrpropertyrr!rrr"r$rrIrIrIrJrvs         #       H  N rv__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)destactionrVrz-iz--indent-guidesrFzdisplay indent guidesz-lz--line-numbersr~zrender line numbers)rrrz-wz--widthrz*width of output (default will auto-detect))typerrVrz-rz--wraprzword wrap long linesz-sz --soft-wrap soft_wrapzenable soft wrapping mode)rrrVrz-tz--themerUzpygments theme)rrVrz-bz--background-colorr]zOveride background colorz-xz--lexerrVrz Lexer name)rVrr)r!)force_terminalr-)rrr~rrUr]r)r~rrUr]r)r)sos.pathrplatformrich.containersrrabcrrtypingrrrrr r r r r pygments.lexersrrpygments.stylerrlpygments.stylesrpygments.tokenrrrrrrrrrr pygments.utilr_looprr-rr rr!r"r#r$jupyterr%measurer&segmentr'rgrr(r[rSsystemWINDOWSrPreprocWordBuiltinFunction NamespaceClass Exception DecoratorVariableConstant AttributeTagDeletedInsertedHeading SubheadingPromptr:rr?ryrBrTrmrvrOargparsesysArgumentParserparser add_argumentr parse_argsargs rich.consolerrrstdinrrrr~rrUr]rsyntaxrprintrrIrIrIrJs  ,  0                                                         *! )