Feature/notification app

BaseDRF/settings.py

@@ -37,6 +37,7 @@
     'django.contrib.sessions',
     'django.contrib.messages',
     'django.contrib.staticfiles',
+    'notification.apps.NotificationConfig'
 ]
 
 MIDDLEWARE = [

notification/admin.py

@@ -0,0 +1,4 @@
+from django.contrib import admin
+from notification.models import *
+admin.site.register(UserNotification)
+admin.site.register(Notification)

notification/apps.py

@@ -0,0 +1,7 @@
+from django.apps import AppConfig
+
+
+class NotificationConfig(AppConfig):
+    name = 'notification'
+    def ready(self):
+        from . import signals

notification/models.py

@@ -0,0 +1,18 @@
+from django.db import models
+from django.contrib.auth.models import User
+class UserNotification(models.Model):
+    user = models.OneToOneField(to=User,on_delete=models.CASCADE)
+
+class Notification(models.Model):
+    user_notification = models.ForeignKey(to=UserNotification,on_delete=models.CASCADE)
+    TYPE_SMS = 'sms'
+    TYPE_EMAIL = 'email'
+    TYPE_IN_APP = 'In app'
+
+    TYPES = (
+        (TYPE_SMS, 'sms'),
+        (TYPE_EMAIL, 'email'),
+        (TYPE_IN_APP, 'In app'),
+    )
+
+    notif_type = models.CharField(max_length=10,choices=TYPES)

notification/serializers.py

@@ -0,0 +1,15 @@
+from django.db import models
+from rest_framework import fields,serializers
+
+
+class NotificationInputsSerializer(serializers.Serializer):
+    TYPE_SMS = 'sms'
+    TYPE_EMAIL = 'email'
+    TYPE_IN_APP = 'In app'
+
+    TYPES = (
+        (TYPE_SMS, 'sms'),
+        (TYPE_EMAIL, 'email'),
+        (TYPE_IN_APP, 'In app'),
+    )
+    notif_type = fields.ChoiceField(choices=TYPES)

notification/signals.py

@@ -0,0 +1,10 @@
+from django.dispatch import receiver
+from django.db.models.signals import post_save
+from django.contrib.auth.models import User
+from notification.models import UserNotification
+
+
+@receiver(post_save, sender=User)
+def auto_create_user_notification(sender, instance, created, **kwargs):
+    if created:
+        UserNotification.objects.create(user=instance)

notification/tests.py

@@ -0,0 +1,3 @@
+from django.test import TestCase
+
+# Create your tests here.

notification/views.py

@@ -0,0 +1,10 @@
+from django.contrib.auth.models import Permission
+from django.shortcuts import render
+from rest_framework import serializers
+from rest_framework.generics import CreateAPIView
+from .serializers import NotificationInputsSerializer
+class Notification(CreateAPIView):
+    serializer_class = NotificationInputsSerializer
+    # Permission_classes = ?
+    def perform_create(self, serializer):
+        return super().perform_create(serializer)