File: //usr/local/lib/python3.7/site-packages/pip/_vendor/rich/__pycache__/traceback.cpython-37.pyc
B
L��g�s � @ 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 �
ee. eeG eGeeH eIeIeGeGeIeeI eIeeeHe f eGeeeJ eJee
gef d!�d"d#�ZKeG d$d%� d%��ZLeG d&d'� d'��ZMeG d(d)� d)��ZNeG d*d+� d+��ZOG d,d-� d-e6�ZPG d.d/� d/�ZQeRd0k�r�d d1l-m.Z. e.� Z-d dlZedd2�d3d4�ZSedd2�d5d6�ZTdd7�d8d9�ZUeU� dS ):� )�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_frames)r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? �returnc
s� | dkrt dd�n| � � jr(�dkr(dn��tt ttt dd���������� �
� � �
f
dd� � tdd�� fdd � }
yt� }|
|� tj S t
k
r� tj }� t_ |S X dS )
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 �>/tmp/pip-unpacked-wheel-hv55ucu3/pip/_vendor/rich/traceback.py�
excepthooka s" zinstall.<locals>.excepthook)�ipr@ c sb i ��j � ttd d�� �fdd� }dd�tttd d����fdd����_|�_ �fd d
��_d S ) N)�args�kwargsr@ c s |�� | |� dS )zEwrap the default ip.showtraceback to store info for ip._showtracebackNr2 )rM rN )�default_showtraceback�tb_datar2 rJ �ipy_show_traceback~ s zCinstall.<locals>.ipy_excepthook_closure.<locals>.ipy_show_tracebackF)� is_syntax)rM rR rN r@ c sv �� � }| rdn|d }��dd�}��d|r2dnd�}x t|�D ]}|dkrPP |j}qBW � |d |d |� i �dS )z2Internally called traceback from ip._showtracebackN� Zrunning_compiled_codeF� tb_offsetr r )Z
_get_exc_info�get�range�tb_next)rR rM rN Z exc_tuple�tbZcompiledrT �_)rK rL rP r2 rJ �ipy_display_traceback� s
zFinstall.<locals>.ipy_excepthook_closure.<locals>.ipy_display_tracebackc s � | ddi|��S )NrR Tr2 )rM rN )rZ r2 rJ �<lambda>� s z9install.<locals>.ipy_excepthook_closure.<locals>.<lambda>)Z
showtracebackr rH Z_showtracebackZshowsyntaxerror)rL rQ )rK )rO rL rZ rP rJ �ipy_excepthook_closurez s z'install.<locals>.ipy_excepthook_closure) r Z
is_jupyterr �
BaseExceptionr
r r Z get_ipython�sysrK � Exception)r3 r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? r\ rL Zold_excepthookr2 )rK r5 r= r; r< r9 r: r? r8 r> r6 rI r4 r7 rJ �install0 s )*)r` 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__�intrf rg r
r
r �Noder2 r2 r2 rJ ra � s
ra c @ s6 e Zd ZU eed< eed< eed< eed< eed< dS )� _SyntaxError�offsetrb rf rc �msgN)rh ri rj rm rl rk r2 r2 r2 rJ ro � s
ro 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)rh ri rj rk rl ru r
ro rv rH r �listrx r ra r2 r2 r2 rJ rr � s
rr c @ s e Zd ZU ee ed< dS )�Trace�stacksN)rh ri rj r rr rl r2 r2 r2 rJ rz � s
rz c @ s e Zd ZdgZdS )�PathHighlighterz(?P<dim>.*/)(?P<bold>.+)N)rh ri rj Z
highlightsr2 r2 r2 rJ r| � s r| c @ sp e Zd ZdZdddddd�Zd$dd dd
d
eed d
d d dd
� ee ee e ee
e e e e e e e e e
e
ef e d�
dd�Zedd dd
d
eed d
d d dd
� ee eee ee e ee
e e e e e e e e e
e
ef e d d�dd��Zed
eed d
d�ee eee e e e e e ed� dd��Zeeed�dd�Ze� e ed�d d ��Z ee
e
e
d �d d ��Z e� e ed!�d"d#��Z dS )%rF 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�pythonZcythonZpyrex)re z.pyz.pxdz.pyxz.pxiNr0 r1 FTr2 ) r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? )
�tracer4 r5 r6 r7 r8 r9 r: r; r< r= r> r? c C s |d krHt �� \}}}|d ks.|d ks.|d kr6td��| j||||d�}|| _|| _|| _t�|pdd�| _ || _
|| _ | | _ || _
|| _| | _|
| _g | _xb| D ]Z}t|t�s�|jd k s�t|�d���tj�|j�}n|}tj�tj�|��}| j�|� q�W |
dk�rt d|
�nd| _ d S )Nz9Value for 'trace' required if not called in except: block)r8 Z ansi_darkz+ must be a module with '__file__' attributer � ) r^ �exc_info�
ValueError�extractr r4 r5 r+ Z get_themer6 r7 r8 r= r9 r: r; r< r> �
isinstancerk �__file__�AssertionError�os�path�dirname�normpath�abspath�append�maxr? )�selfr r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? rs rt rD Zsuppress_entityr� r2 r2 rJ �__init__� s8
zTraceback.__init__)rs rt rD r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? 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� )�clsrs rt rD r4 r5 r6 r7 r8 r9 r: r; r< r= r> r? Zrich_tracebackr2 r2 rJ rG $ s. +zTraceback.from_exception)r8 r9 r: r; r< ) rs rt rD r8 r9 r: r; r< r@ c s� g } d}
ddl m} ttd�dd�} �x�t| |j�| |�|
d�}
t|t�rxt|j pTd|j
p\d|j pdd|j pld |j
d
�|
_| �|
� |
jj}ttttf ttttf d �� �fd d
� }x�t|�D ]�\}}|jj}|r�|�d�s�tj�|�s�tj�| |�}|j � dd��r q�t |�pd||jj |�rB��fdd�||j � � �D �ndd�}||� |j � dd�r�|
jdd�= q�W t!|dd�}|�r�|j"}|}|j#}d}
q(|j$}|�r�t!|dd��s�|j"}|}|j#}d}
q(P q(W 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" yt | �S tk
r dS X dS )z1Don't allow exceptions from __str__ to propagate.z<exception str() failed>N)rk r_ )r� r2 r2 rJ �safe_str� s z#Traceback.extract.<locals>.safe_str)rs rt rv �?re )rp rb rc rf rq )� iter_localsr@ c 3 sV � s�s| E dH dS x:| D ]2\}}� r4|� d�r4q �rD|� d�rDq ||fV q W dS )z-Extract locals from an iterator of key pairs.N�__rY )�
startswith)r� �keyrC )r; r< r2 rJ �
get_locals� s
z%Traceback.extract.<locals>.get_locals�<Z_rich_traceback_omitc s" i | ]\}}t j|� �d �|�qS ))�
max_length�
max_string)r Ztraverse)�.0r� rC )r9 r: r2 rJ �
<dictcomp>� s z%Traceback.extract.<locals>.<dictcomp>N)rb rc rd rg �_rich_traceback_guard� __cause__T�__suppress_context__)r{ )&Zpip._vendor.richr� r rk rr rh r� � SyntaxErrorro rp rb rc r} rq ru r� rx r r �objectr �f_code� co_filenamer� r� r� �isabs�join�f_localsrU ra �co_name�items�getattr� __class__�
__traceback__� __context__rz )r� rs rt rD r8 r9 r: r; r< r{ rv r� r� �stackr� r� �
frame_summaryZline_norb �frame�causer r2 )r; r< r9 r: rJ r� j sj
$
zTraceback.extract)r3 �optionsr@ c c s� | j }|�� }|j}t|t�|t�|t�|tj�|t �|t
�t dd� |t�|t�t dd� |t �|t j
�|t j
�|t j
�|tj�|t�|t�|tj
�t dd� d�dd�}t� }�x:tt| jj��D �]$\}} | j�rt| �| �d|ddd d
�}
t|
| j�}
|�|�� |
V W d Q R X | jd k �r�|�|��. tt| � | j�|d dd | jd �| j�V W d Q R X t � | j � d
�df|| jj ��V n>| j!�r�t � | j � d
�df|| j!��V nt � | j � df�V |s�| j"�r�t �#d�V q�t �#d�V q�W d S )NT)Zdim)Zbold)r z
pygments.textzpygments.stringzpygments.functionzpygments.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)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 Zget_background_styleZget_style_for_tokenr, � TextTokenr r r ZFunctionr r r* r ZConstantZ Delimiterr r' r �reversedr r{ rx r( �
_render_stackr% r4 Z use_themeru �_render_syntax_errorr �assemblers rq rt rv � from_markup) r� r3 r� r6 Zbackground_styleZ token_styleZtraceback_theme� highlighter�lastr� Zstack_renderabler2 r2 rJ �__rich_console__� sv
zTraceback.__rich_console__)ru r@ c c s� t � }t� }|jdkrVtj�|j�rVtjd|j� �dfdt|j �dfdd�}||�V ||j
� � �}d|_ t
|jd t|��}|�d
||� |tjd d| d dd�7 }|V d S )
Nz<stdin>� zpygments.string)�:z
pygments.textzpygments.numberz
pygments.text)r� Tr zbold underline�
u [traceback.offset]▲[/])r' r| rb r� r� �existsr r� rk rc rf �rstripZno_wrap�minrp �lenZstylizer� )r� ru r� �path_highlighterr} Zsyntax_error_textrp r2 r2 rJ r� 5 s$
z Traceback._render_syntax_error)rb �coder@ c C s� t j�|�d }|sP|�d�}|dkr2|d |� n|}|�d�rPd|�� krPdS y| j�|�pht||�j S t
k
r| dS X d S )N���r� z#!r~ r} ) r� r� �splitext�indexr� �lower�LEXERSrU r rd r )r� rb r� �extZnew_line_index�
first_liner2 r2 rJ � _guess_lexerL s
zTraceback._guess_lexer)r� r@ c
# s\ t � }�j}ttd�dd�}ttt d��fdd� }d }�jdkrbt�jd t|j ��jd �}d }�x�t
|j �D �]�\}} |r�||kr�d
}qt|r�|d k s�t �t d t|�� d �d
dd�V d }|dk}
| j
� t� fdd��jD ��} tj�| j
��r,t j|t | j
dd��dt| j�dfd| jdfdd�} n$t jd| jdfdt| j�dfdd�} | j
�d��sj|
�sjdV | V | j
�d��r�|| �E d H qt| styb|| j
�}
|
�s�wt��| j
|
�}t|
||d
| j�j | j�j f| jh�jd �jd d �
}dV W n: t k
�r. } zt �d |� �df�V W d d }~X Y qtX | j �rNt |f|| ��d d �n|V qtW d S )!N)rb 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
re )r� � linecache�getlines)rb r2 r2 rJ � read_codea s z*Traceback._render_stack.<locals>.read_code)r� r@ c 3 s& | j r"t| j d� j� j� jd�V d S )Nrg )r� r= r� r� )rg r) r= r9 r: )r� )r� r2 rJ �
render_localsl s z.Traceback._render_stack.<locals>.render_localsr rS FTz
... z frames hidden ...�centerztraceback.error)Zjustifyr� c 3 s | ]}� � |�V qd S )N)r� )r� r� )�frame_filenamer2 rJ � <genexpr>� s z*Traceback._render_stack.<locals>.<genexpr>zpygments.string)r� )r� z
pygments.textzpygments.numberz in zpygments.functionz
pygments.textzin r� re �X )r6 Z line_numbersZ
line_rangeZhighlight_linesr7 Z
code_widthr= �dedentr� r )r� ) r| r6 rk ra r r" r? rV r� rx � enumerater� r rb �anyr> r� r� r� r� rc rd r� r� r+ r5 r7 r= r_ rg r )r� r� r� r6 r� r� Zexclude_framesZexcludedZ frame_indexr� �firstZ
suppressedr} r� Z
lexer_name�syntax�errorr2 )r� r� rJ r� \ s�
$ zTraceback._render_stack)N)!rh ri rj �__doc__r� �LOCALS_MAX_LENGTH�LOCALS_MAX_STRINGr
rz rm rk rH r r r r� � classmethodr r r] r rG r� r r! r# r� r$ ro r� r� rr r� r2 r2 r2 rJ rF � s` D$@3 rLrF �__main__)r )�ar@ c C s d}t || � d S )Nr )rE )r� Zoner2 r2 rJ �bar� s r� c C s" d}ddddhdd�}t | � d S )NTz
Paul AtreideszVladimir Harkonnenz Thufir Hawatz Duncan Idaho)NFT)�
charactersZ atomic_types)r� )r� r� Zzedr2 r2 rJ �foo� s r� )r@ c C s@ y"y t d� W n t Y nX W n tjdd� Y nX d S )Nr T)r8 )r� Z slfkjsldkfjr3 �print_exceptionr2 r2 r2 rJ r� � s r� )V�
__future__r r� r� �platformr^ Z dataclassesr r rD r �typesr r �typingr r r
r r r
r r r r Zpip._vendor.pygments.lexersr Zpip._vendor.pygments.tokenr r r r r r r r� r Zpip._vendor.pygments.utilr re r Z_loopr �columnsr r3 r r! r" r# r$ Z constrainr% r� r& r' Zpanelr( Zscoper) r� r* r� r+ r} r6 r, �system�WINDOWSr� r� rm rk rH r] r` ra ro rr rz r| rF rh r� r� r� r2 r2 r2 rJ �<module> sx 0
<$p