U f/e/( @sdZddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl mZddlmZddd d d d Zd dddZd dddZeejddeee je jdZd dddZGdd d ZdS)z Module for scope operations ) annotationsN)StringIO) Timestamp) DeepChainMapintScopelevelreturncKst|d||||dS)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r)r r rrrkwargsrrA/tmp/pip-unpacked-wheel-tiezk1ph/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@Pcss|]}t|VqdS)N)r).0rrrr 5sz_raw_hex_id..)structpackidjoin)objpackedrrr _raw_hex_id1sr$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_stringDsr/c@seZdZUdZdddddgZded<ded<ded<d ed<d*dd d dZddddZeddddZ dddddZ d+ddddddZ ddddd Z ddd!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 r scoperrtempsrrdictNr)r cCs|d|_tt|_||_t|trT|j|j|jdk rH|j|_| |jt |j}z>|j |pp|j |_t|ts|j |p|j|_W5~Xt|tr|t|jj7}t||_i|_dS)Nr )r rDEFAULT_GLOBALScopyr0r isinstancerupdate_updatesys _getframe new_child f_globalsf_localsr)rmapsr1)selfr r rrrframerrr__init__rs,          zScope.__init__rrcCs@tt|j}tt|j}t|jd|d|dS)Nz(scope=z , resolvers=))r/r(r0keysrtype__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 )rFlenrr>rrr has_resolverss zScope.has_resolvers)keyis_localc Csz<|r|j|WS|jr$|j|WS|s.|jr2t|j|WStk rz|j|WYStk r}zddlm}||||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 r)UndefinedVariableErrorN)r0rIrAssertionErrorKeyErrorr1Zpandas.core.computation.opsrL)r>rJrKerrrLrrrresolves    z Scope.resolveNone)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)rIrr=r0appendr1)r>rRrS new_valuer=mappingrrrswapkeys  z Scope.swapkeyz list[str])scopesr cCsPt||}|D]:\}\}}}}}}z t|d|}|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) itertoolsproductgetattrr0r:)r>stackrX variablesr0r?_drrr _get_varss zScope._get_varsr c 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\rrrr7 s 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^)rCrDntempsr$r1rM)r>valuenamerrradd_tmp!s   z Scope.add_tmpcCs t|jS)z/The number of temporary variables in this scope)rGr1rHrrrrd9sz 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. )r1rr=r0r)r>r=rrr full_scope>szScope.full_scope)NNrN)N)rD __module__ __qualname____doc__ __slots____annotations__r@rEpropertyrIrPrWr`r7rgrdrhrrrrrWs. + *)NNrN)rk __future__rr%rbiorrYr,rr8ZnumpynpZpandas._libs.tslibsrZpandas.compat.chainmaprrrr$r(r)r*r3r/rrrrrs: