eh' XdZddlmZddlmZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZddlZddlmZddlmZed Zed ZGd d eeefZ d ddZddZddZeej ddeeej:ej:dZddZGdd Z y)z Module for scope operations ) annotations)ChainMapN)StringIO)TypeVar) Timestamp)UndefinedVariableError_KT_VTc eZdZdZddZddZy) DeepChainMapz~ Variant of ChainMap that allows direct updates to inner scopes. Only works when all passed mapping are mutable. c`|jD] }||vs|||<y||jd|<y)Nr)maps)selfkeyvaluemappings [/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/core/computation/scope.py __setitem__zDeepChainMap.__setitem__!s<yy Gg~$  " ! ScL|jD] }||vs||=yt|)z\ Raises ------ KeyError If `key` doesn't exist. N)rKeyError)rrrs r __delitem__zDeepChainMap.__delitem__(s4yy Gg~CL smrN)rr rr returnNone)rr rr)__name__ __module__ __qualname____doc__rrrrr r s " rr Scopec(t|dz||||S)z.Ensure that we are grabbing the correct scope.) global_dict local_dict resolverstarget)r )levelr#r$r%r&s r ensure_scoper(6s$     rcf t|}t|S#t$r|}Yt|SwxYw)z Replace a number with its hexadecimal representation. Used to tag temporary variables with their calling scope's id. )ord TypeErrorhex)xhexins r _replacerr/Cs<A u:  u: s  00ctjdt|}dj|Dcgc] }t |c}Scc}w)z,Return the padded hexadecimal id of ``obj``.z@P)structpackidjoinr/)objpackedr-s r _raw_hex_idr8Ss9[[r#w 'F 77&1QIaL1 221sATF)rdatetimeTrueFalselisttupleinfInfcdt}tj|||jS)z Return a prettier version of obj. Parameters ---------- obj : object Object to pretty print Returns ------- str Pretty print object repr )stream)rpprintgetvalue)r6sios r_get_pretty_stringrEfs% *C MM#c" <<>rceZdZUdZgdZded<ded<ded<ded < d dd Zdd Zedd Z ddZ dddZ ddZ ddZ ddZeddZeddZy )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'scoper&r%tempsintr'r rGr%dictrHNcZ|dz|_ttj|_||_t |tr]|jj|j|j |j |_|j|jtj|j} |jj||n |jj}t||_t |tsG|jj||n |jj}t||_~t |tr"|t|j j"z }t||_i|_y#~wxYw)Nr")r'r DEFAULT_GLOBALScopyrGr& isinstancer update_updatesys _getframe new_child f_globalsf_localsr=r%rrH) rr'r#r$r%r&frame scope_global scope_locals r__init__zScope.__init__sSQY "/"6"6"89  j% ( JJ  j.. /  ,(// LL)) * djj)  ::// + 7U__RRTL&l3DJj%0"jj22#-#9Zu~~SSU *+6  j% ( z33889 9I%y1 s BF''F*ctt|jj}tt|jj}t |j d|d|dS)Nz(scope=z , resolvers=))rEr<rGkeysr%typer)r scope_keysres_keyss r__repr__zScope.__repr__s\'TZZ__->(?@ %d4>>+>+>+@&ABt*%%&gj\hZqQQrc>tt|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 )boollenr%rs r has_resolverszScope.has_resolverssC'((rc |r|j|S|jr|j|S|s |jrJ|j|S#t$r1 |j|cYS#t$r}t |||d}~wwxYwwxYw)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)rGrer%rrHr)rris_localerrs rresolvez Scope.resolves" Ezz#&!!~~c** (:(: ::::c? " E Ezz#& E,S(;D E  Es9A A A BA(%B( B1 A>>BBc|jr.|jj|jjz}n|jj}|j |j |D] }||vs|||<yy)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)rer%rrGappendrH)rold_keynew_key new_valuerrs rswapkeyz Scope.swapkeysj   >>&&8D::??D DJJ G'!#,  rctj||}|D]F\}\}}}}}} t|d|}t|jj ||_~Hy#~wxYw)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 rGrS)rstackscopes variablesrGrV_ds r _get_varszScope._get_varssu%%fe4 -6  )E)E1aAq ERw<0)$***>*>q*AB  s 8A##A&c|dz}tj} |j|d|dg|dd=~y#|dd=~wxYw)z Update the current scope by going back `level` levels. Parameters ---------- level : int r"Nlocals)rv)inspectrurz)rr'slrus rrPz Scope._update&sJQY   NN5":xjN 9a%a%s 9Act|jd|jdt|}||jvsJ||j|<||jvsJ|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. rx)r]rntempsr8rH)rrnames radd_tmpz Scope.add_tmp:sju+&&'q Q{47H6IJ4::%%%  4tzz!!! rc,t|jS)z/The number of temporary variables in this scope)rcrHrds rrz Scope.ntempsRs4::rc|jg|jjz|jjz}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. )rHr%rrGr )rrs r full_scopezScope.full_scopeWs6 |dnn111DJJOOCT""rNNrN)r'rIrrrstr)rrb)rrrgrb)N)rlrrmrrr)rvz list[str]rr)rrI)rr )rrrr __slots____annotations__rYr`propertyrerirorzrPrrrrrrr r ys(CI J  KSW&& &PR  ) )%EN2. (0 # #rr)r'rIrr r)!r __future__r collectionsrr9r}iorrrrBr2rQtypingrnumpynppandas._libs.tslibsr pandas.errorsrr r r r(r/r8r<r=r>rLrEr rrrrs#  )0 en en8CH%:IM       3!!     66 66 &j#j#r