B

�P�eN�@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZdd
lZdd
lZdd
lZdd
l	Z	dd
l
Z
ejdkrldd�Zn6yddl
mZ
Wnek
r�


dd�ZYn
Xdd�ZGdd	�d	�ZGd
d�d�Zd
S)�N)��c

Cs
t|t
�S)
N)�
isinstance�ExceptionGroup)
�exc�r�J/opt/alt/python37/lib/python3.7/site-packages/loguru/_better_exceptions.py�is_exception_groups
r	)
rc


Csd
S)NFr)
rrrrr	s
c

Cs
t|t
�S)
N)rr)
rrrrr	s
c@s�eZ
dZd
dddddddddd�
Zeee�
�
Zd	d
dhZdd
dddddddh	Z	e
jh
ZdZ
ejdkr�e�e
je
je
jh�

e
jZ
ddd�
Zdd�Zedd��
Z
dS)�SyntaxHighlighterz{}z{}z
{}z{}z{}z{}z{})
�comment�keyword�builtin�string�number�operator�punctuation�constant�
identifier�other�True�False�None�
(�
)�
[�
]�
{�
}�
:�
,�
;N)r�cCs|
p|j|_
dS)
N)�_default_style�_style)�self�stylerrr�__init__8s
zSyntaxHighlighter.__init__cCs�
|j}d
\}}d}�
x`|�
|
�
D�
]P}|\}}\}	}
\}}}
||jkr^||�d�
|�d�
7}|t
jkr�||jkr||d}n0t�|�
r�|d}n||jkr�|d}n|d}nn|t
j	kr�||j
kr�|d	}n|d
}nF|t
jkr�|d}n2||jkr�|d}n|t
j
k�
r|d
}n|d}|	|k�
r<|
|d�}
|	d}}|t
jk�
rn||
||
�7}||�|
|
|��
7}|}q W||
|d�7}|S)N)r
r
�rrrrr
rrrrrrrr
)r#�tokenize�_fstring_middle�count�NAME�
_constantsr�	iskeyword�	_builtins�OP�_punctation�NUMBER�_strings�COMMENT�ENCODING�format)r$�sourcer%�row�column�output�token�type_rZ	start_rowZstart_column�
_Z
end_column�line�colorrrr�	highlight;sF










































zSyntaxHighlighter.highlightc

csF|�d
�
}t
�|�
}yt�|j�
EdH
Wntjk
r@


dSXdS)Nzutf-8)�encode�io�BytesIOr(�readline�
TokenError)
r6rrrr(ms





zSyntaxHighlighter.tokenize)
N)�__name__�
__module__�__qualname__r"�set�dir�builtinsr.r,r0r(�STRINGr2r)�sys�version_info�updateZ
FSTRING_STARTZFSTRING_MIDDLEZFSTRING_ENDr&r?�staticmethodrrrrr
 s*

















2r
c@s�eZ
dZd
dddddddddd	d
�Zd/dd�
Zedd��
Zedd�
dd��
Zdd�Zdd�Z	d
dd�dd�Z
d d!�Zd"d#�Zd$d%�Z
d&d'�Zd
ddd(d)�d*d+�Zdd,�
d-d.�Zd
S)0�ExceptionFormatterz{}z
{}z{}z{}z{}z{}z{}z{}z{})�introduction�cause�context�dirname�basenamer=�function�exception_type�exception_value�arrows�valueFTN��asciir'c

Csp|
|_||_
|p|j|_||_t|�
|_||_||_||_	|	|_
|��|_|�
d
d�|_|�
dd�|_d|_dS)Nu│�
|u└z->z <Loguru catch point here>)�	_colorize�	_diagnose�_default_theme�_theme�
_backtracer
�_syntax_highlighter�_max_length�	_encoding�_hidden_frames_filename�_prefix�
_get_lib_dirs�	_lib_dirs�	_get_char�
_pipe_char�	_cap_char�_catch_point_identifier)
r$�colorizeZ	backtraceZdiagnose�themer%�
max_length�encodingZhidden_frames_filename�prefixrrrr&�s













zExceptionFormatter.__init__cs4t�
�}d
dddg��f
dd�|D�
}
dd�|
D�
S)	N�stdlib�
platstdlib�platlib�purelibc
s"h|]}
�D]}t�
||
��qqSr)�	sysconfig�get_path)�.0�scheme�name)
�namesrr�	<setcomp>�sz3ExceptionFormatter._get_lib_dirs.<locals>.<setcomp>c
Ss,g|]$}
|
tj
krtj
�|
�
��tj�qSr)rL�path�os�abspath�lower�sep)ryr~rrr�
<listcomp>�sz4ExceptionFormatter._get_lib_dirs.<locals>.<listcomp>)rw�get_scheme_names)�schemes�pathsr)
r|rrh�s


z ExceptionFormatter._get_lib_dirsz| )
rrc
csH|
d
kr|V
dSx0|�d�
D]"}d|
||}|�
�dV
qWdS)Nr
Tz  �

)�
splitlines�rstrip)�textr*rrr=Zindentedrrr�_indent�s




zExceptionFormatter._indentc	Cs2y|
�|j
�

Wnttfk
r(


|SX|
SdS)
N)r@re�UnicodeEncodeError�LookupError)r$�char�defaultrrrrj�s




zExceptionFormatter._get_charcs8tj
�|
�
�����d
�
sdSt�f
dd�|jD�
�
S)Nz.pyFc
3s|]}
��|
�
V
qdS)
N)
�
startswith)ry�
d)
�filepathrr�	<genexpr>�sz3ExceptionFormatter._is_file_mine.<locals>.<genexpr>)rr~r�r��endswith�anyri)r$�filer)
r�r�
_is_file_mine�s




z ExceptionFormatter._is_file_mine)�limit�from_decoratorcsgd}}|
dks"|dk	r*|d
k
r*||fS�f
dd�}dd�}g}	||
j�
rf|	�
||
j|
j�|
jf�

|op�j}
�jr||s�|
r�|
jj}x4|r�||�
r�|	�d
|||j�|f�
|
r�P|j}q�W|	r�|
s�|	d\\}}
}}}|�j7}||
||f|f|	d<|
j}
x8|
�
r6||
j�
�
r,|	�
||
j|
j�|
jf�

|
j}
�
qW|dk	�
rP|	|d�}	x�|	D]�\\}}
}}}|}|�
r�j	�
o���
|�
}g}|�
r�|�
�j�|�
�

n
|�
|�

�j
�
rڈ�||�}��t|�
|�}|t|�
7}d�|�
}|�
||
||f�

�
qVW||fS)Nr
c

s|jj
�jkS)
N)�f_code�co_filenamerf)
�frame)
r$rr�is_valid�s
z4ExceptionFormatter._extract_frames.<locals>.is_validcSs,|jj
}|jj}t�||
���}||
||fS)
N)r�r��co_name�	linecache�getline�strip)r��lineno�filenamerVr6rrr�get_info�s



z4ExceptionFormatter._extract_frames.<locals>.get_info���z
    )�tb_frame�append�	tb_linenorb�f_back�insert�f_linenorm�tb_nextr^r�rcr?r_�_get_relevant_values�_format_relevant_values�list�join)r$�tb�is_firstr�r��frames�final_sourcer�r��infosZget_parent_onlyr�r�r�rVr6rn�lines�relevant_values�valuesr)
r$r�_extract_frames�sV



































z"ExceptionFormatter._extract_framesc
csx
d}d}d
}d
}d}�
xL|j�
|
�
D�
]:}|^}	}
\}}}|dk	rb|	t
jksX|
dksX|r^|V
d}|	t
jk�
rt�|
�
�
s|s�x�|j|jfD]<}
y|
|
}Wntk
r�


w�Yq�Xd}||�	|�
f}Pq�Wn@|�
r`yt
�||
�}Wntk
r�


d
}YnX||�	|�
fV
q$|	t
jk�
r0|
dk�
r0d}d
}q$|	t
jk�
rT|
dk�
rTd}d
}d
}q$d
}d
}d
}q$W|dk	�
rt|V
dS)NFT�
=�
.r )
rcr(r/r+rr-�f_locals�	f_globals�KeyError�
_format_value�inspect�getattr_static�AttributeError)r$r6r�rZ�pendingZis_attributeZis_valid_valueZ
is_assignmentr:r;rr<�col�	variablesrrrr�
sN
































z'ExceptionFormatter._get_relevant_valuesccs
x�tt
t|
�
�
�
D]�}|
|\}}d
d�|
d|�D�
}d}d}x*|D]"}	|d|	||j7}|	d}qFW|d||7}|�d�
}
xpt|
�
D]d\}}|dkr�||jd}
n|dt|j�
d}
|r�|jd�|
�
}
|jd	�|�
}|
|V
q�WqWdS)
Nc
Ssg|]\}
}|
�qSrr)ryZpcolr<rrrr�4
sz>ExceptionFormatter._format_relevant_values.<locals>.<listcomp>r'r
�
 �
r�rYrZ)	�reversed�range�lenrk�split�	enumeraterlrar5)r$r�rn�
ir�rZZ	pipe_colsZpre_line�indexZpcZvalue_lines�
nZ
value_linerYrrrr�1
s$













z*ExceptionFormatter._format_relevant_valuescCsbyt|
�
}
Wn"t
k
r.


d
t|
�
j}
YnX|j}|dk	r^t|
�
|kr^|
d|d�d}
|
S)Nz<unprintable %s object>rz...)�repr�	Exception�typerErdr�)r$�
vrprrrr�K
s






z ExceptionFormatter._format_valuec
csV
|}d
}�
xF|
D�
]<}t�
||�}|�
rH|�ddd�\}}}	|�|�
}
|	dk	rTd}nd}|jr�|	r�|	�|j�
r�|	dt|j�
�}	d|dd�}|j�
r|
�
rt	j
�|�
\}}
|r�|t	j7}|j
d	�|�
}|j
d
�|
�
}
||
}|j
d�|�
}|j
d�|	�
}	|j�
r"|
�
s|�
r"d|}|�|||	�}|||��d�}|
}|V
qWdS)NzI^  File "(?P<file>.*?)", line (?P<line>[^,]+)(?:, in (?P<function>.*))?\nr�r=rVz  File "{}", line {}, in {}
z  File "{}", line {}
�
>r�rTrUr�)�re�match�groupr�rbr�rmr�r^rr~r�r�rar5r_�end)r$�frames_lines�has_introductionZprepend_with_new_line�regexr�r�r�r=rVZis_mine�patternrTrU�locationrrr�_format_locationsV
s8



















z$ExceptionFormatter._format_locationsr
)�seenr�r��
group_nestingccs�t|
�
|
|}}}	|dkr"t
�}|�t|�
�

|�
r`|jdk	r�t|j�
|kr�|j|j|jj||d
�EdH
d}
|jr�|jd�	|
�
}
|j
r�|�d|
d|�EdH
n|�d|
d|�EdH
n�|jdk	�
r`t|j�
|k�
r`|j
�
s`|j|j|jj||d
�EdH
d}|j�
r"|jd�	|�
}|j
�
rF|�d|d|�EdH
n|�d|d|�EdH
t|
�
}|�
r�|d	k�
r�|j|
||d
||d�EdH
dSy
tj}
Wntk
�
r�


d}
YnX|j|	||
|d�\}}t�||�}x$t|�
D]\}}|�d
�
�
s�P�
q�W||dd�}|j�rvd|k�rf|�dd
�\}}|jd�	|�
}|jd�	|�
}|d|}n|jd�	|�
}|j
�r�|�r�t|t��r�t|�
�s�|�r�|j�r�|j�|�
}|d|7}d|}|d||<|�r�|jV
t|�
}|�rT|�r�d}nd}|j�r|jd�	|�
}|d
k�r>|j|d|dd�EdH
n|�|d|�EdH
t�|�
|}|j�sz|j �sz|j
�r�|j!||d�}|�d�"|�
|�EdH
|�r�x�t|
j#d
d�D]�\}}dd|dk�r�dn|�$dd �}|j|||d
k�r�d!nd"d�EdH
|dk�r:d#t%|
j#�
d}|�||d
�EdH
PnN|d$k�rjt|�
�rjd%}|�||d
�EdH
n|j||j||d
d
�EdH
�q�Wt|�
�r�|d$k�r�|jd&|d
d!d�EdH
dS)'N)r�r�zDThe above exception was the direct cause of the following exception:rRz

z


r�zCDuring handling of the above exception, another exception occurred:rSr
r�)r�r�r�r�)r�r�r�r�rrWrXz: z2Exception Group Traceback (most recent call last):z"Traceback (most recent call last):rQz+ )
rr)
r�r')
�start�
+z %s �z...�#�
-z+-z  zand %d more exceptions
�
z... (max_group_depth is 10)
z#-----------------------------------)&r�rH�add�id�	__cause__�_format_exception�
__traceback__r^rar5r_r��__context__�__suppress_context__r	rL�tracebacklimitr�r��	traceback�format_exception_onlyr�r�r��
issubclass�AssertionError�strrcr?rg�bool�format_listrbr�r��
exceptions�centerr�)r$rZr�r�r�r�r��exc_type�	exc_value�
exc_tracebackrRrSZ
is_groupedZtraceback_limitr�r�Zexception_onlyZerror_message_index�part�
error_messagerWrXr�rQr�r�rZruler�messagerrrr�~
s�

























































"
$












z$ExceptionFormatter._format_exception)
r�c
cs|j||d
|d�EdH
dS)NT)r�r�)
r�)r$r;rZr�r�rrr�format_exceptions
z#ExceptionFormatter.format_exception)	FFTNNr[r\Nr')rErFrGr`r&rOrhr�rjr�r�r�r�r�r�r�r�rrrrrPysB



















@1)rP)rJr�rArr�rr�rLrwr(r�rMr	Zexceptiongroupr�ImportErrorr
rPrrrr�<module>
s&













Y