Merge pull request #4 from qburst/main

feat: Optimized code by introducing bulk_create function

apps/conversation/views.py

@@ -7,7 +7,8 @@
 from rest_framework import generics
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.decorators import permission_classes
-from apps.notification.utilities import create_notification
+from apps.notification.utilities import create_notification_bulk
+from ..notification.models import Notification
 from .models import Conversation, ConversationMessage
 from .serializers import ChatSerializer, ConversationSerializer
 
@@ -42,9 +43,12 @@ def api_add_message(request):
     message = ConversationMessage.objects.create(conversation_id=conversation_id, content=content, created_by=request.user)
 
     if conversation_id:
+        notifications = []
         for user in message.conversation.users.all():
             if user != request.user:
-                create_notification(request, user, 'message')
+                notification = Notification(created_by=request, to_user=user, notification_type='message')
+                notifications.append(notification)
+        create_notification_bulk(notifications)
 
     return JsonResponse({'success': True})
 

apps/feed/views.py

@@ -4,11 +4,12 @@
 from django.http import JsonResponse
 from django.contrib.auth.models import User
 from django.views.decorators.csrf import csrf_exempt
-from apps.notification.utilities import create_notification
+from apps.notification.utilities import create_notification, create_notification_bulk
 from rest_framework.decorators import api_view
 from rest_framework.pagination import PageNumberPagination
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.decorators import permission_classes
+from ..notification.models import Notification
 from .serializers import TweekSerializer
 from apps.feed.models import Tweek, Like, Dislike
 
@@ -64,11 +65,14 @@ def api_add_tweek(request):
 
     # send notification to users mentioned in the tweek
     results = re.findall("(^|[^@\w])@(\w{1,20})", body)
+    notifications = []
     for result in results:
         result = result[1]
         if User.objects.filter(username=result).exists() and result != request.user.username:
             user = User.objects.get(username=result)
-            create_notification(request, user, 'mention')
+            notification = Notification(created_by=request, to_user=user, notification_type='mention')
+            notifications.append(notification)
+    create_notification_bulk(notifications)
 
     return JsonResponse({'success': True})
 

apps/notification/utilities.py

@@ -5,3 +5,7 @@ def create_notification(created_by, to_user, notification_type):
     notification = Notification.objects.create(to_user=to_user, notification_type=notification_type,
                                                created_by=created_by)
     return notification
+
+def create_notification_bulk(notifications):
+    notification = Notification.objects.bulk_create(notifications)
+    return notification