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="注册成功")