U md @sddlmZddlZddlZddlZddlZddlmZddlZej ddddddZ d d d d Z ej ddd ddddddZ dS)) annotationsN) GeneratorstrzGenerator[(None, None, None)])old_namenew_namereturnc csz dVWnrtk r|}zT|js$t|jd}|||}|f}t|jdkrd||jdd}||_W5d}~XYnXdS)z. Rewrite the message of an exception. Nr) Exceptionargsrreplacelen)rrerrmsgr rP/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/pandas/util/_exceptions.pyrewrite_exception s  rint)rcCsjddl}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). rNtestsr) Zpandasospathdirname__file__joininspect currentframegetfile startswithf_back)pdZpkg_dirZtest_dirframenfnamerrrfind_stack_levels  r"z type[Warning]ztype[Warning] | None)target_messagetarget_category new_message new_categoryrc cs|dkr |}tjdd }dVW5QRXt|dkrt|}|D]N}|j|krnt|t|jrn|}|}n|j|j}}tj |||j |j dqDdS)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)messagecategoryfilenamelineno) warningscatch_warningsr recompiler)searchrr( warn_explicitr*r+) r#r$r%r&r'matchwarningr)r(rrrrewrite_warning6s(  r4)N) __future__r contextlibrrr.typingrr,contextmanagerrr"r4rrrrs