a )j@s ddlZddlZddlZddlZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZmZmZmZmZmZmZmZmZmZmZddlmZddlmZmZmZm Z m!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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6d dl7m8Z8d dl9m:Z:m;Z;d dlm?Z?d dl@mAZAd dlBmCZCmDZDd d lEm#Z#d dlFmGZGejHdkZIdZJdZKeDeDeeeLeLeLfdddZMdddd dd!d!eJeKdd"ddd"d#dd$ee0eeLeeLeLeeNeOeOeLeLeeLeOeeOee3eOeeeNe feLeeePePee gefd%d&d'ZQeGd(d)d)ZReGd*d+d+ZSeGd,d-d-ZTeGd.d/d/ZUGd0d1d1e:ZVGd2d3d3ZWeXd4kreQd"d5ddlZedd6d7d8ZYedd6d9d:ZZdd;dN) dataclassfield)islice)walk_tb) ModuleType TracebackType) AnyCallableDictIterableListOptionalSequenceSetTupleTypeUnion)guess_lexer_for_filename)CommentKeywordNameNumberOperatorString)Text)Token) ClassNotFound)pretty)loop_first_last loop_last)Columns)ConsoleConsoleOptionsConsoleRenderableOverflowMethodGroup RenderResultgroup) Constrain)RegexHighlighterReprHighlighter)Panel) render_scope)Style)SyntaxSyntaxPosition)Themewin32 P)startendreturnc csx|\}}|\}}||kr&|||fVnNtt||dD]:\}}}|rT||dfVq8|rf|d|fVq8|ddfVq8dS)zYield start and end positions per line. Args: start: Start position. end: End position. Returns: Iterable of (LINE, COLUMN1, COLUMN2). rrN)rrange) r5r6line1column1line2column2firstlastline_norA4E:\soccer\ml_env\Lib\site-packages\rich/traceback.py_iter_syntax_lines8s rCdXFTrA)consolewidth code_width extra_linestheme word_wrap show_localslocals_max_lengthlocals_max_stringlocals_max_depthlocals_hide_dunderlocals_hide_sunderlocals_overflow indent_guidessuppress max_frames)rGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVr7cs|durtddn|jr(dur(dntttttdd fdd tddfdd }zt}||tj WSt ytj }t_ |YS0dS) 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_max_depth (int, optional): Maximum depths of locals before truncating, or None to disable. Defaults to None. 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. locals_overflow (OverflowMethod, optional): How to handle overflowing locals, or None to disable. Defaults to None. 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 tracebackr7cs@tj|||  t d} |dS)NrHrIrJrKrLrMrNrOrPrQrRrSrTrUrV) Tracebackfrom_exceptionboolprint)rXrYrZexception_traceback)rIrJrTrQrRrPrNrOrSrVrMrUrKtraceback_consolerHrLrArB excepthooks*zinstall..excepthook)ipr7csbijttddfdd }ddtttddfdd_|_fd d _dS) N)argskwargsr7cs||i|dS)zEwrap the default ip.showtraceback to store info for ip._showtracebackNrArdre)default_showtracebacktb_datarArBipy_show_tracebackszCinstall..ipy_excepthook_closure..ipy_show_tracebackF) is_syntax)rdrjrer7cs~}|rdn|d}dd}d}|durB|r>dnd}t|D]}|durZqb|j}qJ|d|d|idS)z2Internally called traceback from ip._showtracebackNrunning_compiled_codeF tb_offsetrr) _get_exc_infogetr9tb_next)rjrdre exc_tupletbcompiledrm_)rbrcrhrArBipy_display_tracebacks    zFinstall..ipy_excepthook_closure..ipy_display_tracebackcs|ddi|S)NrjTrArf)rurArBs z9install..ipy_excepthook_closure..) showtracebackrr^_showtracebackshowsyntaxerror)rcri)rb)rgrcrurhrBipy_excepthook_closuresz'install..ipy_excepthook_closure) r" is_jupyterr BaseExceptionr rr get_ipythonsysrb Exception)rGrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrzrcold_excepthookrA)rIrbrJrTrQrRrPrNrOrSrVrMrUrKrarHrLrBinstallTs*/.+ rc@steZdZUeed<eed<eed<dZeed<dZee ee j fed<dZ ee e eefe eeffed<dS) FramefilenamelinenonamelineNlocalslast_instruction)__name__ __module__ __qualname__str__annotations__intrrr r rNoderrrArArArBrs  rc@sLeZdZUeed<eed<eed<eed<eed<eedZe eed<dS) _SyntaxErroroffsetrrrmsgdefault_factorynotesN) rrrrrrrlistrr rArArArBrs 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 rrr^rrrr rrrrrArArArBrs   rc@seZdZUeeed<dS)rstacksN)rrrr rrrArArArBrs rc@seZdZdgZdS)PathHighlighterz(?P.*/)(?P.+)N)rrr highlightsrArArArBrsrc@seZdZdZddddddZd&dd d dd d eedd d dd d ddeeee ee e ee e e e e ee e e ee e e ee efe dddZedd d dd d eedd d dd d ddeeeeeee ee e ee e e e e ee e e ee e e ee efe ddddZed eedd d ddeeeeee e e ee e e eeeed ddZeeedddZeeedddZee e e d d!d"Z ee!ed#d$d%Z"dS)'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. 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_max_depth (int, optional): Maximum depths of locals before truncating, or None to disable. Defaults to None. 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. locals_overflow (OverflowMethod, optional): How to handle overflowing locals, or None to disable. Defaults to None. 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.pxiNrDrErFFTrA)rHrIrJrKrLrMrNrOrPrQrRlocals_overlowrTrUrV)tracerHrIrJrKrLrMrNrOrPrQrRrrTrUrVcCs*|durHt\}}}|dus.|dus.|dur6td|j||||d}||_||_||_||_t |pjd|_ ||_ ||_ ||_ ||_| |_| |_| |_| |_| |_g|_|D]Z}t|ts|jdusJ|dtj|j}n|}tjtj|}|j|q|dkr td|nd|_dS)Nz9Value for 'trace' required if not called in except: blockrM ansi_darkz+ must be a module with '__file__' attributer) r~exc_info ValueErrorextractrrHrIrJr/ get_themerKrLrMrTrNrOrPrQrRrSrU isinstancer__file__ospathdirnamenormpathabspathappendmaxrV)selfrrHrIrJrKrLrMrNrOrPrQrRrrTrUrVrrrZsuppress_entityrrArArB__init__'sF zTraceback.__init__r[)rrrZrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVr7cCsD|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_depth (int, optional): Maximum depths of locals before truncating, or None to disable. Defaults to None. 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. locals_overflow (OverflowMethod, optional): How to handle overflowing locals, or None to disable. Defaults to None. 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. )rMrNrOrPrQrR)rHrIrJrKrLrMrTrNrOrPrQrRrrUrV)r)clsrrrZrHrIrJrKrLrMrNrOrPrQrRrSrTrUrVrich_tracebackrArArBr]`s:1 zTraceback.from_exception)rMrNrOrPrQrR_visited_exceptions) rrrZrMrNrOrPrQrRrr7c" sg} d} ddlm} t|ddp"g}| dur2tn| }ttddd}t||j||| |d }tj d krt |t t frd |_ |jD]@}||vrq|||jtjt|||j||d qt |trt|jpd|jpd |jpd|jpd|j|d|_| ||jj}tttt ftttt fdfdd }t!|D]2\}}|j"j#}d}tj d kr|j$d}t%t&|j"'||d}|\}}}}|dur|dur|dur|dur||f||ff}|r|(dst)j*+|st)j*,| |}|j-.ddrqJt/|p d ||j"j0|rNfdd||j-1Dnd|d}|||j-.ddrJ|jdd=qJ|st|dd} | dur| |ur| j2}| }| j}d } qF|j3} | durt|dds| j2}| }| j}d} qFqqFt4| d}!|!S)a,Extract 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_max_depth (int, optional): Maximum depths of locals before truncating, or None to disable. Defaults to None. 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)_objectr7cSs$z t|WStyYdS0dS)z1Don't allow exceptions from __str__ to propagate.zN)rr)rrArArBsafe_strs  z#Traceback.extract..safe_str)rrrr)rF T)rMrNrQrRr?r)rrrrrr) iter_localsr7c3sRss|EdHdS|D]2\}}r2|dr2qrB|drBq||fVqdS)z-Extract locals from an iterator of key pairs.N__rt startswith)rkeyrY)rQrRrArB get_locals s  z%Traceback.extract..get_localsrkr<_rich_traceback_omitc s8i|]0\}}t|st|s|tj|dqS)) max_length max_string max_depth)inspect isfunctionisclassrtraverse).0rrY)rPrNrOrArB Dsz%Traceback.extract..)rrrrr_rich_traceback_guard __cause____suppress_context__)r)5richrgetattrsetrrrrr~ version_inforBaseExceptionGroupExceptionGrouprraddrr\rtype __traceback__ SyntaxErrorrrrrrrrrr robjectrf_code co_filenamef_lastinextr co_positionsrrrisabsjoinf_localsrorco_nameitems __class__ __context__r)"rrrrZrMrNrOrPrQrRrrrrrgrouped_exceptionsrstack exceptionrr frame_summaryr@rrinstruction_indexinstruction_position start_lineend_line start_column end_columnframecauserrA)rQrRrPrNrOrBrs!               zTraceback.extract)rGoptionsr7c#sj}||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 dddddtttttdfdd ttjjD]\}}||VqdS) NT)dim)bold)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)inherit)rr?r7c 3s|jr\t|ddddd}t|j}|VWdn1sR0Y|jdur܈8tt|jdddjdjVWdn1s0Yt |j dd f|jj Vn>|j rt |j dd f|j Vnt |j d fV|j D]}t d |Vq |jrt|jd D]\\}}g}t|jD]\}}|||qdd Vttt|d |ddjVqN|s|jrtdVn tdVdS)Nz7[traceback.title]Traceback [dim](most recent call last)ztraceback.borderT)rr)titlestyle border_styleexpandpaddingztraceback.border.syntax_error)rrrr rHz: 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)rH use_themer_render_syntax_errorrassemblerrrrr enumeraterr rrr&r from_markup) rr?stack_renderablenotegroup_nogroup_exceptionr group_last group_stackbackground_stylerG highlighter render_stackrtraceback_themerArBrsv  $   $    z0Traceback.__rich_console__..render_stack)rKget_background_styleget_style_for_tokenr1 TextTokenrrrFunctionrrr.rConstant Delimiterrr+r(rr^r'r reversedrr)rrGrrK token_styler?rrArrB__rich_console__rs8"CzTraceback.__rich_console__)rr7ccst}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+rrrrexistsrr rrrrstripno_wrapminrlenstylizer)rrrpath_highlighterrsyntax_error_textrrArArBr s(   zTraceback._render_syntax_error)rcoder7cCstj|d}|sP|d}|dkr2|d|n|}|drPd|vrPdSz|j|pht||j WSt y~YdS0dS)Nr8r(z#!rr) rrsplitextindexrlowerLEXERSrorrr)rrr1extnew_line_index first_linerArArB _guess_lexers  zTraceback._guess_lexer)rr7c #st}j}tttdfdd }d}jdkrRtjdt|jjd}d}t |jD]\}}|r|||vr|d}q`|r|dusJt dt|d d d d Vd}|dk} |j t fd dj D} tj|j rt j|t |j dddt|jdfd|jdfdd} n$t jd|jdfdt|jdfdd} |j dsV| sVdV| V|j drz||EdHq`| s`zrt|j } d| } | sWq`|j | }t| ||d|jj|jjf|jhjjjdd }dVWn<ty,}z"t d|d fVWYd}~q`d}~00|j dur|j \}}t!||D]\}}}zH|dkr| |d}t|t|"}|dkrt| |d}Wnt#yYqNYn0|j$d||f||fdqN|j%rt&|g||ddn|Vq`dS) N)rr7c 3s.|jr*t|jdjjjjjdVdS)Nr)rrTrrroverflow)rr-rTrNrOrPrS)r)rrArB render_localssz.Traceback._render_stack..render_localsrrkFTz ... z frames hidden ...centerztraceback.error)justifyrc3s|]}|VqdS)Nr)rr)frame_filenamerArB (z*Traceback._render_stack..rr'r%rz in rrzin rr)rK line_numbers line_rangehighlight_linesrLrIrTdedentr(rr8ztraceback.error_range)rr5r6)r )'rrKrr r$rVr9r-rrrranyrUrrr)r rrrr linecachegetlinesrr9r/rJrLrIrTrrrClstrip IndexError stylize_rangerr!)rrr/rKr;exclude_framesexcluded frame_indexrr> suppressedr code_linesr1 lexer_namesyntaxerrorr5r6r:r;r=rrA)r>rrBr s                    zTraceback._render_stack)N)#rrr__doc__r5LOCALS_MAX_LENGTHLOCALS_MAX_STRINGr rrrr^r%r rrr classmethodrrr|rr]rrr"r#r'r#r(rr r9rr rArArArBr\s  9O C fr\__main__r)ar7cCsd}t||dS)Nr)r_)rXonerArArBbarsrZcCsd}hddd}t|dS)NT>z Paul Atreidesz Thufir HawatzVladimir Harkonnenz Duncan Idaho)NFT) charactersZ atomic_types)rZ)rXrZzedrArArBfoos  r\)r7cCs tddS)Nr)r\rArArArBrRsrR)\rrFrr~ dataclassesrr itertoolsrrZrtypesrrtypingrr r r r r rrrrrZpygments.lexersrZpygments.tokenrrrrrrrrrZ pygments.utilrrr_looprr columnsr!rGr"r#r$r%r&r'r( constrainr)rr*r+panelr,scoper-rr.rQr/r0rrKr1platformWINDOWSrTrUrrCrr^r|rrrrrrr\rrZr\rRrArArArBs  4       $