src/api/emitters.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 """
   6 Wrappers for piston Emitter classes.
   7
   8 When outputting a queryset of selected models, instead of returning
   9 XML or JSON stanzas, SSI include statements are returned.
  10
  11 """
  12 from django.core.urlresolvers import reverse
  13 from django.db.models.query import QuerySet
  14 from piston.emitters import Emitter, XMLEmitter, JSONEmitter
  15 from catalogue.models import Book, Fragment, Tag
  16 from django.utils.translation import get_language
  17
  18
  19 class SsiQS(object):
  20     """A wrapper for QuerySet that won't serialize."""
  21
  22     def __init__(self, queryset):
  23         self.queryset = queryset
  24
  25     def __unicode__(self):
  26         raise TypeError("This is not serializable.")
  27
  28     def get_ssis(self, emitter_format):
  29         """Yields SSI include statements for the queryset."""
  30         url_pattern = reverse(
  31             'api_include',
  32             kwargs={
  33                 'model': self.queryset.model.__name__.lower(),
  34                 'pk': '0000',
  35                 'emitter_format': emitter_format,
  36                 'lang': get_language(),
  37             })
  38         for instance in self.queryset:
  39             yield "<!--#include file='%s'-->" % url_pattern.replace('0000', str(instance.pk))
  40
  41
  42 class SsiEmitterMixin(object):
  43     def construct(self):
  44         if isinstance(self.data, QuerySet) and self.data.model in (Book, Fragment, Tag):
  45             return SsiQS(self.data)
  46         else:
  47             return super(SsiEmitterMixin, self).construct()  # WTF
  48
  49
  50 class SsiJsonEmitter(SsiEmitterMixin, JSONEmitter):
  51     def render(self, request):
  52         try:
  53             return super(SsiJsonEmitter, self).render(request)
  54         except TypeError:
  55             return '[%s]' % ",".join(self.construct().get_ssis('json'))
  56
  57 Emitter.register('json', SsiJsonEmitter, 'application/json; charset=utf-8')
  58
  59
  60 class SsiXmlEmitter(SsiEmitterMixin, XMLEmitter):
  61     def render(self, request):
  62         try:
  63             return super(SsiXmlEmitter, self).render(request)
  64         except TypeError:
  65             return '<?xml version="1.0" encoding="utf-8"?>\n' \
  66                 '<response><resource>%s</resource></response>' % \
  67                 '</resource><resource>'.join(self.construct().get_ssis('xml'))
  68
  69 Emitter.register('xml', SsiXmlEmitter, 'text/xml; charset=utf-8')