U ǽbG@s*ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z e eZdd Zd d Zeed ZGd ddeZGdddeZdZGdddejZeeejeeejeeejeeejddZGdddej Z!e!"ee!j#ddZ$GdddeZ%dS)N) OrderedDict) force_bytes)yaml)openapiSwaggerValidationErrorc Csvzddlm}ddlm}Wntk r2YdSXz ||Wn0|k rp}ztt||W5d}~XYnXdS)Nr)parse)ValidationError)Z flex.corer Zflex.exceptionsr ImportErrorrstr)specZ validate_flexr exr3/tmp/pip-unpacked-wheel-o6yr43pd/drf_yasg/codecs.py_validate_flexs  rc CsZddlm}ddlm}z ||Wn0|k rT}ztt||W5d}~XYnXdS)Nrr) validate_spec)Zswagger_spec_validator.commonrZ"swagger_spec_validator.validator20rr )r ZSSVErrZ validate_ssvrrrr _validate_swagger_spec_validators    r)ZflexZssvc@sDeZdZdZddZeddZddZdd Zd d Z d d Z dS) _OpenAPICodecNcCs ||_dSNZ _validators)self validatorsrrr__init__/sz_OpenAPICodec.__init__cCs|jS)z List of validator names to applyr)rrrrr2sz_OpenAPICodec.validatorsc Cst|tjstd||}i}|jD]J}zt|t|Wq(t k rp}zt |||<W5d}~XYq(Xq(|rt d ||||}t t ||t||S)a4Transform an :class:`.Swagger` object to a sequence of bytes. Also performs validation and applies settings. :param openapi.Swagger document: Swagger spec object as generated by :class:`.OpenAPISchemaGenerator` :return: binary encoding of ``document`` :rtype: bytes z%Expected a `openapi.Swagger` instanceNzspec validation failed: {}) isinstancerZSwagger TypeErrorgenerate_swagger_objectr VALIDATORScopydeepcopyrr formatloggerwarningr _dump_dict)rdocumentr errorsZ validatoreexcrrrencode7s    z_OpenAPICodec.encodecCst||S)zDDump an error message into an encoding-appropriate sequence of bytes)rr#)rerrrrr encode_errorTsz_OpenAPICodec.encode_errorcCs tddS)zDump the given dictionary into its string representation. :param dict spec: a python dict :return: string representation of ``spec`` :rtype: str or bytes zoverride this methodN)NotImplementedErrorrr rrrr#Xsz_OpenAPICodec._dump_dictcCs|S)zGenerates the root Swagger object. :param openapi.Swagger swagger: Swagger spec object as generated by :class:`.OpenAPISchemaGenerator` :return: swagger spec as dict :rtype: OrderedDict )Zas_odict)rZswaggerrrrrasz%_OpenAPICodec.generate_swagger_object) __name__ __module__ __qualname__ media_typerpropertyrr(r*r#rrrrrr,s  rcs*eZdZdZdfdd ZddZZS)OpenAPICodecJsonapplication/jsonFcs tt||||_||_dSr)superr2rprettyr0)rrr5r0 __class__rrrnszOpenAPICodecJson.__init__cCs<|jr.tj|ddd}|ddkr*|d7}|St|SdS)z-Dump ``spec`` into JSON. :rtype: str),z: )indent separators N)r5jsondumps)rr outrrrr#ss  zOpenAPICodecJson._dump_dict)Fr3r-r.r/r0rr# __classcell__rrr6rr2ksr2ztag:yaml.org,2002:mapcs<eZdZdZddZd fdd Zd dd Zd d ZZS)SaneYamlDumperzYYamlDumper class usable for dumping ``OrderedDict`` and list instances in a standard way.cCsdS)zDisable YAML references.Tr)rdatarrrignore_aliasesszSaneYamlDumper.ignore_aliasesFc stt|jf|dd|S)zLhttps://stackoverflow.com/a/39681672 Indent list elements. F)flow indentless)r4rCincrease_indent)rrFrGkwargsr6rrrHszSaneYamlDumper.increase_indentNc Cst}g}tj|||d}|jdk r.||j|j<d}t|drD|}|D]V\}}||} ||} t| tj rv| j rzd}t| tj r| j rd}| | | fqH|dkr|j dk r|j |_ n||_ |S)zhttps://gist.github.com/miracle2k/3184458 Make PyYAML output an OrderedDict. It will do so fine if you use yaml.dump(), but that generates ugly, non-standard YAML code. To use yaml.safe_dump(), you need the following. ) flow_styleNTitemsF) YAML_MAP_TAGrZ MappingNodeZ alias_keyZrepresented_objectshasattrrKZrepresent_datarZ ScalarNodestyleappenddefault_flow_stylerJ) rmappingrJtagvaluenodeZ best_styleZitem_keyZ item_valueZnode_keyZ node_valuerrrrepresent_odicts*         zSaneYamlDumper.represent_odictcCs$d|kr|jd|ddS|d|S)Nr=ztag:yaml.org,2002:str|)rN)Zrepresent_scalar)rtextrrrrepresent_textszSaneYamlDumper.represent_text)FF)N) r-r.r/__doc__rErHrUrXrBrrr6rrCs  rCcCstj|td|rdnddS)aDump the given data dictionary into a sane format: * OrderedDicts are dumped as regular mappings instead of non-standard !!odict * multi-line mapping style instead of json-like inline style * list elements are indented into their parents * YAML references/aliases are disabled :param dict data: the data to be dumped :param bool binary: True to return a utf-8 encoded binary object, False to return a string :return: the serialized YAML :rtype: str or bytes Fzutf-8N)ZDumperrPencoding)rdumprC)rDbinaryrrryaml_sane_dumps r]c@seZdZdddZdS)SaneYamlLoaderFcCs||t||Sr)Zflatten_mappingrZconstruct_pairs)rrTdeeprrrconstruct_odicts zSaneYamlLoader.construct_odictN)F)r-r.r/r`rrrrr^sr^cCstj|tdS)zLoad the given YAML stream while preserving the input order for mapping items. :param stream: YAML stream (can be a string or a file-like object) :rtype: OrderedDict )Loader)rloadr^)streamrrryaml_sane_loadsrdcs*eZdZdZdfdd ZddZZS)OpenAPICodecYamlapplication/yamlcstt||||_dSr)r4rerr0)rrr0r6rrrszOpenAPICodecYaml.__init__cCs t|ddS)z/Dump ``spec`` into YAML. :rtype: bytesT)r\)r]r,rrrr#szOpenAPICodecYaml._dump_dict)rfrArrr6rresre)&rr>logging collectionsrZcoreapi.compatrZruamelrrr%r getLoggerr-r!rrrobjectrr2rLZ SafeDumperrCZadd_representerbytesrXr rUZadd_multi_representerr]Z SafeLoaderr^Zadd_constructorr`rdrerrrrs4        ?4