mirror of
https://github.com/QingdaoU/OnlineJudge.git
synced 2025-11-04 14:49:58 +08:00
增加小组成员管理的api;优化权限判断逻辑和精简代码
This commit is contained in:
parent
a586aec451
commit
c772a4bcdc
@ -1,7 +1,8 @@
|
||||
# coding=utf-8
|
||||
from rest_framework import serializers
|
||||
|
||||
from .models import Group
|
||||
from account.serializers import UserSerializer
|
||||
from .models import Group, UserGroupRelation
|
||||
|
||||
|
||||
class CreateGroupSerializer(serializers.Serializer):
|
||||
@ -25,3 +26,16 @@ class GroupSerializer(serializers.ModelSerializer):
|
||||
class Meta:
|
||||
model = Group
|
||||
exclude = ["members"]
|
||||
|
||||
|
||||
class GroupMemberSerializer(serializers.ModelSerializer):
|
||||
user = UserSerializer()
|
||||
|
||||
class Meta:
|
||||
model = UserGroupRelation
|
||||
exclude = ["id", "group"]
|
||||
|
||||
|
||||
class EditGroupMemberSerializer(serializers.Serializer):
|
||||
group_id = serializers.IntegerField()
|
||||
members = serializers.ListField(child=serializers.IntegerField())
|
||||
@ -9,10 +9,34 @@ from account.decorators import login_required
|
||||
|
||||
from .models import Group, JoinGroupRequest, UserGroupRelation
|
||||
from .serializers import (CreateGroupSerializer, EditGroupSerializer,
|
||||
JoinGroupRequestSerializer, GroupSerializer)
|
||||
JoinGroupRequestSerializer, GroupSerializer,
|
||||
GroupMemberSerializer, EditGroupMemberSerializer)
|
||||
|
||||
|
||||
class GroupAdminAPIView(APIView):
|
||||
class GroupAPIViewBase(object):
|
||||
def get_group(request, group_id):
|
||||
"""
|
||||
根据group_id查询指定的小组的信息,结合判断用户权限
|
||||
管理员可以查询所有的小组,其他用户查询自己创建的自傲组
|
||||
"""
|
||||
if request.user.admin_type == SUPER_ADMIN:
|
||||
group = Group.object.get(id=group_id, visible=True)
|
||||
else:
|
||||
group = Group.object.get(id=group_id, visible=True, admin=request.user)
|
||||
return group
|
||||
|
||||
def get_groups(request):
|
||||
"""
|
||||
如果是超级管理员,就返回全部的小组
|
||||
如果是管理员,就返回他创建的全部小组
|
||||
"""
|
||||
if request.user.admin_type == SUPER_ADMIN:
|
||||
groups = Group.objects.filter(visible=True)
|
||||
else:
|
||||
groups = Group.objects.filter(admin=request.user, visible=True)
|
||||
|
||||
|
||||
class GroupAdminAPIView(APIView, GroupAPIViewBase):
|
||||
def post(self, request):
|
||||
"""
|
||||
创建小组的api
|
||||
@ -40,7 +64,7 @@ class GroupAdminAPIView(APIView):
|
||||
if serializer.is_valid():
|
||||
data = serializer.data
|
||||
try:
|
||||
group = Group.objects.get(id=data["id"], admin=request.user)
|
||||
group = self.get_group(request, data["group_id"])
|
||||
except Group.DoesNotExist:
|
||||
return error_response(u"小组不存在")
|
||||
group.name = data["name"]
|
||||
@ -60,21 +84,41 @@ class GroupAdminAPIView(APIView):
|
||||
group_id = request.GET.get("group_id", None)
|
||||
if group_id:
|
||||
try:
|
||||
if request.user.admin_type == SUPER_ADMIN:
|
||||
group = Group.object.get(id=group_id)
|
||||
else:
|
||||
group = Group.object.get(id=group_id, admin=request.user)
|
||||
group = self.get_group(request, group_id)
|
||||
return success_response(GroupSerializer(group).data)
|
||||
except Group.DoesNotExist:
|
||||
return error_response(u"小组不存在")
|
||||
else:
|
||||
if request.user.admin_type == SUPER_ADMIN:
|
||||
groups = Group.objects.filter(visible=True)
|
||||
else:
|
||||
groups = Group.objects.filter(admin=request.user, visible=True)
|
||||
groups = self.get_groups(request)
|
||||
return paginate(request, groups, GroupSerializer)
|
||||
|
||||
|
||||
class GroupMemberAdminAPIView(APIView, GroupAPIViewBase):
|
||||
def get(self, request):
|
||||
group_id = request.GET.get("group_id", None)
|
||||
if not group_id:
|
||||
return error_response(u"参数错误")
|
||||
try:
|
||||
group = self.get_group(request, group_id)
|
||||
except Group.DoesNotExist:
|
||||
return error_response(u"小组不存在")
|
||||
|
||||
return paginate(request, UserGroupRelation.objects.filter(group=group), GroupMemberSerializer)
|
||||
|
||||
def put(self, request):
|
||||
serializer = EditGroupMemberSerializer(data=request.data)
|
||||
if serializer.is_valid():
|
||||
try:
|
||||
group = self.get_group(request, group_id)
|
||||
except Group.DoesNotExist:
|
||||
return error_response(u"小组不存在")
|
||||
user_id_list = serializer.data["members"]
|
||||
UserGroupRelation.objects.delete(group=group, user__id__in=user_id_list)
|
||||
return success_response(u"删除成功")
|
||||
else:
|
||||
return serializer_invalid_response(serializer)
|
||||
|
||||
|
||||
def join_group(user, group):
|
||||
return UserGroupRelation.objects.create(user=user, group=group)
|
||||
|
||||
@ -103,5 +147,6 @@ class JoinGroupAPIView(APIView):
|
||||
keyword = request.GET.get("keyword", None)
|
||||
if not keyword:
|
||||
return error_response(u"参数错误")
|
||||
# 搜索包含这个关键词的 没有解散的 而且允许加入的小组
|
||||
groups = Group.objects.filter(name__contains=keyword, visible=True, join_group_setting__lte=2)
|
||||
return paginate(request, groups, GroupSerializer)
|
||||
|
||||
Loading…
Reference in New Issue
Block a user