aboutsummaryrefslogtreecommitdiffstats
path: root/src/rest/emitters.py
blob: c1ff5cc22b636c4970886d8b275e591bec57ade1 (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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
from django.template import RequestContext, loader
import json
from utils import dict2xml

class BaseEmitter(object):
    uses_forms = False

    def __init__(self, resource):
        self.resource = resource

    def emit(self, output):
        return output

class TemplatedEmitter(BaseEmitter):
    template = None

    def emit(self, output):
        if output is None:
            content = ''
        else:
            content = json.dumps(output, indent=4, sort_keys=True)

        template = loader.get_template(self.template)
        context = RequestContext(self.resource.request, {
            'content': content,
            'resource': self.resource,
        })
        
        ret = template.render(context)

        # Munge DELETE Response code to allow us to return content
        # (Do this *after* we've rendered the template so that we include the normal deletion response code in the output)
        if self.resource.resp_status == 204:
            self.resource.resp_status = 200

        return ret

class JSONEmitter(BaseEmitter):
    def emit(self, output):
        if output is None:
            # Treat None as no message body, rather than serializing
            return ''
        return json.dumps(output)

class XMLEmitter(BaseEmitter):
    def emit(self, output):
        if output is None:
            # Treat None as no message body, rather than serializing
            return ''
        return dict2xml(output)

class HTMLEmitter(TemplatedEmitter):
    template = 'emitter.html'
    uses_forms = True

class TextEmitter(TemplatedEmitter):
    template = 'emitter.txt'