98 lines
3.3 KiB
Python
98 lines
3.3 KiB
Python
# api/views.py
|
|
from rest_framework import generics, viewsets, permissions,status
|
|
from django.contrib.auth import get_user_model
|
|
from rest_framework_simplejwt.views import TokenObtainPairView
|
|
from .serializers import RegisterSerializer, LinkSerializer, LinkGroupSerializer
|
|
from .models import Link, LinkGroup
|
|
from rest_framework.views import APIView
|
|
from rest_framework.response import Response
|
|
from django.shortcuts import get_object_or_404
|
|
from drf_spectacular.utils import extend_schema, OpenApiParameter
|
|
|
|
|
|
User = get_user_model()
|
|
|
|
@extend_schema(
|
|
parameters=[
|
|
OpenApiParameter(
|
|
name='id',
|
|
type=int,
|
|
location=OpenApiParameter.PATH
|
|
)
|
|
]
|
|
)
|
|
|
|
class RegisterView(generics.CreateAPIView):
|
|
queryset = User.objects.all()
|
|
permission_classes = (permissions.AllowAny,)
|
|
serializer_class = RegisterSerializer
|
|
|
|
class LoginView(TokenObtainPairView):
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
class LinkGroupViewSet(viewsets.ModelViewSet):
|
|
queryset = LinkGroup.objects.all()
|
|
serializer_class = LinkGroupSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_queryset(self):
|
|
# возвращаем только группы текущего пользователя
|
|
return self.queryset.filter(owner=self.request.user).order_by('order')
|
|
|
|
class LinkViewSet(viewsets.ModelViewSet):
|
|
serializer_class = LinkSerializer
|
|
permission_classes = (permissions.IsAuthenticated,)
|
|
|
|
def get_queryset(self):
|
|
return Link.objects.filter(owner=self.request.user).order_by('order')
|
|
|
|
def perform_create(self, serializer):
|
|
serializer.save(owner=self.request.user)
|
|
|
|
class UserLinksListView(generics.ListAPIView):
|
|
serializer_class = LinkSerializer
|
|
permission_classes = (permissions.AllowAny,)
|
|
|
|
def get_queryset(self):
|
|
username = self.kwargs['username']
|
|
return Link.objects.filter(owner__username=username).order_by('order')
|
|
|
|
|
|
from .serializers import UserSerializer # нужно завести сериализатор для пользователя
|
|
|
|
User = get_user_model()
|
|
|
|
class UserProfileView(generics.RetrieveAPIView):
|
|
"""
|
|
Возвращает данные авторизованного пользователя.
|
|
GET /api/auth/user/
|
|
"""
|
|
serializer_class = UserSerializer
|
|
permission_classes = [permissions.IsAuthenticated]
|
|
|
|
def get_object(self):
|
|
return self.request.user
|
|
|
|
from users.models import User
|
|
from links.models import LinkGroup
|
|
from .serializers import PublicUserGroupsSerializer
|
|
|
|
@extend_schema(
|
|
responses=PublicUserGroupsSerializer
|
|
)
|
|
class PublicUserGroupsView(generics.GenericAPIView):
|
|
"""
|
|
Возвращает публичные группы и их ссылки для заданного username.
|
|
GET /api/users/{username}/public/
|
|
"""
|
|
serializer_class = PublicUserGroupsSerializer
|
|
permission_classes = [permissions.AllowAny]
|
|
|
|
def get(self, request, username):
|
|
user = get_object_or_404(User, username=username)
|
|
groups = LinkGroup.objects.filter(owner=user, is_public=True).prefetch_related('links')
|
|
# Формируем ответ в виде словаря — GenericAPIView сам вызовет нужный сериализатор:
|
|
return Response({
|
|
'username': user.username,
|
|
'groups': groups
|
|
}) |