Add ability to send any message through the admin panel

gcm/admin.py

@@ -1,22 +1,63 @@
+from functools import update_wrapper
 from django.contrib import admin
+from django.shortcuts import redirect, render_to_response
+from django.template import RequestContext
 from django.utils.translation import ugettext_lazy as _
 
+from .forms import MessageForm
 from .models import get_device_model
 
+Device = get_device_model()
+
 
 class DeviceAdmin(admin.ModelAdmin):
     list_display = ['dev_id', 'name', 'modified_date', 'is_active']
     search_fields = ('dev_id', 'name')
     list_filter = ['is_active']
     date_hierarchy = 'modified_date'
     readonly_fields = ('dev_id', 'reg_id')
-    actions = ['send_test_message']
+    actions = ['send_message_action']
+
+    def get_urls(self):
+        from django.conf.urls import patterns, url
+
+        def wrap(view):
+            def wrapper(*args, **kwargs):
+                return self.admin_site.admin_view(view)(*args, **kwargs)
+            return update_wrapper(wrapper, view)
+
+        urlpatterns = patterns('',
+            url(r'^send-message/$', wrap(self.send_message_view), name=self.build_admin_url('send_message')),
+        )
+        return urlpatterns + super(DeviceAdmin, self).get_urls()
+
+    def build_admin_url(self, url_name):
+        return '%s_%s_%s' % (self.model._meta.app_label, self.model._meta.module_name, url_name)
+
+    def send_message_view(self, request):
+        base_view = 'admin:%s' % self.build_admin_url('changelist')
+        session_key = 'device_ids'
+        device_ids = request.session.get(session_key)
+        if not device_ids:
+            return redirect(base_view)
+
+        form = MessageForm(data=request.POST or None)
+        if form.is_valid():
+            devices = Device.objects.filter(pk__in=device_ids)
+            for device in devices:
+                device.send_message(form.cleaned_data['message'])
+            self.message_user(request, _('Message was sent.'))
+            del request.session[session_key]
+            return redirect(base_view)
+
+        context = {'form': form, 'opts': self.model._meta, 'add': False}
+        return render_to_response('gcm/admin/send_message.html', context, context_instance=RequestContext(request))
 
-    def send_test_message(self, request, queryset):
-        for device in queryset:
-            device.send_message('GCM: Test message')
-        self.message_user(request, _("All messages were sent."))
-    send_test_message.short_description = _("Send test message")
+    def send_message_action(self, request, queryset):
+        request.session['device_ids'] = list(queryset.values_list('id', flat=True))
+        url = 'admin:%s' % self.build_admin_url('send_message')
+        return redirect(url)
+    send_message_action.short_description = _("Send message")
 
 
-admin.site.register(get_device_model(), DeviceAdmin)
+admin.site.register(Device, DeviceAdmin)

gcm/forms.py

@@ -1,4 +1,5 @@
 from django import forms
+from django.utils.translation import ugettext_lazy as _
 
 from .models import get_device_model
 
@@ -23,3 +24,8 @@ class Meta:
     def save(self, commit=True):
         self.instance.is_active = False
         return super(UnregisterDeviceForm, self).save(commit)
+
+
+class MessageForm(forms.Form):
+
+    message = forms.CharField(label=_('Message'), required=True)

gcm/templates/gcm/admin/send_message.html

@@ -0,0 +1,22 @@
+{% extends "admin/change_form.html" %}
+{% load i18n static %}
+
+{% block breadcrumbs %}
+        {% with app_label=opts.app_label original=_('Send message') has_change_permission=1 %}
+            {{ block.super }}
+        {% endwith %}
+{% endblock %}
+
+{% block field_sets %}
+    <h2>{% trans "Send message" %}</h2>
+    <fieldset class="module aligned wide">
+	    {{ form }}
+    </fieldset>
+{% endblock field_sets %}
+
+{% block inline_field_sets %}{% endblock inline_field_sets %}
+
+
+{% block submit_buttons_bottom %}
+	<input type="submit" value="{% trans 'Send' %}" class="default" name="_save" />
+{% endblock %}