Add authentication to chatbox

chat_box/consumers.py

@@ -5,6 +5,7 @@ from django.urls import reverse
 from django.http import HttpResponse, HttpResponseRedirect
 from django.core import serializers
 
+from judge.jinja2.gravatar import gravatar
 from judge.models.profile import Profile
 
 
@@ -33,7 +34,14 @@ class ChatConsumer(AsyncWebsocketConsumer):
         text_data_json = json.loads(text_data)
         message = text_data_json['message']
 
-        message_saved = save_data_and_return(message)
+        author = self.scope['user']
+        author = Profile.objects.get(id=author.id)
+
+        message['author'] = author.username
+        message['css_class'] = author.css_class
+        message['image'] = gravatar(author, 32)
+
+        message_saved = save_data_and_return(message, author)
         message['time'] = message_saved[0]['fields']['time']
         message['id'] = message_saved[0]['pk']
 
@@ -56,10 +64,9 @@ class ChatConsumer(AsyncWebsocketConsumer):
 
 
 # return time
-def save_data_and_return(message):
+def save_data_and_return(message, author):
     new_message = Message(body=message['body'],
-                          author=Profile.objects
+                          author=author,
-                                        .get(pk=message['author_id']),
                           )
     new_message.save()
     json_data = serializers.serialize("json", 

templates/chat/chat.html

@@ -141,10 +141,6 @@
                 
                 message = {
                     'body': body,
-                    'image': img,
-                    'author': '{{request.profile}}',
-                    'author_id': {{request.profile.id}},
-                    'css_class': '{{request.profile.css_class}}',
                 }
 
                 chatSocket.send(JSON.stringify({