U
|
®ý°du
|
ã @ s d dl mZ d dlZd dlZd dlZd dlZd dlmZ d dlZej ddddddZ
|
d d
|
d d Z ej ddd ddddddZ dS )é )Ú annotationsN)Ú GeneratorÚstrz Generator[(None, None, None)])Úold_nameÚnew_nameÚreturnc
|
c s z
|
dV W nr t k
|
r| } zT|js$ t|jd }| | |¡}|f}t|jdkrd||jdd
}||_ W 5 d}~X Y nX dS )z.
|
Rewrite the message of an exception.
|
Nr é )Ú ExceptionÚargsr ÚreplaceÚlen)r r ÚerrÚmsgr
|
© r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/util/_exceptions.pyÚrewrite_exception s
|
r Úint)r c C sj ddl } tj | j¡}tj |d¡}t ¡ }d}|rft |¡}| |¡rf| |¡sf|j
|
}|d7 }q0qfq0|S )zb
|
Find the first place in the stack that is not inside pandas
|
(tests notwithstanding).
|
r NÚtestsr ) ZpandasÚosÚpathÚdirnameÚ__file__ÚjoinÚinspectÚ currentframeÚgetfileÚ
|
startswithÚf_back)ÚpdZpkg_dirZtest_dirÚframeÚnÚfnamer r r Úfind_stack_level s
|
|
r" z type[Warning]ztype[Warning] | None)Útarget_messageÚtarget_categoryÚ new_messageÚ new_categoryr c c s |dkr |}t jdd }dV W 5 Q R X t|dkrt | ¡}|D ]N}|j|krnt |t|j¡rn|}|}n|j|j }}t j |||j
|
|j d qDdS )a
|
Rewrite the message of a warning.
|
|
Parameters
|
----------
|
target_message : str
|
Warning message to match.
|
target_category : Warning
|
Warning type to match.
|
new_message : str
|
New warning message to emit.
|
new_category : Warning or None, default None
|
New warning type to emit. When None, will be the same as target_category.
|
NT)Úrecordr )ÚmessageÚcategoryÚfilenameÚlineno) ÚwarningsÚcatch_warningsr ÚreÚcompiler) Úsearchr r( Ú warn_explicitr* r+ ) r# r$ r% r& r' ÚmatchÚwarningr) r( r r r Úrewrite_warning6 s(
|
ÿür4 )N) Ú
|
__future__r Ú
|
contextlibr r r. Útypingr r, Úcontextmanagerr r" r4 r r r r Ú<module> s ü
|
