File: //usr/local/CyberCP/lib/python3.10/site-packages/jmespath/__pycache__/functions.cpython-310.pyc
o
�h�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� �E/usr/local/CyberCP/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 )N�_func_r )�functionr � )r �
startswith�getattr�FUNCTION_TABLE)r$ �function_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 �UnknownFunctionError�_validate_arguments)�self�
function_name�
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 �VariadictArityError�
ArityError� _type_check)r: �argsr r; r r r r9 S s ���z Functions._validate_argumentsc C s8 t t|��D ]}|| d }|r| �|| ||� qd S )N�types)�rangerB �_type_check_single)r: �actualr r; �i�
allowed_typesr r r rE ] s ���zFunctions._type_checkc C sR | � |�\}}t|�j}||vrt�||| �|�|��|r'| �||||� d S d S r )�_get_allowed_pytypes�typer2 r �JMESPathTypeError�_convert_to_jmespath_type�_subtype_check)r: �currentrG r; rL �allowed_subtypes�actual_typenamer r r rI d s
���z Functions._type_check_singlec C s` g }g }|D ]%}|� dd�}t|�dkr |\}}|�t| � n|d }|�t| � q||fS )N�-� � r )�splitrB �append�REVERSE_TYPES_MAP�extend)r: rG rL rS �t�type_�subtyper r r rM { 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 )NrV r )rB rN r2 r rO )
r: rR rS rG r; �elementrT �first�subtypes�allowedr r r rQ � s:
���� �
��� zFunctions._subtype_checkrG r
c C � t |�S r )�abs�r: �argr r r � _func_abs� � zFunctions._func_absz array-numberc C s |r t |�tt|�� S d S r )�sumr rB re r r r � _func_avg� s zFunctions._func_avgT)rG r@ c G s |D ]
}|d ur | S qd S r r )r: r �argumentr r r �_func_not_null� s
��zFunctions._func_not_nullc C s t |t�r|S |gS r )�
isinstancer
re r r r �_func_to_array� s
zFunctions._func_to_arrayc C s t |t�r|S tj|dtd�S )N)�,�:)�
separators�default)rm � STRING_TYPE�json�dumpsr re 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 )rm r
r r r r �
ValueErrorre r r r �_func_to_number� s
��zFunctions._func_to_numberr r c C s ||v S r r )r: �subject�searchr r r �_func_contains� rh zFunctions._func_contains)r r r c C rc r )rB re r r r � _func_length� rh zFunctions._func_lengthc C �
|� |�S r )�endswith�r: rz �suffixr r r �_func_ends_with� �
zFunctions._func_ends_withc C r} r )r- r r r r �_func_starts_with� r� zFunctions._func_starts_withc C s$ t |t�r |d d d� S tt|��S )Nr? )rm rs r
�reversedre r r r �
_func_reverse� s
zFunctions._func_reversec C �
t �|�S r )�math�ceilre r r r �
_func_ceil� r� zFunctions._func_ceilc C r� r )r� �floorre r r r � _func_floor� r� zFunctions._func_floorz array-stringc C r} r )�join)r: � separatorr r r r �
_func_join� r� zFunctions._func_joinr c C s& g }|D ] }|� |�|j|�� q|S r )rY �visit�
expression)r: r rf �resultr_ r r r � _func_map� s zFunctions._func_mapc C � |rt |�S d S r )�maxre r r r � _func_max� � zFunctions._func_maxr c G s i }|D ]}|� |� q|S r )�update)r: r �mergedrf r r r � _func_merge s zFunctions._func_mergec C r� r )�minre r r r � _func_min
r� zFunctions._func_minc C s t t|��S r )r
�sortedre r r r �
_func_sort � zFunctions._func_sortc C rc r )ri re r r r � _func_sum rh zFunctions._func_sumc C � t |�� �S r )r
�keysre r r r �
_func_keys s zFunctions._func_keysc C r� r )r
�valuesre 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 )rm rs r r
r r r re 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 �sort_byr r
��key)
rP rN r� r� r2 r rO �_create_key_funcr
r� )r: r r �
required_type�keyfuncr r r �
_func_sort_by5 s ���zFunctions._func_sort_byc C �& | � |ddgd�}|rt||d�S d S )Nr
r �min_byr� )r� r� �r: r r r� r r r � _func_min_byI � � zFunctions._func_min_byc C r� )Nr
r �max_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� rN r2 rP r rO )�xr� rT �
jmespath_type�rL r r; r: r r r� ^ s
�z+Functions._create_key_func.<locals>.keyfuncr )r: r rL r; r� r r� r r� ] s zFunctions._create_key_funcc C s t �|d�S )N�unknown)� TYPES_MAPrA )r: �pyobjectr r r rP i s z#Functions._convert_to_jmespath_typeN)'r2 r3 r4 r/ r> r9 rE rI rM rQ r rg rj rl rn rv rx r{ r| r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rP r r r r r6 C s|
r6 )� metaclass)r� rt �jmespathr �jmespath.compatr rs r r� rZ r rN r r6 r r r r �<module> s: ��