File: //usr/local/lib/python3.10/dist-packages/rich/__pycache__/traceback.cpython-310.pyc
o
.��fYs � ! @ s� d dl mZ d dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
d dl m Z m
Z
d dlmZmZmZmZmZmZmZmZmZmZ d dlmZ d dlm Z m Z m Z m Z m Z m!Z! d d lm"Z# d d
lm$Z$ d d l%m&Z& d d
l'm(Z( d dl)m*Z* d dl+m,Z, d dl-m.Z.m/Z/m0Z0m1Z1m2Z2 d dl3m4Z4 d dl5m6Z6m7Z7 d dl8m9Z9 d dl:m;Z; d dl<m=Z= d dl>m?Z? d d l@m"Z" d dlAmBZB e�C� dkZDdZEdZFddd dd d eEeFd dd d dd �
d!ee. d"eeG d#eGd$eeH d%eId&eId'eGd(eGd)eId*eeI d+eId,eeeHe f d-eGd.eeeJ eJee
gef f d/d0�ZKeG d1d2� d2��ZLeG d3d4� d4��ZMeG d5d6� d6��ZNeG d7d8� d8��ZOG d9d:� d:e6�ZPG d;d<� d<�ZQeRd=k�rrd d>l-m.Z. e.� Z-d dlZd?ed.dfd@dA�ZSd?ed.dfdBdC�ZTdFdDdE�ZUeU� dS dS )G� )�absolute_importN)� dataclass�field)�walk_tb)�
ModuleType�
TracebackType)
�Any�Callable�Dict�Iterable�List�Optional�Sequence�Tuple�Type�Union)�guess_lexer_for_filename)�Comment�Keyword�Name�Number�Operator�String)�Text)�Token)�
ClassNotFound� )�pretty)� loop_last)�Columns)�Console�ConsoleOptions�ConsoleRenderable� RenderResult�group)� Constrain)�RegexHighlighter�ReprHighlighter)�Panel)� render_scope)�Style)�Syntax)�Theme�Windows�
�P �d � FT� )
�console�width� extra_lines�theme� word_wrap� show_locals�locals_max_length�locals_max_string�locals_hide_dunder�locals_hide_sunder�
indent_guides�suppress�
max_framesr3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? �returnc
s� | du r t dd�n| � � jr�du rdn��dtt dtdtt ddf��������� �
� � �
f
dd � � d
tddf� fd d � }
z t� }|
|� tj W S t
y\ tj }� t_ | Y S w )
a" Install 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.
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� tracebackr@ c s6 �
� tj| ||� � � � ����t�����d�� d S )N� r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? )�print� Traceback�from_exception�bool)rB rC rD )
r5 r= r; r< r9 r: r? r8 r> r6 �traceback_consoler4 r7 r2 �9/usr/local/lib/python3.10/dist-packages/rich/traceback.py�
excepthooka s& ��zinstall.<locals>.excepthook�ipc sl i ��j � dtdtdd f� �fdd� }dd�dtdtdtdd f���fd d
����_|�_ �fd d ��_d S )
N�args�kwargsr@ c s |�� | i |�� dS )zEwrap the default ip.showtraceback to store info for ip._showtracebackNr2 �rN rO )�default_showtraceback�tb_datar2 rK �ipy_show_traceback~ s zCinstall.<locals>.ipy_excepthook_closure.<locals>.ipy_show_tracebackF)� is_syntaxrT c st �� � }| rdn|d }��dd�}��d|rdnd�}t|�D ] }|du r( n|j}q � |d |d |� i �dS )z2Internally called traceback from ip._showtracebackN� �running_compiled_codeF� tb_offsetr r )�
_get_exc_info�get�range�tb_next)rT rN rO � exc_tuple�tb�compiledrW �_)rL rM rR r2 rK �ipy_display_traceback� s zFinstall.<locals>.ipy_excepthook_closure.<locals>.ipy_display_tracebackc s � | ddi|��S )NrT Tr2 rP )r` r2 rK �<lambda>� s ���z9install.<locals>.ipy_excepthook_closure.<locals>.<lambda>)�
showtracebackr rI �_showtraceback�showsyntaxerror)rM rS )rL )rQ rM r` rR rK �ipy_excepthook_closurez s �����z'install.<locals>.ipy_excepthook_closure) r �
is_jupyterr �
BaseExceptionr
r r � get_ipython�sysrL � Exception)r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? re rM �old_excepthookr2 )rL r5 r= r; r< r9 r: r? r8 r> r6 rJ r4 r7 rK �install0 s4 )������&�) �rl c @ sL e Zd ZU eed< eed< eed< dZeed< dZee ee
j f ed< dS )�Frame�filename�lineno�name� �lineN�locals) �__name__�
__module__� __qualname__�str�__annotations__�intrr rs r
r
r �Noder2 r2 r2 rK rm � s
rm c @ s6 e Zd ZU eed< eed< eed< eed< eed< dS )� _SyntaxError�offsetrn rr ro �msgN)rt ru rv ry rx rw r2 r2 r2 rK r{ � s
r{ c @ sP e Zd ZU eed< eed< dZee ed< dZe ed< e
e d�Z e
e ed< dS ) �Stack�exc_type� exc_valueN� syntax_errorF�is_cause)�default_factory�frames)rt ru rv rw rx r� r
r{ r� rI r �listr� r rm r2 r2 r2 rK r~ � s
r~ c @ s e Zd ZU ee ed< dS )�Trace�stacksN)rt ru rv r r~ rx r2 r2 r2 rK r� � s
r� c @ s e Zd ZdgZdS )�PathHighlighterz(?P<dim>.*/)(?P<bold>.+)N)rt ru rv �
highlightsr2 r2 r2 rK r� � s
r� c " @ s� e Zd ZdZdddddd�Z d4dd dd
d
eed d
d d dd
� dee dee de dee
de de de de de de de de e
e
ef de fdd �Zedd dd
d
eed d
d d dd
� d ee d ed ee dee de dee
de de de de de de de de e
e
ef de d d f d!d"��Zed
eed d
d#�d ee d ed ee de de de de de d efd$d%��Zd&ed'ed efd(d)�Ze� d*e d efd+d,��Z ed-e
d.e
d e
fd/d0��Z e� d1e d efd2d3��Z dS )5rG a� A 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.
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.
�text�python�cython�pyrex)rq z.pyz.pxdz.pyxz.pxiNr0 r1 FTr2 rE �tracer4 r5 r6 r7 r8 r9 r: r; r< r= r> r? c C s |d u r$t �� \}}}|d u s|d u s|d u rtd��| j||||d�}|| _|| _|| _t�|p2d�| _ || _
|| _ | | _ || _
|| _| | _|
| _g | _| D ]-}t|t�sj|jd usbJ |�d���tj�|j�}n|}tj�tj�|��}| j�|� qO|
dkr�td|
�| _ d S d| _ d S )Nz9Value for 'trace' required if not called in except: block�r8 � ansi_darkz+ must be a module with '__file__' attributer � ) ri �exc_info�
ValueError�extractr� r4 r5 r+ � get_themer6 r7 r8 r= r9 r: r; r< r> �
isinstancerw �__file__�os�path�dirname�normpath�abspath�append�maxr? )�selfr� r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r r� rD �suppress_entityr� r2 r2 rK �__init__� s> ��
�"zTraceback.__init__r r� rD r@ c C s< | j ||||| |
| | d�}| |||||||
| |
| | ||d�
S )ae Create 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.
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.
�r8 r9 r: r; r< ) r4 r5 r6 r7 r8 r= r9 r: r; r< r> r? )r� )�clsr r� rD r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? �rich_tracebackr2 r2 rK rH $ s2 +� �zTraceback.from_exceptionr� c s� g } d}
ddl m} dtdtfdd�} t| |j�| |�|
d �}
t|t�r<t|j p*d|j
p.d
|j p2d|j p6d |j
d �|
_| �|
� |
jj}d
ttttf dttttf f� �fdd� }t|�D ]R\}}|jj}|r}|�d�s}tj�|�s}tj�| |�}|j � dd�r�qat |p�d
||jj |r���fdd�||j � � �D �ndd�}||� |j � dd�r�|
jdd�= qat!|dd�}|r�|j"}|}|j#}d}
q|j$}|r�t!|dd�s�|j"}|}|j#}d}
q t%| 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_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�_objectr@ c S s zt | �W S ty Y dS w )z1Don't allow exceptions from __str__ to propagate.z<exception str() failed>)rw rj )r� r2 r2 rK �safe_str� s
�z#Traceback.extract.<locals>.safe_strT)r r� r� �?rq )r| rn ro rr r} � iter_localsc 3 sT � � s �s | E dH dS | D ]\}}� r|� d�rq�r"|� d�r"q||fV qdS )z-Extract locals from an iterator of key pairs.N�__r_ ��
startswith)r� �keyrC )r; r< r2 rK �
get_locals� s �
�z%Traceback.extract.<locals>.get_locals�<�_rich_traceback_omitc s" i | ]
\}}|t j|� �d ��qS ))�
max_length�
max_string)r �traverse)�.0r� rC )r9 r: r2 rK �
<dictcomp>� s ���z%Traceback.extract.<locals>.<dictcomp>N)rn ro rp rs �_rich_traceback_guard� __cause__�__suppress_context__)r� )&�richr� r rw r~ rt r� � SyntaxErrorr{ r| rn ro r� r} r� r� r� r r �objectr �f_code� co_filenamer� r� r� �isabs�join�f_localsrY rm �co_name�items�getattr� __class__�
__traceback__� __context__r� )r� r r� rD r8 r9 r: r; r< r� r� r� r� �stackr� r� �
frame_summary�line_norn �frame�causer� r2 )r; r< r9 r: rK r� j s~ �
�
�� � � � �
zTraceback.extractr3 �optionsc c sR � | j }|�� }|j}ti d|t��d|t��d|t��d|tj��d|t ��d|t
�t dd� �d |t��d
|t�t dd � �d |t ��d
|t j
��d|t j
��d|t j
��d|tj��d|t��d|t��d|tj
�t dd� �dd�}t� }tt| jj��D ]�\}} | jr�t| �| �d|dddd�}
t|
| j�}
|�|�� |
V W d � n1 s�w Y | jd ur�|�|�� tt| � | j�|ddd| jd�| j�V W d � n1 s�w Y t � | j � d �d f|| jj ��V n | j!�rt � | j � d �d f|| j!��V n
t � | j � d f�V |�s&| j"�r t �#d �V q�t �#d �V q�d S ) Nr �
pygments.text�pygments.string�pygments.function�pygments.numberz repr.indentT)�dimzrepr.strz
repr.brace)�boldz repr.numberzrepr.bool_truezrepr.bool_falsez repr.nonez scope.borderz scope.equalsz scope.keyzscope.key.specialF)�inheritz7[traceback.title]Traceback [dim](most recent call last)ztraceback.border)r r )�title�style� border_style�expand�paddingz traceback.border.syntax_error)r� r� r� r� r4 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:
)$r6 �get_background_style�get_style_for_tokenr, � TextTokenr r r �Functionr r r* r �Constant� Delimiterr r' r �reversedr� r� r� r( �
_render_stackr% r4 � use_themer� �_render_syntax_errorr �assembler r} r� r� � from_markup) r� r3 r� r6 �background_style� token_style�traceback_theme� highlighter�lastr� �stack_renderabler2 r2 rK �__rich_console__� s� ����
����� �
�
�
�
������ �
�� �
������zTraceback.__rich_console__r� c c s� � t � }t� }|jdkr,tj�|j�r,tjd|j� �dfdt|j �dfdd�}||�V ||j
� � �}d|_ t
|jd t|��}|�d
||� |tjd d| d dd�7 }|V d S )
Nz<stdin>� r� ��:r� r� r� �r� Tr zbold underline�
u [traceback.offset]▲[/])r' r� rn r� r� �existsr r� rw ro rr �rstrip�no_wrap�minr| �len�stylizer� )r� r� r� �path_highlighterr� �syntax_error_textr| r2 r2 rK r� 5 s* �
�
�
z Traceback._render_syntax_errorrn �codec C s� t j�|�d }|s(|�d�}|dkr|d |� n|}|�d�r(d|�� v r(dS z
| j�|�p4t||�j W S t
y? Y dS w )N���r� z#!r� r� ) r� r� �splitext�indexr� �lower�LEXERSrY r rp r )r� rn r� �ext�new_line_index�
first_liner2 r2 rK � _guess_lexerL s
�zTraceback._guess_lexerr� c
# sV � t � }�j}dtdtfdd�}dtdtt f�fdd� }d }�jdkr4t�jd t|j ��jd �}d
}t
|j �D ]�\}} |rH||v rHd }q;|r`|d usPJ �t d t|�� d
�ddd�V d
}|dk}
| j � t
� fdd��jD ��} tj�| j �r�t j|t | j dd��dt| j�dfd| jdfdd�} nt jd| jdfdt| j�dfdd�} | j �d�s�|
s�d V | V | j �d�r�|| �E d H q;| �s(z1|| j �}
|
s�W q;��| j |
�}t|
||d | j�j | j�j f| jh�jd �jd
d �
}d V W n t�y } zt �d |� �df�V W Y d }~q;d }~ww | j �r%t |g|| ��d d!�n|V q;d S )"Nrn r@ c S s d� t�| ��S )z�Read files, and cache results on filename.
Args:
filename (str): Filename to read
Returns:
str: Contents of file
rq )r� � linecache�getlines)rn r2 r2 rK � read_codea s z*Traceback._render_stack.<locals>.read_coder� c 3 s, � | j rt| j d� j� j� jd�V d S d S )Nrs )r� r= r� r� )rs r) r= r9 r: )r� )r� r2 rK �
render_localsl s � ��z.Traceback._render_stack.<locals>.render_localsr rU FTz
... z frames hidden ...�centerztraceback.error)�justifyr� c 3 s � | ]}� � |�V qd S �Nr� )r� r� )�frame_filenamer2 rK � <genexpr>� s � z*Traceback._render_stack.<locals>.<genexpr>r� r� r� r� z in r� r� zin r� rq �X )r6 � line_numbers�
line_range�highlight_linesr7 �
code_widthr= �dedentr� r )r� ) r� r6 rw rm r r" r? rZ r� r� � enumerater rn �anyr> r� r� r� r� ro rp r� r r+ r5 r7 r= rj rs r )r� r� r� r6 r r �exclude_frames�excluded� frame_indexr� �first�
suppressedr� r� �
lexer_name�syntax�errorr2 )r
r� rK r� \ s� �
� � � �
��
���
�������zTraceback._render_stackr )!rt ru rv �__doc__r� �LOCALS_MAX_LENGTH�LOCALS_MAX_STRINGr
r� ry rw rI r r r r� � classmethodr r rg r rH r� r r! r# r� r$ r{ r� r r~ r� r2 r2 r2 rK rG � s
�
�������� �
� � �
��
�3������� �
� � �
����� �E������� �
� �|��
�NrG �__main__)r �ac C s d}t || � d S )Nr )rF )r! �oner2 r2 rK �bar� s r# c C s d}h d�dd�}t | � d S )NT> � Duncan Idaho� Thufir Hawat�
Paul Atreides�Vladimir Harkonnen)NFT)�
characters� atomic_types)r# )r! r� �zedr2 r2 rK �foo� s
� r+ c C s> zzt d� W W d S t Y W d S tjdd� Y d S )Nr Tr� )r+ � slfkjsldkfjr3 �print_exceptionr2 r2 r2 rK r � s r )r@ N)V�
__future__r r r� �platformri � dataclassesr r rD r �typesr r �typingr r r
r r r
r r r r �pygments.lexersr �pygments.tokenr r r r r r r r� r �
pygments.utilr rq r �_loopr �columnsr r3 r r! r" r# r$ � constrainr% r� r& r' �panelr( �scoper) r� r* r r+ r� r6 r, �system�WINDOWSr r ry rw rI rg rl rm r{ r~ r� r� rG rt r# r+ r r2 r2 r2 rK �<module> s� 0
�������� �
� � �
��
�
�