Files
django-anymail/djrill/views.py
2012-01-16 14:52:44 -08:00

63 lines
1.8 KiB
Python

from django.conf import settings
from django.core.exceptions import ImproperlyConfigured
from django.http import HttpResponse
from django.utils import simplejson as json
from django.views.generic import View, TemplateView
try:
import requests
except ImportError:
raise ImportError("Install the damn requirements!")
class DjrillApiMixin(object):
"""
Simple Mixin to grab the api info from the settings file.
"""
def __init__(self, *args, **kwargs):
self.api_key = getattr(settings, "MANDRILL_API_KEY", None)
self.api_url = getattr(settings, "MANDRILL_API_URL", None)
if not self.api_key:
raise ImproperlyConfigured("You have not set your mandrill api key "
"in the settings file.")
if not self.api_url:
raise ImproperlyConfigured("You have not added the Mandrill api "
"url to your settings.py")
class DjrillApiJsonObjectsMixin(object):
"""
Mixin to grab json objects from the api.
"""
api_uri = None
def get_json_objects(self):
payload = json.dumps({"key": self.api_key})
req = requests.post("%s/%s" % (self.api_url, self.api_uri),
data=payload)
if req.status_code == 200:
return req.content
raise Exception("OH GOD, NO!")
class DjrillIndexView(DjrillApiMixin, View):
def get(self, request):
payload = json.dumps({"key": self.api_key})
r = requests.post("%s/users/info.json" % self.api_url, data=payload)
return HttpResponse(r.content)
class DjrillSendersListView(DjrillApiMixin, DjrillApiJsonObjectsMixin,
TemplateView):
api_uri = "users/senders.json"
template_name = "djrill/senders_list.html"
def get(self, request):
objects = self.get_json_objects()
return self.render_to_response({"objects": objects})