def get_user_data():
page_size = 10
page = 1
all_user = UserModelDao.query_user(0)
all_list = [i for i in all_user]
data = all_list[(page - 1) * page_size:page * page_size]
user_name = []
user_pwd = []
state = []
email = []
for i in data:
user_name.append(i.user_name)
user_pwd.append(i.user_pwd)
state.append(i.user_state)
email.append(i.email)
print(user_name, user_pwd, email, state)
return json.dumps({
'code': 200,
'msg': 'Success',
'data': {
'user_name': user_name,
'user_pwd': user_pwd,
'email': email
}
})
def register_send_email():
"""
注册时，开启另一个线程发送邮箱验证码
:return:
"""
global register_email_code
email = request.json["email"] # 获取用户输入的邮箱
if email:
if not dao.validate_email(email):
# 邮箱格式错误
response_data = response[20307]
elif UserModelDao.query_user(2, email=email):
# 邮箱已被使用
response_data = response[20306]
else:
register_email_code = dao.thread_send_email(email)
print(register_email_code)
# 发送成功
response_data = response[200]
else:
# 缺少参数
response_data = response[20101]
return json.dumps(response_data, ensure_ascii=False)
def alter_send_email():
"""
修改密码时，开启另一个线程发送邮箱验证码
:return:
"""
global alter_email_code
email = request.json['email']
user_name = request.json['user_name']
if email and user_name:
if not dao.validate_email(email):
# 邮箱格式错误
response_data = response[20307]
elif not UserModelDao.query_user(4, user_name=user_name, email=email):
# 用户名和邮箱不匹配
response_data = response[20308]
else:
alter_email_code = dao.thread_send_email(email)
# 发送成功
response_data = response[200]
else:
# 缺少参数
response_data = response[20101]
return json.dumps(response_data, ensure_ascii=False)
def alter_pwd():
"""
用户修改密码
:return:
"""
global alter_email_code
req_json = request.json
user_name = req_json['user_name']
new_password = req_json['new_password']
# re_password = req_json['re_password']
email = req_json['email']
user_email_code = req_json['email_code']
if user_name and new_password and email and user_email_code:
if not UserModelDao.query_user(1, user_name=user_name):
# 用户名不存在
response_data = response[20201]
# elif new_password != re_password:
# 两次输入密码不一致
# response_data = response[20304]
elif user_email_code != alter_email_code:
# 邮箱验证码错误
response_data = response[20305]
else:
# 修改成功
UserModelDao.alter_user_pwd(user_name, new_password)
response_data = response[200]
else:
# 缺少参数
response_data = response[20101]
return json.dumps(response_data, ensure_ascii=False)
def get_user_data():
page_size = request.json['page_size']
page = request.json['page']
group_id = request.json['group_id']
all_user_list = UserModelDao.query_user(3, group_id=group_id)
all_user_list = [i for i in all_user_list]
user_list = all_user_list[(page - 1) * page_size:page * page_size]
user_data = []
for user in user_list:
user_data.append({
'user_name': user.user_name,
'user_pwd': user.user_pwd,
'user_state': user.user_state,
'email': user.email
})
data = {'count': len(all_user_list), 'user_data': user_data}
return json.dumps({'code': 200, 'msg': 'Success', 'data': data})
def get_user_mount():
"""
得到用户数量
:return:
"""
all_user_list = UserModelDao.query_user(0)
data = {'root': 0, 'manager': 0, 'user': 0}
for user_model in all_user_list:
if user_model.group_id == 1:
data['user'] += 1
elif user_model.group_id == 2:
data['manager'] += 1
elif user_model.group_id == 3:
data['root'] += 1
response_data = {'data': data}
response_data.update(response[200])
return json.dumps(response_data)
def register():
"""
用户注册功能
:return:
"""
global register_email_code
data = request.json # 获取表单数据
name = data["user_name"]
pwd = data["password"]
# check_pwd = data["check_password"]
email = data["email"]
user_email_code = data["email_code"]
if name and pwd and email and user_email_code: # and check_pwd:
if not re.search(u'^[_a-zA-Z0-9\u4e00-\u9fa5]+$', name):
# 用户名格式出错
response_data = response[20302]
elif UserModelDao.query_user(1, user_name=name):
# 用户名已存在
response_data = response[20301]
elif len(pwd) < 6:
# 密码长度太短
response_data = response[20303]
# elif check_pwd != pwd:
# 两次密码输入不一致
# response_data = response[20304]
elif user_email_code != register_email_code:
# 邮箱验证码错误
response_data = response[20305]
else:
# 插入新用户
UserModelDao.add_user(name, pwd, email)
response_data = response[200]
else:
# 缺少参数
response_data = response[20101]
return json.dumps(response_data, ensure_ascii=False)