65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
"""
|
|
Models for notifications
|
|
"""
|
|
from django.contrib.auth.models import User # lint-amnesty, pylint: disable=imported-auth-user
|
|
from django.db import models
|
|
from model_utils.models import TimeStampedModel
|
|
|
|
|
|
class NotificationApplication(models.TextChoices):
|
|
"""
|
|
Application choices where notifications are generated from
|
|
"""
|
|
DISCUSSION = 'DISCUSSION'
|
|
|
|
|
|
class NotificationType(models.TextChoices):
|
|
"""
|
|
Notification type choices
|
|
"""
|
|
NEW_CONTRIBUTION = 'NEW_CONTRIBUTION'
|
|
|
|
|
|
class NotificationTypeContent:
|
|
"""
|
|
Notification type content
|
|
"""
|
|
NEW_CONTRIBUTION_NOTIFICATION_CONTENT = 'There is a new contribution. {new_contribution}'
|
|
|
|
|
|
class Notification(TimeStampedModel):
|
|
"""
|
|
Model to store notifications for users
|
|
|
|
.. no_pii:
|
|
"""
|
|
user = models.ForeignKey(User, related_name="notifications", on_delete=models.CASCADE)
|
|
app_name = models.CharField(max_length=64, choices=NotificationApplication.choices)
|
|
notification_type = models.CharField(max_length=64, choices=NotificationType.choices)
|
|
content = models.CharField(max_length=1024)
|
|
content_context = models.JSONField(default={})
|
|
content_url = models.URLField(null=True, blank=True)
|
|
last_read = models.DateTimeField(null=True, blank=True)
|
|
last_seen = models.DateTimeField(null=True, blank=True)
|
|
|
|
def __str__(self):
|
|
return f'{self.user.username} - {self.app_name} - {self.notification_type} - {self.content}'
|
|
|
|
def get_content(self):
|
|
return self.content
|
|
|
|
def get_content_url(self):
|
|
return self.content_url
|
|
|
|
def get_notification_type(self):
|
|
return self.notification_type
|
|
|
|
def get_app_name(self):
|
|
return self.app_name
|
|
|
|
def get_content_context(self):
|
|
return self.content_context
|
|
|
|
def get_user(self):
|
|
return self.user
|