better login/register page (new default login)

[wolnelektury.git] / src / newsletter / admin.py

diff --git a/src/newsletter/admin.py b/src/newsletter/admin.py
index 8c38f3f..9fe387d 100644 (file)
--- a/src/newsletter/admin.py
+++ b/src/newsletter/admin.py
@@ -1,3 +1,28 @@
+# -*- coding: utf-8 -*-
+from django.conf.urls import url
 from django.contrib import admin
+from django.http.response import HttpResponse
+from django.views.decorators.cache import never_cache
 
-# Register your models here.
+from newsletter.models import Subscription
+
+
+class SubscriptionAdmin(admin.ModelAdmin):
+    list_display = ('email', 'active')
+    list_filter = ('active',)
+
+    def get_urls(self):
+        urls = super(SubscriptionAdmin, self).get_urls()
+        my_urls = [
+            url(r'^extract/$', self.extract_subscribers, name='extract_subscribers'),
+        ]
+        return my_urls + urls
+
+    @never_cache
+    def extract_subscribers(self, request):
+        active_subscriptions = Subscription.objects.filter(active=True)
+        return HttpResponse('\n'.join(active_subscriptions.values_list('email', flat=True)),
+                            content_type='text/plain')
+
+
+admin.site.register(Subscription, SubscriptionAdmin)