dashboard: add notification model

circle/dashboard/models.py

@@ -7,7 +7,12 @@ from django.contrib.auth.signals import user_logged_in
 from django.db.models import (
 from django.db.models import (
     Model, ForeignKey, OneToOneField, CharField, IntegerField, TextField
     Model, ForeignKey, OneToOneField, CharField, IntegerField, TextField
 )
 )
-from django.utils.translation import ugettext_lazy as _
+from django.template.loader import render_to_string
+from django.utils.translation import ugettext_lazy as _, override
+
+from model_utils.models import TimeStampedModel
+from model_utils.fields import StatusField
+from model_utils import Choices
 
 
 from vm.models import Instance
 from vm.models import Instance
 from acl.models import AclBase
 from acl.models import AclBase
@@ -20,6 +25,32 @@ class Favourite(Model):
     user = ForeignKey(User)
     user = ForeignKey(User)
 
 
 
 
+class Notification(TimeStampedModel):
+    STATUS = Choices(('new', _('new')),
+                     ('delivered', _('delivered')),
+                     ('read', _('read')))
+
+    status = StatusField()
+    to = ForeignKey(User)
+    subject = CharField(max_length=128)
+    message = TextField()
+
+    class Meta:
+        ordering = ['-created']
+
+    @classmethod
+    def send(cls, user, subject, template, context={}):
+        try:
+            language = user.profile.preferred_language
+        except:
+            language = None
+        with override(language):
+            context['user'] = user
+            rendered = render_to_string(template, context)
+            subject = unicode(subject)
+        return cls.objects.create(to=user, subject=subject, message=rendered)
+
+
 class Profile(Model):
 class Profile(Model):
     user = OneToOneField(User)
     user = OneToOneField(User)
     preferred_language = CharField(verbose_name=_('preferred language'),
     preferred_language = CharField(verbose_name=_('preferred language'),
@@ -31,6 +62,9 @@ class Profile(Model):
         help_text=_('Unique identifier of the person, e.g. a student number.'))
         help_text=_('Unique identifier of the person, e.g. a student number.'))
     instance_limit = IntegerField(default=5)
     instance_limit = IntegerField(default=5)
 
 
+    def notify(self, subject, template, context={}):
+        return Notification.send(self.user, subject, template, context)
+
 
 
 class GroupProfile(AclBase):
 class GroupProfile(AclBase):
     ACL_LEVELS = (
     ACL_LEVELS = (