diff --git a/api/serializers.py b/api/serializers.py index 5aef553..840e1b3 100644 --- a/api/serializers.py +++ b/api/serializers.py @@ -67,7 +67,8 @@ class BillingProjectSerializer(serializers.ModelSerializer): class NotificationSerializer(serializers.ModelSerializer): project = BillingProjectSerializer() + recipient = serializers.CharField() class Meta: model = Notification - fields = ['id', 'project', 'title', 'short_description', 'content', 'sent_status', 'is_read'] + fields = ['id', 'project', 'title', 'short_description', 'content', 'sent_status', 'is_read', 'created_at', 'recipient'] diff --git a/core/models.py b/core/models.py index 15f7f1a..7f98d4f 100644 --- a/core/models.py +++ b/core/models.py @@ -264,6 +264,11 @@ class Notification(BaseModel, TimestampMixin): sent_status = models.BooleanField() is_read = models.BooleanField() + def recipient(self): + if self.project: + return self.project.email_notification + return 'Admin' + def send(self): from core.utils.dynamic_setting import get_dynamic_setting, EMAIL_ADMIN