o d @srddlmZddlZddlZddlZddlZddlmZddlZej dd d Z dd d Z ej ddddZ dS)) annotationsN) Generatorold_namestrnew_namereturnGenerator[None, None, None]c csxzdVWdSty;}z(|jst|jd}|||}|f}t|jdkr3||jdd}||_d}~ww)z. Rewrite the message of an exception. Nr) Exceptionargsrreplacelen)rrerrmsgr rK/app/.heroku/python/lib/python3.10/site-packages/pandas/util/_exceptions.pyrewrite_exception s  rintcCspddl}tj|j}tj|d}t}d}|r6t|}| |r1| |s1|j }|d7}n |S|s|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#target_messagetarget_category type[Warning] new_message new_categorytype[Warning] | Nonec cs|dur|}tjdd }dVWdn1swYt|dkrUt|}|D])}|j|urBt|t|jrB|}|}n|j|j}}tj |||j |j dq-dSdS)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.   r7)rrrrrr)rr)N) r$rr%r&r'rr(r)rr) __future__r contextlibrrr1typingrr/contextmanagerrr#r7rrrrs