File: //lib/python3.6/site-packages/jmespath/__pycache__/visitor.cpython-36.opt-1.pyc
3
g� VF$ � @ sx d dl Z d dlmZ dd� Zdd� ZG dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de�Z G dd� de�Z
dS )� N)� functionsc C s t | |�rdS | |kS d S )NF)�_is_special_integer_case)�x�y� r � /usr/lib/python3.6/visitor.py�_equals s
r c C sD | dks| dkr |dkp |dkS |dks0|dkr@| dkp>| dkS d S )Nr � TFr )r r r r r r
s r c @ s e Zd ZdZdd� ZdS )�Optionsz8Options to control how a JMESPath function is evaluated.c C s
|| _ d S )N)�dict_cls)�selfr r r r �__init__& s zOptions.__init__N)�__name__�
__module__� __qualname__�__doc__r
r r r r r
$ s r
c @ s e Zd Zdd� ZdS )� _Expressionc C s
|| _ d S )N)�
expression)r r r r r r
3 s z_Expression.__init__N)r r r r
r r r r r 2 s r c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Visitorc C s
i | _ d S )N)�
_method_cache)r r r r r
8 s zVisitor.__init__c O sL |d }| j j|�}|d kr<t| d|d | j�}|| j |< ||f|�|�S )N�typezvisit_%s)r �get�getattr�
default_visit)r �node�args�kwargsZ node_type�methodr r r �visit; s
z
Visitor.visitc O s t d��d S )Nr )�NotImplementedError)r r r r r r r r D s zVisitor.default_visitN)r r r r
r r r r r r r 7 s r c s e Zd Zejdd� ejejeejejd�Z e
Z d9� fdd� Z dd� 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%� Z&S ):�TreeInterpreterc C s t | |� S )N)r )r r r r r �<lambda>K s zTreeInterpreter.<lambda>)�le�ne�ltZlte�eq�gtZgteNc sN t t| �j� || _| j| _|d k r8|jd k r8| jj| _tj� | _ | | j _
d S )N) �superr r
Z_options�MAP_TYPE� _dict_clsr r ZRuntimeFunctions�
_functionsZ interpreter)r Zoptions)� __class__r r r
T 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 ]}| j ||�}qW |S )N�children)r )r r �value�resultr r r �visit_subexpressionc s z#TreeInterpreter.visit_subexpressionc C s( y|j |d �S tk
r" d S X d S )Nr- )r �AttributeError)r r r- r r r � visit_fieldi s zTreeInterpreter.visit_fieldc C s8 | j |d }|| j|d d |�| j|d d |��S )Nr- r, r r )�COMPARATOR_FUNCr )r r r- Zcomparator_funcr r r �visit_comparatoro s z TreeInterpreter.visit_comparatorc C s |S )Nr )r r r- r r r �
visit_currentv s z TreeInterpreter.visit_currentc C s t |d d �S )Nr, r )r )r r r- r r r � visit_exprefy s z TreeInterpreter.visit_exprefc C s> g }x&|d D ]}| j ||�}|j|� qW | jj|d |�S )Nr, r- )r �appendr* Z
call_function)r r r- Z
resolved_args�child�currentr r r �visit_function_expression| s
z)TreeInterpreter.visit_function_expressionc C s| | j |d d |�}t|t�s"d S |d d }g }xD|D ]<}| j| j ||��r8| j |d d |�}|d k r8|j|� q8W |S )Nr, r � r )r �
isinstance�list�_is_truer6 )r r r- �baseZcomparator_node� collected�elementr8 r r r �visit_filter_projection� s
z'TreeInterpreter.visit_filter_projectionc C sX | j |d d |�}t|t�s"d S g }x,|D ]$}t|t�rF|j|� q,|j|� q,W |S )Nr, r )r r; r< �extendr6 )r r r- r> Z merged_listr@ r r r �
visit_flatten� s
z TreeInterpreter.visit_flattenc C s |S )Nr )r r r- 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 )Nr- )r; r< �
IndexError)r r r- r r r � visit_index� s
zTreeInterpreter.visit_indexc C s& |}x |d D ]}| j ||�}qW |S )Nr, )r )r r r- r. r r r �visit_index_expression� s z&TreeInterpreter.visit_index_expressionc C s" t |t�sd S t|d � }|| S )Nr, )r; r< �slice)r r r- �sr r r � visit_slice� s
zTreeInterpreter.visit_slicec C s | j |d d |�S )Nr, r )r )r r r- r r r �visit_key_val_pair� s z"TreeInterpreter.visit_key_val_pairc C s |d S )Nr- r )r r r- r r r �
visit_literal� s z TreeInterpreter.visit_literalc C s> |d kr d S | j � }x$|d D ]}| j||�||d < q W |S )Nr, r- )r) r )r r r- r? r7 r r r �visit_multi_select_dict� s z'TreeInterpreter.visit_multi_select_dictc C s8 |d kr d S g }x"|d D ]}|j | j||�� qW |S )Nr, )r6 r )r r r- r? r7 r r r �visit_multi_select_list� s z'TreeInterpreter.visit_multi_select_listc C s6 | j |d d |�}| j|�r2| j |d d |�}|S )Nr, r r )r � _is_false)r r r- �matchedr r r �visit_or_expression� s
z#TreeInterpreter.visit_or_expressionc C s6 | j |d d |�}| j|�r"|S | j |d d |�S )Nr, r r )r rO )r r r- rP r r r �visit_and_expression� s
z$TreeInterpreter.visit_and_expressionc C s& | j |d d |�}|dkr dS | S )Nr, r F)r )r r r- Zoriginal_resultr r r �visit_not_expression� s z$TreeInterpreter.visit_not_expressionc C s& |}x |d D ]}| j ||�}qW |S )Nr, )r )r r r- r. r r r �
visit_pipe� s zTreeInterpreter.visit_pipec C s^ | j |d d |�}t|t�s"d S g }x2|D ]*}| j |d d |�}|d k r,|j|� q,W |S )Nr, r r )r r; r< r6 )r r r- r> r? r@ r8 r r r �visit_projection� s
z TreeInterpreter.visit_projectionc C sr | j |d d |�}y |j� }W n tk
r4 d S X g }x2|D ]*}| j |d d |�}|d k r@|j|� q@W |S )Nr, r r )r �valuesr0 r6 )r r r- r> r? r@ r8 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 r- r r r rO � s zTreeInterpreter._is_falsec C s | j |� S )N)rO )r r- r r r r= s zTreeInterpreter._is_true)N)'r r r �operatorr" r$ r r& �ger2 �dictr( r
r r/ r1 r3 r4 r5 r9 rA rC rD rF rG rJ rK rL rM rN rQ rR rS rT rU rW rO r= �
__classcell__r r )r+ r r H sD
r c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�GraphvizVisitorc s t t| �j� g | _d| _d S )Nr )r' r] r
�_lines�_count)r )r+ r r r
s zGraphvizVisitor.__init__c O sP | j jd� d|d | jf }| jd7 _| j||� | j jd� dj| j �S )Nz
digraph AST {z%s%sr r �}�
)r^ r6 r_ �_visit�join)r r r r r8 r r r r s zGraphvizVisitor.visitc C s| | j jd||d |jdd�f � xT|jdg �D ]D}d|d | jf }| jd7 _| j jd||f � | j||� q0W d S ) Nz%s [label="%s(%s)"]r r- rX r, z%s%sr z
%s -> %s)r^ r6 r r_ rb )r r r8 r7 Z
child_namer r r rb s zGraphvizVisitor._visit)r r r r
r rb r\ r r )r+ r r] s r] ) rY Zjmespathr r r �objectr
r r r r] r r r r �<module> s B