Dec28 changes completed

.gitignore

@@ -1,3 +1,5 @@
 media/
 **/__pycache__
-**/migrations
+**/migrations
+completeApp.log
+config.py

chat/admin.py

@@ -0,0 +1,8 @@
+from django.contrib import admin
+
+from .models import Message, MessageRequest, Room, Friend
+
+admin.site.register(Message)
+admin.site.register(MessageRequest)
+admin.site.register(Room)
+admin.site.register(Friend)

chat/apps.py

@@ -0,0 +1,5 @@
+from django.apps import AppConfig
+
+
+class ChatappConfig(AppConfig):
+    name = 'chat'

chat/consumers.py

@@ -0,0 +1,139 @@
+# chat/consumers.py
+import json
+from asgiref.sync import async_to_sync
+from channels.generic.websocket import WebsocketConsumer
+from .models import Message, Room
+from notifications.models import Notification
+from django.contrib.auth.models import User
+from django.shortcuts import get_object_or_404
+from .views import get_user_contact
+
+class ChatConsumer(WebsocketConsumer):
+
+    def message_to_json(self, message):
+        if message.pdf:
+            return {
+                'author': message.author.username,
+                'content': message.content,
+                'url': message.pdf.url,
+                'timestamp': str(message.timestamp),
+            }
+        else:
+            return {
+                'author': message.author.username,
+                'content': message.content, 
+                'timestamp': str(message.timestamp),
+            }
+
+
+    def messages_to_json(self, messages):
+        """Function that serialize messages object to json"""
+        result = []
+        for message in messages:
+            result.append(self.message_to_json(message))
+        return result
+
+
+    def fetch_messages(self, data):
+        messages = Message.objects.filter(room__id = data['roomname']).order_by('timestamp')
+        content = {
+            'message': self.messages_to_json(messages),
+            'command': 'fetch_messages'
+        }
+
+        self.send_message(content)
+
+
+    def new_message(self, data):
+        author = data['from']
+        room_id = data['roomname']
+        room_id_valid = get_object_or_404(Room, id=room_id)
+
+        if len(data['message']) != 0:
+            message = Message.objects.create(
+                author = get_object_or_404(User, username=author),
+                content = data['message'],
+                room = room_id_valid
+            )
+            authorMember = get_object_or_404(User, username=author)
+            for member in room_id_valid.members.exclude(username=authorMember.username):
+                if Notification.objects.filter(fromUser=authorMember, toUser=member, notificationType='newMessage', viewed=False, identifier=room_id_valid.pk).exists():
+                    notfi = Notification.objects.filter(fromUser=authorMember, toUser=member, notificationType='newMessage', viewed=False, identifier=room_id_valid.pk).first()
+                    count = int(notfi.content)
+                    notfi.content = count+1
+                    notfi.save()
+                else:
+                    notfi = Notification.objects.create(
+                                fromUser=authorMember, 
+                                toUser=member,
+                                notificationType='newMessage', 
+                                content=1,
+                                identifier=room_id_valid.pk,
+                            )
+            content = {
+                'command': 'new_message',
+                'message': self.message_to_json(message)
+            }
+        else:
+            message = get_object_or_404(Message, pk=data['message_id'])
+            content = {
+                'command': 'new_message',
+                'message': self.message_to_json(message)
+            }
+        return self.send_chat_messages(content)
+
+    command = {
+        'fetch_messages': fetch_messages, 
+        'new_message': new_message
+    }
+
+    def connect(self):
+        self.room_name = self.scope['url_route']['kwargs']['room_name']
+        self.room_group_name = 'chat_%s' % self.room_name
+
+        # Join room group
+        async_to_sync(self.channel_layer.group_add)(
+            self.room_group_name,
+            self.channel_name
+        )
+
+        self.accept()
+
+    def disconnect(self, close_code):
+        # Leave room group
+        async_to_sync(self.channel_layer.group_discard)(
+            self.room_group_name,
+            self.channel_name
+        )
+
+    # Receive message from WebSocket
+    def receive(self, text_data):
+        data = json.loads(text_data)
+        # print(data, "!!!!!!!!!!!!!!!!!!!!!!!!!")
+        self.command[data["command"]](self, data)
+
+    def send_chat_messages(self, message):
+        # Send message to room group
+        try:
+            async_to_sync(self.channel_layer.group_send)(
+                self.room_group_name,
+                {
+                    'type': 'chat_message',
+                    'message': message
+                }
+            )
+        except Exception as e:
+            print(e)
+
+    def send_message(self, message):
+        self.send(text_data=json.dumps(message))
+
+
+    # Receive message from room group
+    def chat_message(self, event):
+        message = event['message']
+
+        # Send message to WebSocket
+        self.send(text_data=json.dumps({
+            'message': message
+        }))

chat/models.py

@@ -0,0 +1,61 @@
+from django.db import models
+from django.contrib.auth import get_user_model
+from users.models import UserStudent
+from django.dispatch import receiver
+
+import uuid, os
+
+User = get_user_model()
+
+
+def unique_file_path_generator_chat(instance, filename):
+    """Generate file path for chat attachments"""
+    ext = filename.split('.')[-1]
+    filename = f'{uuid.uuid4()}.{ext}'
+    filepath = ("uploads/chat/{roomid}/").format(
+                    roomid=instance.room.pk, 
+                )
+    return os.path.join(filepath, filename)
+
+
+class Room(models.Model):
+
+    name = models.CharField(max_length = 50, blank=True, null=True)
+    members = models.ManyToManyField(User, related_name='members')
+    room_type = models.BooleanField(default=False, blank=True, null=True)
+
+    def __str__(self):
+        return '{}'.format(self.pk)
+
+
+class Message(models.Model):
+
+    author = models.ForeignKey(User, related_name="author_message", on_delete=models.CASCADE)
+    content = models.TextField()
+    timestamp = models.DateTimeField(auto_now_add=True)
+    room = models.ForeignKey(Room, related_name="room", on_delete=models.CASCADE, null=True)
+    pdf = models.FileField(upload_to=unique_file_path_generator_chat, null=True, blank=True)
+    # pdf = models.TextField(blank=True, null=True)
+
+    def __str__(self):
+        return self.author.username
+
+
+class MessageRequest(models.Model):
+
+    logged_in_user = models.ForeignKey(UserStudent,  related_name = 'message_user', on_delete=models.CASCADE)
+    request_sender = models.ForeignKey(UserStudent, related_name = 'message_request_sender', on_delete=models.CASCADE)
+    status = models.BooleanField(default=False)
+
+    def __str__(self):
+        return self.request_sender.user.user.username
+
+
+class Friend(models.Model):
+    student = models.OneToOneField(UserStudent, on_delete=models.CASCADE)
+    friends = models.ManyToManyField(UserStudent, related_name='friendsList')
+    # from_friend = models.ForeignKey(UserStudent, related_name='from_friends', on_delete=models.CASCADE, blank=True, null=True)
+    # to_friend = models.ForeignKey(UserStudent, related_name='to_friends', on_delete=models.CASCADE, blank=True, null=True)
+
+    def __str__(self):
+        return '{}'.format(self.pk)

chat/routing.py

@@ -0,0 +1,9 @@
+# chat/routing.py
+from django.urls import re_path
+
+from . import consumers
+
+websocket_urlpatterns = [
+    re_path(r'ws/chat/(?P<room_name>\w+)/$', consumers.ChatConsumer),
+    re_path(r'ws/chat/(?P<room_name>\w+)/$', consumers.ChatConsumer),
+]