88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
import binascii
|
|
import json, datetime, base64
|
|
from django.shortcuts import HttpResponse
|
|
from django.middleware.csrf import get_token
|
|
from django.views.decorators.http import require_GET, require_POST
|
|
from apps.auth import models as auth_models
|
|
from django.contrib.auth.hashers import make_password, check_password
|
|
|
|
|
|
# Create your views here.
|
|
@require_GET
|
|
def gettoken(request):
|
|
"""
|
|
获取token
|
|
:param request:
|
|
:return:
|
|
"""
|
|
token = get_token(request)
|
|
return HttpResponse(json.dumps({'token': token}), content_type="application/json,charset=utf-8")
|
|
|
|
|
|
@require_POST
|
|
def search_user(request):
|
|
"""
|
|
查询用户名
|
|
:param request:
|
|
:return:
|
|
"""
|
|
username = request.POST.get('username')
|
|
user = auth_models.User.objects.filter(username=username)
|
|
if user.exists():
|
|
return HttpResponse("用户名已存在")
|
|
else:
|
|
return HttpResponse(True)
|
|
|
|
|
|
@require_POST
|
|
def add_user(request):
|
|
"""
|
|
用户注册
|
|
:param request: POST提交注册信息
|
|
:return: 注册结果
|
|
"""
|
|
try:
|
|
username = request.POST.get("username")
|
|
pwd_base64 = base64.b64encode(request.POST.get("pwd"))
|
|
pwd = make_password(pwd_base64)
|
|
email = request.POST.get("email")
|
|
phone = request.POST.get("phone")
|
|
create_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
last_login_time = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
print(username, pwd, email, phone)
|
|
avatar = request.FILES.get("avatar")
|
|
auth_models.User.objects.create(
|
|
username=username,
|
|
pwd=pwd,
|
|
email=email,
|
|
phone=phone,
|
|
create_time=create_time,
|
|
last_login_time=last_login_time,
|
|
avatar=avatar
|
|
)
|
|
return HttpResponse("添加用户成功")
|
|
except Exception as e:
|
|
return HttpResponse(f"报错了：{e}")
|
|
|
|
|
|
@require_POST
|
|
def login_user(request):
|
|
"""
|
|
用户登录，验证用户密码是否正确，正确返回菜单，错误返回用户名或密码不正确
|
|
:param request:
|
|
:return:
|
|
"""
|
|
try:
|
|
username = request.POST.get("username")
|
|
pwd_input = base64.b64decode(request.POST.get("pwd"))
|
|
pwd_made = auth_models.User.objects.filter(username=username).first().pwd
|
|
pwd = check_password(pwd_input, pwd_made)
|
|
if pwd:
|
|
return HttpResponse(True)
|
|
else:
|
|
return HttpResponse("用户名或密码不正确")
|
|
except binascii.Error as e:
|
|
return HttpResponse("base64解码失败")
|
|
except Exception as e:
|
|
return HttpResponse(f"报错了:{e}")
|