Make the field migrations work.

[wolnelektury.git] / src / push / forms.py

diff --git a/src/push/forms.py b/src/push/forms.py
index b49096d..afe12d1 100644 (file)
--- a/src/push/forms.py
+++ b/src/push/forms.py
@@ -1,4 +1,3 @@
-# -*- coding: utf-8 -*-
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
@@ -17,7 +16,11 @@ class NotificationForm(forms.ModelForm):
 
     def save(self, commit=True):
         notification = super(NotificationForm, self).save(commit=commit)
-        wl_base = u'https://' + Site.objects.get_current().domain
-        notification.message_id = send_fcm_push(notification.title, notification.body, wl_base + notification.image.url)
+        wl_base = 'https://' + Site.objects.get_current().domain
+        if notification.image:
+            image_url = wl_base + notification.image.url
+        else:
+            image_url = None
+        notification.message_id = send_fcm_push(notification.title, notification.body, image_url)
         if commit:
             notification.save()