o Rh#@sddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZmZmZddlmZddlmZmZmZmZm Z m!Z!dd lm"Z#dd lm$Z$dd l%m&Z&d d l'm(Z(d dl)m*Z*m+Z+d dl,m-Z-d dl.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4d dl5m6Z6d dl7m8Z8m9Z9d dl:m;Z;d dlm?Z?d dl@mAZAmBZBd d lCm"Z"d dlDmEZEejFdkZGdZHdZIdeBdeBdeeeJeJeJffddZKdd d!d"dd#d#eHeId$dd$d%d d&d'ee/d(eeJd)eeJd*eJd+eeLd,eMd-eMd.eJd/eJd0eMd1eeMd2eMd3eeeLe fd4eJdeeeNeNee geffd5d6ZOeGd7d8d8ZPeGd9d:d:ZQeGd;d<d<ZReGd=d>d>ZSGd?d@d@e8ZTGdAdBdBZUeVdCkreOd$dDddlZdEeddfdFdGZWdEeddfdHdIZXdLdJdKZYeYdSdS)MN) dataclassfield)islice)walk_tb) ModuleType TracebackType) AnyCallableDictIterableListOptionalSequenceTupleTypeUnion)guess_lexer_for_filename)CommentKeywordNameNumberOperatorString)Text)Token) ClassNotFound)pretty)loop_first_last loop_last)Columns)ConsoleConsoleOptionsConsoleRenderableGroup RenderResultgroup) Constrain)RegexHighlighterReprHighlighter)Panel) render_scope)Style)SyntaxSyntaxPosition)Themewin32 Pstartendreturnc cs||\}}|\}}||kr|||fVdStt||dD]\}}}|r,||dfVq|r5|d|fVq|ddfVqdS)zYield start and end positions per line. Args: start: Start position. end: End position. Returns: Iterable of (LINE, COLUMN1, COLUMN2). rrN)rrange) r3r4line1column1line2column2firstlastline_nor?P/home/air/sanwanet/backup_V2/venv/lib/python3.10/site-packages/rich/traceback.py_iter_syntax_lines6s rAdXFTr?)consolewidth code_width extra_linestheme word_wrap show_localslocals_max_lengthlocals_max_stringlocals_hide_dunderlocals_hide_sunder indent_guidessuppress max_framesrErFrGrHrIrJrKrLrMrNrOrPrQrRcs|dur tddn| jrdurdndttdtdttddf fdd d tddffd d }z t}||tj WSt y]tj }t_ |YSw) aInstall a rich traceback handler. Once installed, any tracebacks will be printed with syntax highlighting and rich formatting. Args: console (Optional[Console], optional): Console to write exception to. Default uses internal Console instance. width (Optional[int], optional): Width (in characters) of traceback. Defaults to 100. code_width (Optional[int], optional): Code width (in characters) of traceback. Defaults to 88. extra_lines (int, optional): Extra lines of code. Defaults to 3. theme (Optional[str], optional): Pygments theme to use in traceback. Defaults to ``None`` which will pick a theme appropriate for the platform. word_wrap (bool, optional): Enable word wrapping of long lines. Defaults to False. show_locals (bool, optional): Enable display of local variables. Defaults to False. locals_max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to 10. locals_max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to 80. locals_hide_dunder (bool, optional): Hide locals prefixed with double underscore. Defaults to True. locals_hide_sunder (bool, optional): Hide locals prefixed with single underscore. Defaults to False. indent_guides (bool, optional): Enable indent guides in code and locals. Defaults to True. suppress (Sequence[Union[str, ModuleType]]): Optional sequence of modules or paths to exclude from traceback. Returns: Callable: The previous exception handler that was replaced. NT)stderrtype_value tracebackr5cs<tj|||  t d} |dS)N rFrGrHrIrJrKrLrMrNrOrPrQrR) Tracebackfrom_exceptionboolprint)rTrUrVexception_traceback)rGrHrPrNrOrLrMrRrKrQrItraceback_consolerFrJr?r@ excepthooks&zinstall..excepthookipc slijdtdtddffdd }dddtdtdtddffd d _|_fd d _dS) Nargskwargsr5cs||i|dS)zEwrap the default ip.showtraceback to store info for ip._showtracebackNr?r`ra)default_showtracebacktb_datar?r@ipy_show_tracebackszCinstall..ipy_excepthook_closure..ipy_show_tracebackF) is_syntaxrfcst}|rdn|d}dd}d|rdnd}t|D] }|dur(n|j}q |d|d|idS)z2Internally called traceback from ip._showtracebackNrunning_compiled_codeF tb_offsetrr) _get_exc_infogetr7tb_next)rfr`ra exc_tupletbcompiledri_)r^r_rdr?r@ipy_display_tracebacks  zFinstall..ipy_excepthook_closure..ipy_display_tracebackcs|ddi|S)NrfTr?rb)rqr?r@sz9install..ipy_excepthook_closure..) showtracebackrrZ_showtracebackshowsyntaxerror)r_re)r^)rcr_rqrdr@ipy_excepthook_closures z'install..ipy_excepthook_closure) r! is_jupyterr BaseExceptionr rr get_ipythonsysr^ Exception)rErFrGrHrIrJrKrLrMrNrOrPrQrRrvr_old_excepthookr?)rGr^rHrPrNrOrLrMrRrKrQrIr]rFrJr@installRs4+() r}c@steZdZUeed<eed<eed<dZeed<dZee ee j fed<dZ ee e eefe eeffed<dS) FramefilenamelinenonamelineNlocalslast_instruction)__name__ __module__ __qualname__str__annotations__intrrr r rNoderrr?r?r?r@r~s  ,r~c@sLeZdZUeed<eed<eed<eed<eed<eedZe eed<dS) _SyntaxErroroffsetrrrmsgdefault_factorynotesN) rrrrrrrlistrr r?r?r?r@rs rc@seZdZUeed<eed<dZeeed<dZe ed<e e dZ e eed<e e dZe eed <dZe ed <e e dZe d ed <dS) Stackexc_type exc_valueN syntax_errorFis_causerframesris_groupTrace exceptions)rrrrrrr rrrZrrrr r~rrrr?r?r?r@rs   rc@seZdZUeeed<dS)rstacksN)rrrr rrr?r?r?r@rs rc@seZdZdgZdS)PathHighlighterz(?P.*/)(?P.+)N)rrr highlightsr?r?r?r@rs rc$@seZdZdZddddddZ d6dd d dd d eed d d d dd deedee dee de dee de de de de de de de de e e efde fddZedd d dd d eed d d d dd deed ed!eedee dee de dee de de de de de de de de e e efde d"df"d#d$Zed eed d d%deed ed!eede de de de de d"efd&d'Zd(ed)ed"efd*d+Zed,ed"efd-d.Zed/e d0e d"e fd1d2Zed3ed"efd4d5Z dS)7rXaA Console renderable that renders a traceback. Args: trace (Trace, optional): A `Trace` object produced from `extract`. Defaults to None, which uses the last exception. width (Optional[int], optional): Number of characters used to traceback. Defaults to 100. code_width (Optional[int], optional): Number of code characters used to traceback. Defaults to 88. extra_lines (int, optional): Additional lines of code to render. Defaults to 3. theme (str, optional): Override pygments theme used in traceback. word_wrap (bool, optional): Enable word wrapping of long lines. Defaults to False. show_locals (bool, optional): Enable display of local variables. Defaults to False. indent_guides (bool, optional): Enable indent guides in code and locals. Defaults to True. locals_max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to 10. locals_max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to 80. locals_hide_dunder (bool, optional): Hide locals prefixed with double underscore. Defaults to True. locals_hide_sunder (bool, optional): Hide locals prefixed with single underscore. Defaults to False. suppress (Sequence[Union[str, ModuleType]]): Optional sequence of modules or paths to exclude from traceback. max_frames (int): Maximum number of frames to show in a traceback, 0 for no maximum. Defaults to 100. textpythoncythonpyrex)rz.pyz.pxdz.pyxz.pxiNrBrCrDFTr?rWtracerFrGrHrIrJrKrLrMrNrOrPrQrRc Cs"|dur$t\}}}|dus|dus|durtd|j||||d}||_||_||_||_t |p5d|_ ||_ ||_ | |_ ||_| |_| |_| |_g|_| D]-}t|tsm|jduseJ|dtj|j}n|}tjtj|}|j|qR|dkrtd||_dSd|_dS)Nz9Value for 'trace' required if not called in except: blockrK ansi_darkz+ must be a module with '__file__' attributer)rzexc_info ValueErrorextractrrFrGrHr- get_themerIrJrKrPrLrMrNrOrQ isinstancer__file__ospathdirnamenormpathabspathappendmaxrR)selfrrFrGrHrIrJrKrLrMrNrOrPrQrRrrrVsuppress_entityrr?r?r@__init__s@  "zTraceback.__init__rrrVr5c Cs>|j|||| | | | | d}|||||||| || | | | ||dS)aCreate a traceback from exception info Args: exc_type (Type[BaseException]): Exception type. exc_value (BaseException): Exception value. traceback (TracebackType): Python Traceback object. width (Optional[int], optional): Number of characters used to traceback. Defaults to 100. code_width (Optional[int], optional): Number of code characters used to traceback. Defaults to 88. extra_lines (int, optional): Additional lines of code to render. Defaults to 3. theme (str, optional): Override pygments theme used in traceback. word_wrap (bool, optional): Enable word wrapping of long lines. Defaults to False. show_locals (bool, optional): Enable display of local variables. Defaults to False. indent_guides (bool, optional): Enable indent guides in code and locals. Defaults to True. locals_max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to 10. locals_max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to 80. locals_hide_dunder (bool, optional): Hide locals prefixed with double underscore. Defaults to True. locals_hide_sunder (bool, optional): Hide locals prefixed with single underscore. Defaults to False. suppress (Iterable[Union[str, ModuleType]]): Optional sequence of modules or paths to exclude from traceback. max_frames (int): Maximum number of frames to show in a traceback, 0 for no maximum. Defaults to 100. Returns: Traceback: A Traceback instance that may be printed. rKrLrMrNrO) rFrGrHrIrJrKrPrLrMrNrOrQrR)r)clsrrrVrFrGrHrIrJrKrLrMrNrOrPrQrRrich_tracebackr?r?r@rYPs4- zTraceback.from_exceptionrc sg} d} ddlm} t|ddpg} dtdtfdd } t| |j| || | d }tjd krQt |t t frQd |_ |j D]}|j tjt|||j|d q;t |trnt|jp[d|jp_d|jpcd|jpgd|j| d|_| ||jj}dttttfdttttfffdd }t|D]\}}|j j!}d}tjd kr|j"d}t#t$|j %||d}|\}}}}|dur|dur|dur|dur||f||ff}|r|&dst'j()|st'j(*| |}|j+,ddrqt-|pd||j j.|rfdd||j+/Dnd|d}|||j+,ddr|jdd=qt|dd}|r4|j0}|}|j}d } q|j1}|rLt|ddsL|j0}|}|j}d} q t2| d}|S)aExtract traceback information. Args: exc_type (Type[BaseException]): Exception type. exc_value (BaseException): Exception value. traceback (TracebackType): Python Traceback object. show_locals (bool, optional): Enable display of local variables. Defaults to False. locals_max_length (int, optional): Maximum length of containers before abbreviating, or None for no abbreviation. Defaults to 10. locals_max_string (int, optional): Maximum length of string before truncating, or None to disable. Defaults to 80. locals_hide_dunder (bool, optional): Hide locals prefixed with double underscore. Defaults to True. locals_hide_sunder (bool, optional): Hide locals prefixed with single underscore. Defaults to False. Returns: Trace: A Trace instance which you can use to construct a `Traceback`. Fr) _IMPORT_CWD __notes__N_objectr5cSs zt|WStyYdSw)z1Don't allow exceptions from __str__ to propagate.z)rr{)rr?r?r@safe_strs   z#Traceback.extract..safe_strT)rrrr)rD )rKrLrNrO?r)rrrrrr iter_localsc3sTs s |EdHdS|D]\}}r|drqr"|dr"q||fVqdS)z-Extract locals from an iterator of key pairs.N__rp startswith)rkeyrU)rNrOr?r@ get_localss   z%Traceback.extract..get_localsrgr<_rich_traceback_omitcs6i|]\}}t|st|s|tj|dqS)) max_length max_string)inspect isfunctionisclassrtraverse).0rrU)rLrMr?r@ "sz%Traceback.extract..)rrrrr_rich_traceback_guard __cause____suppress_context__)r)3richrgetattrrrrrrz version_inforBaseExceptionGroupExceptionGrouprrrrXrtype __traceback__ SyntaxErrorrrrrrrrrr robjectrf_code co_filenamef_lastinextr co_positionsrrrisabsjoinf_localsrkr~co_nameitems __class__ __context__r)rrrrVrKrLrMrNrOrrrrrstack exceptionrr frame_summaryr>rrinstruction_indexinstruction_position start_lineend_line start_column end_columnframecauserr?)rNrOrLrMr@rs               zTraceback.extractrEoptionsc#sFj}||j}tid|td|td|td|tjd|t d|t t ddd |td |tt dd d |t d |t j d|t j d|t j d|tjd|td|td|tj t ddddttdtdtdtffdd ttjjD] \}}||VqdS)Nr pygments.textpygments.stringpygments.functionpygments.numberz repr.indentT)dimzrepr.strz repr.brace)boldz repr.numberzrepr.bool_truezrepr.bool_falsez repr.nonez scope.borderz scope.equalsz scope.keyzscope.key.specialF)inheritrr=r5c 3s|jr/t|ddddd}t|j} |VWdn1s*wY|jdurott|jdddjdjVWdn1sXwYt |j dd f|jj Vn|j rt |j dd f|j Vn t |j d fV|j D] }t d |Vq|jrt|jd D],\}}g}t|jD] \}}|||qd Vttt|d |ddjVq|s|jrtdVdStdVdSdS)Nz7[traceback.title]Traceback [dim](most recent call last)ztraceback.borderT)rr)titlestyle border_styleexpandpaddingztraceback.border.syntax_error)rrrrrFz: ztraceback.exc_type)z[NOTE] ztraceback.noterrzSub-exception #ztraceback.group.border)rrzI [i]The above exception was the direct cause of the following exception: zH [i]During handling of the above exception, another exception occurred: )rr* _render_stackr'rF use_themer_render_syntax_errorrassemblerrrrr enumeraterrrrr$r from_markup) rr=stack_renderablenotegroup_nogroup_exceptiongrouped_exceptions group_last group_stackbackground_stylerE highlighter render_stackrtraceback_themer?r@rks~           z0Traceback.__rich_console__..render_stack)rIget_background_styleget_style_for_tokenr/ TextTokenrrrFunctionrrr,rConstant Delimiterrr)r&rrZr%rreversedrr)rrErrI token_styler=rr?rr@__rich_console__Ls\      &CzTraceback.__rich_console__rccst}t}|jdkr,tj|jr,tjd|jdfdt|j dfdd}||V||j }d|_ t |jd t|}|d |||tjd d|d dd7}|VdS) Nz r:rrrrTrzbold underline u[traceback.offset]▲[/])r)rrrrexistsrrrrrrstripno_wrapminrlenstylizer)rrrpath_highlighterrsyntax_error_textrr?r?r@rs*    zTraceback._render_syntax_errorrcodecCstj|d}|s(|d}|dkr|d|n|}|dr(d|vr(dSz |j|p4t||j WSt y?YdSw)Nr6r z#!rr) rrsplitextindexrlowerLEXERSrkrrr)rrr)extnew_line_index first_liner?r?r@ _guess_lexers  zTraceback._guess_lexerrc #st}j}dtdttffdd }d}jdkr+tjdt|jjd}d}t |jD]G\}}|r@||vr@d}q2|rX|dusHJt d t|d d d d Vd}|dk} |j t fddj D} tj|j rt j|t |j dddt|jdfd|jdfdd} nt jd|jdfdt|jdfdd} |j ds| sdV| V|j dr||EdHq2| szz8t|j } d| } | sWq2|j | }t| ||d|jj|jjf|jhjjjdd }dVWnty}zt d|d fVWYd}~q2d}~ww|j durh|j \}}t!||D]B\}}}z$|dkr@| |d}t|t|"}|dkrMt| |d}Wn t#yYYq%w|j$d||f||fdq%|j%rwt&|g||dd n|Vq2dS)!Nrr5c3s,|jrt|jdjjjdVdSdS)Nr)rrPrr)rr+rPrLrM)r)rr?r@ render_localss z.Traceback._render_stack..render_localsrrgFTz ... z frames hidden ...centerztraceback.error)justifyrc3s|]}|VqdSNr)rr)frame_filenamer?r@ sz*Traceback._render_stack..rrrrz in rrzin rr)rI line_numbers line_rangehighlight_linesrJrGrPdedentr rr6ztraceback.error_range)rr3r4)r)'rrIr~r r#rRr7r%rrrranyrQrrr!rrrrr linecachegetlinesrr1r-rHrJrGrPr{rrAlstrip IndexError stylize_rangerr )rrr'rIr2exclude_framesexcluded frame_indexrr< suppressedr code_linesr) lexer_namesyntaxerrorr3r4r8r9r;rr?)r6rr@rs                    zTraceback._render_stackr5)!rrr__doc__r-LOCALS_MAX_LENGTHLOCALS_MAX_STRINGr rrrrZr rrr classmethodrrrxrrYrr!r"r%rr&rrr1rrr?r?r?r@rXs       5      H   3 frX__main__racCsd}t||dS)Nr)r[)rOoner?r?r@bar^srQcCsd}hddd}t|dS)NT> Duncan Idaho Thufir Hawat Paul AtreidesVladimir Harkonnen)NFT) characters atomic_types)rQ)rOrzedr?r?r@foods  rYcCs tddS)Nr)rYr?r?r?r@rIqs rI)r5N)Zrr=rrz dataclassesrr itertoolsrrVrtypesrrtypingrr r r r r rrrrpygments.lexersrpygments.tokenrrrrrrrrr pygments.utilrrr_looprrcolumnsr rEr!r"r#r$r%r& constrainr'rr(r)panelr*scoper+rr,rHr-r.rrIr/platformWINDOWSrKrLrrArrZrxr}r~rrrrrXrrQrYrIr?r?r?r@s  0                         b