File: //opt/saltstack/salt/lib/python3.10/site-packages/jmespath/__pycache__/functions.cpython-310.pyc
o
;j�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�bases�attrs�� __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)r7 �argsr r8 r r r r6 S s ���z Functions._validate_argumentsc C s8 t t|��D ]}|| d }|r| �|| ||� qd S )N�types)�ranger> �_type_check_single)r7 �actualr r8 �i�
allowed_typesr r r r? ] s ���zFunctions._type_checkc C sR | � |�\}}t|�j}||vrt�||| �|�|��|r'| �||||� d S d S r )�_get_allowed_pytypes�typer0 r �JMESPathTypeError�_convert_to_jmespath_type�_subtype_check)r7 �currentrA r8 rF �allowed_subtypes�actual_typenamer r r rC 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)r7 rA rF rM �t�type_�subtyper r r rG { 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 )NrP r )r> rH r0 r rI )
r7 rL rM rA r8 �elementrN �firstZsubtypes�allowedr r r rK � s:
���� �
��� zFunctions._subtype_checkrA r
c C � t |�S r )�abs�r7 �argr r r � _func_abs� � zFunctions._func_absz array-numberc C s |r
t |�t|� S d S r )�sumr> r^ r r r � _func_avg� s zFunctions._func_avgT)rA r< c G s |D ]
}|d ur | S qd S r r )r7 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)rf � 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 )rf 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 )r7 �subject�searchr r r �_func_contains� ra zFunctions._func_contains)r r r c C r\ r )r> r^ r r r � _func_length� ra zFunctions._func_lengthc C �
|� |�S r )�endswith�r7 rs �suffixr r r �_func_ends_with� �
zFunctions._func_ends_withc C rv r )r, rx r r r �_func_starts_with� r{ zFunctions._func_starts_withc C s$ t |t�r |d d d� S tt|��S )Nr; )rf rl r
�reversedr^ r r r �
_func_reverse� s
zFunctions._func_reversec C �
t �|�S r )�math�ceilr^ r r r �
_func_ceil� r{ zFunctions._func_ceilc C r r )r� �floorr^ r r r � _func_floor� r{ zFunctions._func_floorz array-stringc C rv r )�join)r7 � separatorr r r r �
_func_join� r{ zFunctions._func_joinr c C s& g }|D ] }|� |�|j|�� q|S r )rS �visit�
expression)r7 r r_ �resultrY 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)r7 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 r\ r )rb r^ r r r � _func_sum ra 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 )rf rl 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)
rJ rH r� r� r0 r rI �_create_key_funcr
r� )r7 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� �r7 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� rH r0 rJ r rI )�xr� rN Z
jmespath_type�rF r r8 r7 r r r� ^ s
�z+Functions._create_key_func.<locals>.keyfuncr )r7 r rF r8 r� r r� r r� ] s zFunctions._create_key_funcc C s t �|d�S )N�unknown)� TYPES_MAPr= )r7 Zpyobjectr r r rJ i s z#Functions._convert_to_jmespath_typeN)'r0 r1 r2 r. r: r6 r? rC rG rK r r` rc re rg ro rq rt ru rz r| r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rJ r r r r r4 C s|
r4 )� metaclass)r� rm Zjmespathr Zjmespath.compatr rl r r� rT r rH r r4 r r r r �<module> s: ��