From e79567cf71980a1d65a369a915bb8e7f6a23fc2f Mon Sep 17 00:00:00 2001
From: Fergal Moran <fergal.moran@gmail.com>
Date: Sun, 24 Aug 2014 22:06:25 +0100
Subject: [PATCH] Corrected image in gritter

---
 spa/models/notification.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/spa/models/notification.py b/spa/models/notification.py
index 1b5b7df..418fb0f 100644
--- a/spa/models/notification.py
+++ b/spa/models/notification.py
@@ -7,7 +7,7 @@ from django.template import loader, Context
 
 from core.realtime.notification import post_notification
 from dss import localsettings
-from spa.models import BaseModel
+from spa.models import BaseModel, UserProfile
 
 
 class Notification(BaseModel):
@@ -33,10 +33,14 @@ class Notification(BaseModel):
         self.send_notification_email()
 
         sessions = self.to_user.user.session_set.all()
+        avatar_image = UserProfile.get_default_avatar_image()
+        if self.from_user is not None:
+            avatar_image = self.from_user.get_avatar_image()
+
         for session in sessions:
             post_notification(
                 session.session_key,
-                self.to_user.get_avatar_image(),
+                avatar_image,
                 self.notification_html)
 
         return super(Notification, self).save(force_insert, force_update, using, update_fields)