U dWm@sLddlZddlZddlZddlmZddlmZmZddlmZm Z ddl m Z ddl m Z mZddlmZmZddlmZd d lmZmZmZmZd d lmZd d lmZd d lmZmZdZ GdddeZ!ddZ"ddZ#dCddZ$ddZ%ddZ&ddZ'Gddde(Z)Gdd 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,e)Z/Gd+d,d,e.e)Z0Gd-d.d.e.e)Z1Gd/d0d0e.e)Z2eZ3Gd1d2d2e.e)Z4Gd3d4d4e)Z5Gd5d6d6e-e)Z6Gd7d8d8e6Z7Gd9d:d:e,e)Z8Gd;d<dd>e/Z:Gd?d@d@e*Z;GdAdBdBe)Zint_get_value_for_keycallable_get_value_for_keyssplitkeyr<defaultr1r1r2 get_valueEs   rKcCs@t|dkrt|d||St|ddt|d|||SdS)Nr r)lenrDrF)keysr<rJr1r1r2rFOs  rFc Cslt|r.z ||WStttfk r,YnXt|r`z|t|WStttfk r^YnXt|||Sr*)r= IndexError TypeErrorKeyErrorrArC ValueErrorgetattrrHr1r1r2rDXs rDcCs6|dkr dSt|dr|St|dr,|St|jS)z Helper for converting an object to a dictionary only if it is not dictionary already or an indexable object nor a simple type N__marshallable__ __getitem__)r:rSdict__dict__r;r1r1r2to_marshallable_typefs  rWc@sReZdZdZdZdZdZdddZddZdd Z d d Z e d d Z ddZ dS)ra Raw provides a base field class from which others should extend. It applies no formatting by default, and should only be used in cases where data does not need to be formatted before being serialized. Fields should throw a :class:`MarshallingError` in case of parsing problem. :param default: The default value for the field, if no value is specified. :param attribute: If the public facing value differs from the internal value, use this to retrieve a different attribute from the response than the publicly named value. :param str title: The field title (for documentation purpose) :param str description: The field description (for documentation purpose) :param bool required: Is the field required ? :param bool readonly: Is the field read only ? (for documentation purpose) :param example: An optional data example (for documentation purpose) :param callable mask: An optional mask function to be applied to output objectNc KsB||_||_||_||_||_||_|dk r0|n|j|_||_dSr*) attributerJtitle descriptionrequiredreadonly__schema_example__examplemask) r.rJrYrZr[r\r]r_r`kwargsr1r1r2r,s z Raw.__init__cCs|S)a Formats a field's value. No-op by default - field classes that modify how the value of existing object keys should be presented should override this and apply the appropriate formatting. :param value: The value to format :raises MarshallingError: In case of formatting problem Ex:: class TitleCase(Raw): def format(self, value): return unicode(value).title() r1r.valuer1r1r2formatsz Raw.formatc Kst|jdkr|n|j|}|dkr>|d}|r:||S|Sz||}Wn<tk r}zd||t|}t|W5d}~XYnX|jr|j|S|S)a Pulls the value for the given key from the object, applies the field's formatting and returns the result. If the key is not found in the object, returns the default value. Field classes that create values which do not require the existence of the key in the object should override this and return the desired value. :raises MarshallingError: In case of formatting problem NrJz.Unable to marshal field "{0}" value "{1}": {2})rKrY_vrdr)r-r`apply) r.rIr<rarcrJdataemsgr1r1r2outputs  z Raw.outputcCst||}t|r|S|S)z;Helper for getting a value from attribute allowing callable)rRrE)r.rIrcr1r1r2res zRaw._vcCs t|Sr*)rschemar.r1r1r2 __schema__szRaw.__schema__cCs&|j|j|j|j|j|d|jdS)NrJ)typerdrZr[ZreadOnlyrJr_)__schema_type____schema_format__rZr[r]rer_rlr1r1r2rksz Raw.schema)NNNNNNNN)r3r4r5r6rorpr^r,rdrjrer rmrkr1r1r1r2rws&  rcsReZdZdZdZdfdd ZeddZddd Zfd d Z dd d Z Z S)r!a Allows you to nest one set of fields inside another. See :ref:`nested-field` for more information :param dict model: The model dictionary to nest :param bool allow_null: Whether to return None instead of a dictionary with null keys, if a nested dictionary has all-null keys :param bool skip_none: Optional key will be used to eliminate inner fields which value is None or the inner field's key not exist in data :param kwargs: If ``default`` keyword argument is present, a nested dictionary will be marshaled as its value if nested dictionary is all-null keys (e.g. lets you return an empty JSON object instead of null) NFc s.||_||_||_||_tt|jf|dSr*)modelas_list allow_null skip_noner+r!r,)r.rqrsrtrrrar/r1r2r,s zNested.__init__cCst|jd|jS)Nresolved)rRrqrlr1r1r2nestedsz Nested.nestedcKsPt|jdkr|n|j|}|dkr<|jr,dS|jdk r<|jSt||j|j|dS)Nrtordered)rKrYrsrJrrvrt)r.rIr<rxrarcr1r1r2rjs z Nested.outputcsttt|}d|jj}|jr8d|d<d|i|d<n8t|rh| dg}| d|i||d<n||d<|S)Nz#/definitions/{0}arrayrnz$refitemsallOf) r+r!rkrdrvnamerranyvaluesgetappend)r.rkrefr{r/r1r2rks   z Nested.schemacCs@|j}|d}|r2|t|dr,|jn|}|j|f|S)Nrqru)rVcopypoprfr:rur0r.r`rarqr1r1r2clone!s   z Nested.clone)FFF)F)N) r3r4r5r6ror,propertyrvrjrkrr7r1r1r/r2r!s   r!csHeZdZdZfddZddZdddZfd d Zdd d ZZ S)r"z Field for marshalling lists of other fields. See :ref:`list-field` for more information. :param cls_or_instance: The field type the list will contain. c s|dd|_|dd|_|dd|_tt|jf|d}t|trft |t s\t |||_ nt|t sxt |||_ dS)N min_items max_itemsuniquez>The type of the list elements must be a subclass of fields.Raw) rrrrr+r"r,r>rn issubclassrr) containerr.Zcls_or_instancera error_msgr/r1r2r,2s    z List.__init__cs`ttrttjtp*tjtkfdddkrDgSfddtDS)Ncsjjot|jjSr*)rrYr:)valrlr1r2is_attrHszList.format..is_attrcs8g|]0\}}j|t|ts&|r.s.|nqSr1)rrjr>rU).0idxrrZ is_nestedr.rcr1r2 MszList.format..)r>setr?rr!rnr enumeraterbr1rr2rdAs  z List.formatFcKsXt|jdkr|n|j|}t|r6t|ts6||S|dkrH|dSt||jj gS)NrJ) rKrYr=r>rUrdrerrrv)r.rIrgrxrarcr1r1r2rjWs   z List.outputcsHtt|}|j|d|d|ddd|d<|jj|d<|S)Nrrr)ZminItemsZmaxItemsZ uniqueItemsryrnrz)r+r"rkupdatererrmr.rkr/r1r2rkbs z List.schemaNcCs0|j}|d}|r"||}|j|f|SNr)rVrrrfr0rr1r1r2rms    z List.clone)F)N) r3r4r5r6r,rdrjrkrr7r1r1r/r2r")s   r"cs,eZdZdZfddZfddZZS)r&stringcs@|dd|_|dd|_|dd|_tt|j||dS)N min_length max_lengthpattern)rrrrr+r&r,r.argsrar/r1r2r,xszStringMixin.__init__cs4tt|}|j|d|d|dd|S)Nrrr)Z minLengthZ maxLengthr)r+r&rkrrerr/r1r2rk~szStringMixin.schemar3r4r5ror,rkr7r1r1r/r2r&us r&cs(eZdZfddZfddZZS)r'csN|dd|_|dd|_|dd|_|dd|_tt|j||dS)NminZ exclusiveMinmaxZ exclusiveMax)rminimumexclusiveMinimummaximumexclusiveMaximumr+r'r,rr/r1r2r,s zMinMaxMixin.__init__cs<tt|}|j|d|d|d|dd|S)Nrrrr)rrrr)r+r'rkrrerr/r1r2rkszMinMaxMixin.schema)r3r4r5r,rkr7r1r1r/r2r's r'cs,eZdZdZfddZfddZZS)r(numbercs$|dd|_tt|j||dS)Nmultiple)rrr+r(r,rr/r1r2r,szNumberMixin.__init__cs$tt|}|j|dd|S)Nr)Z multipleOf)r+r(rkrrerr/r1r2rkszNumberMixin.schemarr1r1r/r2r(s r(cs4eZdZdZfddZddZfddZZS)rz& Marshal a value as a string. cs@|dd|_|dd|_tt|j|||jp8|j|_dS)Nenum discriminator)rrrr+rr,r\rr/r1r2r,szString.__init__c Cs:z t|WStk r4}z t|W5d}~XYnXdSr*)r-rQr)r.rcver1r1r2rds z String.formatcsH|d}tt|}|r(|j|d|rD|ddkrD|d|d<|S)Nr)rr_r)rer+rrkr)r.rrkr/r1r2rks   z String.schema)r3r4r5r6r,rdrkr7r1r1r/r2rs rc@seZdZdZdZddZdS)rz Field for outputting an integer value. :param int default: The default value for the field, if no value is specified. integerc CsNz|dkr|jWSt|WSttfk rH}z t|W5d}~XYnXdSr*)rJrCrQrOr)rr1r1r2rds  zInteger.formatNr3r4r5r6rordr1r1r1r2rsrc@seZdZdZddZdS)rz A double as IEEE-754 double precision. ex : 3.141592653589793 3.1415926535897933e-06 3.141592653589793e+24 nan inf -inf c CsNz|dkr|jWSt|WSttfk rH}z t|W5d}~XYnXdSr*)rJfloatrQrOr)rr1r1r2rds  z Float.formatNr3r4r5r6rdr1r1r1r2rsrc@seZdZdZddZdS)rzu A floating point number with an arbitrary precision. ex: 634271127864378216478362784632784678324.23432 cCs tt|Sr*)r-rrbr1r1r2rdszArbitrary.formatNrr1r1r1r2rsrcs*eZdZdZdfdd ZddZZS)r z2 A decimal number with a fixed precision. c s0tt|jf|tdd|dd|_dS)Nz0.0r 1)r+r r,r precision)r.Zdecimalsrar/r1r2r,szFixed.__init__cCs4t|}|s |tkr tdt|j|jtdS)NzInvalid Fixed precision number.)rounding)r is_normalZEROr)r-quantizerr)r.rcZdvaluer1r1r2rdsz Fixed.format)r)r3r4r5r6r,rdr7r1r1r/r2r sr c@seZdZdZdZddZdS)rz Field for outputting a boolean value. Empty collections such as ``""``, ``{}``, ``[]``, etc. will be converted to ``False``. rcCst|Sr*)rrbr1r1r2rd szBoolean.formatNrr1r1r1r2rsrcs^eZdZdZdZdZdfdd ZddZd d Zd d Z d dZ ddZ fddZ Z S)ra5 Return a formatted datetime string in UTC. Supported formats are RFC 822 and ISO 8601. See :func:`email.utils.formatdate` for more info on the RFC 822 format. See :meth:`datetime.datetime.isoformat` for more info on the ISO 8601 format. :param str dt_format: ``rfc822`` or ``iso8601`` rz date-timeiso8601c stt|jf|||_dSr*)r+rr, dt_format)r.rrar/r1r2r,szDateTime.__init__cCsf|dkr dSt|tr0|jdkr$tnt}||St|tr>|St|trZt|j|j|j St ddS)NrzUnsupported DateTime format) r>r-rrrrryearmonthdayrQ)r.rcparserr1r1r2parse"s   zDateTime.parsec Cs|zH||}|jdkr"||WS|jdkr8||WStd|jWn.ttfk rv}z t|W5d}~XYnXdS)NrZrfc822zUnsupported date format %s)rrformat_iso8601 format_rfc822r)AttributeErrorrQ)r.rcrhr1r1r2rd3s     zDateTime.formatcCstt|S)z Turn a datetime object into a formatted date. :param datetime dt: The datetime to transform :return: A RFC 822 formatted date string )rr utctimetupler.dtr1r1r2r?szDateTime.format_rfc822cCs|S)z Turn a datetime object into an ISO8601 formatted date. :param datetime dt: The datetime to transform :return: A ISO 8601 formatted date string ) isoformatrr1r1r2rHszDateTime.format_iso8601cCs"|||}|r||SdSr*)rrerd)r.r|rcr1r1r2 _for_schemaQszDateTime._for_schemacs<tt|}|d|d<|d|d<|d|d<|S)NrJrr)r+rrkrrr/r1r2rkUs zDateTime.schema)r)r3r4r5r6rorpr,rrdrrrrkr7r1r1r/r2rs    rcs,eZdZdZdZfddZddZZS)rz Return a formatted date string in UTC in ISO 8601. See :meth:`datetime.date.isoformat` for more info on the ISO 8601 format. rc s*|ddtt|jfddi|dS)Nrr)rr+rr,)r.rar/r1r2r,fs z Date.__init__cCsJ|dkr dSt|trt|St|tr0|St|tr>|StddS)NzUnsupported Date format)r>r-rrrrQrbr1r1r2rjs   z Date.parse)r3r4r5r6rpr,rr7r1r1r/r2r]s rcs*eZdZdZdfdd ZddZZS) ra? A string representation of a Url :param str endpoint: Endpoint name. If endpoint is ``None``, ``request.endpoint`` is used instead :param bool absolute: If ``True``, ensures that the generated urls will have the hostname included :param str scheme: URL scheme specifier (e.g. ``http``, ``https``) NFc s(tt|jf|||_||_||_dSr*)r+rr,endpointabsolutescheme)r.rrrrar/r1r2r,sz Url.__init__c Kszt|}|jdk r|jntj}tt|fd|ji|}|jrp|jdk rP|jn|j}t||j|j dddfWStdd|j dddfWSt k r}z t |W5d}~XYnXdS)NZ _external) rWrr rr rrr netlocpathrOr)) r.rIr<rargrorter1r1r2rjsz Url.output)NFNr3r4r5r6r,rjr7r1r1r/r2rwsrcs(eZdZdZfddZddZZS)ra FormattedString is used to interpolate other values from the response into this field. The syntax for the source string is the same as the string :meth:`~str.format` method from the python stdlib. Ex:: fields = { 'name': fields.String, 'greeting': fields.FormattedString("Hello {name}") } data = { 'name': 'Doug', } marshal(data, fields) :param str src_str: the string to format with the other values from the response. c s tt|jf|t||_dSr*)r+rr,r-src_str)r.rrar/r1r2r,szFormattedString.__init__c KsLzt|}|jjf|WSttfk rF}z t|W5d}~XYnXdSr*)rWrrdrOrNr))r.rIr<rargerrorr1r1r2rjs zFormattedString.outputrr1r1r/r2rs rcs*eZdZdZdfdd ZddZZS)r#z Return the serialized object class name as string. :param bool dash: If `True`, transform CamelCase to kebab_case. Fc stt|jf|||_dSr*)r+r#r,dash)r.rrar/r1r2r,szClassName.__init__cKs&|jj}|dkrdS|jr"t|S|S)NrUrX)r0r3rr)r.rIr<ra classnamer1r1r2rjszClassName.output)Frr1r1r/r2r#sr#cs>eZdZdZd fdd Zd ddZddZdd d ZZS)r$a A Nested field handling inheritance. Allows you to specify a mapping between Python classes and fields specifications. .. code-block:: python mapping = { Child1: child1_fields, Child2: child2_fields, } fields = api.model('Thing', { owner: fields.Polymorph(mapping) }) :param dict mapping: Maps classes to their model/fields representation Fc s:||_|t|}tt|j|fd| i|dS)Nrs)mappingresolve_ancestorr?r~r+r$r,)r.rr\raparentr/r1r2r,szPolymorph.__init__c st|jdkr|n|j|dkr<|jr,dS|jdk r<|jStdsNtdfdd|jD}t|dkrtdj j n6t|dkrtdj j nt |dj |j |d SdS) Nr0z+Polymorph field only accept class instancescs g|]\}}t|kr|qSr1)rn)rclsfieldsrcr1r2rs z$Polymorph.output..rzUnknown class: r z%Unable to determine a candidate for: )r`rx)rKrYrsrJr:rQrrzrLr0r3rrur`)r.rIr<rxra candidatesr1rr2rjs.      zPolymorph.outputcCsZdd|D}tj|}t|dkrDdd|D}tdd||}|d|S)zp Resolve the common ancestor for all models. Assume there is only one common ancestor. cSsg|] }|jqSr1) ancestors)rmr1r1r2rsz.Polymorph.resolve_ancestor..r cSsg|] }|jqSr1)r|)rfr1r1r2rsz-Unable to determine the common ancestor for: z, r)r intersectionrLrQjoinrZ get_parent)r.modelsrr field_names parent_namer1r1r2rs   zPolymorph.resolve_ancestorNcCs>|j}|d}dD]}||dq||d<t|f|S)Nr)rsrqr`)rVrrr$)r.r`rgrfieldr1r1r2r s   zPolymorph.clone)F)F)N) r3r4r5r6r,rjrrr7r1r1r/r2r$s  r$csreZdZdZeZdZdZeZdZ fddZ ddZ e ddZ d d Zdd d ZfddZddZZS)r%z{ Field for marshalling list of "unkown" fields. :param cls_or_instance: The field type the list will contain. Nc sXtt|jf|d}t|trrnrrr)rrr/r1r2r,$s    zWildcard.__init__cs|dkr dS||jkr&|jdk r&|jSt|trFdd|D|_n&ddt|}fdd|D|_t|_||_|jS)NcSsg|]}|qSr1r1rxr1r1r2r6sz%Wildcard._flatten..cSs.|\}}t|s&|dr*|dr*dSdS)N__FT)inspect isroutine startswithendswith)rY attr_nameZattr_objr1r1r2Z__match_attributes9s z-Wildcard._flatten..__match_attributescsg|]}|r|qSr1r1rZ_Wildcard__match_attributesr1r2rBs) _obj_flatr>rUrzr getmembersr_cache)r.r< attributesr1rr2_flatten0s  zWildcard._flattencCs|jSr*)_lastrlr1r1r2rIHsz Wildcard.keycCs&t|_d|_d|_t|_d|_dSr*)rexcluderrrrrlr1r1r2resetLs zWildcard.resetFcCsd}t|}||rzf|r0|jd\}}n|j\}}||jkr|||jkr|t||tj r||}|j |||_ WqWqt k rYqYqXq|dkr|j dk r|j|j SdSt|jtrt||jj|jj|dS|j|S)Nrrw)fnmatch translaterrrrrrematch IGNORECASEaddrrNrJrrdr>r!rrvrt)r.rIr<rxrcregZobjkeyrr1r1r2rjSs>      zWildcard.outputcs&tt|}d|d<|jj|d<|S)NrXrnZadditionalProperties)r+r%rkrrmrr/r1r2rk}s zWildcard.schemacCs"|j}|d}|j|f|Sr)rVrrr0)r.rarqr1r1r2rs  zWildcard.clone)F)r3r4r5r6rrrrrrr,rrrIrrjrkrr7r1r1r/r2r%s   * r%)N)=rrrcalendarrrrdecimalrr email.utilsr urllib.parserr Zflaskr r Zwerkzeug.utilsr inputsrrrrerrorsrZ marshallingrutilsrr__all__r)r=rArKrFrDrWrXrr!r"r&r'r(rrrrrr rrrrrr#r$r%r1r1r1r2sR        pBL   M!P