一：新建modles名为loginModels.py

  1. # -*- coding: utf-8 -*-
  2. from django.db import models
  4. '''===================================
  5. @Project：wisdomShop
  6. @Author：班婕妤
  7. @Date：10/3/2020 下午2:14
  8. @Company：深圳市智慧养老宝科技有限公司
  9. @Motto：心有猛虎，细嗅蔷薇
  10. @Python_Version：3.7.3
  11. @Django_Version：2.1.5
  12. ======================================='''
  13. class Userinfo(models.Model):
  14.     USER_TYPE = (
  15.         (1, '普通用户'),
  16.         (2, 'VIP'),
  17.         (3, 'SVIP')
  18.     )
  19.     user_type = models.IntegerField(choices=USER_TYPE, blank=True, null=True)
  20.     userName = models.CharField(max_length=10)
  21.     userPwd = models.CharField(max_length=100)
  22.     userTelphone = models.CharField(max_length=10)
  23.     userAddress = models.CharField(max_length=10)
  24.     userAge = models.CharField(max_length=4)
  26. class UserToken(models.Model):
  27.     user = models.OneToOneField(Userinfo, on_delete=models.CASCADE)
  28.     token = models.CharField(max_length=64)

二：执行迁移脚本命令migrate

  1. python manage.py makemigrations zhylbwg
  2. python manage.py migrate zhylbwg

三：新建views名为login.py

  1. # -*- coding: utf-8 -*-
  2. '''===================================
  3. @Project：wisdomShop
  4. @Author：班婕妤
  5. @Date：5/3/2020 下午1:50
  6. @Company：深圳市智慧养老宝科技有限公司
  7. @Motto：心有猛虎，细嗅蔷薇
  8. @Python_Version：3.7.3
  9. @Django_Version：2.1.5
  10. ======================================='''
  11. from django.shortcuts import render,HttpResponse
  12. from django.http import JsonResponse
  13. import pandas as pd
  14. import json
  15. from zhylbwg.models import loginModels
  16. from zhylbwg.views import md5  # 导入自定义md5加密函数
  17. from zhylbwg.views import requestResult  # 导入自定义的统一返回函数
  19. from rest_framework.views import APIView
  20. from zhylbwg.util.MySchemaGenerator import DocParam
  23. # Create your views here.
  25. def login(request):
  26.     if request.method == 'POST':
  27.         # 获取请求头数据，请求以json的格式传输
  28.         loginInfor = request.body
  29.         loginInformation = json.loads(loginInfor)
  30.         # 获取用户名
  31.         userName = loginInformation.get('userName')
  32.         # 判断该用户名是否存在
  33.         userNameDB = loginModels.Userinfo.objects.filter(userName=userName)
  34.         print(userNameDB.values_list('userName')[0])
  35.         if len(userNameDB) == 0:
  36.             return HttpResponse(json.dumps(requestResult.result_json('202', '该用户名不存在，请注册', '')),
  37.                                 content_type="application/json,charset=utf-8")
  38.         else:
  39.             '''
  40.                 判断密码是否正确
  41.             '''
  42.             # 得到前端传过来的密码
  43.             userPwd = loginInformation.get('userPwd')
  44.             # 使用mdb进行加密
  45.             userPwdMd5 = md5.Md5(userPwd)
  46.             # 根据同用户名从数据库中获取密码
  47.             checkLogin = loginModels.Userinfo.objects.get(userName=userName)
  48.             # 比较2个密码是否一致
  49.             if userPwdMd5 == checkLogin.userPwd:
  50.                 # 若一致，则登录成功
  51.                 return HttpResponse(json.dumps(requestResult.result_json('203', '登录成功', '')),
  52.                                     content_type="application/json,charset=utf-8")
  53.             else:
  54.                 return HttpResponse(json.dumps(requestResult.result_json('504', '密码错误', '')),
  55.                                     content_type="application/json,charset=utf-8")
  56. def register(request):
  57.     # 判断是否为post请求
  58.     if request.method == "POST":
  59.         # 获取请求头数据，请求以json的格式传输
  60.         registerinformation = request.body
  61.         # 将请求头数据转化为json格式
  62.         registerinformationData = json.loads(registerinformation)
  63.         print(registerinformationData)
  64.         # 获取用户名
  65.         userName = registerinformationData.get('userName')
  66.         # 从数据库中查找是否存在该用户名
  67.         userNameDB = loginModels.Userinfo.objects.filter(userName=userName)
  68.         # 判断用户名是否存在，若存在，则提示已有该用户，若不存在，则进行密码加密后存储到数据库中
  69.         if userNameDB:
  70.             return HttpResponse(json.dumps(requestResult.result_json('312', '该用户名已经存在', '')),
  71.                                 content_type="application/json,charset=utf-8")
  72.         else:
  73.             # 获取用户密码
  74.             #
  75.             userPwd = registerinformationData.get('userPwd')
  76.             # 密码加密操作md5，md5加密功能具体看md5加密代码
  77.             userPwdMd5 = md5.Md5(userPwd)
  78.             # 将加密后的密码赋值给请求头中的密码参数
  79.             registerinformationData["userPwd"] = userPwdMd5
  80.             # 将json格式数据，类型为dict 存储到数据库中，表明为Userinfo，将注册请求存储到数据库中
  81.             loginModels.Userinfo.objects.create(**registerinformationData)
  82.             return HttpResponse(json.dumps(requestResult.result_json('201', '注册成功，请登录', '')),
  83.                                 content_type="application/json,charset=utf-8")
  84.     else:
  85.         return HttpResponse(json.dumps(requestResult.result_json('501', '不是post请求', '')),
  86.                             content_type="application/json,charset=utf-8")

四：url映射

  1. path('zhylbwg/register/', login.register),
  2. path('zhylbwg/login/', login.login),

五：测试

图片.png

图片.png