o h' @sdZddlmZddlmZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZddlZddlmZddlmZed Zed ZGd d d eeefZ dd ddZd!ddZd!ddZeejddeeejejdZd!ddZGdddZ dS)"z Module for scope operations ) annotations)ChainMapN)StringIO)TypeVar) Timestamp)UndefinedVariableError_KT_VTc@s$eZdZdZd dd Zdd d Zd S) DeepChainMapz~ Variant of ChainMap that allows direct updates to inner scopes. Only works when all passed mapping are mutable. keyrvaluer returnNonecCs4|jD] }||vr|||<dSq||jd|<dS)Nr)maps)selfr r mappingrO/var/www/html/env/lib/python3.10/site-packages/pandas/core/computation/scope.py __setitem__!s zDeepChainMap.__setitem__cCs(|jD] }||vr||=dSqt|)z\ Raises ------ KeyError If `key` doesn't exist. N)rKeyError)rr rrrr __delitem__(s zDeepChainMap.__delitem__N)r rr r r r)r rr r)__name__ __module__ __qualname____doc__rrrrrrr s r rlevelintr ScopecCst|d||||dS)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r)rrr r!r"rrr ensure_scope6sr#strcCs0z t|}Wt|Sty|}Yt|Sw)z Replace a number with its hexadecimal representation. Used to tag temporary variables with their calling scope's id. )ord TypeErrorhex)xhexinrrr _replacerCs  r*cCs$tdt|}ddd|DS)z,Return the padded hexadecimal id of ``obj``.z@PcSsg|]}t|qSr)r*).0r(rrr Wsz_raw_hex_id..)structpackidjoin)objpackedrrr _raw_hex_idSsr4TF)rdatetimeTrueFalselisttupleinfInfcCst}tj||d|S)z Return a prettier version of obj. Parameters ---------- obj : object Object to pretty print Returns ------- str Pretty print object repr )stream)rpprintgetvalue)r2siorrr_get_pretty_stringfsr@c@seZdZUdZgdZded<ded<ded<ded < d*d+ddZd,ddZed-ddZ d.ddZ d/d0ddZ d1d d!Z d+d"d#Z d,d$d%Zed2d&d'Zed3d(d)Zd S)4ra 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 )rscoper"r!tempsrrr rAr!dictrBNrr rc Cs|d|_tt|_||_t|tr*|j|j|jdur$|j|_| |jt |j}z0|j |dur:|n|j }t||_t|ts^|j |durS|n|j}t||_W~n~wt|trp|t|jj7}t||_i|_dS)Nr)rr DEFAULT_GLOBALScopyrAr" isinstancerupdate_updatesys _getframe new_child f_globalsf_localsr9r!rrB) rrrr r!r"frame scope_global scope_localrrr__init__s2           zScope.__init__r$cCs@tt|j}tt|j}t|jd|d|dS)Nz(scope=z , resolvers=))r@r8rAkeysr!typer)r scope_keysres_keysrrr__repr__szScope.__repr__boolcCstt|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 )rXlenr!rrrr has_resolverss zScope.has_resolversr is_localc Csz|r |j|WS|jr|j|WS|s|jrJ|j|WSty@z|j|WYSty?}zt|||d}~www)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)rAr[r!rrBr)rr r\errrrrresolves     z Scope.resolveold_keynew_keycCsP|jr |jj|jj}n|jj}||j|D] }||vr%|||<dSqdS)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)r[r!rrAappendrB)rr_r` new_valuerrrrrswapkeys  z Scope.swapkeyscopes list[str]cCsVt||}|D] \}\}}}}}}zt|d|}t|j||_W~q~wdS)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 rArK)rstackrd variablesrArN_drrr _get_varss zScope._get_varsc CsJ|d}t}z|j|d|dgdW|dd=~dS|dd=~w)z Update the current scope by going back `level` levels. Parameters ---------- level : int rNlocals)rd)inspectrjrn)rrslrjrrrrH&s z Scope._updatecCsJt|jd|jdt|}||jvsJ||j|<||jvs#J|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. rl)rTrntempsr4rB)rr namerrradd_tmp:s  z Scope.add_tmpcCs t|jS)z/The number of temporary variables in this scope)rYrBrZrrrrrRs z Scope.ntempscCs |jg|jj|jj}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. )rBr!rrAr )rrrrr full_scopeWs zScope.full_scopeNNrN)rrr rr r$)r rX)r r$r\rX)N)r_r$r`r$r r)rdrer r)r r)r r )rrrr __slots____annotations__rQrWpropertyr[r^rcrnrHrtrrrurrrrrys*  (  '    rv)rrr rrw)!r __future__r collectionsrr5rpiorrgr=r.rItypingrnumpynppandas._libs.tslibsr pandas.errorsrrr r r#r*r4r8r9r:rDr@rrrrrs@