80 lines
3.1 KiB
Python
80 lines
3.1 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from django.contrib.auth.models import User
|
||
from django.conf import settings
|
||
from django.db import transaction
|
||
from django.views.decorators.csrf import csrf_exempt
|
||
|
||
from account.models import Profile
|
||
from business.models import BusinessAccountInfo
|
||
|
||
from utils.response import json_response
|
||
from utils.errors import BizError, UserError
|
||
|
||
|
||
def check_biz(request):
|
||
email = request.GET.get('email', '') # 获取邮箱
|
||
try: # 检查数据库中是否由该邮箱注册过的数据
|
||
biz = BusinessAccountInfo.objects.get(email=email)
|
||
except BusinessAccountInfo.DoesNotExist:
|
||
biz = None
|
||
return json_response(200, 'OK', { # 返回是否已经被注册过和是否已经有此用户
|
||
'userexists': User.objects.filter(email=email).exists(),
|
||
'bizaccountexists': bool(biz)
|
||
})
|
||
|
||
|
||
@csrf_exempt
|
||
@transaction.atomic
|
||
def registry_biz(request):
|
||
email = request.POST.get('email', '') # 获取填写的邮箱
|
||
name = request.POST.get('name', '') # 获取填写的机构名
|
||
username = request.POST.get('username', '') # 获取填写的机构联系人
|
||
phone = request.POST.get('phone', '') # 获取填写的手机号
|
||
ctype = request.POST.get('type', BusinessAccountInfo.INTERNET) # 获取机构类型
|
||
flag = int(request.POST.get('flag', 2)) # 获取一个标记位，代表用户是创建新用户还是使用绑定老用户的方式
|
||
uname = email.split('@')[0] # 和之前的注册逻辑没什么区别，创建一个账户名
|
||
if not User.objects.filter(username__exact=name).exists():
|
||
final_name = username
|
||
elif not User.objects.filter(username__exact=uname).exists():
|
||
final_name = uname
|
||
else:
|
||
final_name = email
|
||
if flag == 2: # 如果标记位是2，那么将为他创建新用户
|
||
user = User.objects.create_user(
|
||
username=final_name,
|
||
email=email,
|
||
password=settings.INIT_PASSWORD,
|
||
is_active=False,
|
||
is_staff=False
|
||
)
|
||
if flag == 1: # 如果标记位是1，那么为他绑定老用户
|
||
try:
|
||
user = User.objects.get(email=email)
|
||
except User.DoesNotExist:
|
||
return json_response(*UserError.UserNotFound)
|
||
pvalues = {
|
||
'phone': phone,
|
||
'name': final_name,
|
||
'user_src': Profile.COMPANY_USER,
|
||
}
|
||
profile, _ = Profile.objects.select_for_update().get_or_create(email=email) # 获取或创建用户信息
|
||
for k, v in pvalues.items():
|
||
setattr(profile, k, v)
|
||
profile.save()
|
||
bizvalues = {
|
||
'company_name': name,
|
||
'company_username': username,
|
||
'company_phone': phone,
|
||
'company_type': ctype,
|
||
}
|
||
biz, _ = BusinessAccountInfo.objects.select_for_update().get_or_create( # 获取或创建机构账户信息
|
||
email=email,
|
||
defaults=bizvalues
|
||
)
|
||
return json_response(200, 'OK', { # 响应JSON格式数据，这个标记位在发送验证邮件的时候还有用
|
||
'name': final_name,
|
||
'email': email,
|
||
'flag': flag
|
||
})
|