diff options
Diffstat (limited to 'src/notifier/models.py')
| -rw-r--r-- | src/notifier/models.py | 25 |
1 files changed, 25 insertions, 0 deletions
diff --git a/src/notifier/models.py b/src/notifier/models.py new file mode 100644 index 0000000..49189e8 --- /dev/null +++ b/src/notifier/models.py @@ -0,0 +1,25 @@ +############################################################################## +# Copyright (c) 2018 Sawyer Bergeron, Parker Berberian, and others. +# +# All rights reserved. This program and the accompanying materials +# are made available under the terms of the Apache License, Version 2.0 +# which accompanies this distribution, and is available at +# http://www.apache.org/licenses/LICENSE-2.0 +############################################################################## + +from django.db import models +from account.models import UserProfile + + +class Notification(models.Model): + title = models.CharField(max_length=150) + content = models.TextField() + recipients = models.ManyToManyField(UserProfile, related_name='notifications') + is_html = models.BooleanField(default=True) + read_by = models.ManyToManyField(UserProfile, related_name='read_notifications') + + def __str__(self): + return self.title + + def to_preview_html(self): + return "<h3>" + self.title + "</h3>" # TODO - template? |