L�X�9 � @ sK d d l m Z d d l m Z m Z m Z m Z 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 Gd d � d e � Z Gd d
� d
e � Z Gd d � d e � Z Gd
d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z Gd d � d e � Z e � e � Z Z d S)� )�absolute_import)�Callable�Hashable�Iterable�Mapping�MutableMapping)�copy)�errors)� _str_type)�get_Validator_class�validator_factory� mapping_hashc @ s e Z d Z d Z d S)� SchemaErrorz` Raised when the validation schema is missing, has the wrong format or
contains errors. N)�__name__�
__module__� __qualname__�__doc__� r r �
/schema.pyr s r c s e Z d Z d Z � f d d � Z i d d � Z d d � Z d d � Z d
d � Z d d
� Z d d � Z
d d � Z d d � Z d d � Z
e d d � � Z e d d � � Z e d d � � Z d d � Z d d � Z d d! d" � Z d# d$ � Z � S)%�DefinitionSchemaz3 A dict-subclass for caching of validated schemas. c s4 d t � k r t d t � a t t | � j | � S)N�SchemaValidator)�globalsr �SchemaValidatorMixinr �superr �__new__)�cls�args�kwargs)� __class__r r r s zDefinitionSchema.__new__c C s� t | t � � s t d � � | | _ t | t � rK | j j | | � } t | t � s� y t | � } Wn* t k
r� t
t j j
| � � � Yn Xt | � | _ t d d | j d t j d | d | �| _ | j | � } | j | � | | _ d S)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$ r0 r r r �__init__ s&
zDefinitionSchema.__init__c C sw | j j � } y | | =WnL t k
r> t d | � � Yn5 t k
rh } z | � WYd d } ~ Xn X| j | =d S)Nz Schema has no field '%s' defined)r0 r �
ValueErrorr r( )r1 �key� _new_schema�er r r � __delitem__; s
z DefinitionSchema.__delitem__c C s | j | S)N)r0 )r1 �itemr r r � __getitem__F s z DefinitionSchema.__getitem__c C s
t | j � S)N)�iterr0 )r1 r r r �__iter__I s zDefinitionSchema.__iter__c C s
t | j � S)N)�lenr0 )r1 r r r �__len__L s zDefinitionSchema.__len__c C s
t | � S)N)�str)r1 r r r �__repr__O s zDefinitionSchema.__repr__c C s= | j d | i � d } | j | | i � | | j | <d S)Nr )r. r/ r0 )r1 r4 �valuer r r � __setitem__R s z DefinitionSchema.__setitem__c C s
t | j � S)N)r> r0 )r1 r r r �__str__W s zDefinitionSchema.__str__c C s | j | j | j j � � S)N)r r$ r0 r )r1 r r r r Z s zDefinitionSchema.copyc C s; y" | j | � } | j | � } Wn t k
r6 Yn X| S)N)�_expand_logical_shortcuts�_expand_subschemasr( )r r0 r r r r. ] s
zDefinitionSchema.expandc s� d d � � x� | D]� } x� � f d d � | | D� D]q } | j d � \ } } | | j | g i � x1 | | | D]! } | | | j | | i � qx W| | | =q7 Wq W| 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 | j d � S)N�allof_�anyof_�noneof_�oneof_)rE rF rG rH )r" r
�
startswith)�xr r r �
is_of_rulem s z>DefinitionSchema._expand_logical_shortcuts.<locals>.is_of_rulec 3 s! | ] } � | � r | Vq d S)Nr )�.0rJ )rK r r � <genexpr>r s z=DefinitionSchema._expand_logical_shortcuts.<locals>.<genexpr>�_)�split�update�append)r r0 �fieldZof_rule�operator�ruler@ r )rK r rC f s
$ z*DefinitionSchema._expand_logical_shortcutsc sW � � f d d � } � � f d d � } x&� D] � | � sC nT | � rn | j � � d � � � d <n) | j d � � d i � d � � d <xG d D]? } | � � k r� | j d � � | i � d � � | <q� Wxk d D]c } | � � k r� g } x6 � � | D]& } | j | j d | i � d � qW| � � | <q� Wq1 W� S)Nc s! t � � t � o d � � k S)Nr0 )r" r r )rR r0 r r �has_schema_rule| s z<DefinitionSchema._expand_subschemas.<locals>.has_schema_rulec sF y) t d d � � � d j � D� � SWn t k
rA d SYn Xd S)zc Tries to determine heuristically if the schema-constraints are
aimed to mappings. c s s | ] } t | t � Vq d S)N)r" r )rL rJ r r r rM � s zRDefinitionSchema._expand_subschemas.<locals>.has_mapping_schema.<locals>.<genexpr>r0 FN)�all�values� TypeErrorr )rR r0 r r �has_mapping_schema� s
z?DefinitionSchema._expand_subschemas.<locals>.has_mapping_schemar0 r � keyschema� valueschema�allof�anyof�items�noneof�oneof)rZ r[ )r\ r] zitemsr_ r` )r. rQ )r r0 rU rY rT Znew_rules_definitionr8 r )rR r0 r rD z s$
")
-
$z#DefinitionSchema._expand_subschemasc C s� y< | j | � } | j j � } | j | � | j | � WnT t k
rh t t j j | � � � Yn4 t
k
r� } z | � WYd d } ~ Xn
X| | _ d S)N) r. r0 r rP r/ r3 r r r) r* r( )r1 r0 r5 r6 r r r rP � s
zDefinitionSchema.updatec C s t | j � | _ d S)N)r+ r$ r, )r1 r r r � regenerate_validation_schema� s z-DefinitionSchema.regenerate_validation_schemaNc C sW | d k r | j } t | � } | | j j k rS | j | � | j j j | � d S)N)r0 r
r$ �_valid_schemas� _validate�add)r1 r0 �_hashr r r r/ � s
zDefinitionSchema.validatec C s� t | t � r' | j j j | | � } | d k rB t t j � � t | � } x? | D]7 } t | | t � rU t j | | | | � | | <qU W| j
| d d �s� t | j
j � � d S)z� 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 this Validator object.
N� normalizeF) r" r
r$ r% r& r r ZSCHEMA_ERROR_MISSINGr �rules_set_registryr- )r1 r0 rR r r r rc � s
zDefinitionSchema._validate)r r r r r r2 r7 r9 r; r= r? rA rB r � classmethodr. rC rD rP ra r/ rc r r )r r r s$ % r c @ s+ e Z d Z i d d � Z d d � Z d S)�UnvalidatedSchemac C s( t | t � s t | � } | | _ d S)N)r" r r' r0 )r1 r0 r r r r2 � s zUnvalidatedSchema.__init__c C s d S)Nr )r1 r0 r r r r/ � s zUnvalidatedSchema.validateN)r r r r2 r/ r r r r ri � s ri c @ s e Z d Z d d � Z d S)r+ c C s" d d d | j d d i | _ d S)Nr Fr0 �typer' )Zrulesr0 )r1 r$ r r r r2 � s z SchemaValidationSchema.__init__N)r r r r2 r r r r r+ � s r+ c @ s� e Z d Z d Z e d d � � Z e j d d � � Z e d d � � Z e j d d � � Z e d d � � Z e d
d � � Z d d
� Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d S) r zX This validator is extended to validate schemas passed to a Cerberus
validator. c C s | j j d f � S)N�known_rules_set_refs)�_configr& )r1 r r r rk � s z)SchemaValidatorMixin.known_rules_set_refsc C s | | j d <d S)Nrk )rl )r1 r@ r r r rk � s c C s | j j d f � S)N�known_schema_refs)rl r& )r1 r r r rm � s z&SchemaValidatorMixin.known_schema_refsc C s | | j d <d S)Nrm )rl )r1 r@ r r r rm � s c C s | j d S)z% The schema that is being validated. r )rl )r1 r r r r � s z"SchemaValidatorMixin.target_schemac C s | j d S)z0 The validator whose schema is being validated. r! )rl )r1 r r r r! � s z%SchemaValidatorMixin.target_validatorc C s� | j d | d | j d d | j d � } xt | D]l } t d | i � } | | j j k r` q3 | | d d �| j r� | j | j � q3 | j j j | � q3 Wd S)z4 {'allowed': ('allof', 'anyof', 'noneof', 'oneof')} �document_crumbr0 r �turingrf FN)�_get_child_validator�root_allow_unknownr
r! rb �_errors�_errorrd )r1 rT rR r@ r$ Z constraintsre r r r �_validate_logical� s
z&SchemaValidatorMixin._validate_logicalc C s t | t � r d Sd S)NT)r" r )r1 r@ r r r �_validate_type_callable s z,SchemaValidatorMixin._validate_type_callablec C s t | t � r d Sd S)NT)r" r )r1 r@ r r r �_validate_type_hashable s z,SchemaValidatorMixin._validate_type_hashablec s0 � j | � r, t � f d d � | D� � Sd S)Nc 3 s | ] } � j | � Vq d S)N)rv )rL rJ )r1 r r rM s z@SchemaValidatorMixin._validate_type_hashables.<locals>.<genexpr>)Z_validate_type_listrV )r1 r@ r )r1 r �_validate_type_hashables s z-SchemaValidatorMixin._validate_type_hashablesc C s t | t � r� | | j k r" d S| j | f 7_ | j j j | � } | d k r} | j | f } | j | d | � d S| } t d | i � } | | j j k r� d S| j
d | d | j d d | j d � } | | d d �| j r| j | j � n | j j j
| � d S)Nz"Rules set definition %s not found.ro rn r0 r rf F)r" r
rk r! rg r& �
document_pathrs r
rb rp rq rr rd )r1 rR r@ �
definition�pathre r$ r r r �_validator_bulk_schema s*
z+SchemaValidatorMixin._validator_bulk_schemac C s� t | t � r d St | t � rU | | j j | j j k r� | j | d | � n0 t | t � r� x | D] } | j | | � qk Wd S)Nz%s is no valid coercer) r" r r
r! Z
validatorsZcoercersrs r �_validator_handler)r1 rR r@ �handlerr r r r| 7 s
z'SchemaValidatorMixin._validator_handlerc C s7 x0 t | � D]" \ } } | j | | f | � q
Wd S)N)� enumerater{ )r1 rR r@ �ir0 r r r �_validator_itemsB s z%SchemaValidatorMixin._validator_itemsc C s t | t � r� | | j k r" d S| j | f 7_ | j j j | � } | d k r} | j | f } | j | d | � d S| } t | � } | | j j k r� d S| j
d | d d d | j � } | | d d �| j r� | j | j � n | j j j
| � d S)Nz Schema definition %s not found.rn r0 r rf F)r" r
rm r! r% r& rx rs r
rb rp rq rr rd )r1 rR r@ ry rz re r$ r r r �_validator_schemaF s( z&SchemaValidatorMixin._validator_schemaN)r r r r �propertyrk �setterrm r r! rt ru rv rw r{ r| r� r� r r r r r � s r c @ sp e Z d Z d Z i d d � Z d d � Z d d � Z d d � Z d
d � Z d d
d � Z d d � 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 | _ | j | � d S)N)�_storage�extend)r1 � definitionsr r r r2 l s zRegistry.__init__c C s | j | � | j | <d S)aA 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� )r1 �namery r r r rd p s z Registry.addc C s | j S)zW Returns a :class:`dict` with all registered definitions mapped to
their name. )r� )r1 r r r rV { s z Registry.allc C s7 x0 t | � j � D] \ } } | j | | � q Wd S)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^ rd )r1 r� r� ry r r r r� � s zRegistry.extendc C s | j j � d S)z( Purge all definitions in the registry. N)r� �clear)r1 r r r r� � s zRegistry.clearNc C s | j 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& )r1 r� �defaultr r r r& � s z Registry.getc G s( x! | D] } | j j | d � q Wd S)z� Unregister definitions from the registry.
:param names: The names of the definitions that are to be
unregistered. N)r� �pop)r1 �namesr� r r r �remove� s
zRegistry.remove) r r r r r2 rd rV r� r� r& r� r r r r r� e s
r� c @ s" e Z d Z e d d � � Z d S)�SchemaRegistryc C s
t j | � S)N)r r. )r ry r r r r� � s z!SchemaRegistry._expand_definitionN)r r r rh r� r r r r r� � s r� c @ s" e Z d Z e d d � � Z d S)�RulesSetRegistryc C s t j d | i � d S)Nr )r r. )r ry r r r r� � s z#RulesSetRegistry._expand_definitionN)r r r rh r� r r r r r� � s r� N) Z
__future__r � collectionsr r r r r r Zcerberusr Zcerberus.platformr
Zcerberus.utilsr r r
r( r r ri r+ �objectr r� r� r� r% rg r r r r �<module> s ( �
�: