a Of' @sdZddlmZddlmZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZddlZddlmZddlmZed Zed ZGd d d eeefZddddddZddddZddddZeejddeeejejdZddddZGdddZ dS) z Module for scope operations ) annotations)ChainMapN)StringIO)TypeVar) Timestamp)UndefinedVariableError_KT_VTc@s2eZdZdZddddddZdddd d Zd S) DeepChainMapz~ Variant of ChainMap that allows direct updates to inner scopes. Only works when all passed mapping are mutable. rr None)keyvaluereturncCs4|jD]}||vr|||<dSq||jd|<dS)Nr)maps)selfr r mappingrV/var/www/ai-form-bot/venv/lib/python3.9/site-packages/pandas/core/computation/scope.py __setitem__!s  zDeepChainMap.__setitem__)r rcCs,|jD]}||vr||=dSqt|dS)z\ Raises ------ KeyError If `key` doesn't exist. N)rKeyError)rr rrrr __delitem__(s  zDeepChainMap.__delitem__N)__name__ __module__ __qualname____doc__rrrrrrr sr rintScopelevelrcCst|d||||dS)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r)rr r!r"r#rrr ensure_scope6sr$strrcCs,z t|}Wnty"|}Yn0t|S)z Replace a number with its hexadecimal representation. Used to tag temporary variables with their calling scope's id. )ord TypeErrorhex)xZhexinrrr _replacerCs    r+cCs$tdt|}ddd|DS)z,Return the padded hexadecimal id of ``obj``.z@PcSsg|] }t|qSr)r+).0r*rrr Wz_raw_hex_id..)structpackidjoin)objpackedrrr _raw_hex_idSsr6TF)rdatetimeTrueFalselisttupleinfZInfcCst}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)r4siorrr_get_pretty_stringfsrAc@seZdZUdZgdZded<ded<ded<ded <d)dd d ddZddddZeddddZ dddddZ d*ddd dddZ dd ddd Z dd d d!d"Z ddd#d$Zeddd%d&Zeddd'd(Zd S)+ra 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 rBr"dictrCNrr rc Cs|d|_tt|_||_t|trT|j|j|jdurH|j|_| |jt |j}z`|j |durt|n|j }t||_t|ts|j |dur|n|j}t||_W~n~0t|tr|t|jj7}t||_i|_dS)Nr)rr DEFAULT_GLOBALScopyrBr# isinstancerupdate_updatesys _getframe new_child f_globalsf_localsr;r"rrC) rrr r!r"r#frameZ scope_globalZ scope_localrrr__init__s0          zScope.__init__r%r&cCs@tt|j}tt|j}t|jd|d|dS)Nz(scope=z , resolvers=))rAr:rBkeysr"typer)rZ scope_keysZres_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 )rUlenr"rrrr has_resolverss zScope.has_resolvers)r is_localc Csz<|r|j|WS|jr$|j|WS|s.|jr2J|j|WStyz|j|WYSty}zt|||WYd}~n d}~00Yn0dS)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)rBrXr"rrCr)rr rYerrrrrresolves    z Scope.resolve)old_keynew_keyrcCsP|jr|jj|jj}n|jj}||j|D]}||vr0|||<dSq0dS)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)rXr"rrBappendrC)rr\r] new_valuerrrrrswapkeys  z Scope.swapkeyz list[str])scopesrcCsXt||}|D]B\}\}}}}}}z(t|d|}t|j||_W~q~0qdS)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') Zf_N) itertoolsproductgetattrr rBrL)rstackra variablesrBrO_drrr _get_varss zScope._get_varsc CsL|d}t}z(|j|d|dgdW|dd=~n|dd=~0dS)z Update the current scope by going back `level` levels. Parameters ---------- level : int rNlocals)ra)inspectreri)rrslrerrrrI&s z Scope._updatecCsJt|jd|jdt|}||jvs.J||j|<||jvsFJ|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. rg)rSrntempsr6rC)rr namerrradd_tmp:s   z Scope.add_tmpcCs t|jS)z/The number of temporary variables in this scope)rVrCrWrrrrmRsz 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. )rCr"rrBr )rrrrr full_scopeWs zScope.full_scope)NNrN)N)rrrr __slots____annotations__rPrTpropertyrXr[r`rirIrormrprrrrrys( ( ')NNrN)!r __future__r collectionsrr7rkiorrbr>r0rJtypingrnumpynpZpandas._libs.tslibsrZ pandas.errorsrrr r r$r+r6r:r;r<rErArrrrrs>