File: //usr/lib/python3.6/site-packages/jmespath/__pycache__/functions.cpython-36.pyc
3
�f V�2 � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ ddddd d d
d
ddd d � Z d d d!d"d#d$d%d�Z
dd� Z dd � Z e G d d � d e
��ZdS )&� N)�
exceptions)� string_type)� get_methods�boolean�array�object�null�string�number�expref) �bool�list�dictZNoneType�unicode�str�float�int� OrderedDict� _Projection� _Expressionr r
r r r �Noner r r r r )r r r r r r
r c C sJ | j }x>t| �D ]2\}}t|dd �}|d k r||d�||dd � < qW | S )N� signature)�functionr � )�FUNCTION_TABLEr �getattr)�clsZ
func_table�name�methodr � r � /usr/lib/python3.6/functions.py�populate_function_table& s r! c s � fdd�}|S )Nc s
� | _ | S )N)r )�func)� argumentsr r �
_record_arity1 s z'builtin_function.<locals>._record_arityr )r# r$ r )r# r �builtin_function0 s r% c @ s� e Zd Zi Zdd� Zedd� �Zejdd� �Zdd� Zdd � Z d
d � Z
d d
� Z dd� Z dd� Z
eddgi�dd� �Zeddgi�dd� �Zeg dd��dd � �Zedg i�d d � �Zedg i�d d � �Zedg i�d!d"� �Zedd#d$gidg i�d%d&� �Zedd$d#d'gi�d(d)� �Zedd$gidd$gi�d*d+� �Zedd$gidd$gi�d,d-� �Zedd#d$gi�d.d/� �Zeddgi�d0d1� �Zeddgi�d2d3� �Zedd$gidd4gi�d5d6� �Z edd7gidd#gi�d8d9� �Z eddd4gi�d:d;� �Z ed'gdd��d<d=� �Z eddd4gi�d>d?� �Z edd4dgi�d@dA� �Z!eddgi�dBdC� �Z"edd'gi�dDdE� �Z#edd'gi�dFdG� �Z$edg i�dHdI� �Z%edd#gidd7gi�dJdK� �Z&edd#gidd7gi�dLdM� �Z'edd#gidd7gi�dNdO� �Z(dPdQ� Z)dRdS� Z*dTS )U�RuntimeFunctionsc C s
d | _ d S )N)� _interpreter)�selfr r r �__init__? s zRuntimeFunctions.__init__c C s | j d krd S | j � S d S )N)r' )r( r r r � interpreterB s
z RuntimeFunctions.interpreterc C s t j|�| _d S )N)�weakref�refr' )r( �valuer r r r* I s c C s^ y| j | }W n" tk
r0 tjd| ��Y nX |d }|d }| j|||� || f|�� S )NzUnknown function: %s()r r )r �KeyErrorr ZUnknownFunctionError�_validate_arguments)r( �
function_nameZ
resolved_args�specr r r r r �
call_functionQ s z RuntimeFunctions.call_functionc C sn |r:|d j d�r:t|�t|�k r`tjt|�t|�|��n&t|�t|�kr`tjt|�t|�|��| j|||�S )N� �variadic���)�get�lenr ZVariadictArityErrorZ
ArityError� _type_check)r( �argsr r0 r r r r/ \ s z$RuntimeFunctions._validate_argumentsc C s< x6t t|��D ]&}|| d }|r| j|| ||� qW d S )N�types)�ranger7 �_type_check_single)r( �actualr r0 �i�
allowed_typesr r r r8 f s
z RuntimeFunctions._type_checkc C sN | j |�\}}t|�j}||kr6tj||| j|�|��|rJ| j||||� d S )N)�_get_allowed_pytypes�type�__name__r �JMESPathTypeError�_convert_to_jmespath_type�_subtype_check)r( �currentr: r0 r? �allowed_subtypes�actual_typenamer r r r< m s
z#RuntimeFunctions._type_check_singlec C sd g }g }xR|D ]J}|j dd�}t|�dkrB|\}}|jt| � n|d }|jt| � qW ||fS )N�-r3 � r )�splitr7 �append�REVERSE_TYPES_MAP�extend)r( r: r? rG �tZtype_�subtyper r r r@ � s
z%RuntimeFunctions._get_allowed_pytypesc
C s� t |�dkrF|d }x�|D ]&}t|�j}||krtj||||��qW n~t |�dkr�|r�t|d �j}x.|D ]}||krj|} P qjW tj||d ||��x.|D ]&}t|�j}|| kr�tj||||��q�W d S )Nr3 r )r7 rA rB r rC )
r( rF rG r: r0 �elementrH �firstZsubtypesZallowedr r r rE � s(
z RuntimeFunctions._subtype_checkr: r
c C s t |�S )N)�abs)r( �argr r r � _func_abs� s zRuntimeFunctions._func_absz array-numberc C s t |�tt|�� S )N)�sumr r7 )r( rT r r r � _func_avg� s zRuntimeFunctions._func_avgT)r: r4 c G s x|D ]}|d k r|S qW d S )Nr )r( r# Zargumentr r r �_func_not_null� s
z RuntimeFunctions._func_not_nullc C s t |t�r|S |gS d S )N)�
isinstancer
)r( rT r r r �_func_to_array� s
z RuntimeFunctions._func_to_arrayc C s" t |t�r|S tj|dtd�S d S )N�,�:)Z
separators�default)r[ r\ )rY � STRING_TYPE�json�dumpsr )r( rT r r r �_func_to_string� s
z RuntimeFunctions._func_to_stringc C sh t |tttf�rd S |d kr d S t |ttf�r2|S y d|krDt|�S t|�S W n tk
rb d S X d S )N�.)rY r
r r r r �
ValueError)r( rT r r r �_func_to_number� s z RuntimeFunctions._func_to_numberr r c C s ||kS )Nr )r( Zsubject�searchr r r �_func_contains� s z RuntimeFunctions._func_containsr c C s t |�S )N)r7 )r( rT r r r � _func_length� s z RuntimeFunctions._func_lengthc C s
|j |�S )N)�endswith)r( re �suffixr r r �_func_ends_with� s z RuntimeFunctions._func_ends_withc C s
|j |�S )N)�
startswith)r( re ri r r r �_func_starts_with� s z"RuntimeFunctions._func_starts_withc C s( t |t�r|d d d� S tt|��S d S )Nr3 r5 )rY r^ r
�reversed)r( rT r r r �
_func_reverse� s
z RuntimeFunctions._func_reversec C s
t j|�S )N)�mathZceil)r( rT r r r �
_func_ceil� s zRuntimeFunctions._func_ceilc C s
t j|�S )N)ro Zfloor)r( rT r r r � _func_floor� s z RuntimeFunctions._func_floorz array-stringc C s
|j |�S )N)�join)r( Z separatorr r r r �
_func_join� s zRuntimeFunctions._func_joinr c C s, g }x"|D ]}|j | jj|j|�� q
W |S )N)rL r* �visit�
expression)r( r rT �resultrQ r r r � _func_map� s
zRuntimeFunctions._func_mapc C s |r t |�S d S d S )N)�max)r( rT r r r � _func_max s zRuntimeFunctions._func_maxc G s i }x|D ]}|j |� q
W |S )N)�update)r( r# ZmergedrT r r r � _func_merge s
z RuntimeFunctions._func_mergec C s |r t |�S d S d S )N)�min)r( rT r r r � _func_min s zRuntimeFunctions._func_minc C s t t|��S )N)r
�sorted)r( rT r r r �
_func_sort s zRuntimeFunctions._func_sortc C s t |�S )N)rV )r( rT r r r � _func_sum s zRuntimeFunctions._func_sumc C s t |j� �S )N)r
�keys)r( rT r r r �
_func_keys" s zRuntimeFunctions._func_keysc C s t |j� �S )N)r
�values)r( rT r r r � _func_values( s z RuntimeFunctions._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 )rY r^ r r
r r r )r( rT r r r �
_func_type, s
zRuntimeFunctions._func_typec C sj |s|S | j t| jj|j|d ��j�}|dkrHtjd|d |ddg��| j|j|gd�}t t
||d��S )Nr r
r Zsort_by)�key)r
r ) rD rA r* rt ru rB r rC �_create_key_funcr
r~ )r( r r Z
required_type�keyfuncr r r �
_func_sort_by; s z RuntimeFunctions._func_sort_byc C s | j |jddgd�}t||d�S )Nr
r �min_by)r� )r� ru r| )r( r r r� r r r � _func_min_byO s z RuntimeFunctions._func_min_byc C s | j |jddgd�}t||d�S )Nr
r r� )r� )r� ru rx )r( r r r� r r r � _func_max_byV s z RuntimeFunctions._func_max_byc s �j �� ����fdd�}|S )Nc s< �j �| �}t|�j}�j|�}|� kr8tj�||� ��|S )N)rt rA rB rD r rC )�xrv rH Z
jmespath_type)r? � expr_noder0 r* r( r r r� ` s
z2RuntimeFunctions._create_key_func.<locals>.keyfunc)r* )r( r� r? r0 r� r )r? r� r0 r* r( r r� ] s z!RuntimeFunctions._create_key_funcc C s t j|d�S )N�unknown)� TYPES_MAPr6 )r( Zpyobjectr r r rD k s z*RuntimeFunctions._convert_to_jmespath_typeN)+rB �
__module__� __qualname__r r) �propertyr* �setterr2 r/ r8 r< r@ rE r% rU rW rX rZ ra rd rf rg rj rl rn rp rq rs rw ry r{ r} r r� r� r� r� r� r� r� r� rD r r r r r& 7 sL
r& )r )r
r )r r )r )r r )r r )r )ro r_ r+ Zjmespathr Zjmespath.compatr r^ r r� rM r! r% r r& r r r r �<module> s6