File: //lib/python3/dist-packages/jmespath/__pycache__/visitor.cpython-38.pyc
U
@`a\* � @ 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_integer_case��x�y� r �2/usr/lib/python3/dist-packages/jmespath/visitor.py�_equals s
r c C s\ t | �tkr,| dks | dkr,|dkp*|dkS t |�tkrX|dksH|dkrX| dkpV| dkS d S )Nr � TF)�type�intr r r r
r s r c C s t | �pt| t�S �N)�_is_actual_number�
isinstancer �r r r r
�_is_comparable& s r c C s | dks| dkrdS t | t�S )NTF)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 r )�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 r )�
expression� interpreter)r r r r r r
r K s z_Expression.__init__c O s | j j|f|�|�S r )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 r )�
_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 )Nr
zvisit_%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* ��NotImplementedErrorr! 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 r )r r r r r
�<lambda>g � zTreeInterpreter.<lambda>)�eq�ne�lt�gtZlteZgter1 r2 Nc sb t t| ��� | j| _|d kr$t� }|| _|jd k r>| jj| _|jd k rT| jj| _ n
t
� � | _ d S r ) �superr. r �MAP_TYPE� _dict_clsr Z_optionsr r �
_functionsr Z Functions)r Zoptions�� __class__r r
r p s
zTreeInterpreter.__init__c O s t |d ��d S )Nr
r, r! r r r
r* } s z TreeInterpreter.default_visitc C s" |}|d D ]}| � ||�}q |S �N�children�r �r r" �value�resultr r r
�visit_subexpression� s z#TreeInterpreter.visit_subexpressionc C s, z|� |d �W S tk
r& Y d S X d S �Nr? )r( �AttributeError�r r" r? 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 )Nr? r< r r )�COMPARATOR_FUNC�
_EQUALITY_OPSr r �floatr )r r" r? Zcomparator_func�left�rightZ num_typesr r r
�visit_comparator� s ��z TreeInterpreter.visit_comparatorc C s |S r r rD r r r
�
visit_current� s z TreeInterpreter.visit_currentc C s t |d d | �S �Nr< r )r rD r r r
� visit_expref� s z TreeInterpreter.visit_exprefc C s: g }|d D ]}| � ||�}|�|� q | j�|d |�S �Nr< r? )r �appendr8 Z
call_function)r r" r? Z
resolved_args�child�currentr r r
�visit_function_expression� s
z)TreeInterpreter.visit_function_expressionc C sx | � |d d |�}t|t�s"d S |d d }g }|D ]<}| �| � ||��r6| � |d d |�}|d k r6|�|� q6|S )Nr< r � r )r r �list�_is_truerP )r r" r? �baseZcomparator_node� collected�elementrR r r r
�visit_filter_projection� s
z'TreeInterpreter.visit_filter_projectionc C sT | � |d d |�}t|t�s"d S g }|D ]$}t|t�rD|�|� q*|�|� q*|S rM )r r rU �extendrP )r r" r? rW Z merged_listrY r r r
�
visit_flatten� s
z TreeInterpreter.visit_flattenc C s |S r r rD r r r
�visit_identity� s z TreeInterpreter.visit_identityc C s8 t |t�sd S z||d W S tk
r2 Y d S X d S rB )r rU �
IndexErrorrD r r r
� visit_index� s
zTreeInterpreter.visit_indexc C s" |}|d D ]}| � ||�}q |S r; r= r> r r r
�visit_index_expression� s z&TreeInterpreter.visit_index_expressionc C s" t |t�sd S t|d � }|| S r; )r rU �slice)r r" r? �sr r r
� visit_slice� s
zTreeInterpreter.visit_slicec C s | � |d d |�S rM r= rD r r r
�visit_key_val_pair� s z"TreeInterpreter.visit_key_val_pairc C s |d S rB r rD r r r
�
visit_literal� s z TreeInterpreter.visit_literalc C s: |d kr d S | � � }|d D ]}| �||�||d < q |S rO )r7 r �r r" r? rX rQ r r r
�visit_multi_select_dict� s z'TreeInterpreter.visit_multi_select_dictc C s4 |d kr d S g }|d D ]}|� | �||�� q|S r; )rP r rf r r r
�visit_multi_select_list� s z'TreeInterpreter.visit_multi_select_listc C s6 | � |d d |�}| �|�r2| � |d d |�}|S �Nr< r r �r � _is_false�r r" r? Zmatchedr r r
�visit_or_expression� s
z#TreeInterpreter.visit_or_expressionc C s6 | � |d d |�}| �|�r"|S | � |d d |�S ri rj rl r r r
�visit_and_expression� s
z$TreeInterpreter.visit_and_expressionc C s2 | � |d d |�}t|�tkr,|dkr,dS | S )Nr< r F)r r
r )r r" r? Zoriginal_resultr r r
�visit_not_expression s z$TreeInterpreter.visit_not_expressionc C s" |}|d D ]}| � ||�}q |S r; r= r> r r r
�
visit_pipe
s zTreeInterpreter.visit_pipec C sZ | � |d d |�}t|t�s"d S g }|D ]*}| � |d d |�}|d k r*|�|� q*|S ri )r r rU rP �r r" r? rW rX rY rR r r r
�visit_projection s
z TreeInterpreter.visit_projectionc C sp | � |d d |�}z |�� }W n tk
r6 Y d S X g }|D ]*}| � |d d |�}|d k r@|�|� q@|S ri )r �valuesrC rP rq 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
rk ( s �zTreeInterpreter._is_falsec C s | � |� S r )rk rv r r r
rV / s zTreeInterpreter._is_true)N)(r r r r �operatorr3 r4 �le�gerF rG �dictr6 r r* rA rE rK rL rN rS rZ r\ r] r_ r` rc rd re rg rh rm rn ro rp rr rt rk rV �
__classcell__r r r9 r
r. d sF �
r. c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�GraphvizVisitorc s t t| ��� g | _d| _d S )Nr )r5 r| r �_lines�_countr' r9 r r
r 4 s zGraphvizVisitor.__init__c O sP | j �d� d|d | jf }| jd7 _| �||� | j �d� d�| j �S )Nz
digraph AST {�%s%sr
r �}�
)r} rP r~ �_visit�join)r r" r# r$ rR r r r
r 9 s zGraphvizVisitor.visitc C sx | j �d||d |�dd�f � |�dg �D ]D}d|d | jf }| jd7 _| j �d||f � | �||� q.d S ) Nz%s [label="%s(%s)"]r
r? ru r< r r z
%s -> %s)r} rP r( r~ r� )r r" rR rQ Z
child_namer r r
r� A s
�zGraphvizVisitor._visit)r r r r r r� r{ r r r9 r
r| 3 s r| )rw Zjmespathr Zjmespath.compatr Znumbersr r r r r �objectr r r% r. r| r r r r
�<module> s
P