관리-도구
편집 파일: decorator.cpython-38.pyc
U /?�fc( � @ sF 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 er�d d lm Z e d e d �Z e d e e ee ef f Z ed d d�d ded gd f d�dd��Z!ed d d�dd��Z!d d d�e d ded�dd�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� ConfigType)�funcr �returnc C s d S �N� )r r r r �C/opt/hc_python/lib/python3.8/site-packages/pydantic/v1/decorator.pyr s r )r r c C s d S r r )r r r r! r s c s( ddd�� fdd� }| r || �S |S dS )zC Decorator to validate the arguments passed to a function. r )�_funcr c sJ t | ��� t| �tttd�� fdd� �}� |_� j|_� j|_� j|_|S )N��args�kwargsr c s � j | |�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>.validateNr )r r r, r r r! r s Zv__argsZ v__kwargsZv__positional_onlyZv__duplicate_kwargsc @ s� e Zd Zddd�dd�Zeeed�dd�Zeeed�d d �Zeed f e e ef e e ef d�d d �Z eed�dd�Z e e ef e e ddd�dd�ZdS )r* r r )�functionr c C sH ddl m}m} ||�j}|�� tttth@ rXt dt� dt� dt� dt� dt j � d� ��|| _ i | _ t� | _d| _d | _t|�}d }d }i } t|�� �D �]4\} \} } | j| jkr�t} n|| } | j| jkr�d n| j}| j|jk�r| | j | < | |f| | < tt d f| t< | j� | � q�| j|j k�rN| | j | < | |f| | < tt d f| t< q�| j|j k�rj| |f| | < q�| j|j k�r�| | _t!| d f d f| | < d }q�| j|j"k�s�t#| j��| | _t$t | f d f| | < d }q�|�s�| j| k�r�t| _|�s | j| k�r t| _|�stt d f| | j< |�s4t$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)&�inspectr0 r1 � 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_KEYWORD�AssertionErrorr r )�selfr/ r r0 r1 r4 Z type_hints� takes_args� takes_kwargs�fields�i�name�prB rD r r r! �__init__= s` $� zValidatedFunction.__init__r# c O s | � ||�}| jf |�S r )� build_valuesr. )rN r$ r% �valuesr r r! r+ � s z%ValidatedFunction.init_model_instancec O s | j ||�}| �|�S r )r+ �execute)rN r$ r% �mr r r! r& � s zValidatedFunction.call.c s@ i }|rvt |�}zt|�\}}W n tk r8 Y qvY nX � j�|�}|d k rX|||< q|gdd� |D � |� j< qvqi }g } g } � fdd�� jj�� D �} t � jj�� j� j h } |�� D ]R\} }| | ks�| | k�r| � j kr�| � | � | |kr�| � | � ||| < q�||| < q�|�r ||� j < | �r.| |t < | �r<| |t< |S )Nc S s g | ] \}}|�qS r r )�.0�_�ar r r! � <listcomp>� s z2ValidatedFunction.build_values.<locals>.<listcomp>c s& g | ] \}}|� j � jfkr|j�qS r )r> r? �alias)rZ rS �field�rN r r! r] � s �)r@ �next� StopIterationr; �getr> r. � __fields__rA r< r? r= �appendr8 r9 )rN r$ r% rW Zarg_iterrR r\ Zarg_name� var_kwargsZwrong_positional_argsZduplicate_kwargsZ fields_aliasZnon_var_fields�k�vr r` r! rV � sD � z ValidatedFunction.build_values)rY r c s� � fdd�� � � D �}|�| ji �}| j|kr�g }d}i }|�� D ]8\}}|rX|||< qB|| jkrp||7 }d}qB|�|� qB| j|||��S | jr�g }i }|�� D ]&\}}|| jkr�|�|� q�|||< q�| j|||��S | jf ||��S d S )Nc s, i | ]$\}}|� j ks"� j| jr||�qS r )Z__fields_set__rd �default_factory)rZ rg rh �rY r r! � <dictcomp>� s z-ValidatedFunction.execute.<locals>.<dictcomp>FT)Z_iter�popr? r> rA re r- r= ) rN rY �drf Zargs_Z in_kwargsr% rS �valuer rj r! rX � s. zValidatedFunction.executeN)rQ rO rP r r c s� t �j��G dd� d�� ts@t|t�r4tdd|�� n |d k r@|� t� d�sTt� 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 rq �Configr rQ Zalias_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�eee eee d���fdd� �Ze�j ddd�ee e ef ee e ef d��fdd� �Z ee ddd�eee d d�d d ��Zeeddd�eee d d�d d ��ZG � fdd�d� �Zd S )z:ValidatedFunction.create_model.<locals>.DecoratorBaseModelFT)Z check_fieldsZ allow_reuse)rh r c s0 �s |d kr|S t � � d� t|� � d���d S )Nz# positional arguments expected but z given)� TypeError�len)�clsrh )�pos_argsrO r r! � check_args� s zEValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_argsc sP � s |d kr|S t |�dkr dnd}d�tt|�� ��}td|� d|� ���d S )N� � �s�, zunexpected keyword argument�: )rt �join�map�reprr5 rs �ru rh �pluralr5 )rP r r! � check_kwargs� s zGValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_kwargsNc S sN |d kr d S t |�dkr dnd}d�tt|��}td|� d|� d|� ���d S )Nrx ry rz r{ zpositional-only argumentz passed as keyword argumentr| �rt r} r~ r rs r� r r r! �check_positional_only� s zPValidatedFunction.create_model.<locals>.DecoratorBaseModel.check_positional_onlyc S sH |d kr d S t |�dkr dnd}d�tt|��}td|� d|� ���d S )Nrx ry rz 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: ro rp �getattrr Zforbidr� r )rq r r! rr s rr )r: ro rp r r> r r r rw r? r rG r� r8 r� r9 r� rr r �rq rv rN rO rP r r! �DecoratorBaseModel� s (. r� �__base__)rt r; r � isinstance�dict�type�hasattrr r r r r- r: r. )rN rQ rO rP r r� r r� r! r � s � &z ValidatedFunction.create_model)r: ro rp rU r r r+ r& r r rG rV rX �boolr r r r r! r* < s C*+ r* )N)N)'� functoolsr �typingr r r r r r r r r r r r Z pydantic.v1r Zpydantic.v1.configr Zpydantic.v1.errorsr Zpydantic.v1.mainr r Zpydantic.v1.typingr Zpydantic.v1.utilsr �__all__r r rG r r r6 r7 r8 r9 r* r r r r! �<module> s, 8 &