LimsBackend/apps/user/views/userjoinview.py

import logging

from django.http import HttpResponse
from rest_framework.views import APIView

from apps.group.models import Group
from apps.log.models import get_log, gen_log
from apps.user.middleware.rolecontrol import RoleControl
from apps.user.models import User

logger = logging.getLogger('django')


class UserJoinView(APIView):
    authentication_classes = []

    @staticmethod
    @RoleControl
    def post(request, user_id, *args, **kwargs):
        code = request.POST.get('code')
        group = Group.objects.filter(code=code, delete=False)
        if not group:
            logger.error("分组不存在")
            return HttpResponse(status=500, content='分组不存在')
        group = group[0]

        user = User.objects.filter(id=user_id)
        if not user:
            logger.error("用户不存在")
            return HttpResponse(status=404, content='用户不存在')
        user = user[0]

        group.users.add(user)
        executor, action, method = get_log(request)
        gen_log(action, "加入分组", group.name, method, executor)
        return HttpResponse(status=200, content='加入成功')