new models, frontend functions, public pages
This commit is contained in:
@@ -1,32 +1,52 @@
|
||||
# api/views.py
|
||||
from rest_framework import generics, viewsets, permissions
|
||||
from django.contrib.auth import get_user_model
|
||||
# coding: utf-8
|
||||
from rest_framework import generics, viewsets, permissions, status
|
||||
from rest_framework.parsers import MultiPartParser, FormParser, JSONParser
|
||||
from rest_framework_simplejwt.views import TokenObtainPairView
|
||||
from .serializers import RegisterSerializer, LinkSerializer, LinkGroupSerializer
|
||||
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
|
||||
from django.contrib.auth import get_user_model
|
||||
|
||||
from .models import Link, LinkGroup
|
||||
from .serializers import (
|
||||
RegisterSerializer,
|
||||
UserSerializer,
|
||||
LinkSerializer,
|
||||
LinkGroupSerializer,
|
||||
)
|
||||
|
||||
User = get_user_model()
|
||||
|
||||
|
||||
class RegisterView(generics.CreateAPIView):
|
||||
queryset = User.objects.all()
|
||||
permission_classes = (permissions.AllowAny,)
|
||||
permission_classes = [permissions.AllowAny]
|
||||
serializer_class = RegisterSerializer
|
||||
|
||||
|
||||
class LoginView(TokenObtainPairView):
|
||||
permission_classes = (permissions.AllowAny,)
|
||||
permission_classes = [permissions.AllowAny]
|
||||
|
||||
|
||||
class LinkGroupViewSet(viewsets.ModelViewSet):
|
||||
queryset = LinkGroup.objects.all()
|
||||
serializer_class = LinkGroupSerializer
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
parser_classes = [MultiPartParser, FormParser, JSONParser]
|
||||
|
||||
def get_queryset(self):
|
||||
# возвращаем только группы текущего пользователя
|
||||
return self.queryset.filter(owner=self.request.user).order_by('order')
|
||||
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(owner=self.request.user)
|
||||
|
||||
|
||||
class LinkViewSet(viewsets.ModelViewSet):
|
||||
queryset = Link.objects.all()
|
||||
serializer_class = LinkSerializer
|
||||
permission_classes = (permissions.IsAuthenticated,)
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
parser_classes = [MultiPartParser, FormParser, JSONParser]
|
||||
|
||||
def get_queryset(self):
|
||||
return Link.objects.filter(owner=self.request.user).order_by('order')
|
||||
@@ -34,26 +54,66 @@ class LinkViewSet(viewsets.ModelViewSet):
|
||||
def perform_create(self, serializer):
|
||||
serializer.save(owner=self.request.user)
|
||||
|
||||
|
||||
class UserProfileView(generics.RetrieveAPIView):
|
||||
serializer_class = UserSerializer
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
def get_object(self):
|
||||
return self.request.user
|
||||
|
||||
|
||||
class UserLinksListView(generics.ListAPIView):
|
||||
serializer_class = LinkSerializer
|
||||
permission_classes = (permissions.AllowAny,)
|
||||
|
||||
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):
|
||||
class PublicUserGroupsView(APIView):
|
||||
"""
|
||||
Возвращает данные авторизованного пользователя.
|
||||
GET /api/auth/user/
|
||||
GET /api/users/{username}/public/
|
||||
"""
|
||||
serializer_class = UserSerializer
|
||||
permission_classes = [permissions.IsAuthenticated]
|
||||
permission_classes = [permissions.AllowAny]
|
||||
|
||||
def get_object(self):
|
||||
return self.request.user
|
||||
def get(self, request, username):
|
||||
# 1. Ищем пользователя
|
||||
user = get_object_or_404(User, username=username)
|
||||
|
||||
# 2. Берём его группы со ссылками
|
||||
groups_qs = LinkGroup.objects.filter(owner=user).prefetch_related('links')
|
||||
|
||||
result = {
|
||||
"username": user.username,
|
||||
"groups": []
|
||||
}
|
||||
|
||||
for grp in groups_qs:
|
||||
# icon у группы (абсолютный URL)
|
||||
grp_icon_url = None
|
||||
if grp.icon:
|
||||
grp_icon_url = request.build_absolute_uri(grp.icon.url)
|
||||
|
||||
grp_data = {
|
||||
"id": grp.id,
|
||||
"name": grp.name,
|
||||
"icon": grp_icon_url,
|
||||
"links": [],
|
||||
}
|
||||
|
||||
for ln in grp.links.all():
|
||||
# icon у ссылки
|
||||
ln_icon_url = None
|
||||
if ln.icon:
|
||||
ln_icon_url = request.build_absolute_uri(ln.icon.url)
|
||||
|
||||
grp_data["links"].append({
|
||||
"id": ln.id,
|
||||
"title": ln.title,
|
||||
"url": ln.url,
|
||||
"icon": ln_icon_url,
|
||||
})
|
||||
|
||||
result["groups"].append(grp_data)
|
||||
|
||||
return Response(result, status=status.HTTP_200_OK)
|
||||
Reference in New Issue
Block a user