aboutsummaryrefslogtreecommitdiffstats
path: root/src/rest/emitters.py
blob: 5dad624c359f4ca1ad3e7f45ee1c6c6ab5c111e3 (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
from django.template import RequestContext, loader
from django.core.handlers.wsgi import STATUS_CODE_TEXT
import json
from utils import dict2xml

class BaseEmitter(object):
    def __init__(self, resource, request, status, headers):
        self.request = request
        self.resource = resource
        self.status = status
        self.headers = headers
    
    def emit(self, output):
        return output

class TemplatedEmitter(BaseEmitter):
    template = None

    def emit(self, output):
        content = json.dumps(output, indent=4)
        template = loader.get_template(self.template)
        context = RequestContext(self.request, {
            'content': content,
            'status': self.status,
            'reason': STATUS_CODE_TEXT.get(self.status, ''),
            'headers': self.headers,
            'resource_name': self.resource.__class__.__name__,
            'resource_doc': self.resource.__doc__,
            'create_form': self.resource.create_form and self.resource.create_form() or None,
            'update_form': self.resource.update_form and self.resource.update_form() or None,
            'allowed_methods': self.resource.allowed_methods,
            'request': self.request,
            'resource': self.resource,
        })
        return template.render(context)
    
class JSONEmitter(BaseEmitter):
    def emit(self, output):
        return json.dumps(output)

class XMLEmitter(BaseEmitter):
    def emit(self, output):
        return dict2xml(output)

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

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