관리-도구
편집 파일: _signature.cpython-38.pyc
U /?�f� � @ s� d dl mZ d dlZd dlmZmZmZ d dlmZm Z m Z d dl m Z ddl mZ ddlmZ erpd d lmZ d d d d �dd�Zddd�dd�Zddddd�dd�Zd!dddd d d �d d �ZdS )"� )� annotationsN)� Parameter� Signature� signature)� TYPE_CHECKING�Any�Callable)�PydanticUndefined� )� ConfigWrapper)�is_valid_identifier� �� FieldInfo�strr )� field_name� field_info�returnc C s( ddd�dd�}||j �p&||j�p&| S )a� Extract the correct name to use for the field when generating a signature. Assuming the field has a valid alias, this will return the alias. Otherwise, it will return the field name. First priority is given to the validation_alias, then the alias, then the field name. Args: field_name: The name of the field field_info: The corresponding FieldInfo object. Returns: The correct name to use when generating a signature. r z str | None)�xr c S s t | t�rt| �r| S dS )zBReturn the alias if it is a valid alias and identifier, else None.N)� isinstancer r )r � r �K/opt/hc_python/lib/python3.8/site-packages/pydantic/_internal/_signature.py�_alias_if_valid s z2_field_name_for_signature.<locals>._alias_if_valid)�aliasZvalidation_alias)r r r r r r �_field_name_for_signature s r r )�paramr c C sp ddl m} | j}t||�rl| j}|dkr.t}|j}|tkrT|jtkrNtj }nt j }| j |t | j|�|d�S | S )z�Modify the signature for a parameter in a dataclass where the default value is a FieldInfo instance. Args: param (Parameter): The parameter Returns: Parameter: The custom processed parameter r r r )� annotation�name�default)�fieldsr r r r r r �default_factoryr �empty� dataclasses�_HAS_DEFAULT_FACTORY�replacer r )r r Z param_defaultr r r r r �_process_param_defaults% s" �r% zCallable[(Ellipsis, None)]zdict[str, FieldInfo]r zdict[str, Parameter])�initr �config_wrapperr c C s� ddl m} t| �j�� }i }d}d}||dd�D ]t}|�|j�rtt||j dd�dkrZq2|jt |j||j �d�}|j d kr�|jt d �}|j |j kr�|}q2|||j< q2|�r8|j} |�� D ]z\} } t | | �} | |ks�| |kr�q�t| �s�| r�| } nd}q�| �� �ri nd | jdd �i} t| tjfd | �� i| ��|| < q�|jdk�rHd}|�r�|�r�dtjfdtj fg}dd� |D �|k�r�d}n|j}||k�r�|d7 }�q�|j|d�||< |S )zaGenerate a mapping of parameter names to Parameter objects for a pydantic BaseModel or dataclass.r )�isliceNFr r& T)r r )r r )Zcall_default_factoryr Zallow�self�datac S s g | ]}|j |jf�qS r )r �kind)�.0�pr r r � <listcomp>� s z2_generate_signature_parameters.<locals>.<listcomp>� extra_data�_)� itertoolsr( r � parameters�values�getr �getattrr$ r r r r+ � VAR_KEYWORDZpopulate_by_name�itemsr Z is_required� get_defaultr � KEYWORD_ONLYZrebuild_annotation�extra�POSITIONAL_ONLY)r& r r' r( Zpresent_params� merged_paramsZvar_kwZ use_var_kwr Z allow_namesr �field� param_name�kwargsZdefault_model_signatureZ var_kw_namer r r � _generate_signature_parametersG sb ��� � r@ F�boolr )r& r r' � is_dataclassr c C s6 t | ||�}|r"dd� |�� D �}tt|�� �dd�S )a8 Generate signature for a pydantic BaseModel or dataclass. Args: init: The class init. fields: The model fields. config_wrapper: The config wrapper instance. is_dataclass: Whether the model is a dataclass. Returns: The dataclass/BaseModel subclass signature. c S s i | ]\}}|t |��qS r )r% )r, �k�vr r r � <dictcomp>� s z/generate_pydantic_signature.<locals>.<dictcomp>N)r2 �return_annotation)r@ r7 r �listr3 )r& r r' rB r< r r r �generate_pydantic_signature� s rH )F)� __future__r r" �inspectr r r �typingr r r Z pydantic_corer �_configr Z_utilsr r r r r% r@ rH r r r r �<module> s "K �