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") #展示出中文