File: //usr/local/lib/python3.10/dist-packages/pydantic/deprecated/__pycache__/decorator.cpython-310.pyc
o
|��g[* � @ sv d dl Z d dlmZ d dlmZmZmZmZmZm Z m
Z
m Z m Z m
Z
mZmZ d dlmZ ddlmZmZ ddlmZ ddlmZ dd lmZ dd
lm Z m Z dd l m Z esXe Z d Z erued
ef Z!e
de!d�Z"ede e ee#ef f Z$e d!dd�dddddedgdf fdd��Z%ed"dd��Z%eddd�d!dd�de
d dddefdd��Z%dZ&d Z'd Z(d Z)G d d � d �Z*dS )#� N)�wraps) �
TYPE_CHECKING�Any�Callable�Dict�List�Mapping�Optional�Tuple�Type�TypeVar�Union�overload)�
deprecated� )�_config�
_typing_extra)� to_pascal)�PydanticUserError)�field_validator)� BaseModel� create_model)�PydanticDeprecatedSince20)�validate_arguments.� AnyCallableT)�bound��config�funcr �
ConfigType�returnc C � d S �N� )r r r# r# �H/usr/local/lib/python3.10/dist-packages/pydantic/deprecated/decorator.pyr s r c C r! r" r# )r r# r# r$ r " s �KThe `validate_arguments` method is deprecated; use `validate_call` instead.)�categoryc s. t jdtdd� d
� fdd� }| r|| �S |S ) z9Decorator to validate the arguments passed to a function.r% r )�
stacklevel�_func� AnyCallabler 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_function5 s z>validate_arguments.<locals>.validate.<locals>.wrapper_function)�ValidatedFunctionr r r. �init_model_instance�validate� raw_function�model)r( r/ r r- r$ r2 2 s
z$validate_arguments.<locals>.validateN)r( r) r r) )�warnings�warnr )r r r2 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 )r0 �functionr) r r c C s> ddl m}m} ||�j}|�� tttth@ r.t dt� dt� dt� dt� dt
j � d� d d��|| _ i | _
t� | _d | _d
| _tj|d d �}d
}d
}i } t|�� �D ]�\}
\} } | j| ju r`t}
n|| }
| j| ju rldn| j}| j|jkr�| | j
|
< |
|f| | < t t d f| t< | j� | � qQ| j|j kr�| | j
|
< |
|f| | < t t d f| t< qQ| j|j kr�|
|f| | < qQ| j|j!kr�| | _t"|
df d f| | < d }qQ| j|j#ks�J | j��| | _t$t |
f d f| | < d }qQ|s�| j| v r�t| _|s�| j| v r�t| _|�st t 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 " decorator��coder* r+ T)�include_extrasF.)&�inspectr< r= �
parameters�keys�
ALT_V_ARGS� ALT_V_KWARGS�V_POSITIONAL_ONLY_NAME�V_DUPLICATE_KWARGSr r �__name__r3 � arg_mapping�set�positional_only_args� v_args_name�
v_kwargs_namer �get_type_hints� enumerate�items�
annotation�emptyr �default�kind�POSITIONAL_ONLYr �str�add�POSITIONAL_OR_KEYWORD� KEYWORD_ONLY�VAR_POSITIONALr
� VAR_KEYWORDr r )�selfr; r r< r= rC �
type_hints�
takes_args� takes_kwargs�fields�i�name�prR rT r# r# r$ �__init__L sf
��
zValidatedFunction.__init__r* r+ r c O s | � ||�}| jdi |��S )Nr# )� build_valuesr4 )r] r* r+ �valuesr# r# r$ r1 � s z%ValidatedFunction.init_model_instancec O s | j |i |��}| �|�S r" )r1 �execute)r] 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# )rM rN �alias)rj rc �field�r] r# r$ rm � s
�)rP �next�
StopIterationrJ �getrM r4 �__pydantic_fields__rQ rK rN rL �appendrG rH )r] r* r+ rg �arg_iterrb rl �arg_name�
var_kwargs�wrong_positional_args�duplicate_kwargs� fields_alias�non_var_fields�k�vr# rp r$ rf � sJ �
�
�
z ValidatedFunction.build_valuesri c s� � fdd�� j �� D �}|�| ji �}| j|v rIg }d}i }|�� D ] \}}|r-|||< q"|| jkr9||7 }d}q"|�|� q"| j|i |�|��S | jrrg }i }|�� D ]\}}|| jv rc|�|� qT|||< qT| j|i |�|��S | jdi |�|��S )Nc s, i | ]\}}|� j v s� j| jr||�qS r# )�__pydantic_fields_set__rt �default_factory)rj r} r~ �ri r# r$ �
<dictcomp>� s
�z-ValidatedFunction.execute.<locals>.<dictcomp>FTr# )�__dict__rQ �poprN rM ru r3 rL ) r] ri �drx �args_� in_kwargsr+ rc �valuer# r� r$ rh � s2
�
zValidatedFunction.executera r_ r` Nc sx t �j��t�|�� � jrtdd d��� jd u r d� jd< G � ����fdd�dt�}t t
�j j �fd|i|���_
d S )NztSetting the "alias_generator" property on custom Config for @validate_arguments is not yet supported, please remove.r? �forbid�extrac s� e Zd Ze�jdd�edeee deee f��fdd� ��Z e�j
dd�edee e ef dee e ef f�fdd� ��Z
eedd�edeee dd fd
d ���Zeedd�edeee dd fd d
���Z� jZd S )z:ValidatedFunction.create_model.<locals>.DecoratorBaseModelF)� check_fieldsr~ r c s, �s|d u r|S t � � d� t|� � d���)Nz# positional arguments expected but z given)� TypeError�len)�clsr~ )�pos_argsr_ 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�reprrD r� �r� r~ �pluralrD )r` 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_kwargs)rI �
__module__� __qualname__r rM � classmethodr r r r� rN r rW r� rG r� rH r� � config_dict� model_configr# ��config_wrapperr� r] r_ r` r# r$ �DecoratorBaseModel� s , 2
r� �__base__)r� rJ r �
ConfigWrapper�alias_generatorr r� r� r r r r3 rI r4 )r] ra r_ r` r r� r# r� r$ r � s
�
")z ValidatedFunction.create_model)r; r) r r )rI r� r� re r r r1 r, r
r rW rf rh �boolr r# r# r# r$ r0 K s
D.+*!r0 r" )r r r r )+r5 � functoolsr �typingr r r r r r r r
r r r
r �typing_extensionsr � _internalr r �alias_generatorsr �errorsr �functional_validatorsr �mainr r r �DeprecationWarning�__all__r) r rW r r rE rF rG rH r0 r# r# r# r$ �<module> sP 8 ���� � �$