From 2634ae627659b71790147581880529dad741f425 Mon Sep 17 00:00:00 2001
From: jdlugosz963 <jdlugosz963@gmail.com>
Date: Sun, 28 Nov 2021 22:57:10 +0100
Subject: change login api

---
 chat_api/views.py | 55 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 53 insertions(+), 2 deletions(-)

(limited to 'chat_api/views.py')

diff --git a/chat_api/views.py b/chat_api/views.py
index f94f3a0..eca6d45 100644
--- a/chat_api/views.py
+++ b/chat_api/views.py
@@ -5,6 +5,8 @@ from rest_framework.serializers import Serializer
 from rest_framework.views import APIView
 from rest_framework import permissions
 from rest_framework import generics
+from rest_framework.views import Http404
+from rest_framework.generics import get_object_or_404
 from knox.auth import TokenAuthentication
 
 from django.db.models.query import Q
@@ -12,6 +14,17 @@ from django.db.models.query import Q
 from .models import *
 from .serializers import *
 
+
+def set_users(request):
+    users_list = request.data.get('users', [])
+    user_pk = request.user.pk
+
+    if isinstance(users_list, list):
+        if user_pk not in users_list:
+            users_list.append(user_pk)
+
+    request.data['users'] = users_list
+
 class GroupView(generics.ListCreateAPIView):
     permission_classes = (permissions.IsAuthenticated, )
     authentication_classes = (TokenAuthentication, )
@@ -20,9 +33,47 @@ class GroupView(generics.ListCreateAPIView):
     def get_queryset(self):
         return Group.objects.filter(
                     Q(owner=self.request.user) |
-                    Q(users__in=(self.request.user, ))
+                    Q(users__in=(self.request.user.pk, ))
                 ).distinct()
+    
+    def get_serializer_context(self):
+        context = super(GroupView, self).get_serializer_context()
+        context.update({"request": self.request})
+        return context
 
     def create(self, request, *args, **kwargs):
         request.data['owner'] = request.user.pk
-        return super().create(request, *args, **kwargs)
\ No newline at end of file
+        set_users(request)
+        return super().create(request, *args, **kwargs)
+
+class GroupDetailView(generics.RetrieveUpdateDestroyAPIView):
+    permission_classes = (permissions.IsAuthenticated, )
+    authentication_classes = (TokenAuthentication, )
+    serializer_class = GroupSerializer
+
+    def get_queryset(self):
+        return Group.objects.filter(
+                    Q(owner=self.request.user)
+                ).distinct()
+
+    def get_serializer_context(self):
+        context = super(GroupDetailView, self).get_serializer_context()
+        context.update({"request": self.request})
+        return context
+
+    def update(self, request, *args, **kwargs):
+        request.data['owner'] = request.user.pk
+        set_users(request)
+        return super().update(request, *args, **kwargs)
+
+    def retrieve(self, request, pk):
+        try:
+            group = Group.objects.filter(
+                Q(owner=request.user) |
+                Q(users__in=(request.user.pk, ))
+            ).distinct().get(Q(pk=pk))
+        except Group.DoesNotExist:
+            raise Http404
+
+        serializer = GroupSerializer(group, context={"request": self.request})
+        return Response(serializer.data)
-- 
cgit v1.2.3