Views for testing notifications. Should not be present in production
from django.conf import settings
from django.http import HttpResponse
from django.views.generic import TemplateView, View

from . import registry

class TestNotificationList(TemplateView):
    View for listing out all notifications with links to view rendered versions of them

    template_name = 'herald/test/notification_list.html'

    def get_context_data(self, **kwargs):
        context = super(TestNotificationList, self).get_context_data(**kwargs)

        context['notifications'] = [
            (index, x.__name__, x.render_types, (y.__name__ for y in x.__bases__)) for index, x in enumerate(registry._registry)  # pylint: disable=W0212

        return context

class TestNotification(View):
    View for showing rendered test notification

    def get(self, request, *args, **kwargs):  # pylint: disable=W0613
        GET request

        index = int(kwargs['index'])
        render_type = kwargs['type']

        obj = registry._registry[index](*registry._registry[index].get_demo_args())  # pylint: disable=W0212

        context = obj.get_context_data()

        content = obj.render(render_type, context)

        render_type = "plain" if render_type == "text" else render_type
        charset = settings.DEFAULT_CHARSET

        return HttpResponse(content, content_type='text/{}; charset={}'.format(render_type, charset))