U ;{j5|@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*d dl+m,Z,d dl-m.Z.m/Z/m0Z0m1Z1m2Z2d dl3m4Z4d dl5m6Z6m7Z7d dl8m9Z9d dl:m;Z;d dlm?Z?d d l@m"Z"d dlAmBZBejCdkZDdZEdZFdddddddeEeFdddd dd!ee.eeGeeGeGeeHeIeIeGeGeIeeIeIeeeHe feGeeeJeJee gefd"d#d$ZKeGd%d&d&ZLeGd'd(d(ZMeGd)d*d*ZNeGd+d,d,ZOGd-d.d.e6ZPGd/d0d0ZQeRd1kreKdd2ddlZedd3d4d5ZSedd3d6d7ZTdd8d9d:ZUeUdS);N) dataclassfield)islice)walk_tb) ModuleType TracebackType) AnyCallableDictIterableListOptionalSequenceTupleTypeUnion)guess_lexer_for_filename)CommentKeywordNameNumberOperatorString)Text)Token) ClassNotFound)pretty) loop_last)Columns)ConsoleConsoleOptionsConsoleRenderable RenderResultgroup) Constrain)RegexHighlighterReprHighlighter)Panel) render_scope)Style)Syntax)Themewin32 PdXFT)consolewidth code_width extra_linestheme word_wrap show_localslocals_max_lengthlocals_max_stringlocals_hide_dunderlocals_hide_sunder indent_guidessuppress max_frames)r4r5r6r7r8r9r:r;r<r=r>r?r@rAreturncs|dkrtddn| jr(dkr(dntttttdd fdd tddfdd }zt}||tj WSt k rtj }t_ |YSXdS) 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)stderr)type_value tracebackrBcs8 tj|||  t ddS)N r5r6r7r8r9r:r;r<r=r>r?r@rA)print Tracebackfrom_exceptionbool)rDrErF)r6r7r?r=r>r;r<rAr:r@r8traceback_consoler5r9r3;/tmp/pip-install-zsn1f9xv/pip/pip/_vendor/rich/traceback.py excepthookbs(zinstall..excepthook)iprBcsbijttddfdd }ddtttddfdd_|_fd d _dS) N)argskwargsrBcs|||dS)zEwrap the default ip.showtraceback to store info for ip._showtracebackNr3rPrQ)default_showtracebacktb_datar3rMipy_show_tracebackszCinstall..ipy_excepthook_closure..ipy_show_tracebackF) is_syntax)rPrVrQrBcst}|rdn|d}dd}d|r2dnd}t|D]}|dkrPqX|j}q@|d|d|idS)z2Internally called traceback from ip._showtracebackNZrunning_compiled_codeF tb_offsetrr)Z _get_exc_infogetrangetb_next)rVrPrQZ exc_tupletbZcompiledrX_)rNrOrTr3rMipy_display_tracebacks  zFinstall..ipy_excepthook_closure..ipy_display_tracebackcs|ddi|S)NrVTr3rR)r^r3rMs z9install..ipy_excepthook_closure..)Z showtracebackrrKZ_showtracebackZshowsyntaxerror)rOrU)rN)rSrOr^rTrMipy_excepthook_closure|sz'install..ipy_excepthook_closure) r Z is_jupyterr BaseExceptionr rrZ get_ipythonsysrN Exception)r4r5r6r7r8r9r:r;r<r=r>r?r@rAr`rOold_excepthookr3)r6rNr7r?r=r>r;r<rAr:r@r8rLr5r9rMinstall/s*+*)rec@steZdZUeed<eed<eed<dZeed<dZee ee j fed<dZ ee e eefe eeffed<dS) FramefilenamelinenonamelineNlocalslast_instruction)__name__ __module__ __qualname__str__annotations__intrkrlr r rNodermrr3r3r3rMrfs  rfc@s6eZdZUeed<eed<eed<eed<eed<dS) _SyntaxErroroffsetrgrkrhmsgN)rnrorprsrrrqr3r3r3rMrus ruc@sPeZdZUeed<eed<dZeeed<dZe ed<e e dZ e eed<dS) Stackexc_type exc_valueN syntax_errorFis_cause)default_factoryframes)rnrorprqrrr{r rur|rKrlistr~r rfr3r3r3rMrxs  rxc@seZdZUeeed<dS)TracestacksN)rnrorpr rxrrr3r3r3rMrs rc@seZdZdgZdS)PathHighlighterz(?P.*/)(?P.+)N)rnrorpZ highlightsr3r3r3rMrsrc@seZdZdZddddddZd%dd d dd d eed d d d dd eeee ee e ee e e e e e e e e e e efe dddZedd d dd d eed d d d dd eeeeeee ee e ee e e e e e e e e e e efe ddddZed eed d deeeeee e e e e ed ddZeeedddZeeedddZee e e dd d!Zeeed"d#d$Z dS)&rIaA 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. textpythonZcythonZpyrex)rjz.pyz.pxdz.pyxz.pxiNr0r1r2FTr3rG)tracer5r6r7r8r9r:r;r<r=r>r?r@rAc Cs|dkrHt\}}}|dks.|dks.|dkr6td|j||||d}||_||_||_||_t |pjd|_ ||_ ||_ | |_ ||_| |_| |_| |_g|_| D]Z}t|ts|jdk st|dtj|j}n|}tjtj|}|j|q|dkrtd|nd|_dS)Nz9Value for 'trace' required if not called in except: blockr:Z ansi_darkz+ must be a module with '__file__' attributer)rbexc_info ValueErrorextractrr5r6r7r+Z get_themer8r9r:r?r;r<r=r>r@ isinstancerq__file__AssertionErrorospathdirnamenormpathabspathappendmaxrA)selfrr5r6r7r8r9r:r;r<r=r>r?r@rAryrzrFZsuppress_entityrr3r3rM__init__sH zTraceback.__init__)ryrzrFr5r6r7r8r9r:r;r<r=r>r?r@rArBc 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. r:r;r<r=r>) r5r6r7r8r9r:r?r;r<r=r>r@rA)r)clsryrzrFr5r6r7r8r9r:r;r<r=r>r?r@rAZrich_tracebackr3r3rMrJ*s4- zTraceback.from_exceptionr) ryrzrFr:r;r<r=r>rBc sVg} d} ddlm} ttddd} t| |j| || d} t|trtt|j pPd|j pXd|j p`d|j phd |j d | _| | | jj}ttttfttttfd fd d }t|D]*\}}|jj}d}tjdkrF|jd}tt|j||d}|\}}}}|dk rF|dk rF|dk rF|dk rF||f||ff}|rt|dsttj |sttj!| |}|j"#ddrqt$|pd||jj%|rfdd||j"&Dnd|d}|||j"#ddr| jdd=qt'|dd}|r|j(}|}|j)}d} q$|j*}|rHt'|ddsH|j(}|}|j)}d} q$qHq$t+| 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)_objectrBcSs&z t|WStk r YdSXdS)z1Don't allow exceptions from __str__ to propagate.zN)rqrc)rr3r3rMsafe_strs z#Traceback.extract..safe_str)ryrzr|?rj)rvrgrhrkrw) iter_localsrBc3sRss|EdHdS|D]2\}}r2|dr2qrB|drBq||fVqdS)z-Extract locals from an iterator of key pairs.N__r] startswith)rkeyrE)r=r>r3rM get_localss  z%Traceback.extract..get_localsN)r2 rWrs z%Traceback.extract..)rgrhrirlrm_rich_traceback_guard __cause__T__suppress_context__)r),Zpip._vendor.richrrrqrxrnr SyntaxErrorrurvrgrhrrwr{rr~r robjectrf_code co_filenamerb version_infof_lastinextrZ co_positionsrrrisabsjoinf_localsrYrfco_nameitemsgetattr __class__ __traceback__ __context__r)rryrzrFr:r;r<r=r>rr|rrstackrr frame_summaryZline_norgrmZinstruction_indexZinstruction_positionZ start_lineZend_lineZ start_columnZ end_columnframecauserr3)r=r>r;r<rMrss           zTraceback.extract)r4optionsrBc cs|j}|}|j}t|t|t|t|tj|t |t t dd|t|tt dd|t |t j |t j |t j |tj|t|t|tj t ddddd}t}tt|jjD]$\}} | jrt|| d|ddd d } t| |j} || | VW5QRX| jdk r||.tt|| j|d dd |jd |jVW5QRXt| jd df|| jj Vn>| j!rt| jd df|| j!Vnt| jdfV|s| j"rt#dVqt#dVqdS)NT)Zdim)Zbold)r pygments.textpygments.stringpygments.functionpygments.numberz repr.indentzrepr.strz repr.bracez repr.numberzrepr.bool_truezrepr.bool_falsez repr.nonez scope.borderz scope.equalsz scope.keyzscope.key.specialF)Zinheritz7[traceback.title]Traceback [dim](most recent call last)ztraceback.border)rr)titlestyle border_styleexpandpaddingztraceback.border.syntax_error)rrrrr5z: 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: )$r8Zget_background_styleZget_style_for_tokenr, TextTokenrrrZFunctionrrr*rConstantZ Delimiterrr'rreversedrrr~r( _render_stackr%r5Z use_themer{_render_syntax_errorrassembleryrwrzr| from_markup) rr4rr8Zbackground_styleZ token_styleZtraceback_theme highlighterlastrZstack_renderabler3r3rM__rich_console__s       zTraceback.__rich_console__)r{rBccst}t}|jdkrVtj|jrVtjd|jdfdt|j dfdd}||V||j }d|_ t |jd t|}|d |||tjd d|d dd7}|VdS) Nz r:rrrrTrzbold underline u[traceback.offset]▲[/])r'rrgrrexistsrrrqrhrkrstripZno_wrapminrvlenZstylizer)rr{rpath_highlighterrZsyntax_error_textrvr3r3rMr_s(   zTraceback._render_syntax_error)rgcoderBcCstj|d}|sP|d}|dkr2|d|n|}|drPd|krPdSz|j|pht||j WSt k rYdSXdS)Nrz#!rr) rrsplitextindexrlowerLEXERSrYrrir)rrgrextZnew_line_index first_liner3r3rM _guess_lexervs zTraceback._guess_lexer)rrBc #st}j}ttddd}tttdfdd }d}jdkrbtjdt|j jd}d }t |j D] \}} |r||krd }qp|r|dk st t d t|d d ddVd }|dk} | j tfddjD} tj| j r(t j|t | j dddt| jdfd| jdfdd} n$t jd| jdfdt| jdfdd} | j dsf| sfdV| V| j dr|| EdHqp| spzf|| j } | sWqp| j | }t| ||d | jj| jjf| jhjjjd d }dVWn:tk r.}zt d|dfVW5d}~XYqpX| jdk rX| j\}}|jd||d d| j rvt!|f|| d d!n|VqpdS)"N)rgrBcSsdt|S)zRead files, and cache results on filename. Args: filename (str): Filename to read Returns: str: Contents of file rj)r linecachegetlines)rgr3r3rM read_codes z*Traceback._render_stack..read_code)rrBc3s&|jr"t|jdjjjdVdS)Nrl)rr?rr)rlr)r?r;r<)r)rr3rM render_localssz.Traceback._render_stack..render_localsrrWFTz ... z frames hidden ...centerztraceback.error)Zjustifyrc3s|]}|VqdS)Nr)rr)frame_filenamer3rM sz*Traceback._render_stack..rrrrz in rrzin rrj)r8Z line_numbersZ line_rangeZhighlight_linesr9r6r?dedentrztraceback.error_range)rstartendZ style_beforer)r)"rr8rqrfr r"rArZrr~ enumeraterrrganyr@rrrrrhrirrr+r7r9r6r?rcrmZ stylize_rangerlr)rrrr8rrZexclude_framesZexcludedZ frame_indexrfirstZ suppressedrrZ lexer_namesyntaxerrorrrr3)rrrMrs               zTraceback._render_stack)N)!rnrorp__doc__rLOCALS_MAX_LENGTHLOCALS_MAX_STRINGr rrsrqrKr rrr classmethodrrrarrJrr r!r#rr$rurrrxrr3r3r3rMrIs  5H NrI__main__r)arBcCsd}t||dS)Nr)rH)rZoner3r3rMbarsrcCs"d}ddddhdd}t|dS)NTz Paul AtreideszVladimir Harkonnenz Thufir Hawatz Duncan Idaho)NFT) charactersZ atomic_types)r)rrZzedr3r3rMfoo s r)rBcCs tddS)Nr)rr3r3r3rMrsr)VrrrrbZ dataclassesrr itertoolsrrFrtypesrrtypingrr r r r r rrrrZpip._vendor.pygments.lexersrZpip._vendor.pygments.tokenrrrrrrrrrZpip._vendor.pygments.utilrrjrZ_looprcolumnsrr4r r!r"r#r$Z constrainr%rr&r'Zpanelr(scoper)rr*rr+rr8r,platformWINDOWSrrrsrqrKrarerfrurxrrrIrnrrrr3r3r3rMs  0                  1