Lims-back2/TestLaboratory/apps/user/views/userview.py

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

from apps.user.middleware.rolecontrol import RoleControl
from apps.user.models import User


class UserView(APIView):
    authentication_classes = []

    # register
    @staticmethod
    @RoleControl
    def put(request, user_id, *args, **kwargs):
        username = request.POST.get('username')
        tel = request.POST.get('tel')
        password = request.POST.get('password')

        if not username:
            return HttpResponse(status=500, content='用户名不能为空')

        if not password:
            return HttpResponse(status=500, content='密码不能为空')

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

        user.save()
        return HttpResponse(status=200, content='修改已保存')

    @staticmethod
    @RoleControl
    def get(request, user_id, *args, **kwargs):
        user = User.objects.filter(id=user_id)
        if not user:
            return HttpResponse(status=404, content='用户不存在')
        user = user[0]
        return Response({
            'username': user.username,
            'name': user.name,
            'identify': user.identify,
            'tel': user.tel,
        })