Source code for api.media_handler

from functools import partial
from xml.parsers.expat import ExpatError

import xmltodict
import yaml
from dicttoxml import dicttoxml
from falcon.errors import HTTPBadRequest
from falcon.media import JSONHandler
from yaml import FullLoader
from yaml.parser import ParserError, ScannerError


class XMLHandler(JSONHandler):
    def __init__(self, dumps=None, loads=None):
        self.dumps = dumps or partial(
            dicttoxml, custom_root="guard", attr_type=False
        )
        self.loads = loads or partial(xmltodict.parse, force_list="item")

    def deserialize(self, stream, content_type, content_length):
        try:
            return self.loads(stream.read().decode("utf-8"))["guard"]
        except ExpatError:
            raise HTTPBadRequest("Invalid XML")


class YAMLHandler(JSONHandler):
    def __init__(self, dumps=None, loads=None):
        self.dumps = dumps or partial(yaml.dump, sort_keys=True, indent=3)
        self.loads = loads or partial(yaml.load, Loader=FullLoader)

    def deserialize(self, stream, content_type, content_length):
        try:
            x = stream.read().decode("utf-8")
            return self.loads(x)
        except (ParserError, ScannerError):
            raise HTTPBadRequest("Invalid YAML")