LVfX ddlmZddlZddlZddlZddlZddlmZddlZerddl m Z ejddZ d dZ ej d d dZy) ) annotationsN) TYPE_CHECKING) Generatorc#K dy#t$rq}|jst|jd}|j||}|f}t |jdkDr||jddz}||_d}~wwxYww)z. Rewrite the message of an exception. Nr) Exceptionargsstrreplacelen)old_namenew_nameerrmsgr s `/var/www/html/software/conda/envs/catlas/lib/python3.12/site-packages/pandas/util/_exceptions.pyrewrite_exceptionrs~   xx #((1+kk(H-!$ sxx=1 #((12,&D  s%B B BA,A>>BBcnddl}tjj|j}tjj |d}t j}d}|rOt j|}|j|r#|j|s|j}|dz }n |S|rO|S)zb Find the first place in the stack that is not inside pandas (tests notwithstanding). rNtestsr) pandasospathdirname__file__joininspect currentframegetfile startswithf_back)pdpkg_dirtest_dirframenfnames rfind_stack_levelr&!s ggoobkk*Gww||GW-H  "E A &   G $U-=-=h-GLLE FA  H  Hc#K||}tjd5}ddddtdkDrtj|}|D]}|j |ur.tj |t|jr|}|}n|j |j}}tj|||j|jyy#1swYxYww)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+searchr r* warn_explicitr,r-) target_messagetarget_category new_message new_categoryr)matchwarningr+r*s rrewrite_warningr:9s*&   -  6{Q >* G?2ryys7??+8()4$+$4$4goo'  " "! ))~~   sC#CB5C#C C#)r r rr returnGenerator[None, None, None])r;int)N) r4r r5z type[Warning]r6r r7ztype[Warning] | Noner;r<) __future__r contextlibrrr0typingrr.collections.abcrcontextmanagerrr&r:r'rrDs" ) $ 0  *. ''"''' ' ! ''r'