3
���h�# � @ s� d Z ddlZddlZddlmZ ddlZddlZddlZddlZddl m
Z
ddl Z ddl
mZ ddlmZ ddf dfedd�d d
�Zed �d d
�Zed �dd�Zeejddeee je jd�Zed �dd�ZG dd� d�ZdS )z
Module for scope operations
� N)�StringIO)�List)� Timestamp)� DeepChainMap�Scope)�level�returnc K s t | d ||||d�S )z.Ensure that we are grabbing the correct scope.� )� global_dict�
local_dict� resolvers�target)r )r r
r r r
�kwargs� r �?/tmp/pip-build-5_djhm0z/pandas/pandas/core/computation/scope.py� ensure_scope s r )r c C s. y 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 � _replacer! s
r c C s$ t jdt| ��}djdd� |D ��S )z,Return the padded hexadecimal id of ``obj``.z@P� c s s | ]}t |�V qd S )N)r )�.0r r r r � <genexpr>5 s z _raw_hex_id.<locals>.<genexpr>)�struct�pack�id�join)�obj�packedr r r � _raw_hex_id1 s r TF)r �datetime�True�False�list�tuple�infZInfc C s t � }tj| |d� |j� 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)r �sior r r �_get_pretty_stringD s r+ c @ s� e Zd ZdZdddddgZddf dfdd �Zed
�d d �Zee d
�d
d��Z
ee d�dd�Z d!eed�dd�Z e
e d�dd�Zed�dd�Zed
�dd �Zeed
�d d ��Zed d � �ZdS )"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 �tempsNc C s� |d | _ ttj� �| _|| _t|t�rT| jj|j� |jd k rH|j| _| j |j � t
j | j �}z>| jj |pp|j
j� �| _t|t�s�| jj |p�|jj� �| _W d ~X t|t�r�|t|jj�7 }t|� | _i | _d S )Nr )r r �_DEFAULT_GLOBALS�copyr, r
�
isinstancer �update�_update�sys� _getframe� new_child� f_globals�f_localsr% r �mapsr- )�selfr r
r r r
�framer r r �__init__n s$
zScope.__init__)r c C sD t t| jj� ��}t t| jj� ��}t| �j� d|� d|� d�}|S )Nz(scope=z , resolvers=�))r+ r$ r, �keysr �type�__name__)r9 Z
scope_keysZres_keysZ unicode_strr r r �__repr__� s zScope.__repr__c C s t t| j��S )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
)�bool�lenr )r9 r r r �
has_resolvers� s zScope.has_resolvers)�key�is_localc C s� y:|r| j | S | jr | j| S | r.| j s2t�| j | S tk
r� y
| j| S tk
r� } z ddlm} |||�|�W Y dd}~X 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
r )�UndefinedVariableErrorN)r, rC r �AssertionError�KeyErrorr- Zpandas.core.computation.opsrF )r9 rD rE �errrF r r r �resolve� s
z
Scope.resolve)�old_key�new_keyc C sR | j r| jj| jj }n| jj}|j| j� x |D ]}||kr2|||< dS q2W dS )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)rC r r8 r, �appendr- )r9 rK rL Z new_valuer8 �mappingr r r �swapkey� s
z
Scope.swapkey)�scopesc C sT t j||�}xB|D ]:\}\}}}}}}z t|d| �}| jj|�| _W d~X qW dS )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, r5 )r9 �stackrP � variablesr, r: �_�dr r r � _get_vars� s zScope._get_vars)r c C s@ |d }t j� }z | j|d|� dgd� W d|dd�= ~X dS )z�
Update the current scope by going back `level` levels.
Parameters
----------
level : int
r N�locals)rP )�inspectrT rX )r9 r �slrT r r r r2 � 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.
rV )r> r? �ntempsr r- rG )r9 �value�namer r r �add_tmp s
z
Scope.add_tmpc C s
t | j�S )z/The number of temporary variables in this scope)rB r- )r9 r r r r\ * 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.
)r- r r8 r, r )r9 r8 r r r �
full_scope/ s zScope.full_scope)N)r? �
__module__� __qualname__�__doc__� __slots__r; �strr@ �propertyrA rC rJ rO r rX �intr2 r_ r\ r` r r r r r W s !
*) rc r! rZ �ior rQ r( r r3 Ztypingr Znumpy�npZpandas._libs.tslibsr Zpandas.compat.chainmapr rg r re r r r$ r% r& r. r+ r r r r r �<module> s2
|