登录注册的接口模版(最终版本)

from django.http import JsonResponse
from rest_framework.views import APIView
from demo import models


class Login(APIView):

    def post(self, request):

        ret = {}

        try:
            # 接受参数
            user = request._request.POST.get('username')
            pwd = request._request.POST.get('password')

            #  拿User对象,拿到了表示登录成功
            obj = models.UserInfo.objects.filter(username=user, password=pwd).first()

            if not obj:
                ret['code'] = 10005
                ret['msg'] = '用户名或密码错误'
                return JsonResponse(ret)
            else:
                ret['code'] = 200
                ret['msg'] = '登录成功'
                return JsonResponse(ret)

        except Exception as e:
            ret['code'] = 10004
            ret['msg'] = '请求异常'
            return JsonResponse(ret)


class Register(APIView):

    def post(self, request):

        ret = {}

        try:
            # 获取数据
            user = request._request.POST.get('username')
            pwd = request._request.POST.get('password')

            try:
                models.UserInfo.objects.create(username=user, password=pwd)
            except Exception as e:
                ret = {'code': 10006, 'msg': "用户名已存在"}
                return JsonResponse(ret)

            ret['code'] = 200
            ret['msg'] = '注册成功'
            return JsonResponse(ret)

        except Exception as e:
            ret['code'] = 10004
            ret['msg'] = '请求异常'
            return JsonResponse(ret)
            # return HttpResponse(json.dumps(ret, ensure_ascii=False), content_type="application/json,charset=utf-8")  #展示出中文

 

 

posted @ 2022-05-28 17:58  费皿啊  阅读(116)  评论(0)    收藏  举报