466 lines
13 KiB
Python
466 lines
13 KiB
Python
import json
|
||
import os
|
||
import time
|
||
import random
|
||
|
||
import requests
|
||
from werkzeug.security import generate_password_hash, check_password_hash
|
||
|
||
from DBHelper.MongoHelper import MongoHelper
|
||
from Utils.ErrorUtil import ReturnConditionCheckFailed
|
||
from user.user_auth import create_token
|
||
from user.user_obj import User
|
||
from user.user_utils import check_mail_fmt, decrypt_data
|
||
from user.user_db import DELETE_DATA, UPDATE_INFO
|
||
|
||
|
||
class UserImpl(User):
|
||
"""管理端用户实现"""
|
||
|
||
db = MongoHelper("tfse_v0.21")
|
||
|
||
def create(self):
|
||
def check_email_registered():
|
||
"""检查邮箱是否已注册"""
|
||
|
||
email_is_existed = self.db.find_single_column(
|
||
"管理端",
|
||
"用户",
|
||
{"email": self.email},
|
||
"email"
|
||
)
|
||
|
||
if email_is_existed:
|
||
raise ReturnConditionCheckFailed("邮箱已被注册", 200)
|
||
|
||
def generate_new_uid():
|
||
"""生成新的用户ID"""
|
||
|
||
def uid_maker(num):
|
||
"""
|
||
用户ID生成器
|
||
num: ID长度 int
|
||
"""
|
||
|
||
choices = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
|
||
salt = ''
|
||
for i in range(num):
|
||
salt += random.choice(choices)
|
||
return salt
|
||
|
||
def uid_checker(__uid):
|
||
"""
|
||
用户ID重复检查
|
||
__uid: 用户ID str
|
||
return:
|
||
True 用户ID可用
|
||
False 用户ID不可用
|
||
"""
|
||
is_uid_existed = self.db.find_single_column(
|
||
"管理端",
|
||
"用户",
|
||
{"UID": __uid},
|
||
"UID"
|
||
)
|
||
return True if is_uid_existed is None else False
|
||
|
||
uid = uid_maker(8)
|
||
while not uid_checker(uid):
|
||
uid = uid_maker(8)
|
||
self.uid = uid
|
||
|
||
def __main__():
|
||
check_email_registered()
|
||
generate_new_uid()
|
||
|
||
self.status = "normal"
|
||
self.create_time = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
|
||
self.db.upsert_single_data(
|
||
"管理端",
|
||
"用户",
|
||
{"UID": self.uid},
|
||
self.dict_to_save()
|
||
)
|
||
|
||
__main__()
|
||
|
||
|
||
class UserManageImpl:
|
||
|
||
@staticmethod
|
||
def list_user(param1, param2, param3, param4):
|
||
"""
|
||
查询用户列表
|
||
param1:
|
||
type: dict
|
||
desc: 用户搜索条件 支持模糊查询条件
|
||
demo: {"姓名": "xxx", "邮箱": "xxx"}
|
||
param2:
|
||
type: dict
|
||
desc: 排序条件 asc正序 desc倒序
|
||
demo: {"姓名": asc}
|
||
param3:
|
||
type: int
|
||
desc: 每页数据数量
|
||
param4:
|
||
type: int
|
||
desc: 当前页数
|
||
"""
|
||
# 查询体构造方法
|
||
def make_search_body(param):
|
||
body = dict()
|
||
search_keys = list(param.keys())
|
||
|
||
if "uid" in search_keys:
|
||
body['UID'] = {"UID": param['uid']}
|
||
|
||
if "姓名" in search_keys:
|
||
body['name'] = {"$regex": param['姓名']}
|
||
|
||
if "邮箱" in search_keys:
|
||
body['email'] = {"$regex": param['邮箱']}
|
||
|
||
if "状态" in search_keys:
|
||
if param['状态'] == "正常":
|
||
body['status'] = "normal"
|
||
elif param['状态'] == "停用":
|
||
body['status'] = "disable"
|
||
else:
|
||
pass
|
||
|
||
if "角色" in search_keys:
|
||
roles = list()
|
||
role_map = {"管理员": "admin", "分析师": "analysts", "开发者": "developer", "运营人员": "operator", "访客": "guest"}
|
||
for role in list(set(param['角色'])):
|
||
roles.append(role_map[role])
|
||
body['role'] = {"$in": roles}
|
||
|
||
return body
|
||
|
||
# 顺序条件构造方法
|
||
def make_sort_body(param):
|
||
if param != {}:
|
||
columns_map = {"姓名": "name", "邮箱": "email", "状态": "status", "角色": "role", "创建时间": "create_time"}
|
||
asc_or_desc = 1 if list(param.values())[0] == "asc" else -1
|
||
sort_column = columns_map[list(param.keys())[0]]
|
||
body = {sort_column: asc_or_desc}
|
||
else:
|
||
body = {"create_time": -1}
|
||
|
||
return body
|
||
|
||
# 连接数据库
|
||
db = MongoHelper("tfse_v0.21")
|
||
|
||
search_body = make_search_body(param1)
|
||
sort = make_sort_body(param2)
|
||
page_size = 10 if int(param3) > 10 else int(param3)
|
||
page_no = int(param4)
|
||
|
||
total = db.find_all_data_with_count(
|
||
"管理端",
|
||
"用户",
|
||
search_body
|
||
)
|
||
|
||
records = db.find_data_by_page_with_sort(
|
||
"管理端",
|
||
"用户",
|
||
search_body,
|
||
["UID", "email", "name", "status", "role", "create_time"],
|
||
sort,
|
||
page_size,
|
||
page_no
|
||
)
|
||
|
||
if records:
|
||
for record in records:
|
||
record['uid'] = record.pop('UID')
|
||
record['邮箱'] = record.pop('email')
|
||
record['姓名'] = record.pop('name')
|
||
record['状态'] = "正常" if record.pop('status') == "normal" else "停用"
|
||
record['角色'] = record.pop('role')
|
||
record['创建时间'] = record.pop('create_time')
|
||
else:
|
||
records = []
|
||
|
||
result = {
|
||
"records": records,
|
||
"total": total
|
||
}
|
||
|
||
return result
|
||
|
||
@staticmethod
|
||
def send_vcode_to_user_impl(email):
|
||
|
||
def gen_vcode():
|
||
choices = '0123456789'
|
||
salt = ''
|
||
for i in range(6):
|
||
salt += random.choice(choices)
|
||
return salt
|
||
|
||
def send_email():
|
||
email_api = 'http://116.63.130.34:30001'
|
||
vcode = gen_vcode()
|
||
timestamp = round(time.time())
|
||
headers = {"Content-Type": "application/json;charset=UTF-8"}
|
||
data = {"title": "【远东资信】{}".format("登录验证码"),
|
||
"sender": 'fecribd@fecr.com.cn',
|
||
"recipients": [email],
|
||
"msg_body": "您{}的验证码为 【{}】,5分钟内有效。".format("登录", vcode)}
|
||
requests.post(url=email_api + '/send_mail', headers=headers, data=json.dumps(data))
|
||
|
||
db = MongoHelper("tfse_v0.21")
|
||
db.upsert_single_data(
|
||
"管理端",
|
||
"邮箱验证码记录",
|
||
{"email": email},
|
||
{"vcode": vcode, "timestamp": timestamp}
|
||
)
|
||
return True
|
||
|
||
def __main__():
|
||
res = send_email()
|
||
if res is not True:
|
||
return res
|
||
|
||
return True
|
||
|
||
return __main__()
|
||
|
||
@staticmethod
|
||
def create_user_impl(email, name, pwd, role):
|
||
"""
|
||
创建新用户流程
|
||
Parameters:
|
||
email str 邮箱
|
||
name str 姓名
|
||
pwd str 密码
|
||
role str 角色
|
||
Returns:
|
||
执行成功 bool True
|
||
执行失败 str 异常信息
|
||
"""
|
||
db = MongoHelper("tfse_v0.21")
|
||
|
||
def check_params():
|
||
"""
|
||
参数检查
|
||
"""
|
||
email_is_existed = db.find_single_column(
|
||
"管理端",
|
||
"用户",
|
||
{"email": email},
|
||
"email"
|
||
)
|
||
|
||
if email_is_existed:
|
||
return "邮箱已被注册"
|
||
|
||
return True
|
||
|
||
def gen_new_uid():
|
||
"""
|
||
生成新的用户ID,如果该ID存在,则重新生成
|
||
Returns:
|
||
生成的用户ID
|
||
"""
|
||
|
||
def make_id(num):
|
||
"""
|
||
随机生成字符串
|
||
"""
|
||
choices = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
|
||
salt = ''
|
||
for i in range(num):
|
||
salt += random.choice(choices)
|
||
return salt
|
||
|
||
def gen_id():
|
||
"""
|
||
生成新ID,
|
||
检查新ID是否存在,如果存在则继续生成新ID
|
||
若新ID可使用,返回新ID
|
||
"""
|
||
new_id = make_id(8)
|
||
|
||
case = db.find_single_column(
|
||
"管理端",
|
||
"用户",
|
||
{"UID": new_id},
|
||
"UID"
|
||
) is not None
|
||
|
||
while case:
|
||
new_id = make_id(8)
|
||
return new_id
|
||
|
||
return gen_id()
|
||
|
||
def __main__():
|
||
"""
|
||
执行流程
|
||
"""
|
||
check_param_result = check_params()
|
||
if check_param_result is not True:
|
||
return check_param_result
|
||
|
||
user = {
|
||
'UID': gen_new_uid(),
|
||
'email': email,
|
||
'name': name,
|
||
'pwd': generate_password_hash(decrypt_data(encrypt_msg=pwd)),
|
||
'status': 'normal',
|
||
'role': role,
|
||
'create_time': time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
||
}
|
||
|
||
db.upsert_single_data(
|
||
"管理端",
|
||
"用户",
|
||
{"UID": user['UID']},
|
||
user
|
||
)
|
||
|
||
return True
|
||
|
||
return __main__()
|
||
|
||
|
||
class UserOperateImpl:
|
||
|
||
@staticmethod
|
||
def login_impl(email, pwd, vcode):
|
||
db = MongoHelper("tfse_v0.21")
|
||
|
||
user_info = db.find_single_data(
|
||
"管理端",
|
||
"用户",
|
||
{"email": email},
|
||
["UID", "name", "pwd", "status", "role"]
|
||
)
|
||
|
||
def check_email():
|
||
if not user_info:
|
||
return "不存在该邮箱"
|
||
if not check_mail_fmt(email):
|
||
return "邮箱格式错误"
|
||
return True
|
||
|
||
def check_disable():
|
||
if user_info['status'] != 'normal':
|
||
return "账户已禁用"
|
||
return True
|
||
|
||
def check_vcode():
|
||
record = db.find_single_data(
|
||
"管理端",
|
||
"邮箱验证码记录",
|
||
{"email": email},
|
||
["vcode", "timestamp"]
|
||
)
|
||
|
||
if not record:
|
||
return "无验证信息"
|
||
|
||
if record['vcode'] != vcode:
|
||
return "验证码错误"
|
||
|
||
if time.time() - record['timestamp'] > 300:
|
||
return "验证码过期"
|
||
|
||
db.delete_single_data(
|
||
"管理端",
|
||
"邮箱验证码记录",
|
||
{"email": email}
|
||
)
|
||
|
||
return True
|
||
|
||
def check_pwd():
|
||
hash_pwd = user_info['pwd']
|
||
try:
|
||
if not check_password_hash(hash_pwd, decrypt_data(encrypt_msg=pwd)):
|
||
return "密码错误"
|
||
except Exception:
|
||
return "密码错误"
|
||
return True
|
||
|
||
def make_menus():
|
||
role = user_info['role']
|
||
|
||
with open(os.path.abspath(os.path.dirname(__file__)+'/static/menus.json'), "r", encoding='utf-8') as f:
|
||
duties = json.load(f)
|
||
|
||
return duties[role]
|
||
|
||
def make_session_id():
|
||
choices = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789'
|
||
session_id = ''
|
||
for i in range(4):
|
||
session_id += random.choice(choices)
|
||
|
||
db.update_single_data(
|
||
"管理端",
|
||
"token记录",
|
||
{"UID": user_info['UID']},
|
||
{"session_id": session_id}
|
||
)
|
||
|
||
return session_id
|
||
|
||
def make_login_return_data():
|
||
return_data = dict()
|
||
return_data['name'] = user_info["name"]
|
||
return_data['token'] = create_token(make_session_id())
|
||
return_data['menus'] = make_menus()
|
||
return return_data
|
||
|
||
def __main__():
|
||
result = check_email()
|
||
if result is not True:
|
||
return result
|
||
|
||
result = check_disable()
|
||
if result is not True:
|
||
return result
|
||
|
||
result = check_pwd()
|
||
if result is not True:
|
||
return result
|
||
|
||
result = check_vcode()
|
||
if result is not True:
|
||
return result
|
||
|
||
return make_login_return_data()
|
||
|
||
return __main__()
|
||
|
||
@staticmethod
|
||
def disable_user_impl(uid):
|
||
"""
|
||
禁用用户
|
||
"""
|
||
result = UPDATE_INFO("用户", "用户信息", {"UID": uid}, {"status": "disable"})
|
||
return result
|
||
|
||
@staticmethod
|
||
def delete_user_impl(uid):
|
||
"""
|
||
删除用户
|
||
"""
|
||
result = DELETE_DATA("用户", "用户信息", {"UID": uid})
|
||
return result
|
||
|
||
@staticmethod
|
||
def manage_role_of_user_impl(uid, role):
|
||
"""
|
||
管理用户角色
|
||
"""
|
||
result = UPDATE_INFO("用户", "用户信息", {"UID": uid}, {"role": role})
|
||
return result
|