summaryrefslogtreecommitdiffstats
path: root/chat_api/views.py
blob: f94f3a0745e187ae76c369c8425d7fd163a5c225 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from django.db.models.query import QuerySet
from django.http.response import Http404
from rest_framework.response import Response
from rest_framework.serializers import Serializer
from rest_framework.views import APIView
from rest_framework import permissions
from rest_framework import generics
from knox.auth import TokenAuthentication

from django.db.models.query import Q

from .models import *
from .serializers import *

class GroupView(generics.ListCreateAPIView):
    permission_classes = (permissions.IsAuthenticated, )
    authentication_classes = (TokenAuthentication, )
    serializer_class = GroupSerializer

    def get_queryset(self):
        return Group.objects.filter(
                    Q(owner=self.request.user) |
                    Q(users__in=(self.request.user, ))
                ).distinct()

    def create(self, request, *args, **kwargs):
        request.data['owner'] = request.user.pk
        return super().create(request, *args, **kwargs)