|
|
|
|
from django.contrib.auth.models import User
|
|
|
|
|
from rest_framework import viewsets
|
|
|
|
|
from rest_framework.permissions import AllowAny, IsAuthenticatedOrReadOnly
|
|
|
|
|
|
|
|
|
|
from user_info.serializers import UserRegisterSerializer
|
|
|
|
|
from user_info.permissions import IsSelfOrReadOnly
|
|
|
|
|
|
|
|
|
|
from rest_framework.decorators import action
|
|
|
|
|
from rest_framework.response import Response
|
|
|
|
|
from user_info.serializers import UserDetailSerializer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class UserViewSet(viewsets.ModelViewSet):
|
|
|
|
|
queryset = User.objects.all()
|
|
|
|
|
serializer_class = UserRegisterSerializer
|
|
|
|
|
lookup_field = 'username' # 和UserRegisterSerializer中的url中lookup_field对应
|
|
|
|
|
|
|
|
|
|
# http://127.0.0.1:8000/api/user/admin/info/
|
|
|
|
|
@action(detail=True, methods=['get'])
|
|
|
|
|
def info(self, request, username=None):
|
|
|
|
|
queryset = User.objects.get(username=username)
|
|
|
|
|
serializer = UserDetailSerializer(queryset, many=False)
|
|
|
|
|
return Response(serializer.data)
|
|
|
|
|
|
|
|
|
|
# http://127.0.0.1:8000/api/user/sorted/
|
|
|
|
|
@action(detail=False)
|
|
|
|
|
def sorted(self, request):
|
|
|
|
|
users = User.objects.all().order_by('-username')
|
|
|
|
|
# 是否分页
|
|
|
|
|
page = self.paginate_queryset(users)
|
|
|
|
|
if page is not None:
|
|
|
|
|
serializer = self.get_serializer(page, many=True)
|
|
|
|
|
return self.get_paginated_response(serializer.data)
|
|
|
|
|
serializer = self.get_serializer(users, many=True)
|
|
|
|
|
return Response(serializer.data)
|
|
|
|
|
|
|
|
|
|
def get_permissions(self):
|
|
|
|
|
# 注册用户的POST请求是允许所有人都可以操作的
|
|
|
|
|
if self.request.method == 'POST':
|
|
|
|
|
self.permission_classes = [AllowAny]
|
|
|
|
|
else:
|
|
|
|
|
self.permission_classes = [IsAuthenticatedOrReadOnly, IsSelfOrReadOnly]
|
|
|
|
|
|
|
|
|
|
return super(UserViewSet, self).get_permissions()
|
