File: //opt/alt/python37/lib/python3.7/site-packages/jmespath/__pycache__/visitor.cpython-37.pyc
B
�P�eR* � @ s� d dl Z d dlmZ d dlmZ d dlmZ dd� Zdd� Zd d
� Z d d � Z
G d
d� de �Z G dd� de �Z
G dd� de �ZG dd� de�ZG dd� de�ZdS )� N)� functions)� string_type)�Numberc C s t | |�rdS | |kS d S )NF)�_is_special_number_case)�x�y� r �A/opt/alt/python37/lib/python3.7/site-packages/jmespath/visitor.py�_equals s
r
c C s8 t | �r| dkrt|t�S t |�r4|dkr4t| t�S d S )N)r � )�_is_actual_number�
isinstance�bool)r r r r r r s
r c C s t | �pt| t�S )N)r r
r )r r r r �_is_comparable& s r c C s t | t�rdS t | t�S )NF)r
r r )r r r r r . s
r c @ s e Zd ZdZddd�ZdS )�Optionsz8Options to control how a JMESPath function is evaluated.Nc C s || _ || _d S )N)�dict_cls�custom_functions)�selfr r r r r �__init__= s zOptions.__init__)NN)�__name__�
__module__� __qualname__�__doc__r r r r r r ; s r c @ s e Zd Zdd� Zdd� ZdS )� _Expressionc C s || _ || _d S )N)�
expression� interpreter)r r r r r r r K s z_Expression.__init__c O s | j j|f|�|�S )N)r �visit)r �node�args�kwargsr r r r O s z_Expression.visitN)r r r r r r r r r r J s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Visitorc C s
i | _ d S )N)�
_method_cache)r r r r r T s zVisitor.__init__c O sL |d }| j �|�}|d kr<t| d|d | j�}|| j |< ||f|�|�S )N�typezvisit_%s)r! �get�getattr�
default_visit)r r r r Z node_type�methodr r r r W s
z
Visitor.visitc O s t d��d S )Nr% )�NotImplementedError)r r r r r r r r% ` s zVisitor.default_visitN)r r r r r r% r r r r r S s r c s e Zd Zedd� ejejejejd�Z ddgZ
e Z d;� fdd� Z
d d
� Zd d � Zd
d� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Z d'd(� Z d)d*� Z d+d,� Z d-d.� Z d/d0� Z!d1d2� Z"d3d4� Z#d5d6� Z$d7d8� Z%d9d:� Z&� Z'S )<�TreeInterpreterc C s t | |� S )N)r
)r r r r r �<lambda>g � zTreeInterpreter.<lambda>)�eq�ne�lt�gtZlteZgter+ r, Nc sb t t| ��� | j| _|d kr$t� }|| _|jd k r>| jj| _|jd k rT| jj| _ n
t
� � | _ d S )N) �superr( r �MAP_TYPE� _dict_clsr �_optionsr r �
_functionsr Z Functions)r �options)� __class__r r r p s
zTreeInterpreter.__init__c O s t |d ��d S )Nr" )r' )r r r r r r r r% } s z TreeInterpreter.default_visitc C s& |}x |d D ]}| � ||�}qW |S )N�children)r )r r �value�resultr r r �visit_subexpression� s z#TreeInterpreter.visit_subexpressionc C s( y|� |d �S tk
r" d S X d S )Nr7 )r# �AttributeError)r r r7 r r r � visit_field� s zTreeInterpreter.visit_fieldc C s� | j |d }|d | jkrF|| �|d d |�| �|d d |��S | �|d d |�}| �|d d |�}ttf}t|�r�t|�s�d S |||�S d S )Nr7 r6 r r )�COMPARATOR_FUNC�
_EQUALITY_OPSr �int�floatr )r r r7 Zcomparator_func�left�rightZ num_typesr r r �visit_comparator� s z TreeInterpreter.visit_comparatorc C s |S )Nr )r r r7 r r r �
visit_current� s z TreeInterpreter.visit_currentc C s t |d d | �S )Nr6 r )r )r r r7 r r r � visit_expref� s z TreeInterpreter.visit_exprefc C s> g }x&|d D ]}| � ||�}|�|� qW | j�|d |�S )Nr6 r7 )r �appendr3 Z
call_function)r r r7 Z
resolved_args�child�currentr r r �visit_function_expression� s
z)TreeInterpreter.visit_function_expressionc C s| | � |d d |�}t|t�s"d S |d d }g }xD|D ]<}| �| � ||��r8| � |d d |�}|d k r8|�|� q8W |S )Nr6 r � r )r r
�list�_is_truerE )r r r7 �baseZcomparator_node� collected�elementrG r r r �visit_filter_projection� s
z'TreeInterpreter.visit_filter_projectionc C sX | � |d d |�}t|t�s"d S g }x,|D ]$}t|t�rF|�|� q,|�|� q,W |S )Nr6 r )r r
rJ �extendrE )r r r7 rL Z merged_listrN r r r �
visit_flatten� s
z TreeInterpreter.visit_flattenc C s |S )Nr )r r r7 r r r �visit_identity� s z TreeInterpreter.visit_identityc C s4 t |t�sd S y ||d S tk
r. d S X d S )Nr7 )r
rJ �
IndexError)r r r7 r r r � visit_index� s
zTreeInterpreter.visit_indexc C s& |}x |d D ]}| � ||�}qW |S )Nr6 )r )r r r7 r8 r r r �visit_index_expression� s z&TreeInterpreter.visit_index_expressionc C s" t |t�sd S t|d � }|| S )Nr6 )r
rJ �slice)r r r7 �sr r r � visit_slice� s
zTreeInterpreter.visit_slicec C s | � |d d |�S )Nr6 r )r )r r r7 r r r �visit_key_val_pair� s z"TreeInterpreter.visit_key_val_pairc C s |d S )Nr7 r )r r r7 r r r �
visit_literal� s z TreeInterpreter.visit_literalc C s> |d kr d S | � � }x$|d D ]}| �||�||d < q W |S )Nr6 r7 )r1 r )r r r7 rM rF r r r �visit_multi_select_dict� s z'TreeInterpreter.visit_multi_select_dictc C s8 |d kr d S g }x"|d D ]}|� | �||�� qW |S )Nr6 )rE r )r r r7 rM rF r r r �visit_multi_select_list� s z'TreeInterpreter.visit_multi_select_listc C s6 | � |d d |�}| �|�r2| � |d d |�}|S )Nr6 r r )r � _is_false)r r r7 �matchedr r r �visit_or_expression� s
z#TreeInterpreter.visit_or_expressionc C s6 | � |d d |�}| �|�r"|S | � |d d |�S )Nr6 r r )r r] )r r r7 r^ r r r �visit_and_expression� s
z$TreeInterpreter.visit_and_expressionc C s. | � |d d |�}t|�r(|dkr(dS | S )Nr6 r F)r r )r r r7 Zoriginal_resultr r r �visit_not_expression s z$TreeInterpreter.visit_not_expressionc C s& |}x |d D ]}| � ||�}qW |S )Nr6 )r )r r r7 r8 r r r �
visit_pipe
s zTreeInterpreter.visit_pipec C s^ | � |d d |�}t|t�s"d S g }x2|D ]*}| � |d d |�}|d k r,|�|� q,W |S )Nr6 r r )r r
rJ rE )r r r7 rL rM rN rG r r r �visit_projection s
z TreeInterpreter.visit_projectionc C sr | � |d d |�}y |�� }W n tk
r4 d S X g }x2|D ]*}| � |d d |�}|d k r@|�|� q@W |S )Nr6 r r )r �valuesr: rE )r r r7 rL rM rN rG r r r �visit_value_projection s
z&TreeInterpreter.visit_value_projectionc C s( |dkp&|g kp&|i kp&|d kp&|dkS )N� Fr )r r7 r r r r] ( s zTreeInterpreter._is_falsec C s | � |� S )N)r] )r r7 r r r rK / s zTreeInterpreter._is_true)N)(r r r r
�operatorr- r. �le�ger< r= �dictr0 r r% r9 r; rB rC rD rH rO rQ rR rT rU rX rY rZ r[ r\ r_ r` ra rb rc re r] rK �
__classcell__r r )r5 r r( d sD
r( c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�GraphvizVisitorc s t t| ��� g | _d| _d S )Nr )r/ rl r �_lines�_count)r )r5 r r r 4 s zGraphvizVisitor.__init__c O sP | j �d� d|d | jf }| jd7 _| �||� | j �d� d�| j �S )Nz
digraph AST {z%s%sr" r �}�
)rm rE rn �_visit�join)r r r r rG r r r r 9 s zGraphvizVisitor.visitc C s| | j �d||d |�dd�f � xT|�dg �D ]D}d|d | jf }| jd7 _| j �d||f � | �||� q0W d S ) Nz%s [label="%s(%s)"]r" r7 rf r6 z%s%sr z
%s -> %s)rm rE r# rn rq )r r rG rF Z
child_namer r r rq A s zGraphvizVisitor._visit)r r r r r rq rk r r )r5 r rl 3 s rl )rg �jmespathr Zjmespath.compatr �numbersr r
r r r �objectr r r r( rl r r r r �<module> s
P