U
|
¬ý°d6) ã @ s d Z ddlmZ ddlZddlZddlmZ ddlZddlZddl Z ddl
|
Z
|
ddl m Z m Z ddlZddlmZ ddlmZ e dZe d ZG d
|
d d e eef Zd d ddddZddddZddddZeejddeeej ej dZ dddd Z G d d dZ dS ) z
|
Module for scope operations
|
é )Ú annotationsN)ÚStringIO)ÚChainMapÚTypeVar)Ú Timestamp)ÚUndefinedVariableErrorÚ_KTÚ_VTc @ s2 e Zd ZdZddddddZdddd d
|
Zd S ) Ú DeepChainMapz~
|
Variant of ChainMap that allows direct updates to inner scopes.
|
|
Only works when all passed mapping are mutable.
|
r r ÚNone)ÚkeyÚvalueÚreturnc C s4 | j D ]}||kr|||< d S q|| j d |< d S )Nr )Úmaps)Úselfr r Úmapping© r úTd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\pandas/core/computation/scope.pyÚ __setitem__# s
|
|
zDeepChainMap.__setitem__)r r c C s, | j D ]}||kr||= dS qt|dS )z\
|
Raises
|
------
|
KeyError
|
If `key` doesn't exist.
|
N)r ÚKeyError)r r r r r r Ú __delitem__* s
|
|
zDeepChainMap.__delitem__N)Ú__name__Ú
|
__module__Ú __qualname__Ú__doc__r r r r r r r
|
s r
|
r ÚintÚScope©Úlevelr c C s t | d ||||dS )z.Ensure that we are grabbing the correct scope.é )Ú global_dictÚ
|
local_dictÚ resolversÚtarget)r )r r r! r" r# r r r Ú ensure_scope8 s ûr$ Ústr©r c C s. z t | }W n tk
|
r$ | }Y nX t|S )z
|
Replace a number with its hexadecimal representation. Used to tag
|
temporary variables with their calling scope's id.
|
)ÚordÚ TypeErrorÚhex)ÚxZhexinr r r Ú _replacerE s
|
|
r+ c C s$ t dt| ¡}d dd |D ¡S )z,Return the padded hexadecimal id of ``obj``.z@PÚ c S s g | ] }t |qS r )r+ )Ú.0r* r r r Ú
|
<listcomp>Y s z _raw_hex_id.<locals>.<listcomp>)ÚstructÚpackÚidÚjoin)ÚobjÚpackedr r r Ú _raw_hex_idU s r5 TF)r ÚdatetimeÚTrueÚFalseÚlistÚtupleÚinfZInfc C s t }tj| |d | ¡ S )z¼
|
Return a prettier version of obj.
|
|
Parameters
|
----------
|
obj : object
|
Object to pretty print
|
|
Returns
|
-------
|
str
|
Pretty print object repr
|
)Ústream)r ÚpprintÚgetvalue)r3 Úsior r r Ú_get_pretty_stringh s r@ c @ sè e Zd ZU dZdddddgZded< ded< ded< d ed< d)dd d ddZddddZeddddZ dddddZ
|
d*ddd ddd Z d d d d d 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 Úscoper# r" Útempsr r
|
ÚdictNr r r c C sò |d | _ tt ¡ | _|| _t|trT| j |j¡ |jd k rH|j| _| |j ¡ t
|
| j ¡}z^| j |d k rt|n|j ¡ ¡}t|| _t|ts¼| j |d k r¦|n|j ¡ ¡}t|| _W 5 ~X t|trÞ|t|jj7 }t| | _i | _d S )Nr )r r
|
ÚDEFAULT_GLOBALSÚcopyrA r# Ú
|
isinstancer ÚupdateÚ_updateÚsysÚ _getframeÚ new_childÚ f_globalsÚf_localsr: r" r rB ) r r r r! r" r# ÚframeZ scope_globalZ scope_localr r r Ú__init__ s0
|
|
|
ÿ
|
|
ÿ
|
|
zScope.__init__r% r& c C s@ t t| j ¡ }t t| j ¡ }t| j d| d| dS )Nz(scope=z , resolvers=ú))r@ r9 rA Úkeysr" Útyper )r Z
|
scope_keysZres_keysr r r Ú__repr__¾ s zScope.__repr__Úboolc C s t t| 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
|
)rT Úlenr" ©r r r r Ú has_resolversà s zScope.has_resolvers)r Úis_localc C s z<|r| j | W S | jr$| j| W S |s.| jr2t| j | W S tk
|
r z| j| W Y S tk
|
r } zt|||W 5 d}~X Y nX Y nX dS )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)rA rW r" ÚAssertionErrorr rB r )r r rX Úerrr r r ÚresolveÑ s z Scope.resolve)Úold_keyÚnew_keyr c C sP | j r| jj| jj }n| jj}| | j¡ |D ]}||kr0|||< dS q0dS )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)rW r" r rA ÚappendrB )r r\ r] Ú new_valuer r r r r Úswapkeyø s z Scope.swapkeyz list[str])Úscopesr c C sV t ||¡}|D ]@\}\}}}}}}z&t|d| }t| j |¡| _W 5 ~X qdS )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)Ú itertoolsÚproductÚgetattrr
|
rA rK )r Ústackra Ú variablesrA rN Ú_Údr r r Ú _get_vars s zScope._get_varsc C s@ |d }t ¡ }z | j|d|
dgd W 5 |dd
= ~X dS )z
|
Update the current scope by going back `level` levels.
|
|
Parameters
|
----------
|
level : int
|
r NÚlocals)ra )Úinspectre ri )r r Úslre r r r rH ( s
|
z Scope._updatec C sJ t |j d| j dt| }|| 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 )rR r Úntempsr5 rB rY )r r Únamer r r Úadd_tmp< s
|
|
z Scope.add_tmpc C s
|
t | jS )z/The number of temporary variables in this scope)rU rB rV r r r rm T s z Scope.ntempsc C s | j g| 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.
|
)rB r" r rA r
|
)r r r r r Ú
|
full_scopeY s zScope.full_scope)NNr N)N)r r r r Ú __slots__Ú__annotations__rO rS ÚpropertyrW r[ r` ri rH ro rm rp r r r r r { s.
|
ÿ( ')NNr N) r Ú
|
__future__r r6 rk Úior rb r= r/ rI Útypingr r ÚnumpyÚnpZpandas._libs.tslibsr Z pandas.errorsr r r r
|
r$ r+ r5 r9 r: r; rD r@ r r r r r Ú<module> sB ÿ ø
|
