U c% @sdZddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl mZddlmZddlmZdd d d d d ZddddZddddZeejddeee je jdZddddZGdd d ZdS)z Module for scope operations ) annotationsN)StringIO) Timestamp) DeepChainMap)UndefinedVariableErrorintScopelevelreturncKst|d||||dS)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r )r rrrrkwargsrrA/tmp/pip-unpacked-wheel-g7fro6k3/pandas/core/computation/scope.py ensure_scopesrstrr cCs.z t|}Wntk r$|}YnXt|S)z Replace a number with its hexadecimal representation. Used to tag temporary variables with their calling scope's id. )ord TypeErrorhex)xZhexinrrr _replacer"s   rcCs$tdt|}ddd|DS)z,Return the padded hexadecimal id of ``obj``.z@PcSsg|] }t|qSr)r).0rrrr 6sz_raw_hex_id..)structpackidjoin)objpackedrrr _raw_hex_id2sr%TF)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)r#siorrr_get_pretty_stringEsr0c@seZdZUdZdddddgZded<ded<ded<d ed<d)dd d ddZddddZeddddZ dddddZ d*ddd dddZ dd ddd Z dd d d!d"Z ddd#d$Zeddd%d&Zeddd'd(Zd 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 scoperrtempsrrdictNrNoner c Cs|d|_tt|_||_t|trT|j|j|jdk rH|j|_| |jt |j}z^|j |dk rt|n|j }t||_t|ts|j |dk r|n|j}t||_W5~Xt|tr|t|jj7}t||_i|_dS)Nr )r rDEFAULT_GLOBALScopyr1r isinstancer update_updatesys _getframe new_child f_globalsf_localsr*rmapsr2) selfr rrrrframeZ scope_globalZ scope_localrrr__init__ss0         zScope.__init__rrcCs@tt|j}tt|j}t|jd|d|dS)Nz(scope=z , resolvers=))r0r)r1keysrtype__name__)r@Z 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 )rHlenrr@rrr has_resolverss zScope.has_resolvers)keyis_localc Csz<|r|j|WS|jr$|j|WS|s.|jr2t|j|WStk rz|j|WYStk r}zt|||W5d}~XYnXYnXdS)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)r1rKrAssertionErrorKeyErrorr2r)r@rLrMerrrrrresolves   z Scope.resolve)old_keynew_keyr cCsP|jr|jj|jj}n|jj}||j|D]}||kr0|||<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)rKrr?r1appendr2)r@rRrS new_valuer?mappingrrrswapkeys  z Scope.swapkeyz list[str])scopesr cCsTt||}|D]>\}\}}}}}}z$t|d|}t|j||_W5~XqdS)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) itertoolsproductgetattrrr1r<)r@stackrX variablesr1rA_drrr _get_varss zScope._get_varsc Cs@|d}t}z|j|d|dgdW5|dd=~XdS)z Update the current scope by going back `level` levels. Parameters ---------- level : int r Nlocals)rX)inspectr\r`)r@r slr\rrrr9s z Scope._updatecCsJt|jd|jdt|}||jks.t||j|<||jksFt|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^)rErFntempsr%r2rN)r@valuenamerrradd_tmps   z Scope.add_tmpcCs t|jS)z/The number of temporary variables in this scope)rIr2rJrrrrd1sz 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. )r2rr?r1r)r@r?rrr full_scope6szScope.full_scope)NNrN)N)rF __module__ __qualname____doc__ __slots____annotations__rBrGpropertyrKrQrWr`r9rgrdrhrrrrr Xs. ( ')NNrN)rk __future__rr&rbiorrYr-rr:ZnumpynpZpandas._libs.tslibsrZpandas.compat.chainmaprZ pandas.errorsrrrr%r)r*r+r5r0r rrrrs<