summaryrefslogtreecommitdiffstats
path: root/custodia/message/simple.py
blob: c7f32ee54ffd93cd620055dd032e5c4022923517 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
# Copyright (C) 2015  Custodia Project Contributors - see LICENSE file

from custodia.message.common import InvalidMessage
from custodia.message.common import MessageHandler
from six import string_types
import json


class SimpleKey(MessageHandler):
    """Handles 'simple' messages"""

    def parse(self, msg):
        """Parses a simple message

        :param req: ignored
        :param msg: the json-decoded value

        :raises UnknownMessageType: if the type is not 'simple'
        :raises InvalidMessage: if the message cannot be parsed or validated
        """

        # On requests we imply 'simple' if there is no input message
        if msg is None:
            return

        if not isinstance(msg, string_types):
            raise InvalidMessage("The 'value' attribute is not a string")

        self.payload = msg

    def reply(self, output):
        return json.dumps({'type': 'simple', 'value': output},
                          separators=(',', ':'))