o
    pf                     @  s   d Z ddlmZ ddlZddlmZmZ ddlmZ ddl	m
Z
 ddlmZmZ dd	lmZmZ ejr_dd
lmZ ddlmZmZ ddlmZ edefdG dd dejZejdddZnejZdZedZG dd deeje edZ dS )z%RootModel class and type definitions.    )annotationsN)copydeepcopy)PydanticUndefined   )PydanticUserError)_model_construction_repr)	BaseModel_object_setattr)Any)Literaldataclass_transform)FieldF)kw_only_defaultfield_specifiersc                   @  s   e Zd ZdS )_RootModelMetaclassN)__name__
__module____qualname__ r   r   L/home/ertert/spirit/venv/lib/python3.10/site-packages/pydantic/root_model.pyr      s    r   Modelr
   )bound)	RootModelRootModelRootTypec                
      s   e Zd ZU dZdZdZdZded<  fddZe	fd7d
dZ
de
_ed8d9 fddZd:ddZd;ddZd<ddZd8d=ddZejrSd ddd!d!d!d!d!dd"	d>d/d0Zd? fd2d3Zd@d5d6Z  ZS )Ar   a  Usage docs: https://docs.pydantic.dev/2.6/concepts/models/#rootmodel-and-custom-root-types

    A Pydantic `BaseModel` for the root object of the model.

    Attributes:
        root: The root object of the model.
        __pydantic_root_model__: Whether the model is a RootModel.
        __pydantic_private__: Private fields in the model.
        __pydantic_extra__: Extra fields in the model.

    TNr   rootc                   s6   | j d}|d urtdddt jdi | d S )Nextraz<`RootModel` does not support setting `model_config['extra']`zroot-model-extra)coder   )model_configgetr   super__init_subclass__)clskwargsr   	__class__r   r   r"   9   s   zRootModel.__init_subclass__returnNonec                K  s0   d}|r|t urtd|}| jj|| d d S )NTz_"RootModel.__init__" accepts either a single positional argument or arbitrary keyword arguments)self_instance)r   
ValueError__pydantic_validator__validate_python)selfr   data__tracebackhide__r   r   r   __init__A   s   zRootModel.__init__r#   type[Model]_fields_setset[str] | Noner   c                   s   t  j||dS )aS  Create a new model using the provided root object and update fields set.

        Args:
            root: The root object of the model.
            _fields_set: The set of fields to be updated.

        Returns:
            The new model.

        Raises:
            NotImplemented: If the model is not a subclass of `RootModel`.
        )r   r2   )r!   model_construct)r#   r   r2   r%   r   r   r4   M   s   zRootModel.model_constructdict[Any, Any]c                 C  s   | j | jdS )N__dict____pydantic_fields_set__r6   r-   r   r   r   __getstate__]   s   zRootModel.__getstate__statec                 C  s$   t | d|d  t | d|d  d S )Nr8   r7   )r   )r-   r;   r   r   r   __setstate__c   s   zRootModel.__setstate__r-   c                 C  s:   t | }||}t|dt| j t|dt| j |S )z$Returns a shallow copy of the model.r7   r8   )type__new__r   r   r7   r8   )r-   r#   mr   r   r   __copy__g   s
   
zRootModel.__copy__memodict[int, Any] | Nonec                 C  s>   t | }||}t|dt| j|d t|dt| j |S )z!Returns a deep copy of the model.r7   )rA   r8   )r=   r>   r   r   r7   r   r8   )r-   rA   r#   r?   r   r   r   __deepcopy__o   s
   
zRootModel.__deepcopy__pythonF)	modeincludeexcludeby_aliasexclude_unsetexclude_defaultsexclude_none
round_tripwarningsrE   Literal['json', 'python'] | strrF   r   rG   rH   boolrI   rJ   rK   rL   rM   c       	   
      C  s   dS )a(  This method is included just to get a more accurate return type for type checkers.
            It is included in this `if TYPE_CHECKING:` block since no override is actually necessary.

            See the documentation of `BaseModel.model_dump` for more details about the arguments.
            Nr   )
r-   rE   rF   rG   rH   rI   rJ   rK   rL   rM   r   r   r   
model_dump{   s   zRootModel.model_dumpotherc                   s2   t |tstS | jd j|jd jkot |S Nr   )
isinstancer   NotImplementedmodel_fields
annotationr!   __eq__)r-   rQ   r%   r   r   rW      s   
$zRootModel.__eq___repr.ReprArgsc                 c  s    d| j fV  d S rR   )r   r9   r   r   r   __repr_args__   s   zRootModel.__repr_args__)r   r   r'   r(   )N)r#   r1   r   r   r2   r3   r'   r   )r'   r5   )r;   r5   r'   r(   )r-   r   r'   r   )r-   r   rA   rB   r'   r   )rE   rN   rF   r   rG   r   rH   rO   rI   rO   rJ   rO   rK   rO   rL   rO   rM   rO   r'   r   )rQ   r   r'   rO   )r'   rX   )r   r   r   __doc____pydantic_root_model____pydantic_private____pydantic_extra____annotations__r"   r   r0   __pydantic_base_init__classmethodr4   r:   r<   r@   rC   typingTYPE_CHECKINGrP   rW   rY   __classcell__r   r   r%   r   r   &   s8   
 




r   )	metaclass)!rZ   
__future__r   _annotationsra   r   r   pydantic_corer    r   	_internalr   r	   mainr
   r   rb   r   typing_extensionsr   r   fieldsr   PydanticModelFieldModelMetaclassr   TypeVarr   __all__r   Genericr   r   r   r   r   <module>   s&    
 