File: //opt/alt/python37/lib/python3.7/site-packages/jmespath/__pycache__/functions.cpython-37.pyc
B
�P�e�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� �C/opt/alt/python37/lib/python3.7/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 )N)�_populate_function_table�superr �__init__)�cls�name�bases�attrs)� __class__r r r! . s zFunctionRegistry.__init__c C sZ i }xJt | �D ]>\}}|�d�s"qt|dd �}|d k r||d�||dd � < qW || _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
d d d"gi�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
d"gdd��d7d8� �Ze
d
dd/gi�d9d:� �Z e
d
d/dgi�d;d<� �Z e
d
dgi�d=d>� �Z e
d
d"gi�d?d@� �Z e
d
d"gi�dAdB� �Z e
d
g i�dCdD� �Z!e
d
d gid
d2gi�dEdF� �Z"e
d
d gid
d2gi�dGdH� �Z#e
d
d gid
d2gi�dIdJ� �Z$dKdL� Z%dMdN� Z&dOS )P� Functionsc C s^ y| j | }W n" tk
r0 t�d| ��Y nX |d }|d }| �|||� || f|�� 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|�|��n&t|�t|�kr`t�t|�t|�|��| �|||�S )N����variadic)�get�lenr ZVariadictArityErrorZ
ArityError� _type_check)r4 �argsr r5 r r r r3 S s z Functions._validate_argumentsc C s< x6t t|��D ]&}|| d }|r| �|| ||� qW d S )N�types)�ranger; �_type_check_single)r4 �actualr r5 �i�
allowed_typesr r r r< ] s
zFunctions._type_checkc C sN | � |�\}}t|�j}||kr6t�||| �|�|��|rJ| �||||� d S )N)�_get_allowed_pytypes�typer- r �JMESPathTypeError�_convert_to_jmespath_type�_subtype_check)r4 �currentr> r5 rC �allowed_subtypes�actual_typenamer r r r@ d s
z Functions._type_check_singlec C sd g }g }xR|D ]J}|� dd�}t|�dkrB|\}}|�t| � n|d }|�t| � qW ||fS )N�-� � r )�splitr; �append�REVERSE_TYPES_MAP�extend)r4 r> rC rJ �t�type_�subtyper r r rD { s
z Functions._get_allowed_pytypesc
C s� t |�dkrF|d }x�|D ]&}t|�j}||krt�||||��qW n~t |�dkr�|r�t|d �j}x.|D ]}||krj|} P qjW t�||d ||��x.|D ]&}t|�j}|| kr�t�||||��q�W d S )NrM r )r; rE r- r rF )
r4 rI rJ r> r5 �elementrK �first�subtypes�allowedr r r rH � s(
zFunctions._subtype_checkr> r
c C s t |�S )N)�abs)r4 �argr r r � _func_abs� s zFunctions._func_absz array-numberc C s |rt |�tt|�� S d S d S )N)�sumr r; )r4 r[ r r r � _func_avg� s zFunctions._func_avgT)r> r9 c G s x|D ]}|d k r|S qW d S )Nr )r4 r �argumentr r r �_func_not_null� s
zFunctions._func_not_nullc C s t |t�r|S |gS d S )N)�
isinstancer
)r4 r[ r r r �_func_to_array� s
zFunctions._func_to_arrayc C s" t |t�r|S tj|dtd�S d S )N)�,�:)�
separators�default)ra � STRING_TYPE�json�dumpsr )r4 r[ r r r �_func_to_string� s
zFunctions._func_to_stringc C sr t |tttf�rd S |d kr d S t |ttf�r2|S yt|�S tk
rl yt|�S tk
rf d S X Y nX d S )N)ra r
r r r r �
ValueError)r4 r[ r r r �_func_to_number� s zFunctions._func_to_numberr r c C s ||kS )Nr )r4 �subject�searchr r r �_func_contains� s zFunctions._func_containsr c C s t |�S )N)r; )r4 r[ r r r � _func_length� s zFunctions._func_lengthc C s
|� |�S )N)�endswith)r4 rn �suffixr r r �_func_ends_with� s zFunctions._func_ends_withc C s
|� |�S )N)r) )r4 rn rr r r r �_func_starts_with� s zFunctions._func_starts_withc C s( t |t�r|d d d� S tt|��S d S )Nr8 )ra rg r
�reversed)r4 r[ r r r �
_func_reverse� s
zFunctions._func_reversec C s
t �|�S )N)�math�ceil)r4 r[ r r r �
_func_ceil� s zFunctions._func_ceilc C s
t �|�S )N)rw �floor)r4 r[ r r r � _func_floor� s zFunctions._func_floorz array-stringc C s
|� |�S )N)�join)r4 � separatorr r r r �
_func_join� s zFunctions._func_joinr c C s* g }x |D ]}|� |�|j|�� q
W |S )N)rP �visit�
expression)r4 r r[ �resultrV r r r � _func_map� s
zFunctions._func_mapc C s |r t |�S d S d S )N)�max)r4 r[ r r r � _func_max� s zFunctions._func_maxc G s i }x|D ]}|� |� q
W |S )N)�update)r4 r �mergedr[ r r r � _func_merge s
zFunctions._func_mergec C s |r t |�S d S d S )N)�min)r4 r[ r r r � _func_min
s zFunctions._func_minc C s t t|��S )N)r
�sorted)r4 r[ r r r �
_func_sort s zFunctions._func_sortc C s t |�S )N)r] )r4 r[ r r r � _func_sum s zFunctions._func_sumc C s t |�� �S )N)r
�keys)r4 r[ r r r �
_func_keys s zFunctions._func_keysc C s t |�� �S )N)r
�values)r4 r[ r r r � _func_values" s zFunctions._func_valuesc C sZ t |t�rdS t |t�r dS t |t�r*dS t |t�r8dS t |ttf�rJdS |d krVdS d S )Nr r r r r
r )ra rg r r
r r r )r4 r[ r r r �
_func_type& s
zFunctions._func_typec C sf |s|S | � t|�|j|d ��j�}|dkrFt�d|d |ddg��| �||gd�}tt ||d��S )Nr )r
r Zsort_byr r
)�key)
rG rE r r� r- r rF �_create_key_funcr
r� )r4 r r Z
required_type�keyfuncr r r �
_func_sort_by5 s zFunctions._func_sort_byc C s* | � |ddgd�}|r"t||d�S d S d S )Nr
r Zmin_by)r� )r� r� )r4 r r r� r r r � _func_min_byI s zFunctions._func_min_byc C s* | � |ddgd�}|r"t||d�S d S d S )Nr
r Zmax_by)r� )r� r� )r4 r r r� r r r � _func_max_byS s zFunctions._func_max_byc s � ���fdd�}|S )Nc s> �� �j| �}t|�j}��|�}|� kr:t��||� ��|S )N)r r� rE r- rG r rF )�xr� rK Z
jmespath_type)rC r r5 r4 r r r� ^ s
z+Functions._create_key_func.<locals>.keyfuncr )r4 r rC r5 r� r )rC r r5 r4 r r� ] s zFunctions._create_key_funcc C s t �|d�S )N�unknown)� TYPES_MAPr: )r4 Zpyobjectr r r rG i s z#Functions._convert_to_jmespath_typeN)'r- r. r/ r+ r7 r3 r< r@ rD rH r r\ r^ r` rb rj rl ro rp rs rt rv ry r{ r~ r� r� r� r� r� r� r� r� r� r� r� r� r� rG r r r r r1 C sF
r1 )� metaclass)rw rh �jmespathr Zjmespath.compatr rg r r� rQ r rE r r1 r r r r �<module> s4