File: //usr/local/lib/python3.10/dist-packages/pydantic/v1/__pycache__/decorator.cpython-310.pyc
o
|��gc( � @ sH d dl mZ d dlmZmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZ d dlmZ d dlmZ d dlmZ d dlmZmZ d dlmZ d dlmZ d Z ered d
lm Z e d e d �Z e
d
e e ee ef f Z ed d
d�dd
ddded gd f fdd��Z!ed dd��Z!d d
d�de d dddefdd�Z!dZ"dZ#dZ$dZ%G dd � d �Z&d
S ) � )�wraps) �
TYPE_CHECKING�Any�Callable�Dict�List�Mapping�Optional�Tuple�Type�TypeVar�Union�overload)� validator)�Extra)� ConfigError)� BaseModel� create_model)�get_all_type_hints)�to_camel)�validate_arguments)� AnyCallable� AnyCallableT)�boundN��config�funcr �
ConfigType�returnc C � d S �N� )r r r! r! �@/usr/local/lib/python3.10/dist-packages/pydantic/v1/decorator.pyr � r c C r r r! )r r! r! r" r r# c s d� fdd� }| r
|| �S |S )zC
Decorator to validate the arguments passed to a function.
�_funcr r c sN t | ��� t| �dtdtdtf� fdd� �}� |_� j|_� j|_� j|_|S )N�args�kwargsr c s � j | i |��S r )�call)r% r& ��vdr! r" �wrapper_function& s z>validate_arguments.<locals>.validate.<locals>.wrapper_function)�ValidatedFunctionr r r) �init_model_instance�validate� raw_function�model)r$ r* r r( r" r- # s
z$validate_arguments.<locals>.validateN)r$ r r r r! )r r r- r! r r" r s
�v__args� v__kwargs�v__positional_only�v__duplicate_kwargsc
@ s� e Zd Zddd�Zdeded efd
d �Zdeded efd d
�Zdeedf de e
ef d e e
ef fdd�Z ded efdd�Z de e
ef de
de
ddd df
dd�ZdS )r+ �functionr r r c C s4 ddl m}m} ||�j}|�� tttth@ r,t dt� dt� dt� dt� dt
j � d� ��|| _ i | _
t� | _d| _d | _t|�}d
}d
}i } t|�� �D ]�\}
\} } | j| ju r[t}
n|| }
| j| ju rgd n| j}| j|jkr�| | j
|
< |
|f| | < tt d f| t< | j� | � qL| j|j kr�| | j
|
< |
|f| | < tt d f| t< qL| j|j kr�|
|f| | < qL| j|j kr�| | _t!|
d f d f| | < d }qL| j|j"ks�J | j��| | _t#t |
f d f| | < d }qL|s�| j| v r�t| _|s�| j| v r�t| _|�stt d f| | j< |�st#ttf d f| | j< | �$| |||� d S )
Nr )� Parameter� signature�"z", "z" and "z6" are not permitted as argument names when using the "z " decoratorr% r& F.T)%�inspectr5 r6 �
parameters�keys�
ALT_V_ARGS� ALT_V_KWARGS�V_POSITIONAL_ONLY_NAME�V_DUPLICATE_KWARGSr r �__name__r. � arg_mapping�set�positional_only_args� v_args_name�
v_kwargs_namer � enumerate�items�
annotation�emptyr �default�kind�POSITIONAL_ONLYr �str�add�POSITIONAL_OR_KEYWORD� KEYWORD_ONLY�VAR_POSITIONALr
� VAR_KEYWORDr r )�selfr4 r r5 r6 r9 �
type_hints�
takes_args� takes_kwargs�fields�i�name�prG rI r! r! r" �__init__= sd
��
zValidatedFunction.__init__r% r& r c O s | � ||�}| jdi |��S )Nr! )� build_valuesr/ )rR r% r& �valuesr! r! r" r, � s z%ValidatedFunction.init_model_instancec O s | j |i |��}| �|�S r )r, �execute)rR r% r& �mr! r! r" r' � s
zValidatedFunction.call.c s4 i }|r9t |�} zt|�\}}W n ty Y n w � j�|�}|d ur*|||< n|gdd� |D � |� j< nq i }g } g }
� fdd�� jj�� D �} t � jj�� j� j
h } |�� D ](\}
}|
| v sh|
| v r�|
� j v rr| � |
� |
|v r{|
� |
� |||
< q\|||
< q\|r�||� j
< | r�| |t
< |
r�|
|t< |S )NTc S s g | ]\}}|�qS r! r! )�.0�_�ar! r! r" �
<listcomp>� s z2ValidatedFunction.build_values.<locals>.<listcomp>c s& g | ]\}}|� j � jfvr|j�qS r! )rC rD �alias)r_ rX �field�rR r! r" rb � s
�)rE �next�
StopIterationr@ �getrC r/ �
__fields__rF rA rD rB �appendr= r> )rR r% r& r\ �arg_iterrW ra �arg_name�
var_kwargs�wrong_positional_args�duplicate_kwargs� fields_alias�non_var_fields�k�vr! re r" r[ � sJ �
�
�
z ValidatedFunction.build_valuesr^ c s� � fdd�� � � D �}|�| ji �}| j|v rHg }d}i }|�� D ] \}}|r,|||< q!|| jkr8||7 }d}q!|�|� q!| j|i |�|��S | jrqg }i }|�� D ]\}}|| jv rb|�|� qS|||< qS| j|i |�|��S | jdi |�|��S )Nc s, i | ]\}}|� j v s� j| jr||�qS r! )�__fields_set__ri �default_factory)r_ rr rs �r^ r! r" �
<dictcomp>� s , z-ValidatedFunction.execute.<locals>.<dictcomp>FTr! )�_iter�poprD rC rF rj r. rB ) rR r^ �drm �args_� in_kwargsr& rX �valuer! rv r" r] � s.
zValidatedFunction.executerV rT rU Nc s� t �j��G dd� d�� ts t|t�rtdd|�� n|d ur |� t� d�s*t� d�r.td��G � ����fdd �d t�}t t
�j j �fd
|i|���_
d S ) Nc @ s e Zd ZdS )z4ValidatedFunction.create_model.<locals>.CustomConfigN)r? �
__module__� __qualname__r! r! r! r" � CustomConfig� s r� �Configr! rV �alias_generatorz�Setting the "fields" and "alias_generator" property on custom Config for @validate_arguments is not yet supported, please remove.c s� e Zd Ze�jddd�deee deee f��fdd� �Ze�j ddd�dee
e ef dee
e ef f�fdd � �Z ee
ddd�deee dd
fd d ��Zeeddd�deee dd
fd
d��ZG � fdd�d� �Zd
S )z:ValidatedFunction.create_model.<locals>.DecoratorBaseModelFT)� check_fields� allow_reusers r c s, �s|d u r|S t � � d� t|� � d���)Nz# positional arguments expected but z given)� TypeError�len)�clsrs )�pos_argsrT r! r" �
check_args� s zEValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_argsc sL � s|d u r|S t |�dkrdnd}d�tt|�� ��}td|� d|� ���)N� � �s�, zunexpected keyword argument�: )r� �join�map�reprr: r� �r� rs �pluralr: )rU r! r" � check_kwargs� s
zGValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_kwargsNc S sJ |d u rd S t |�dkrdnd}d�tt|��}td|� d|� d|� ���)Nr� r� r� r� zpositional-only argumentz passed as keyword argumentr� �r� r� r� r� r� r� r! r! r" �check_positional_only� s
zPValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_positional_onlyc S sD |d u rd S t |�dkrdnd}d�tt|��}td|� d|� ���)Nr� r� r� r� z multiple values for argumentr� r� r� r! r! r" �check_duplicate_kwargs� s
zQValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_duplicate_kwargsc s e Zd Ze� dej�ZdS )zAValidatedFunction.create_model.<locals>.DecoratorBaseModel.Config�extraN)r? r~ r �getattrr �forbidr� r! )r� r! r" r� s r� )r? r~ r r rC r r r r� rD r rL r� r= r� r> r� r� r! �r� r� rR rT rU r! r" �DecoratorBaseModel� s *0 r� �__base__)r� r@ r �
isinstance�dict�type�hasattrr r r r r. r? r/ )rR rV rT rU r r� r! r� r" r � s
� "&z ValidatedFunction.create_model)r4 r r r )r? r~ r rZ r r r, r' r
r rL r[ r] �boolr r! r! r! r" r+ < s
C.+* r+ r )r r r r )'� functoolsr �typingr r r r r r r r
r r r
r � pydantic.v1r �pydantic.v1.configr �pydantic.v1.errorsr �pydantic.v1.mainr r �pydantic.v1.typingr �pydantic.v1.utilsr �__all__r r rL r r r; r< r= r> r+ r! r! r! r" �<module> s. 8 * "