A lot of websites display notifications in one form or another these days. Facebook shows the new message notification, Gmail have the new mail notifications (with the support of browser notifications).

I was asked today how hard it would be to implement a real-time notification system. The answer, as it turns out, is: not very hard at all.

It was in fact so easy that I decided to write up this blog post on the subject to show how to implement a simple notification system in a Django app.

This has been tested with Django 1.6.8 and 1.7.1, and python 2.7.4 and python 3.4, with the following browsers:

Chrome 39.0.2171.71 (64-bit)

Safari 8.0 (10600.1.25.1)

Firefox developer edition 36.0a2 (2014-12-07)

Note: you need redis-server running. If you don't have Redis installed see redis.io, or use your default package manager to install it.

You can download the full source code here.

Setup

This could easily be implemented in an existing application but since I'm doing this from scratch I'll go through the steps of setting everything up.

First step is to install SwampDragon .

pip install swampdragon

The next thing is to create a new project.

dragon-admin startproject notifications cd notifications django-admin.py startapp demo

That's our project files and directories setup. Now we add paths to settings.py for the static files and html templates. Open notifications/settings.py in a text editor and add the following lines

STATIC_ROOT = os.path.join(BASE_DIR, 'static_root') MEDIA_URL = '/media/' MEDIA_ROOT = os.path.join(BASE_DIR, 'media') # Additional locations of static files STATICFILES_DIRS = [ os.path.join(BASE_DIR, 'static'), ] TEMPLATE_DIRS = [ os.path.join(BASE_DIR, 'templates') ]

Make sure you add demo to INSTALLED_APPS as well.

This tells Django where to find our static resources and templates.

That's all we need to do for our setup step.

Models

We store the notification as a model. This way we can load existing notification when the user loads the page the first time.

Open demo/models.py in a text editor and add the following code

from django.db import models from swampdragon.models import SelfPublishModel from .serializers import NotificationSerializer class Notification(SelfPublishModel, models.Model): serializer_class = NotificationSerializer message = models.TextField()

We have yet to create the NotificationSerializer so we will do that in the next step.

We add the SelfPublishModel mixin, so we don't have to worry about actually publishing the model, as that will happen as soon as it's created (note: for this post I have omitted handling updates to notifications).

The only field on this model is a message field, but you could easily add a title to the notification.

The notifications we are creating here are global, so everyone on the site will receive them. You could tailor the notifications to be on a per-user basis by adding a foreign key to the User model, and using swampdragon-auth to subscribe each user to their own notification channels (based on their username or some other unique identifier).

I will add some notes about this at the end of this post, but for now I will keep it simple.

Serializers

Add a new file: demo/serializers.py and open it in a text editor and add the following code

from swampdragon.serializers.model_serializer import ModelSerializer class NotificationSerializer(ModelSerializer): class Meta: model = 'demo.Notification' publish_fields = ['message']

We only want to publish the message field, so we specify this in the publish_fields property.

Routers

Without routers SwampDragon won't know how to deal with the incoming data.

Create a new file: demo/routers.py and add the following code

from swampdragon import route_handler from swampdragon.route_handler import ModelPubRouter from .models import Notification from .serializers import NotificationSerializer class NotificationRouter(ModelPubRouter): valid_verbs = ['subscribe'] route_name = 'notifications' model = Notification serializer_class = NotificationSerializer route_handler.register(NotificationRouter)

Admin

Since we don't have anything generating notifications, we'll add the Notification model to django admin so we can create them from there.

Open demo/admin.py in a text editor and replace the content with the following:

from django.contrib import admin from demo.models import Notification admin.site.register(Notification)

Views

We want to load existing notifications, and be able to show these notifications even if the SwampDragon server is not responding.

Open demo/views.py in a text editor and add the following code

from django.views.generic import ListView from .models import Notification class Notifications(ListView): model = Notification template_name = 'home.html' def get_queryset(self): return self.model.objects.order_by('-pk')[:5]

We use a standard Django CBV (class based view) to load the five last notifications. We set the model to Notification and specify the template to be home.html .

Open notifications/urls.py in a text editor and change it to the following code

from django.conf.urls import patterns, include, url from django.contrib import admin from demo.views import Notifications admin.autodiscover() urlpatterns = patterns('', url(r'^$', Notifications.as_view(), name='home'), url(r'^admin/', include(admin.site.urls)), )

That's all the python code we have to write for this. Now we need to add the template and a bit of JavaScript.

HTML

Create two new directories:

mkdir templates mkdir static

Create a new html template: templates/home.html and add the following

<!DOCTYPE html> <html> <head lang="en"> <meta charset="UTF-8"> <title></title> </head> <body> <h1>Notifications demo</h1> <!-- This is our list of notifications --> <ul id="notifications"> {% for notification in object_list %} <li>{{ notification.message }}</li> {% endfor %} </ul> <!-- SwampDragon --> <script type="text/javascript" src="{{ STATIC_URL }}swampdragon/js/vendor/sockjs-0.3.4.min.js"></script> <script type="text/javascript" src="{{ STATIC_URL }}swampdragon/js/swampdragon.js"></script> <script type="text/javascript" src="{{ STATIC_URL }}swampdragon/js/datamapper.js"></script> <script type="text/javascript" src="{{ STATIC_URL }}swampdragon/js/swampdragon-vanilla.js"></script> <script type="text/javascript" src="http://localhost:9999/settings.js"></script> <!-- notifications --> <script type="text/javascript" src="{{ STATIC_URL }}notifications.js"></script> </body> </html>

If you want to run this on anything but the local dev server, see this post on adding a context processor for the DRAGON_URL .

Javascript

The last piece of the puzzle is the JavaScript.

Create a new JavaScript file: static/notifications.js and add the following:

// Ask the browser for permission to show notifications // Taken from https://developer.mozilla.org/en-US/docs/Web/API/Notification/Using_Web_Notifications window.addEventListener('load', function () { Notification.requestPermission(function (status) { // This allows to use Notification.permission with Chrome/Safari if (Notification.permission !== status) { Notification.permission = status; } }); }); // Create an instance of vanilla dragon var dragon = new VanillaDragon({onopen: onOpen, onchannelmessage: onChannelMessage}); // This is the list of notifications var notificationsList = document.getElementById("notifications"); // New channel message received function onChannelMessage(channels, message) { // Add the notification addNotification((message.data)); } // SwampDragon connection open function onOpen() { // Once the connection is open, subscribe to notifications dragon.subscribe('notifications', 'notifications'); } // Add new notifications function addNotification(notification) { // If we have permission to show browser notifications // we can show the notifiaction if (window.Notification && Notification.permission === "granted") { new Notification(notification.message); } // Add the new notification var li = document.createElement("li"); notificationsList.insertBefore(li, notificationsList.firstChild); li.innerHTML = notification.message; // Remove excess notifications while (notificationsList.getElementsByTagName("li").length > 5) { notificationsList.getElementsByTagName("li")[5].remove(); } }

Finally

Create a database

If you are using Django 1.7+

python manage.py makemigrations python manage.py migrate

Otherwise

python manage.py syncdb

Create a super user. This is done by either answering "yes" to the question when you run syncdb or migrate . You can do it manually by running python manage.py createsuperuser .

Open a new terminal and type:

python manage.py runserver

and another terminal and type

python server.py

To test this out, open a web browser to http://localhost:8000. The browser will ask if localhost can have permission to show notifications. If you answer no, you will only see the notifications on the page.

Open another browser window to http://localhost:8000/admin/. Log in and add a new notification instance and you should be able to see the new notification.

Additional Notes

As mentioned above, you can have user specific notifications if you use something like swampdragon-auth.

Update the model to the following:

class Notification(SelfPublishModel, models.Model): serializer_class = NotificationSerializer message = models.TextField() user = models.ForeignKey(User)

You only have to make a small change to the router by adding get_subscription_contexts and setting the @login_required decorator on the subscribe verb.

@login_required def subscribe(self, **kwargs): super().subscribe(**kwargs) def get_subscription_contexts(self, **kwargs): return {'user_id': self.connection.user.pk}

Now notifications are user specific (and users who are not signed in can simply not subscribe).

The example code for this post is available at https://github.com/wildfish/swampdragon-django-notifications-demo