U
���`�G � @ s� d dl mZ d dlmZ d dlmZ d dlmZmZm Z m
Z
m Z m Z d dl
mZmZmZmZ G dd� de�ZG dd � d e�ZG d
d � d e �ZG d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z e� e � Z Z dS )� )�absolute_import)�warn)�errors)� _str_type�Callable�Hashable�Mapping�MutableMapping�Sequence)�get_Validator_class�validator_factory� mapping_hash�TypeDefinitionc @ s e Zd ZdS )�_AbortN)�__name__�
__module__� __qualname__� r r �@/opt/alt/python38/lib/python3.8/site-packages/cerberus/schema.pyr s r c @ s e Zd ZdZdS )� SchemaErrorz_
Raised when the validation schema is missing, has the wrong format or contains
errors.N)r r r �__doc__r r r r r s r c s� e Zd ZdZ� fdd�Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
edd� �Zedd� �Zedd� �Zd+d d �Zd d � Zd!d"� Zed#d$� �Zd%d&� Zd,d'd(�Zd)d*� Z� ZS )-�DefinitionSchemaz1A dict-subclass for caching of validated schemas.c sX dt � krHtdt�atj�� }|�tdtfd�tdt fd�d�� |t_t
t | �� | �S )N�SchemaValidator�callabler �hashable)r r )
�globalsr �SchemaValidatorMixinr �
types_mapping�copy�updater r r �superr �__new__)�cls�args�kwargsr �� __class__r r r! % s
��zDefinitionSchema.__new__c C s� t |t� �std��|| _t |t�r2|j�||�}t |t�snz t|�}W n$ t k
rl t
t j �
|���Y nX t|�| _td| jt j||d�| _| �|�}| �|� || _dS )z�
:param validator: An instance of Validator-(sub-)class that uses this
schema.
:param schema: A definition-schema as ``dict``. Defaults to an empty
one.
z0validator argument must be a Validator-instance.N)�
allow_unknownZ
error_handler�
target_schema�target_validator)�
isinstancer � RuntimeError� validatorr �schema_registry�getr �dict� Exceptionr r � SCHEMA_ERROR_DEFINITION_TYPE�format�SchemaValidationSchema�validation_schemar ZSchemaErrorHandler�schema_validator�expand�validate�schema)�selfr, r8 r r r �__init__4 s*
�
zDefinitionSchema.__init__c
C sf | j �� }z
||= W nD tk
r4 td| ��Y n. tk
rX } z|�W 5 d }~X Y n
X | j |= d S )Nz Schema has no field '%s' defined)r8 r �
ValueErrorr r0 )r9 �key� _new_schema�er r r � __delitem__U s
z DefinitionSchema.__delitem__c C s
| j | S �N)r8 )r9 �itemr r r � __getitem__` s z DefinitionSchema.__getitem__c C s
t | j�S r@ )�iterr8 �r9 r r r �__iter__c s zDefinitionSchema.__iter__c C s
t | j�S r@ )�lenr8 rD r r r �__len__f s zDefinitionSchema.__len__c C s t | �S r@ )�strrD r r r �__repr__i s zDefinitionSchema.__repr__c C s. | � d|i�d }| �||i� || j|< d S �Nr )r6 r7 r8 )r9 r< �valuer r r � __setitem__l s z DefinitionSchema.__setitem__c C s t | d�rt| j�S dS d S )Nr8 zNo schema data is set yet.)�hasattrrH r8 rD r r r �__str__q s
zDefinitionSchema.__str__c C s | � | j| j�� �S r@ )r& r, r8 r rD r r r r w s zDefinitionSchema.copyc C s< z| � |�}| �|�}W n tk
r, Y nX | �|�}|S r@ )�_expand_logical_shortcuts�_expand_subschemasr0 � _rename_deprecated_rulenames)r" r8 r r r r6 z s
zDefinitionSchema.expandc sz dd� � |� � D ]d\}}� fdd�|D �D ]H}|�dd�\}}|�|g i� || D ]}|| �||i� qT||= q*q|S )z�
Expand agglutinated rules in a definition-schema.
:param schema: The schema-definition to expand.
:return: The expanded schema-definition.
c S s t | t�o| �d�S )N)Zallof_Zanyof_Znoneof_Zoneof_)r* r �
startswith)�xr r r �
is_of_rule� s �z>DefinitionSchema._expand_logical_shortcuts.<locals>.is_of_rulec s g | ]}� |�r|�qS r r ��.0rS �rT r r �
<listcomp>� s z>DefinitionSchema._expand_logical_shortcuts.<locals>.<listcomp>�_� )�items�splitr �append)r" r8 �field�rulesZof_rule�operator�rulerK r rW r rO � s
z*DefinitionSchema._expand_logical_shortcutsc s � �fdd�}� �fdd�}�D ]� |� s,nD|� rN| � �� d ��� d< n"| � d�� d i�d �� d< dD ]2}|�� krt| � d�� | i�d �� |< qtdD ]^}|�� kr�t�� | t�s�q�g }�� | D ] }|�| � d|i�d � q�|�� |< q�q �S ) Nc s t �� t�od�� kS )Nr8 �r* r r �r^ r8 r r �has_schema_rule� s z<DefinitionSchema._expand_subschemas.<locals>.has_schema_rulec s< z t dd� �� d �� D ��W S tk
r6 Y dS X dS )zw
Tries to determine heuristically if the schema-constraints are aimed to
mappings.
c s s | ]}t |t�V qd S r@ rb rU r r r � <genexpr>� s zRDefinitionSchema._expand_subschemas.<locals>.has_mapping_schema.<locals>.<genexpr>r8 FN)�all�values� TypeErrorr rc r r �has_mapping_schema� s �
z?DefinitionSchema._expand_subschemas.<locals>.has_mapping_schemar8 r )� keysrules� valuesrules� keyschema� valueschema)ZallofZanyofr[ ZnoneofZoneof)r6 r* r
r] )r" r8 rd ri ra Znew_rules_definitionrA r rc r rP � s( " $ z#DefinitionSchema._expand_subschemasNc C s | j �||�S r@ )r8 r. )r9 rA �defaultr r r r. � s zDefinitionSchema.getc C s
| j �� S r@ )r8 r[ rD r r r r[ � s zDefinitionSchema.itemsc
C s� z,| � |�}| j�� }|�|� | �|� W nH tk
rP ttj� |���Y n, t
k
rt } z|�W 5 d }~X Y nX || _d S r@ ) r6 r8 r r r7 r; r r r1 r2 r0 )r9 r8 r= r> r r r r � s
zDefinitionSchema.updatec C s� | � � D ]r\}}t|t�r qdD ]X\}}||kr2q ||krBtd��tdj||d�t� | | | | | |< | | �|� q q| S )N))rl rj )r, Z
check_with)rm rk zTThe rule '{new}' is also present with its old name '{old}' in the same set of rules.zrThe rule '{old}' was renamed to '{new}'. The old name will not be available in the next major release of Cerberus.)�old�new)r[ r* rH r+ r r2 �DeprecationWarning�pop)r8 r^ r_ ro rp r r r rQ � s(
� ��z-DefinitionSchema._rename_deprecated_rulenamesc C s t | j�| _d S r@ )r3 r, r4 rD r r r � regenerate_validation_schema� s z-DefinitionSchema.regenerate_validation_schemac C sJ |dkr| j }t|�t| jj�f}|| jjkrF| �|� | jj�|� dS )ap
Validates a schema that defines rules against supported rules.
:param schema: The schema to be validated as a legal cerberus schema
according to the rules of the assigned Validator object.
Raises a :class:`~cerberus.base.SchemaError` when an invalid
schema is encountered.
N)r8 r
r, r �_valid_schemas� _validate�add)r9 r8 �_hashr r r r7 � s
zDefinitionSchema.validatec C s� t |t�r| jj�||�}i }|�� D ]R\}}t |t�rJt�||�||< q&i }|�� D ]\}}|||�dd�< qV|||< q&| j|dd�s�t | jj
��d S )N� rY F�Z normalize) r* r r, r- r. r[ �rules_set_registry�replacer5 r r )r9 r8 Z test_schemar^ r_ Z
test_rulesra Z
constraintr r r ru s
zDefinitionSchema._validate)N)N)r r r r r! r: r? rB rE rG rI rL rN r � classmethodr6 rO rP r. r[ r � staticmethodrQ rs r7 ru �
__classcell__r r r% r r " s2 !
'
r c @ s( e Zd Zi fdd�Zdd� Zdd� ZdS )�UnvalidatedSchemac C s t |t�st|�}|| _d S r@ )r* r r/ r8 �r9 r8 r r r r: ! s
zUnvalidatedSchema.__init__c C s d S r@ r r� r r r r7 & s zUnvalidatedSchema.validatec C s | � | j�� �S r@ )r& r8 r rD r r r r ) s zUnvalidatedSchema.copyN)r r r r: r7 r r r r r r s r c @ s e Zd Zdd� ZdS )r3 c C s d|j dd�| _d S )NFr/ )r' r8 �type)r_ r8 )r9 r, r r r r: 0 s �z SchemaValidationSchema.__init__N)r r r r: r r r r r3 / s r3 c s� e Zd ZdZ� fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Z� ZS ) r zi
This validator mixin provides mechanics to validate schemas passed to a Cerberus
validator.
c s2 |� dt� � |� dt� � tt| �j||� d S )N�known_rules_set_refs�known_schema_refs)�
setdefault�setr r r: )r9 r# r$ r% r r r: > s z SchemaValidatorMixin.__init__c C s
| j d S )z7The encountered references to rules set registry items.r� �Z_configrD r r r r� C s z)SchemaValidatorMixin.known_rules_set_refsc C s
| j d S )z4The encountered references to schema registry items.r� r� rD r r r r� H s z&SchemaValidatorMixin.known_schema_refsc C s
| j d S )z#The schema that is being validated.r( r� rD r r r r( M s z"SchemaValidatorMixin.target_schemac C s
| j d S )z.The validator whose schema is being validated.r) r� rD r r r r) R s z%SchemaValidatorMixin.target_validatorc C s� t |t�rR|| jkrd S | j�|� | jj�|�}|d krN| �|d| � d S |}td|i�t| jj �f}|| jj
krzd S | j |d| jj d�}||dd� |j
r�| �|j
� n| jj
�|� d S )Nz"Rules set definition %s not found.�turingF��document_crumbr' r8 ry )r* r r� rv r) rz r. �_errorr
r rt �_get_child_validatorr_ �_errors)r9 r^ rK �
definitionrw r, r r r �_check_with_bulk_schemaW s.
� � z,SchemaValidatorMixin._check_with_bulk_schemac C s� t |t�r npt |t�rH| j|dddiidd�}||dd�s|| �|j� n4t |t�r|tdd � |D ��s|| j|f }| �|d
� d S ) Nrk r� �listT�r� r8 r' Fry c s s | ]}t |t�V qd S r@ )r* r rU r r r re � s z@SchemaValidatorMixin._check_with_dependencies.<locals>.<genexpr>z)All dependencies must be a hashable type.) r* r r r� r� r� r
rf �
document_path)r9 r^ rK r, �pathr r r �_check_with_dependenciesw s
�
z-SchemaValidatorMixin._check_with_dependenciesc C s& t |�D ]\}}| �||f|� qd S r@ )� enumerater� )r9 r^ rK �ir8 r r r �_check_with_items� s z&SchemaValidatorMixin._check_with_itemsc C s� z| � ||�}W n tk
r& Y d S X t|�t| jj�f}|| jjkrLd S | j|d | jd�}|| �|�dd� |j r�| �
|j � n| jj� |� d S )Nr� Fry ) �&_handle_schema_reference_for_validatorr r
r) r rt r� Zroot_allow_unknown�_expand_rules_set_refsr� r� rv )r9 r^ rK rw r, r r r �_check_with_schema� s �z'SchemaValidatorMixin._check_with_schemac C sL t |t�rt|f�nt|�}|t| jj� }|rH| �|d�d�|��� d S )NzUnsupported types: {}z, )r* r r� r) �typesr� r2 �join)r9 r^ rK Zinvalid_constraintsr r r �_check_with_type� s �z%SchemaValidatorMixin._check_with_typec C s@ i }|� � D ].\}}t|t�r2| jj�|�||< q |||< q |S r@ )r[ r* r r) rz r. )r9 r8 �result�k�vr r r r� � s
z+SchemaValidatorMixin._expand_rules_set_refsc C sd t |t�s|S || jkr t�| j�|� | jj�|�}|d kr`| j|f }| � |d�
|�� t�|S )Nz Schema definition {} not found.) r* r r� r rv r) r- r. r� r� r2 )r9 r^ rK r� r� r r r r� � s
z;SchemaValidatorMixin._handle_schema_reference_for_validatorc C s� t |t�s | �|tj� dS | j|d| jjd�}|D ]X}td|i�t| jj �f}|| jj
kr^q4||dd� |j r~| �|j � q4| jj
� |� q4dS )z2{'allowed': ('allof', 'anyof', 'noneof', 'oneof')}NFr� r� ry )
r* r
r� r ZBAD_TYPEr� r) Zvalidation_rulesr
r rt r� rv )r9 ra r^ rK r, Z constraintsrw r r r �_validate_logical� s$
�
� z&SchemaValidatorMixin._validate_logical)r r r r r: �propertyr� r� r( r) r� r� r� r� r� r� r� r� r~ r r r% r r 8 s$
r c @ sN e Zd ZdZi fdd�Zdd� Zdd� Zdd � Zd
d � Zdd
d�Z dd� Z
d S )�Registryz�
A registry to store and retrieve schemas and parts of it by a name that can be used
in validation schemas.
:param definitions: Optional, initial definitions.
:type definitions: any :term:`mapping`
c C s i | _ | �|� d S r@ )�_storage�extend)r9 � definitionsr r r r: � s zRegistry.__init__c C s | � |�| j|< dS )aQ
Register a definition to the registry. Existing definitions are replaced
silently.
:param name: The name which can be used as reference in a validation
schema.
:type name: :class:`str`
:param definition: The definition.
:type definition: any :term:`mapping`
N)�_expand_definitionr� )r9 �namer� r r r rv � s z Registry.addc C s | j S )z_
Returns a :class:`dict` with all registered definitions mapped to their name.
)r� rD r r r rf � s z Registry.allc C s | j �� dS )z&Purge all definitions in the registry.N)r� �clearrD r r r r� � s zRegistry.clearc C s& t |��� D ]\}}| �||� q dS )a
Add several definitions at once. Existing definitions are
replaced silently.
:param definitions: The names and definitions.
:type definitions: a :term:`mapping` or an :term:`iterable` with
two-value :class:`tuple` s
N)r/ r[ rv )r9 r� r� r� r r r r� � s zRegistry.extendNc C s | j �||�S )z�
Retrieve a definition from the registry.
:param name: The reference that points to the definition.
:type name: :class:`str`
:param default: Return value if the reference isn't registered.
)r� r. )r9 r� rn r r r r. s z Registry.getc G s |D ]}| j �|d� qdS )z�
Unregister definitions from the registry.
:param names: The names of the definitions that are to be
unregistered.
N)r� rr )r9 �namesr� r r r �remove s zRegistry.remove)N) r r r r r: rv rf r� r� r. r� r r r r r� � s
r� c @ s e Zd Zedd� �ZdS )�SchemaRegistryc C s
t �|�S r@ �r r6 �r" r� r r r r� ! s z!SchemaRegistry._expand_definitionN�r r r r| r� r r r r r� s r� c @ s e Zd Zedd� �ZdS )�RulesSetRegistryc C s t �d|i�d S rJ r� r� r r r r� ' s z#RulesSetRegistry._expand_definitionNr� r r r r r� & s r� N) Z
__future__r �warningsr Zcerberusr Zcerberus.platformr r r r r r
Zcerberus.utilsr r r
r r0 r r r r r3 �objectr r� r� r� r- rz r r r r �<module> s" $E