U d @sddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZdd lmZdd lmZdd lmZed ejejBZd dZGdddeZGdddeZGdddeeeZGdddeeeZGdddeZ dS)N) OrderedDict)MutableMapping)cached_property)Mask)abort)Draft4Validator)ValidationError)not_none) HTTPStatusz)u?\'(?P.*)\' is a required propertycCst|tr|S|SN) isinstancetype)clsr5/tmp/pip-unpacked-wheel-dt_sn2ih/flask_restx/model.pyinstances rcsjeZdZdZfddZeddZddZedd Ze d d Z dd dZ ddZ ddZ e ZZS) ModelBasez Handles validation and swagger style inheritance for both subclasses. Subclass must define `schema` attribute. :param str name: The model public name cs>ttj||d|i_|_g_fdd}|_dS)Nnamecsjj|f|Sr ) __class__inheritrparentsselfrrinstance_inherit+sz,ModelBase.__init__..instance_inherit)superr__init__Z __apidoc__r __parents__r)rrargskwargsrrrrr%s   zModelBase.__init__cCs(dd|jD}tjt|jgf|S)z+ Return the ancestors tree cSsg|] }|jqSr) ancestors.0prrr 5sz'ModelBase.ancestors..)rsetunionr)rr"rrrr"0szModelBase.ancestorscCsD|j|kr|S|jD]}||}|r|Sqtd|ddS)NzParent z not found)rr get_parent ValueError)rrparentfoundrrrr)8s    zModelBase.get_parentcCs2|j}|jr*dd|jD}d||giS|SdS)NcSsg|]}dd|jiqS)z$refz#/definitions/{0})formatr)r$r+rrrr&Gsz(ModelBase.__schema__..ZallOf)_schemar)rschemarefsrrr __schema__BszModelBase.__schema__cGs |||d}|dd|_|S)z Inherit this model (use the Swagger composition pattern aka. allOf) :param str name: The new model name :param dict fields: The new model extra fields N)r)rrrmodelrrrrPszModelBase.inheritNc s`tj||d}z||Wn<tk rZttjdtfdd||DdYnXdS)N)resolverformat_checkerzInput payload validation failedc3s|]}|VqdSr ) format_error)r$errr esz%ModelBase.validate..)messageerrors) rr1validater rr BAD_REQUESTdictZ iter_errors)rdatar4r5 validatorrrrr;[szModelBase.validatecCsNt|j}|jdkr0t|jd}||ddd|D}||jfS)Nrequiredr.css|]}t|VqdSr )strr#rrrr8msz)ModelBase.format_error..) listpathr? RE_REQUIREDmatchr9groupappendjoin)rerrorrDrkeyrrrr6hs    zModelBase.format_errorcCsdj|jd|dS)NzModel({name},{{{fields}}}),)rfields)r-rrIkeysrrrr __unicode__ps zModelBase.__unicode__)NN)__name__ __module__ __qualname____doc__rpropertyr"r)r1 classmethodrr;r6rO__str__ __classcell__rrr!rrs     rcsXeZdZdZeZfddZeddZe ddZ dd Z e d d Z d d ZZS)RawModela: A thin wrapper on ordered fields dict to store API doc metadata. Can also be used for response marshalling. :param str name: The model public name :param str mask: an optional default model mask :param bool strict: validation should raise error when there is param not provided in schema csh|dd_|dd_jr:tjts:tj_ttj|f||fdd}|_dS)NmaskstrictFcsjj|f|Sr )rclonerrrrinstance_clonesz)RawModel.__init__..instance_clone) pop__mask__ __strict__r rrrXrr[)rrrr r\r!rrrs  zRawModel.__init__cCs|}t}d}|D]:\}}t|}|j||<|jrD||t|ddr|}qtt |pdd|||j rxt |j nddd}|j rd|d<t |S)N discriminatorFobject)r@ propertiesr`zx-maskrZadditionalProperties)wrapperr'itemsrr1r@addgetattrsortedrCr^rBr_r )rrbr@r`rfieldZ definitionrrrr.s&   zRawModel._schemacCsft|}|jD]}||jqdd|D}t|dkrJtdnt|dkrb|j|d_ |S)zG Resolve real fields before submitting them to marshal cSsg|]}t|ddr|qS)r`N)rf)r$frrrr&s z%RawModel.resolved..rz-There can only be one discriminator by schemar) copydeepcopyrupdateresolvedvalueslenr*rdefault)rrmr+ candidatesrrrrms      zRawModel.resolvedcCs>tjdtddt|ttfr.|j|f|S|||SdS)z Extend this model (Duplicate all fields) :param str name: The new model name :param dict fields: The new model extra fields :deprecated: since 0.9. Use :meth:`clone` instead. z*extend is is deprecated, use clone instead) stacklevelN)warningswarnDeprecationWarningr rCtupler[)rrrMrrrextends zRawModel.extendcGs,|}|D]}|t|q |||S)a\ Clone these models (Duplicate all fields) It can be used from the class >>> model = Model.clone(fields_1, fields_2) or from an Instanciated model >>> new_model = model.clone(fields_1, fields_2) :param str name: The new model name :param dict parents: The new model extra fields )rcrlrjrk)rrrrMr+rrrr[szRawModel.clonecs6|j|jfdd|D|j|jd}|j|_|S)Ncs g|]\}}|t|fqSr)rjrk)r$rKvaluememorrr&sz)RawModel.__deepcopy__..)rYrZ)rrrdr^r_r)rr{objrrzr __deepcopy__szRawModel.__deepcopy__)rPrQrRrSr=rcrrTr.rrmrxrUr[r}rWrrr!rrXxs     rXc@seZdZdZdS)Modelz A thin wrapper on fields dict to store API doc metadata. Can also be used for response marshalling. :param str name: The model public name :param str mask: an optional default model mask N)rPrQrRrSrrrrr~sr~c@seZdZdZeZdS) OrderedModelz A thin wrapper on ordered fields dict to store API doc metadata. Can also be used for response marshalling. :param str name: The model public name :param str mask: an optional default model mask N)rPrQrRrSrrcrrrrrsrcs.eZdZdZdfdd ZddZeZZS) SchemaModelz Stores API doc metadata based on a json schema. :param str name: The model public name :param dict schema: The json schema we are documenting Ncstt|||pi|_dSr )rrrr.)rrr/r!rrrszSchemaModel.__init__cCsdj|j|jdS)NzSchemaModel({name},{schema}))rr/)r-rr.rrrrrOszSchemaModel.__unicode__)N)rPrQrRrSrrOrVrWrrr!rr sr)!rjrert collectionsrcollections.abcrZwerkzeug.utilsrrYrr:rZ jsonschemarZjsonschema.exceptionsr utilsr Z_httpr compileIUrErrarrXr=r~rrrrrrs$         [|