U MZf' @sdZddlmZddlZddlZddlmZddlZddlZddl Z ddl Z ddl m Z m Z ddlZddlmZddlmZe dZe d ZGd d d e eefZdd ddddZddddZddddZeejddeeejejdZddddZGdddZdS)z Module for scope operations ) annotationsN)StringIO)ChainMapTypeVar) 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]}||kr|||<dSq||jd|<dS)Nr)maps)selfr r mappingrA/tmp/pip-unpacked-wheel-nbcvw55c/pandas/core/computation/scope.py __setitem__#s  zDeepChainMap.__setitem__)r rcCs,|jD]}||kr||=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_scope8sr$strrcCs.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 _replacerEs   r+cCs$tdt|}ddd|DS)z,Return the padded hexadecimal id of ``obj``.z@PcSsg|] }t|qSr)r+).0r*rrr Ysz_raw_hex_id..)structpackidjoin)objpackedrrr _raw_hex_idUsr5TF)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)r3siorrr_get_pretty_stringhsr@c@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)+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"tempsrr dictNrr rc 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)rr DEFAULT_GLOBALScopyrAr# isinstancerupdate_updatesys _getframe new_child f_globalsf_localsr:r"rrB) 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=))r@r9rAkeysr"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 )rTlenr"rrrr has_resolverss zScope.has_resolvers)r is_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)rArWr"AssertionErrorrrBr)rr rXerrrrrresolves   z Scope.resolve)old_keynew_keyrcCsP|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)rWr"rrAappendrB)rr\r] new_valuerrrrrswapkeys  z Scope.swapkeyz list[str])scopesrcCsVt||}|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) itertoolsproductgetattrr rArK)rstackra variablesrArN_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 rNlocals)ra)inspectreri)rrslrerrrrH(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. rg)rRrntempsr5rBrY)rr namerrradd_tmp<s   z Scope.add_tmpcCs t|jS)z/The number of temporary variables in this scope)rUrBrVrrrrmTsz 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_scopeYs zScope.full_scope)NNrN)N)rrrr __slots____annotations__rOrSpropertyrWr[r`rirHrormrprrrrr{s. ( ')NNrN) r __future__rr6rkiorrbr=r/rItypingrrZnumpynpZpandas._libs.tslibsrZ pandas.errorsrrr r r$r+r5r9r:r;rDr@rrrrrsB