a Pf< @sddlmZddlZddlZddlZddlZddlmZddlZer\ddl m Z ddl m Z ej ddddd d Zd d d dZej dddddddddZdS)) annotationsN) TYPE_CHECKING) Generator) FrameTypestrzGenerator[None, None, None])old_namenew_namereturnc csz dVWntty~}z\|js"t|jd}|||}|f}t|jdkrb||jdd}||_WYd}~n d}~00dS)z. Rewrite the message of an exception. Nr) Exceptionargsrreplacelen)rrerrmsgr rP/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/util/_exceptions.pyrewrite_exceptions  rint)r cCsvddl}tj|j}tj|d}t}z@d}|rht|}| |rh| |sh|j }|d7}q2qhq2W~n~0|S)zb Find the first place in the stack that is not inside pandas (tests notwithstanding). rNZtestsr ) Zpandasospathdirname__file__joininspect currentframegetfile startswithf_back)pdZpkg_dirZtest_dirframenfilenamerrrfind_stack_level"s  r#z type[Warning]ztype[Warning] | None)target_messagetarget_category new_message new_categoryr c cs|dur |}tjdd}dVWdn1s40Yt|dkrt|}|D]N}|j|urt|t|jr|}|}n|j|j}}tj |||j |j dqXdS)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)messagecategoryr"lineno) warningscatch_warningsrrecompiler*searchrr) warn_explicitr"r+) r$r%r&r'r(matchwarningr*r)rrrrewrite_warning?s&$   r4)N) __future__r contextlibrrr.typingrr,collections.abcrtypesrcontextmanagerrr#r4rrrrs