o
    pf3                     @  sd  d Z ddlmZ ddlZddlZddlZddlmZ ddlm	Z	 ddl
mZmZ ddlmZ ddlmZ d	d
lmZ d	dlmZ d	dlmZ d	dlmZmZmZmZ errddlmZ ddlmZ ddlm Z  d	dl!m"Z" d	dl#m$Z$ 		dEdFddZ%G dd  d eZ&dGd#d$Z'e	dd%dHd'd(Z(dd)dId4d5Z)dJd9d:Z*dd)dKd=d>Z+dLdAdBZ,dLdCdDZ-dS )MzmPrivate logic related to fields (the `Field()` function and `FieldInfo` class), and arguments to `Annotated`.    )annotationsN)copy)	lru_cache)TYPE_CHECKINGAny)PydanticUndefined)PydanticUserError   )_typing_extra)ConfigWrapper)Representation)get_cls_type_hints_lenientget_type_hintsis_classvaris_finalvarBaseMetadata   	FieldInfo	BaseModel)StandardDataclass)DecoratorInfosFobjr   localnsdict[str, Any] | Noneinclude_extrasboolreturndict[str, Any]c                 C  sH   t | dd}d}|rztj| j}W n	 ty   Y nw t| |||dS )a  Gets type hints for an object by inferring the global namespace.

    It uses the `typing.get_type_hints`, The only thing that we do here is fetching
    global namespace from `obj.__module__` if it is not `None`.

    Args:
        obj: The object to get its type hints.
        localns: The local namespaces.
        include_extras: Whether to recursively include annotation metadata.

    Returns:
        The object type hints.
    
__module__N)globalnsr   r   )getattrsysmodules__dict__KeyErrorr   )r   r   r   module_namer"    r)   S/home/ertert/spirit/venv/lib/python3.10/site-packages/pydantic/_internal/_fields.pyget_type_hints_infer_globalns   s   r+   c                   @  s   e Zd ZdZdZdS )PydanticMetadataz0Base class for annotation markers like `Strict`.r)   N)__name__r!   __qualname____doc__	__slots__r)   r)   r)   r*   r,   :   s    r,   metadatar   c                  K  s
   t  | S )zCreate a new `_PydanticGeneralMetadata` class with the given metadata.

    Args:
        **metadata: The metadata to add.

    Returns:
        The new `_PydanticGeneralMetadata` class.
    )_general_metadata_cls)r1   r)   r)   r*   pydantic_general_metadata@   s   
	r3   )maxsizetype[BaseMetadata]c                  C  s"   ddl m}  G dd dt| }|S )zCDo it this way to avoid importing `annotated_types` at import time.r   r   c                   @  s   e Zd ZdZdddZdS )z7_general_metadata_cls.<locals>._PydanticGeneralMetadataz,Pydantic general metadata like `max_digits`.r1   r   c                 S  s
   || _ d S N)r&   )selfr1   r)   r)   r*   __init__T   s   
z@_general_metadata_cls.<locals>._PydanticGeneralMetadata.__init__N)r1   r   )r-   r!   r.   r/   r8   r)   r)   r)   r*   _PydanticGeneralMetadataQ   s    r9   )annotated_typesr   r,   )r   r9   r)   r)   r*   r2   L   s   r2   )typevars_mapclstype[BaseModel]basestuple[type[Any], ...]config_wrapperr   types_namespacer;   dict[Any, Any] | None%tuple[dict[str, FieldInfo], set[str]]c                  s  ddl m} t| |}| jdi }i }t }	| D ];\ }
 dkr%q|jD ]N} |rv|D ](}t	| rYddl
m} t||rH |jv sYtd  dt|  d| d	q1t fd
d|jD }td  d| d| dt q(t|
r|	  qt|
t|  tr|	  qt sq| jr dkrtd dt| di d}|D ]0}dd t|rt |ndD }t	| r||u rq |v rqtd  d|j dt qzt|  t}|tu rtW n= ty+    |v r||
}n(i }| jddd D ]}| t|di  q |v r$t!|  }n||
}Y nw |"|
|}zt#|   W n
 tyC   Y nw | jd } |j$v rSt%d|| < q|ri|& D ]	}|'|| q_||	fS )a[  Collect the fields of a nascent pydantic model.

    Also collect the names of any ClassVars present in the type hints.

    The returned value is a tuple of two items: the fields dict, and the set of ClassVar names.

    Args:
        cls: BaseModel or dataclass.
        bases: Parents of the class, generally `cls.__bases__`.
        config_wrapper: The config wrapper instance.
        types_namespace: Optional extra namespace to look for types in.
        typevars_map: A dictionary mapping type variables to their concrete types.

    Returns:
        A tuple contains fields and class variables.

    Raises:
        NameError:
            - If there is a conflict between a field name and protected namespaces.
            - If there is a field other than `root` in `RootModel`.
            - If a field shadows an attribute in the parent model.
    r   r   __annotations__model_configr   zField "z" conflicts with member z of protected namespace "z".c                 3  s    | ]
}  |s|V  qd S r6   
startswith).0xann_namer)   r*   	<genexpr>   s    

z'collect_model_fields.<locals>.<genexpr>z)" has conflict with protected namespace "z_".

You may be able to resolve this warning by setting `model_config['protected_namespaces'] = z`.rootzUnexpected field with name z4; only 'root' is allowed as a field of a `RootModel`__pydantic_generic_metadata__originc                 S  s   h | ]}|j qS r)   name)rH   fieldr)   r)   r*   	<setcomp>   s    z'collect_model_fields.<locals>.<setcomp>r)   zField name "z"" shadows an attribute in parent "z"; Nmodel_fields__pydantic_decorators__z0you can't override a field with a computed field)(fieldsr   r   r&   getsetitemsprotected_namespacesrG   hasattrmainr   
issubclassrU   	NameErrorr#   tuplewarningswarnUserWarningr   add_is_finalvar_with_default_valr   is_valid_field_name__pydantic_root_model__dataclassesis_dataclassr.   AttributeErrorfrom_annotation	__bases__updater   from_annotated_attributedelattrcomputed_fields
ValueErrorvaluesapply_typevars_map)r<   r>   r@   rA   r;   r   
type_hintsr   rW   
class_varsann_typeprotected_namespacebr   valid_namespacesgeneric_originbasedataclass_fieldsdefault
field_infomodel_fields_lookuprI   
decoratorsrR   r)   rJ   r*   collect_model_fieldsZ   s   












r   type_	type[Any]valc                 C  sJ   ddl m} t| sdS |tu rdS t||r#|jtu r#|jd u r#dS dS )Nr   r   FT)rW   r   r   r   
isinstancer}   default_factory)r   r   r   r)   r)   r*   re      s   re   type[StandardDataclass]dict[str, FieldInfo]c                C  s<  ddl m} i }| j}tt| }tj| j}|dur%i |j	|p#i }|
 D ]c\}}	t|	j||}
t|
r:q)|	jsJ|	jtjkrJ|	jtjkrJq)t|	j|rm|	jjre|	jjdu rdtd| dddq)||
|	j}n||
|	}|||< |jturtt| |||rt| ||j q)|r| D ]}||| q|S )	a  Collect the fields of a dataclass.

    Args:
        cls: dataclass.
        types_namespace: Optional extra namespace to look for types in.
        typevars_map: A dictionary mapping type variables to their concrete types.

    Returns:
        The dataclass fields.
    r   r   NFzDataclass field zD has init=False and init_var=True, but these are mutually exclusive.zclashing-init-and-init-var)code)rW   r   __dataclass_fields__dictvarsr$   r%   rX   r!   r&   rZ   r
   eval_type_lenienttyper   initr}   rh   MISSINGr   r   init_varr   rn   r   r#   setattrrr   rs   )r<   rA   r;   r   rW   r|   cls_localnssource_modulerK   dataclass_fieldrv   r~   rR   r)   r)   r*   collect_dataclass_fields   sD   
r   rQ   strc                 C  s   |  d S )N_rF   rP   r)   r)   r*   rf   :  s   rf   c                 C  s   |  do
|  d S )Nr   __rF   rP   r)   r)   r*   is_valid_privateattr_name>  s   r   )NF)r   r   r   r   r   r   r   r    )r1   r   r   r   )r   r5   )r<   r=   r>   r?   r@   r   rA   r   r;   rB   r   rC   )r   r   r   r   r   r   )r<   r   rA   r   r;   rB   r   r   )rQ   r   r   r   ).r/   
__future__r   _annotationsrh   r$   ra   r   	functoolsr   typingr   r   pydantic_corer   pydantic.errorsr    r
   _configr   _reprr   r   r   r   r   r:   r   rW   r   r]   r   _dataclassesr   _decoratorsr   r+   r,   r3   r2   r   re   r   rf   r   r)   r)   r)   r*   <module>   sF    
 

@