o MEgZ@sddlmZddlZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZmZmZmZddlmZddlmZmZmZmZmZmZdd lmZdd lmZd d l m!Z!d d l"m#Z#m$Z$d dl%m&Z&d dl'm(Z(m)Z)m*Z*m+Z+m,Z,d dl-m.Z.d dl/m0Z0m1Z1d dl2m3Z3d dl4m5Z5d dl6m7Z7d dl8m9Z9d d l:mZd dl;mdZ?dZ@dddddddddee(d eeAd!eAd"eeBd#eCd$eCd%eCd&eeeDeDee ge ffd'd(ZEeGd)d*d*ZFeGd+d,d,ZGeGd-d.d.ZHeGd/d0d0ZIGd1d2d2e0ZJGd3d4d4ZKeLd5krHd d6l'm(Z(e(Z'ddlZd7e d&dfd8d9ZMd7e d&dfd:d;ZNd>d.excepthookipc slijdtdtddffdd }dddtdtdtddffd d _|_fd d _dS) Nargskwargsr5cs||i|dS)zEwrap the default ip.showtraceback to store info for ip._showtracebackNr>rBrC)default_showtracebacktb_datar>r?ipy_show_traceback_szCinstall..ipy_excepthook_closure..ipy_show_tracebackF) is_syntaxrHcst}|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_infogetrangetb_next)rHrBrC exc_tupletbcompiledrK_)r@rArFr>r?ipy_display_tracebackes  zFinstall..ipy_excepthook_closure..ipy_display_tracebackcs|ddi|S)NrHTr>rD)rTr>r?sz9install..ipy_excepthook_closure..) showtracebackrbool_showtracebackshowsyntaxerror)rArG)r@)rErArTrFr?ipy_excepthook_closure[s z'install..ipy_excepthook_closure) rsysstderrr BaseExceptionr rr get_ipythonr@ Exception) r.r/r0r1r2r3r4rZrAold_excepthookr>)r@r0r4r3r1r=r/r2r?install)s() rac@sLeZdZUeed<eed<eed<dZeed<dZee ee j fed<dS)FramefilenamelinenonamelineNlocals) __name__ __module__ __qualname__str__annotations__intrgrhr r rNoder>r>r>r?rbs  rbc@s6eZdZUeed<eed<eed<eed<eed<dS) _SyntaxErroroffsetrcrgrdmsgN)rirjrkrnrmrlr>r>r>r?rps  rpc@sPeZdZUeed<eed<dZeeed<dZe ed<e e dZ e eed<dS) Stackexc_type exc_valueN syntax_errorFis_cause)default_factoryframes)rirjrkrlrmrvr rprwrWrlistryr rbr>r>r>r?rss  rsc@seZdZUeeed<dS)TracestacksN)rirjrkr rsrmr>r>r>r?r{s r{c@seZdZdgZdS)PathHighlighterz(?P.*/)(?P.+)N)rirjrk highlightsr>r>r>r?r}s r}c@sheZdZdZddddddZddd dd d d eef d eed ee de dee de de de de de fddZ e dd dd d d eefdeededeed ee de dee de de de de de ddfddZe d eefdeededeede de de defddZded edefd!d"Zed#edefd$d%Ze d&e d'e de fd(d)Zed*edefd+d,ZdS)-r;aA 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. 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. textpythoncythonpyrex)rfz.pyz.pxdz.pyxz.pxiNr,r-FTtracer/r0r1r2r3r4locals_max_lengthlocals_max_stringc Cs|dur$t\} } } | dus| dus| durtd|j| | | |d}||_||_||_t|p2d|_ ||_ ||_ ||_ ||_ | |_dS)Nz9Value for 'trace' required if not called in except: blockr3 ansi_dark)r[exc_info ValueErrorextractrr/r0r' get_themer1r2r3r4rr) selfrr/r0r1r2r3r4rrrtrur9r>r>r?__init__s$  zTraceback.__init__rtrur9r5c Cs,|j||||d} || |||||| | | d S)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. 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. Returns: Traceback: A Traceback instance that may be printed. r)r/r0r1r2r3r4rr)r) clsrtrur9r/r0r1r2r3r4rrrich_tracebackr>r>r?r<s"zTraceback.from_exceptioncsg}d}ddlm} dtdtfdd} t| |j| ||d } t|tr)rlr_)rr>r>r?safe_str;s   z#Traceback.extract..safe_strT)rtrurw?rf)rqrcrdrgrrr? ^sz%Traceback.extract..N)rcrdrerh __cause____suppress_context__)r|)"richrrrlrsri isinstance SyntaxErrorrprqrcrdrrrrvappendryrf_code co_filename startswithospathisabsjoinrbco_namef_localsitemsgetattr __traceback__ __class__ __context__r{)rrtrur9r3rrr|rwrrstackr frame_summaryline_norcframecauserr>rr?rsx           zTraceback.extractr.optionsc csN|j}|}|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 dd}t}tt|jjD]\}} | jrt|| d|dddd} t| |j} || | VWdn1swY| jdur||tt|| j|ddd|jd|jVWdn1swYt| jddf|| jj Vn| j!rt| jddf|| j!Vn t| jdfV|s$| j"rt#dVqt#dVqdS)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.specialz7[traceback.title]Traceback [dim](most recent call last)ztraceback.border.syntax_error)rr)titlestyle border_styleexpandpaddingztraceback.border)rrrrr/z: ztraceback.exc_typezI [i]The above exception was the direct cause of the following exception: zH [i]During handling of the above exception, another exception occurred: )$r1get_background_styleget_style_for_tokenr( TextTokenrrrFunctionrrr&rConstant Delimiterrr#rreversedrr|ryr$ _render_stackr!r/ use_themerv_render_syntax_errorrassemblertrrrurw from_markup) rr.rr1background_style token_styletraceback_theme highlighterlastrstack_renderabler>r>r?__rich_console__s             zTraceback.__rich_console__rvccst}t}|jdkr%tjd|jdfdt|jdfdd}||V||j}d|_ t |j d t |}| d ||d |tjd d|d dd7}|VdS) Nz r:rrrrTrzbold underline u[traceback.offset]▲[/])r#r}rcrrrlrdrgrstripno_wrapminrqlenstylizer)rrvrpath_highlighterrsyntax_error_textrqr>r>r?rs(    zTraceback._render_syntax_errorrccodecCsltj|d}|s(|d}|dkr|d|n|}|dr(d|vr(dS|j|p3t||j }|S)Nrz#!r) rrsplitextindexrlowerLEXERSrMrre)rrcrextnew_line_index first_line lexer_namer>r>r? _guess_lexers zTraceback._guess_lexerrc #s~t}j}idtdtffdd }dtdttffdd }t|jD]\}}tj |t|j dd d t|j d fd |j d fdd }|j dsO|sOdV|V|j dr`||EdHq&z-||j } |j | } t| | |d|j j|j jf|j hjdjdd } dVWnty} zt d| dfVWYd} ~ q&d} ~ ww|jrt| g||ddn| Vq&dS)Nrcr5csV|}|dur)t|dddd }|}Wdn1s wY||<|S)zRead files, and cache results on filename. Args: filename (str): Filename to read Returns: str: Contents of file Nrtzutf-8replace)encodingerrors)rMopenread)rcr code_file) code_cacher>r? read_codes  z*Traceback._render_stack..read_coderc3s,|jrt|jdjjjdVdSdS)Nrh)rr4rr)rhr%r4rr)r)rr>r? render_localss z.Traceback._render_stack..render_localsrrrrz in rrrrfTXF)r1 line_numbers line_rangehighlight_linesr2 code_widthr4dedentrztraceback.errorr)r)r}r1rlrbr rrryrrrcrdrerrr'r0r2r4r_rhr) rrrr1rrfirstrrrrsyntaxerrorr>)rrr?rsp         zTraceback._render_stack)rirjrk__doc__rLOCALS_MAX_LENGTHLOCALS_MAX_STRINGr r{rnrlrWr classmethodr rr]rr<rrrrrr rprrrsrr>r>r>r?r;s        0 h Mr;__main__)racCsd}t||dS)Nr)r:)roner>r>r?barVsrcCshddd}t|dS)N> Duncan Idaho Thufir Hawat Paul AtreidesVladimir Harkonnen)NFT) characters atomic_types)r)rzedr>r>r?fooZs r cCs>zztdWWdStYWdStjddYdS)NrTr)r  slfkjsldkfjr.print_exceptionr>r>r>r?rgs r)r5N)P __future__rrplatformr[ dataclassesrrr9rtypesrtypingrrr r r r r pygments.lexersrpygments.tokenrrrrrrrrrrfr_looprrcolumnsrr.rrrrr constrainr!rr"r#panelr$scoper%rr&rr'rr1r(systemWINDOWSrrrnrlrWr]rarbrprsr{r}r;rirr rr>r>r>r?s   $               g