def register():
"""注册"""
if request.method == 'POST':
phone = request.form.get('mobile')
pwd = request.form.get('password')
password2 = request.form.get('passwd2')
#验证完整性
if not all([phone, pwd, password2]):
return jsonify(status_code.USER_REGISTER_DATA_NOT_NULL)
#验证手机正确性
if not re.match(r'1[34578]\d{9}$', phone):
return jsonify(status_code.USER_REGISTER_MOBILE_ERROR)
#验证密码
if pwd != password2:
return jsonify(status_code.USER_REGISTER_PASSWORD_NOT_VALID)
#4.保存数据
user = User.query.filter(User.phone == phone).first()
if user:
return jsonify(status_code.USER_REGISTER_MPBILE_EXSITS)
else:
user = User()
user.phone = phone
user.password = pwd
user.name = phone
user.add_update()
return jsonify(status_code.SUCCESS)
def register():
"""注册"""
if request.method == 'GET':
return render_template('register.html')
if request.method == 'POST':
mobile = request.form.get('mobile')
# imagecode = request.form.get('imagecode')
password = request.form.get('password')
password2 = request.form.get('password2')
# 验证参数是否填完
if not all([mobile, password, password2]):
return jsonify(status_code.PARAMS_ERROR)
# 验证手机号码格式是否正确
if not re.fullmatch(r'^1[34578]\d{9}$', mobile):
return jsonify(status_code.USER_LOGIN_PHONE_ERROR)
# 验证手机号是否已经注册过
if User.query.filter_by(phone=mobile).first():
return jsonify(status_code.USER_REGISTER_USER_PHONE_EXSITS)
# 若全部验证成功
user = User()
user.name = mobile
user.phone = mobile
user.password = password
try:
user.add_update()
return jsonify(status_code.SUCCESS)
except:
return jsonify(status_code.USER_REGISTER_USER_ERROR)
def register_p():
username = request.form.get('username')
pwd1 = request.form.get('pwd1')
pwd2 = request.form.get('pwd2')
img_file = request.files.get('avatar')
company = request.form.get('company')
weibo = request.form.get('weibo')
job = request.form.get('job')
we_chat = request.form.get('we_chat')
e_mail = request.form.get('e_mail')
we_public = request.form.get('we_public')
info = request.form.get('info')
if not all([username, pwd1, pwd2]):
return '请填写必填字段'
if pwd1 != pwd2:
return '两次输入的密码不符合'
user = User.query.filter_by(name=username).first()
if user:
return '该用户名已经存在'
else:
user = User()
user.name = username
user.password = pwd1
image_path = os.path.join(UPLOAD_DIR, img_file.filename)
img_file.save(image_path)
user.avatar = os.path.join('upload', img_file.filename)
user.company = company
user.weibo = weibo
user.job = job
user.we_chat = we_chat
user.e_mail = e_mail
user.info = info
try:
user.add_update()
except Exception as e:
db.session.rollback()
return '创建用户失败'
return redirect(url_for('user.login_g'))
def sure_register():
# 取出提交页面中的内容，是个字典
register_dict = request.form
mobile = register_dict.get('mobile')
password = register_dict.get('password')
password2 = register_dict.get('password')
# 填写信息不全
if not all([mobile, password, password2]):
return jsonify(status_code.USER_REGISTER_PARAMS_ERROR)
# 不符合号码规则
if not re.match(r'^1[24578]\d{9}$', mobile):
return jsonify(status_code.USER_REGISTE_MOBILE_ERROR)
# 用户是否已注册
if User.query.filter(User.phone == mobile).count():
return jsonify(status_code.USER_REGISTE_MOBILE_IS_EXISTS)
# 两次输入的密码不相等
if password != password2:
return jsonify(status_code.USER_REGISTER_PASSWORD_IS_ERROR)
# 创建字段
user = User()
user.phone = mobile
user.name = mobile
user.password = password
try:
# 保存字段
user.add_update()
return jsonify(status_code.SUCCESS)
except Exception as e:
return jsonify(status_code.DATABASE_ERROR)
