File: //opt/saltstack/salt/lib/python3.10/site-packages/jmespath/__pycache__/visitor.cpython-310.pyc
o
�N�gR* � @ 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 �NF)�_is_special_number_case��x�y� r
�D/opt/saltstack/salt/lib/python3.10/site-packages/jmespath/visitor.py�_equals s
r c C s< t | �r
| dv r
t|t�S t |�r|dv r t| t�S d S d S )N)r � )�_is_actual_number�
isinstance�boolr 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 r )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 � || _ || _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 r r )�
expression� interpreter)r r r r
r
r r K s
z_Expression.__init__c O s | j j|g|�R i |��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 sR |d }| j �|�}|d u r t| d|d | j�}|| j |< ||g|�R i |��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��)Nr, ��NotImplementedErrorr" r
r
r r, ` � 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 s zTreeInterpreter.<lambda>)�eq�ne�lt�gtZlteZgter3 r4 Nc sd t t| ��� | j| _|d u rt� }|| _|jd ur | jj| _|jd ur+| jj| _ d S t
� � | _ d S r ) �superr1 r �MAP_TYPE� _dict_clsr Z_optionsr r �
_functionsr Z Functions)r �options�� __class__r
r r p s
zTreeInterpreter.__init__c O s t |d ��)Nr) r. r" r
r
r r, } � z TreeInterpreter.default_visitc C �" |}|d D ]}| � ||�}q|S �N�children�r! �r r# �value�resultr
r
r �visit_subexpression� � z#TreeInterpreter.visit_subexpressionc C s& z|� |d �W S ty Y d S w �NrD )r* �AttributeError�r r# rD r
r
r � visit_field� s
�zTreeInterpreter.visit_fieldc C s� | j |d }|d | jv r#|| �|d d |�| �|d d |��S | �|d d |�}| �|d d |�}ttf}t|�rCt|�sEd S |||�S )NrD rA r r
)�COMPARATOR_FUNC�
_EQUALITY_OPSr! �int�floatr )r r# rD Zcomparator_func�left�rightZ num_typesr
r
r �visit_comparator� s ��
z TreeInterpreter.visit_comparatorc C � |S r r
rJ r
r
r �
visit_current� � z TreeInterpreter.visit_currentc C s t |d d | �S �NrA r )r rJ r
r
r � visit_expref� s z TreeInterpreter.visit_exprefc C s: g }|d D ]
}| � ||�}|�|� q| j�|d |�S �NrA rD )r! �appendr: �
call_function)r r# rD Z
resolved_args�child�currentr
r
r �visit_function_expression� s
z)TreeInterpreter.visit_function_expressionc C sx | � |d d |�}t|t�sd S |d d }g }|D ] }| �| � ||��r9| � |d d |�}|d ur9|�|� q|S )NrA r � r
)r! r �list�_is_truerY )r r# rD �baseZcomparator_node� collected�elementr\ r
r
r �visit_filter_projection� s
�z'TreeInterpreter.visit_filter_projectionc C sT | � |d d |�}t|t�sd S g }|D ]}t|t�r"|�|� q|�|� q|S rV )r! r r_ �extendrY )r r# rD ra Z merged_listrc r
r
r �
visit_flatten� s
z TreeInterpreter.visit_flattenc C rS r r
rJ r
r
r �visit_identity� rU z TreeInterpreter.visit_identityc C s2 t |t�sd S z||d W S ty Y d S w rH )r r_ �
IndexErrorrJ r
r
r � visit_index� s
�zTreeInterpreter.visit_indexc C r? r@ rB rC r
r
r �visit_index_expression� rG z&TreeInterpreter.visit_index_expressionc C s" t |t�sd S t|d � }|| S r@ )r r_ �slice)r r# rD �sr
r
r � visit_slice� s
zTreeInterpreter.visit_slicec C s | � |d d |�S rV rB rJ r
r
r �visit_key_val_pair� s z"TreeInterpreter.visit_key_val_pairc C s |d S rH r
rJ r
r
r �
visit_literal� r0 z TreeInterpreter.visit_literalc C s: |d u rd S | � � }|d D ] }| �||�||d < q|S rX )r9 r! �r r# rD rb r[ r
r
r �visit_multi_select_dict� s z'TreeInterpreter.visit_multi_select_dictc C s4 |d u rd S g }|d D ] }|� | �||�� q |S r@ )rY r! rp r
r
r �visit_multi_select_list� s z'TreeInterpreter.visit_multi_select_listc C s6 | � |d d |�}| �|�r| � |d d |�}|S �NrA r r
�r! � _is_false�r r# rD Zmatchedr
r
r �visit_or_expression� s
z#TreeInterpreter.visit_or_expressionc C s6 | � |d d |�}| �|�r|S | � |d d |�S rs rt rv r
r
r �visit_and_expression� s
z$TreeInterpreter.visit_and_expressionc C s. | � |d d |�}t|�r|dkrdS | S )NrA r F)r! r )r r# rD Zoriginal_resultr
r
r �visit_not_expression s z$TreeInterpreter.visit_not_expressionc C r? r@ rB rC r
r
r �
visit_pipe
rG zTreeInterpreter.visit_pipec C sZ | � |d d |�}t|t�sd S g }|D ]}| � |d d |�}|d ur*|�|� q|S rs )r! r r_ rY �r r# rD ra rb rc r\ r
r
r �visit_projection s
�z TreeInterpreter.visit_projectionc C sn | � |d d |�}z|�� }W n
ty Y d S w g }|D ]}| � |d d |�}|d ur4|�|� q |S rs )r! �valuesrI rY r{ r
r
r �visit_value_projection s �
�z&TreeInterpreter.visit_value_projectionc C s( |dkp|g kp|i kp|d u p|du S )N� Fr
�r rD r
r
r ru ( s �zTreeInterpreter._is_falsec C s | � |� S r )ru r� r
r
r r` / r> zTreeInterpreter._is_truer )(r r r r �operatorr5 r6 �le�gerL rM �dictr8 r r, rF rK rR rT rW r] rd rf rg ri rj rm rn ro rq rr rw rx ry rz r| r~ ru r` �
__classcell__r
r
r< r r1 d sH �
r1 c s, e Zd Z� fdd�Zdd� Zdd� Z� ZS )�GraphvizVisitorc s t t| ��� g | _d| _d S )Nr
)r7 r� r �_lines�_countr( r< 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� rY r� �_visit�join)r r# r$ r% r\ r
r
r r! 9 s zGraphvizVisitor.visitc C sx | j �d||d |�dd�f � |�dg �D ]"}d|d | jf }| jd7 _| j �d||f � | �||� qd S ) Nz%s [label="%s(%s)"]r) rD r rA r� r
z
%s -> %s)r� rY r* r� r� )r r# r\ r[ Z
child_namer
r
r r� A s ��zGraphvizVisitor._visit)r r r r r! r� r� r
r
r< r r� 3 s r� )r� Zjmespathr Zjmespath.compatr Znumbersr r r r r �objectr r r& r1 r� r
r
r
r �<module> s
P