146 lines
5.2 KiB
Python
146 lines
5.2 KiB
Python
from django.shortcuts import render
|
||
from django.http import HttpResponse, JsonResponse
|
||
from django.contrib.auth import get_user_model, authenticate, login
|
||
from django.views.decorators.csrf import csrf_exempt
|
||
from django_token.models import Token
|
||
import datetime
|
||
|
||
from .models import VerifyCode
|
||
#这个地方sent_sms_code原先是send_sms_code，报错
|
||
from .utils import generate_code, sent_sms_code
|
||
|
||
|
||
def index(request):
|
||
username = request.GET.get('username')
|
||
password = request.GET.get('password')
|
||
print(username, password)
|
||
return JsonResponse({'status': 'error', 'message': '用户名或密码错误'})
|
||
|
||
|
||
def status_500(request):
|
||
return HttpResponse(status=500)
|
||
|
||
|
||
def status_401(request):
|
||
return HttpResponse(status=401)
|
||
|
||
|
||
@csrf_exempt
|
||
def polls_login(request):
|
||
if request.method == 'POST':
|
||
phone = request.POST.get('phone')
|
||
password = request.POST.get('password')
|
||
UserModel = get_user_model()
|
||
user = UserModel.objects.filter(username=phone).first()
|
||
if not user:
|
||
return JsonResponse({'status': 'error', 'message': '用户名或密码错误'})
|
||
|
||
u = authenticate(request, username=phone, password=password)
|
||
if u is not None:
|
||
login(request, u)
|
||
token = Token.objects.get_or_create(user=u)
|
||
result = dict()
|
||
profile = u.userprofile_set.first()
|
||
result['phone'] = u.first_name
|
||
result['token'] = token
|
||
if profile:
|
||
result['name'] = profile.name
|
||
result['gender'] = profile.sex
|
||
result['thumbnail'] = profile.image.path
|
||
result['organization'] = profile.organization.name
|
||
return JsonResponse({'status': 'success', 'message': result})
|
||
else:
|
||
return JsonResponse({'status': 'error', 'message': '用户名或密码错误'})
|
||
else:
|
||
return HttpResponse(status=405)
|
||
|
||
|
||
@csrf_exempt
|
||
def send_code(request):
|
||
if request.method == 'GET':
|
||
return HttpResponse(status=405)
|
||
phone = request.POST.get('phone')
|
||
category = request.POST.get('category', 0)
|
||
if not phone:
|
||
return JsonResponse({'status': 'error', 'message': '手机号不正确'})
|
||
|
||
exist_code = VerifyCode.objects.filter(
|
||
phone=phone, category=category).first()
|
||
if exist_code and exist_code.in_progress():
|
||
return JsonResponse({'status': 'error', 'message': '验证码使用中'})
|
||
|
||
code = generate_code()
|
||
now = datetime.datetime.now()
|
||
after_10mins = now + datetime.timedelta(minutes=10)
|
||
response = sent_sms_code(phone, code)
|
||
result = response.decode('utf8')
|
||
if "OK" in result:
|
||
VerifyCode.objects.create(code=code, phone=phone, category=category)
|
||
return JsonResponse({'status': 'success'})
|
||
return JsonResponse({'status': 'error', 'message': '验证码发送失败'})
|
||
|
||
|
||
@csrf_exempt
|
||
def register_step_one(request):
|
||
if request.method == 'GET':
|
||
return HttpResponse(status=405)
|
||
phone = request.POST.get('phone')
|
||
code = request.POST.get('code')
|
||
|
||
UserModel = get_user_model()
|
||
is_exists = UserModel.objects.filter(username=phone).exists()
|
||
if is_exists:
|
||
return JsonResponse({'status': 'error', 'message': '该手机号已经使用'})
|
||
|
||
verify_code = VerifyCode.objects.filter(
|
||
phone=phone, code=code, category=0).first()
|
||
if verify_code and verify_code.in_progress():
|
||
return JsonResponse({'status': 'success', 'message': {phone: phone}})
|
||
else:
|
||
return JsonResponse({'status': 'error', 'message': '验证码超时，请重发'})
|
||
|
||
|
||
@csrf_exempt
|
||
def register_step_two(request):
|
||
if request.method == 'GET':
|
||
return HttpResponse(status=405)
|
||
phone = request.POST.get('phone')
|
||
name = request.POST.get('name')
|
||
gender = request.POST.get('gender')
|
||
password = request.POST.get('password')
|
||
organization = request.POST.get('organization')
|
||
UserModel = get_user_model()
|
||
UserModel.objects.create_user(username=phone, password=password)
|
||
return JsonResponse({'status': 'success', 'message': '注册成功'})
|
||
|
||
|
||
@csrf_exempt
|
||
def password_recover_step_one(request):
|
||
if request.method == 'GET':
|
||
return HttpResponse(status=405)
|
||
phone = request.POST.get('phone')
|
||
code = request.POST.get('code')
|
||
verify_code = VerifyCode.objects.filter(
|
||
phone=phone, code=code, category=1).first()
|
||
if verify_code and verify_code.in_progress():
|
||
return JsonResponse({'status': 'success', 'message': {phone: phone}})
|
||
else:
|
||
return JsonResponse({'status': 'error', 'message': '验证码超时，请重发'})
|
||
|
||
|
||
@csrf_exempt
|
||
def password_recover_step_two(request):
|
||
if request.method == 'GET':
|
||
return HttpResponse(status=405)
|
||
phone = request.POST.get('phone')
|
||
password = request.POST.get('password')
|
||
password_confirm = request.POST.get('password_confirm')
|
||
if password != password_confirm:
|
||
return JsonResponse({'status': 'error', 'message': '两次密码输入不一致'})
|
||
UserModel = get_user_model()
|
||
user = UserModel.objects.filter(username=phone).first()
|
||
if not user:
|
||
return JsonResponse({'status': 'error', 'message': '用户名不存在'})
|
||
user.set_password(password)
|
||
return JsonResponse({'status': 'success', 'message': {phone: phone}})
|