def add_test_users():
"""
添加测试用户
:return:
"""
import datetime
users = []
now = datetime.datetime.now()
for num in range(0, 10001):
try:
from info.models import User
from info import mysql_db
from manage import app
user = User()
user.nick_name = "%011d" % num
user.mobile = "%011d" % num
user.password_hash = "pbkdf2:sha256:50000$SgZPAbEj$a253b9220b7a916e03bf27119d401c48ff4a1c81d7e00644e0aaf6f3a8c55829"
user.create_time = now - datetime.timedelta(
seconds=random.randint(0, 2678400))
user.last_login = now - datetime.timedelta(
seconds=random.randint(0, 2678400))
users.append(user)
print(user.mobile)
except Exception as e:
print(e)
with app.app_context():
for i in range(0, 101):
start_index = i * 100
if i != 0:
start_index += 1
# mysql_db.session.add_all(users)
session_user = users[start_index:(i + 1) * 100 + 1]
mysql_db.session.add_all(session_user)
mysql_db.session.commit()
print('OK')
def register():
"""
点击注册实现验证并注册
:return:
"""
# 获取参数和判断是否有值
resp = request.json
mobile = resp.get('mobile')
sms_code = resp.get('sms_code')
password = resp.get('password')
if not all([mobile, sms_code, password]):
return jsonify(errno=RET.PARAMERR, errmsg='参数不全')
# 从Redis中获取指定手机号对应的短信验证码的
try:
real_sms_code = redis_store.get('SMSCode_' + mobile)
except Exception as e:
current_app.logger.error(e)
return jsonify(errno=RET.DATAERR, errmsg='查询失败')
# 判断验证码是否过期
if not real_sms_code:
return jsonify(errno=RET.NODATA, errmsg='验证码已失效')
# 校验验证码
if sms_code != real_sms_code:
return jsonify(errno=RET.DATAERR, errmsg='验证码错误')
# 初始化user模型，并设置数据并添加到数据库
user = User()
user.mobile = mobile
user.nick_name = mobile
user.create_time = datetime.now()
user.last_login = datetime.now()
# 设置password属性会自动设置password_hash
user.password = password
# 保存当前用户的状态
db.session.add(user)
db_commit(db)
# 保存用户登录状态
session['id'] = user.id
session['password'] = user.password_hash
session['is_admin'] = user.is_admin
# 返回注册的结果
return jsonify(errno=RET.OK, errmsg='注册成功')
def add_test_users():
from info import db
from info.models import User
users = []
now = datetime.datetime.now()
for num in range(0, 10000):
try:
user = User()
user.nick_name = "%011d" % num
user.mobile = "%011d" % num
user.password_hash = "pbkdf2:sha256:50000$SgZPAbEj$a253b9220b7a916e03bf27119d401c48ff4a1c81d7e00644e0aaf6f3a8c55829"
user.create_time = now - datetime.timedelta(
seconds=random.randint(0, 2678400))
users.append(user)
print(user.mobile)
except Exception as e:
print(e)
db.session.add_all(users)
db.session.commit()
print('OK')
def add_test_users():
"""批量生成测试用户"""
users = []
now = datetime.datetime.now()
for num in range(10010, 80000):
try:
user = User()
user.nick_name = "%011d" % num
user.mobile = "%011d" % num
user.password_hash = "pbkdf2:sha256:50000$SgZPAbEj$a253b9220b7a916e03bf27119d401c48ff4a1c81d7e00644e0aaf6f3a8c55829"
user.create_time = now - datetime.timedelta(
seconds=random.randint(0, 2678400))
user.last_login = now - datetime.timedelta(
seconds=random.randint(0, 2678400))
users.append(user)
print(user.mobile)
except Exception as e:
print(e)
with app.app_context():
# add_all 批量添加用户，参数就是一个列表
db.session.add_all(users)
db.session.commit()
print('OK')
def register():
"""注册接口"""
# 1. 获取数据
param_dict = request.json
mobile = param_dict.get("mobile")
smscode = param_dict.get("smscode")
password = param_dict.get("password")
# 2. 数据判断
# 2.1 判断值是否都已经输入了
if not all([mobile, smscode, password]):
return jsonify(erron=RET.PARAMERR, errmsg="参数不足")
# 2.2 验证手机号码格式
if not re.match("^1[356789][0-9]{9}$", mobile):
return jsonify(erron=RET.PARAMERR, errmsg="手机格式有误")
# 3.1 根据手机号编码获取数据库中的真实验证码
try:
real_sms_code = redis_store.get("sms_%s" % mobile)
if real_sms_code:
redis_store.delete("sms_%s" % mobile)
except Exception as e:
current_app.logger.error(e)
return jsonify(errno=RET.DBERR, errmsg="获取短信验证码数据库异常")
if not real_sms_code:
# 没有值表示短信验证码过期了
return jsonify(errno=RET.NODATA, errmsg="短信验证码过期")
# 3.2 对比用户添加的短信验证码和真实的短信验证码对比
if smscode != real_sms_code:
return jsonify(errno=RET.PARAMERR, errmsg="短信验证码填写错误")
# 3.3 一致表示用户输入的手机号和验证码都是正确的
# 创建用户对象 给对应属性赋值
user = User()
user.mobile = mobile
user.nick_name = mobile
# 需要将加密的密码加密
user.password = password
# 记录一下最后一次保存的登录时间
user.create_time = datetime.now()
# 将信息保存到数据库中
try:
db.session.add(user)
db.session.commit()
except Exception as e:
current_app.logger.error(e)
# 回滚
db.session.rollback()
return jsonify(errno=RET.DBERR, errmsg="保存用户对象到数据库异常")
# 3.4 使用session保存用户信息，以便自动登录
session["user_id"] = user.id
session["mobile"] = user.mobile
session["nick_name"] = user.nick_name
# 返回响应对象
return jsonify(errno=RET.OK, errmsg="注册成功！")
def register():
"""
注册的逻辑
1. 获取参数
2. 校验参数
3. 取到服务器保存的真实的短信验证码内容
4. 校验用户输入的短信验证码内容和真实验证码内容是否一致
5. 如果一致，初始化 User 模型，并且赋值属性
6. 将 user 模型添加数据库
7. 返回响应
:return:
"""
# 1. 获取参数
param_dict = request.json
nickname = param_dict.get("nickname")
mobile = param_dict.get("mobile")
password = param_dict.get("password")
image_code = param_dict.get("img_code")
image_code_id = param_dict.get("img_code_id")
# 2. 校验参数
if not all([nickname, mobile, password, image_code, image_code_id]):
return jsonify(errno=RET.PARAMERR, errmsg="参数")
# 判断用户名
if not re.match(r'^[0-9A-Za-z_]{5,15}$', nickname):
return jsonify(errno=RET.PARAMERR, errmsg="用户名格式由5-15数字、字母、下滑、下线组成")
# 校验手机号是否正确
if not re.match('1[35678]\\d{9}', mobile):
return jsonify(errno=RET.PARAMERR, errmsg="手机号格式不正确")
# 3. 先从redis中取出真实的验证码内容
try:
real_image_code = redis_store.get("ImageCodeId_" + image_code_id)
except Exception as e:
current_app.logger.error(e)
return jsonify(errno=RET.DBERR, errmsg="数据查询失败")
if not real_image_code:
return jsonify(errno=RET.NODATA, errmsg="图片验证码已过期")
# 4. 与用户的验证码内容进行对比，如果对比不一致，那么返回验证码输入错误
if real_image_code.upper() != image_code.upper():
return jsonify(errno=RET.DATAERR, errmsg="验证码输入错误")
# 5. 如果一致，初始化 User 模型，并且赋值属性
user = User()
user.nick_name = nickname
user.mobile = mobile
# 记录用户最后一次登录时间
user.create_time = datetime.now()
# 对密码做处理
# 需求：在设置 password 的时候，去对 password 进行加密，并且将加密结果给 user.password_hash 赋值
user.password = password
# 6. 添加到数据库
try:
db.session.add(user)
db.session.commit()
except Exception as e:
current_app.logger.error(e)
db.session.rollback()
return jsonify(errno=RET.DBERR, errmsg="数据保存失败")
# 往 session 中保存数据表示当前已经登录
session["user_id"] = user.id
session["mobile"] = user.mobile
session["nick_name"] = user.nick_name
session['is_admin'] = user.is_admin
# 7. 返回响应
return jsonify(errno=RET.OK, errmsg="注册成功")
