File: //opt/saltstack/salt/lib/python3.10/site-packages/jmespath/__pycache__/functions.cpython-310.pyc
o
�N�g�1 �
@ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ ddddd d d
d
d
ddd d � Zd
ddddddd�Z dd� Z
G dd� de �Z G dd� de d�Z
dS ) � N)�
exceptions)� string_type)� get_methods�boolean�array�object�null�string�number�expref) �bool�list�dict�NoneType�unicode�str�float�int�long� OrderedDict� _Projection� _Expression)r )r
r )r r )r )r r )r r r )r )r r r r r r
r c s � fdd�}|S )Nc s
� | _ | S �N)� signature)�func�� arguments� �F/opt/saltstack/salt/lib/python3.10/site-packages/jmespath/functions.py�_record_signature' s z$signature.<locals>._record_signaturer )r r r r r r & s r c s$ e Zd Z� fdd�Zdd� Z� ZS )�FunctionRegistryc s | � � tt| ��|||� d S r )�_populate_function_table�superr �__init__)�cls�name�basesZattrs�� __class__r r r# . s zFunctionRegistry.__init__c C sV i }t | �D ] \}}|�d�sqt|dd �}|d ur%||d�||dd � < q|| _d S )NZ_func_r )�functionr � )r �
startswith�getattr�FUNCTION_TABLE)r$ Zfunction_tabler% �methodr r r r r! 2 s
��
z)FunctionRegistry._populate_function_table)�__name__�
__module__� __qualname__r# r! �
__classcell__r r r' r r - s r c @ s� e Zd Zi Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Z e
d
dgi�dd� �Z e
d
dgi�dd� �Z e
g dd��dd� �Z
e
d
g i�dd� �Ze
d
g i�dd� �Ze
d
g i�d d � �Ze
d
d d gid
g i�d d!� �Ze
d
g d"�i�d#d$� �Ze
d
d gid
d gi�d%d&� �Ze
d
d gid
d gi�d'd(� �Ze
d
d d gi�d)d*� �Ze
d
dgi�d+d,� �Ze
d
dgi�d-d.� �Ze
d
d gid
d/gi�d0d1� �Ze
d
d2gid
d gi�d3d4� �Ze
d
dd/gi�d5d6� �Ze
d7gdd��d8d9� �Ze
d
dd/gi�d:d;� �Z e
d
d/dgi�d<d=� �Z e
d
dgi�d>d?� �Z e
d
d7gi�d@dA� �Z e
d
d7gi�dBdC� �Z e
d
g i�dDdE� �Z!e
d
d gid
d2gi�dFdG� �Z"e
d
d gid
d2gi�dHdI� �Z#e
d
d gid
d2gi�dJdK� �Z$dLdM� Z%dNdO� Z&dPS )Q� Functionsc C sZ z| j | }W n ty t�d| ��w |d }|d }| �|||� || g|�R � S )NzUnknown function: %s()r) r )r- �KeyErrorr ZUnknownFunctionError�_validate_arguments)�self�
function_nameZ
resolved_args�specr) r r r r �
call_functionH s ��zFunctions.call_functionc C sn |r |d � d�r t|�t|�k r t�t|�t|�|��nt|�t|�kr0t�t|�t|�|��| �|||�S )N����variadic)�get�lenr ZVariadictArityErrorZ
ArityError� _type_check)r6 �argsr r7 r r r r5 S s ���z Functions._validate_argumentsc C s8 t t|��D ]}|| d }|r| �|| ||� qd S )N�types)�ranger= �_type_check_single)r6 �actualr r7 �i�
allowed_typesr r r r> ] s ���zFunctions._type_checkc C sR | � |�\}}t|�j}||vrt�||| �|�|��|r'| �||||� d S d S r )�_get_allowed_pytypes�typer/ r �JMESPathTypeError�_convert_to_jmespath_type�_subtype_check)r6 �currentr@ r7 rE �allowed_subtypes�actual_typenamer r r rB d s
���z Functions._type_check_singlec C s` g }g }|D ]%}|� dd�}t|�dkr |\}}|�t| � n|d }|�t| � q||fS )N�-� � r )�splitr= �append�REVERSE_TYPES_MAP�extend)r6 r@ rE rL �tZtype_�subtyper r r rF { s z Functions._get_allowed_pytypesc
C s� t |�dkr"|d }|D ]}t|�j}||vr t�||||��q d S t |�dkr^|r`t|d �j}|D ]
}||v r=|} n q3t�||d ||��|D ]}t|�j}|| vr]t�||||��qJd S d S d S )NrO r )r= rG r/ r rH )
r6 rK rL r@ r7 �elementrM �firstZsubtypes�allowedr r r rJ � s:
���� �
��� zFunctions._subtype_checkr@ r
c C � t |�S r )�abs�r6 �argr r r � _func_abs� � zFunctions._func_absz array-numberc C s |r t |�tt|�� S d S r )�sumr r= r\ r r r � _func_avg� s zFunctions._func_avgT)r@ r; c G s |D ]
}|d ur | S qd S r r )r6 r �argumentr r r �_func_not_null� s
��zFunctions._func_not_nullc C s t |t�r|S |gS r )�
isinstancer
r\ r r r �_func_to_array� s
zFunctions._func_to_arrayc C s t |t�r|S tj|dtd�S )N)�,�:)�
separators�default)rd � STRING_TYPE�json�dumpsr r\ r r r �_func_to_string� s
�zFunctions._func_to_stringc C sr t |tttf�r
d S |d u rd S t |ttf�r|S zt|�W S ty8 zt|�W Y S ty7 Y Y d S w w r )rd r
r r r r �
ValueErrorr\ r r r �_func_to_number� s
��zFunctions._func_to_numberr r c C s ||v S r r )r6 �subject�searchr r r �_func_contains� r_ zFunctions._func_contains)r r r c C rZ r )r= r\ r r r � _func_length� r_ zFunctions._func_lengthc C �
|� |�S r )�endswith�r6 rq �suffixr r r �_func_ends_with� �
zFunctions._func_ends_withc C rt r )r+ rv r r r �_func_starts_with� ry zFunctions._func_starts_withc C s$ t |t�r |d d d� S tt|��S )Nr: )rd rj r
�reversedr\ r r r �
_func_reverse� s
zFunctions._func_reversec C �
t �|�S r )�math�ceilr\ r r r �
_func_ceil� ry zFunctions._func_ceilc C r} r )r~ �floorr\ r r r � _func_floor� ry zFunctions._func_floorz array-stringc C rt r )�join)r6 � separatorr r r r �
_func_join� ry zFunctions._func_joinr c C s& g }|D ] }|� |�|j|�� q|S r )rR �visit�
expression)r6 r r] �resultrW r r r � _func_map� s zFunctions._func_mapc C � |rt |�S d S r )�maxr\ r r r � _func_max� � zFunctions._func_maxr c G s i }|D ]}|� |� q|S r )�update)r6 r Zmergedr] r r r � _func_merge s zFunctions._func_mergec C r� r )�minr\ r r r � _func_min
r� zFunctions._func_minc C s t t|��S r )r
�sortedr\ r r r �
_func_sort � zFunctions._func_sortc C rZ r )r` r\ r r r � _func_sum r_ zFunctions._func_sumc C � t |�� �S r )r
�keysr\ r r r �
_func_keys s zFunctions._func_keysc C r� r )r
�valuesr\ r r r � _func_values" r� zFunctions._func_valuesc C sZ t |t�rdS t |t�rdS t |t�rdS t |t�r dS t |ttf�r%dS |d u r+dS d S )Nr r r r r
r )rd rj r r
r r r r\ r r r �
_func_type& s
�zFunctions._func_typec C sf |s|S | � t|�|j|d ��j�}|dvr#t�d|d |ddg��| �||gd�}tt ||d��S )Nr )r
r Zsort_byr r
��key)
rI rG r� r� r/ r rH �_create_key_funcr
r� )r6 r r Z
required_type�keyfuncr r r �
_func_sort_by5 s ���zFunctions._func_sort_byc C �& | � |ddgd�}|rt||d�S d S )Nr
r Zmin_byr� )r� r� �r6 r r r� r r r � _func_min_byI � � zFunctions._func_min_byc C r� )Nr
r Zmax_byr� )r� r� r� r r r � _func_max_byS r� zFunctions._func_max_byc s � ���fdd�}|S )Nc s> �� �j| �}t|�j}��|�}|� vr t��||� ��|S r )r� r� rG r/ rI r rH )�xr� rM Z
jmespath_type�rE r r7 r6 r r r� ^ s
�z+Functions._create_key_func.<locals>.keyfuncr )r6 r rE r7 r� r r� r r� ] s zFunctions._create_key_funcc C s t �|d�S )N�unknown)� TYPES_MAPr< )r6 Zpyobjectr r r rI i s z#Functions._convert_to_jmespath_typeN)'r/ r0 r1 r- r9 r5 r> rB rF rJ r r^ ra rc re rm ro rr rs rx rz r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rI r r r r r3 C s|
r3 )� metaclass)r~ rk Zjmespathr Zjmespath.compatr rj r r� rS r rG r r3 r r r r �<module> s: ��