E g' TdZddlmZddlmZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZddlZddlmZddlmZed Zed ZGd d eeefZ dddZddZddZeejddeeejejdZddZGddZ dS)z Module for scope operations ) annotations)ChainMapN)StringIO)TypeVar) Timestamp)UndefinedVariableError_KT_VTc"eZdZdZd dZd d Zd S) DeepChainMapz~ Variant of ChainMap that allows direct updates to inner scopes. Only works when all passed mapping are mutable. keyr valuer returnNonecR|jD]}||vr|||<dS||jd|<dS)Nr)maps)selfr rmappings R/var/www/sysmax/venv/lib/python3.11/site-packages/pandas/core/computation/scope.py __setitem__zDeepChainMap.__setitem__!sIy  Gg~~$ " ! ScH|jD] }||vr||=dS t|)z\ Raises ------ KeyError If `key` doesn't exist. N)rKeyError)rr rs r __delitem__zDeepChainMap.__delitem__(s?y  Gg~~CLsmmrN)r r rr rr)r r rr)__name__ __module__ __qualname____doc__rrrrr r sF """"      rr rlevelintrScopec0t|dz||||S)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r")r r%r&r'r(s r ensure_scoper)6s.       rstrcf t|}n#t$r|}YnwxYwt|S)z Replace a number with its hexadecimal representation. Used to tag temporary variables with their calling scope's id. )ord TypeErrorhex)xhexins r _replacerr1CsFA  u::s  !!ctjdt|}dd|DS)z,Return the padded hexadecimal id of ``obj``.z@Pc,g|]}t|Sr)r1).0r/s r z_raw_hex_id..Ws111QIaLL111r)structpackidjoin)objpackeds r _raw_hex_idr=Ss<[r#ww ' 'F 7711&111 2 22rTF)rdatetimeTrueFalselisttupleinfInfcrt}tj|||S)z Return a prettier version of obj. Parameters ---------- obj : object Object to pretty print Returns ------- str Pretty print object repr )stream)rpprintgetvalue)r;sios r_get_pretty_stringrJfs0 **C M#c"""" <<>>rceZdZUdZgdZded<ded<ded<ded < d d!dZd"dZed#dZ d$dZ d%d&dZ d'dZ d!dZ d"dZed(dZed)dZd S)*r"a Object to hold scope, with a few bells to deal with some custom syntax and contexts added by pandas. Parameters ---------- level : int global_dict : dict or None, optional, default None local_dict : dict or Scope or None, optional, default None resolvers : list-like or None, optional, default None target : object Attributes ---------- level : int scope : DeepChainMap target : object temps : dict )r scoper(r'tempsr!r r rLr'dictrMNrrrc^|dz|_tt|_||_t |trL|j|j|j |j|_| |jtj |j} |j ||n|j }t||_t |tsI|j ||n|j}t||_~n#~wxYwt |tr|t|jjz }t||_i|_dS)Nr$)r r DEFAULT_GLOBALScopyrLr( isinstancer"update_updatesys _getframe new_child f_globalsf_localsrBr'rrM) rr r%r&r'r(frame scope_global scope_locals r__init__zScope.__init__sQY "/"6"6"8"899  j% ( ( + J  j. / / / ,(/ LL) * * * dj))  :// + 7U_RRTTL&l33DJj%00 7"j22#-#9ZZu~SSUU *+66 IIII j% ( ( : z3899 9I%y1 s 8B'E!!E$r*ctt|j}tt|j}t |jd|d|dS)Nz(scope=z , resolvers=))rJrArLkeysr'typer)r scope_keysres_keyss r__repr__zScope.__repr__sn'TZ__->->(?(?@@ %d4>+>+>+@+@&A&ABBt**%QQjQQhQQQQrboolcDtt|jS)z Return whether we have any extra scope. For example, DataFrames pass Their columns as resolvers during calls to ``DataFrame.eval()`` and ``DataFrame.query()``. Returns ------- hr : bool )relenr'rs r has_resolverszScope.has_resolverssC''(((rr is_localc |r |j|S|jr |j|S|s|jrJ|j|S#t$r4 |j|cYS#t$r}t |||d}~wwxYwwxYw)a Resolve a variable name in a possibly local context. Parameters ---------- key : str A variable name is_local : bool Flag indicating whether the variable is local or not (prefixed with the '@' symbol) Returns ------- value : object The value of a particular variable N)rLrir'rrMr)rr rjerrs rresolvez Scope.resolves" E 'z#&! +~c** :(: : : ::c? " E E E Ez#&&& E E E,S(;;D E  Es5=== A; AA; A7!A22A77A;old_keynew_keyc|jr|jj|jjz}n |jj}||j|D]}||vr|||<dSdS)a] Replace a variable name, with a potentially new value. Parameters ---------- old_key : str Current variable name to replace new_key : str New variable name to replace `old_key` with new_value : object Value to be replaced along with the possible renaming N)rir'rrLappendrM)rrnro new_valuerrs rswapkeyz Scope.swapkeys{   #>&8DD:?D DJ  G'!!#, "  rscopes list[str]ctj||}|D]S\}\}}}}}} t|d|}t|j||_~N#~wxYwdS)a Get specifically scoped variables from a list of stack frames. Parameters ---------- stack : list A list of stack frames as returned by ``inspect.stack()`` scopes : sequence of strings A sequence containing valid stack frame attribute names that evaluate to a dictionary. For example, ('locals', 'globals') f_N) itertoolsproductgetattrr rLrW)rstackrt variablesrLrZ_ds r _get_varszScope._get_varss%fe44 -6   )E)E1aAq E<<<00)$**>*>q*A*ABB EE   s ?A&&A)c|dz}tj} ||d|dg|dd=~dS#|dd=~wxYw)z Update the current scope by going back `level` levels. Parameters ---------- level : int r$Nlocals)rt)inspectr{r)rr slr{s rrTz Scope._update&siQY   NN5":xjN 9 9 9aaa%%%aaa%    s AA ct|jd|jdt|}||jvsJ||j|<||jvsJ|S)a# Add a temporary variable to the scope. Parameters ---------- value : object An arbitrary object to be assigned to a temporary variable. Returns ------- str The name of the temporary variable created. r})rarntempsr=rM)rrnames radd_tmpz Scope.add_tmp:smu++&JJJJ{47H7HJJ4:%%%%  4tz!!!! rc*t|jS)z/The number of temporary variables in this scope)rgrMrhs rrz Scope.ntempsRs4:rcX|jg|jjz|jjz}t |S)z Return the full scope for use with passing to engines transparently as a mapping. Returns ------- vars : DeepChainMap All variables in this scope. )rMr'rrLr )rrs r full_scopezScope.full_scopeWs+ |dn11DJOCT""rNNrN)r r!rrrr*)rre)r r*rjre)N)rnr*ror*rr)rtrurr)rr!)rr )rrrr __slots____annotations__r]rdpropertyrirmrsrrTrrrrrrr"r"ys_(CBBIJJJKKKSW&&&&&PRRRR  ) ) )X )%E%E%E%EN2.    (0X # # #X # # #rr)r r!rr"r)!r __future__r collectionsrr>riorrxrGr7rUtypingrnumpynppandas._libs.tslibsr pandas.errorsrr r r r)r1r=rArBrCrPrJr"rrrrs#"""""  ))))))000000 genn genn8CH%:IM          3333!     6 6  &j#j#j#j#j#j#j#j#j#j#r