def update_platform(cls, _id, p_name, p_logo, p_type, user_id, **kargs):
"""
更新应用
"""
try:
# 除当前修改的名称以外，是否还存在要修改的名称
data = platform_col.find({"p_name": p_name, "_id": {"$ne": _id}})
if data.count() != 0:
return -1001
else:
data = platform_col.update({"_id": _id}, {
"$set": {
"p_name": p_name,
"p_logo": p_logo,
"p_type": p_type,
"update_time": Util.timeFormat()
}
})
if data["ok"] == 1:
log_str = "应用名称：{}；应用ID：{}；应用logo：{}；应用类型：{}".format(
p_name, _id, p_logo, p_type)
log_result = LogModel.add_log("修改应用", log_str, user_id,
"update")
return 1
else:
return 0
except Exception as e:
current_app.logger.error(e)
return 0
def insert_platform(cls, p_name, p_logo, p_type, user_id, **kargs):
"""
添加应用
"""
try:
data = platform_col.find({"p_name": p_name})
if data.count() != 0:
return -1001
else:
p_id = str(uuid.uuid1())
platform_col.insert({
"_id": p_id,
"p_name": p_name,
"p_logo": p_logo,
"p_type": p_type,
"create_time": Util.timeFormat(),
"deleted": 0,
"creator_id": user_id
})
log_str = "应用名称：{}；应用ID：{}；应用logo：{}；应用类型：{}".format(
p_name, p_id, p_logo, p_type)
log_result = LogModel.add_log("添加应用", log_str, user_id,
"insert")
return 1
except Exception as e:
current_app.logger.error(e)
return 0
def update_test_info(cls, p_id, t_id, t_name, t_str, t_desc, user_id):
"""
修改实验信息
"""
try:
data = test_col.find({
"p_id": p_id,
"t_name": t_name,
})
test_col.update({
"p_id": p_id,
"_id": t_id,
"t_id": {
"$ne": t_id
}
}, {
"$set": {
"t_name": t_name,
"t_str": t_str,
"t_desc": t_desc,
"update_time": Util.timeFormat()
}
})
log_str = "实验名称：{}；实验ID：{}；实验描述：{}；加盐字符：{}".format(
t_name, t_id, t_desc, t_str)
log_result = LogModel.add_log("修改实验", log_str, user_id, "update")
return 1
except Exception as e:
current_app.logger.error(e)
return 0
def add_test(cls, p_id, t_name, t_str, t_desc, user_id):
"""
创建实验
"""
try:
data = test_col.find({"p_id": p_id, "t_name": t_name})
data_len = len(list(data))
if data_len != 0:
return -2001
else:
t_id = str(uuid.uuid1())
test_col.insert({
"_id": t_id,
"p_id": p_id,
"t_name": t_name,
"t_str": t_str,
"t_desc": t_desc,
"t_status": 1,
"create_time": Util.timeFormat(),
"creator_id": user_id
})
log_str = "实验名称：{}；实验ID：{}；实验描述：{}；加盐字符：{}".format(
t_name, t_id, t_desc, t_str)
log_result = LogModel.add_log("创建实验", log_str, user_id,
"insert")
return 1
except Exception as e:
current_app.logger.error(e)
return 0
def add_log(cls, title, content, user_id, l_type):
"""添加日志
:param title: 标题
:param content: 内容
:param user_id: 用户ID
:param type: 类型 insert、remove、update
:return:
"""
try:
log_col.insert({
"_id": str(uuid.uuid1()),
"l_title": title,
"l_content": content,
"user_id": user_id,
"l_type": l_type,
"create_time": Util.timeFormat()
})
return 1
except Exception as e:
current_app.logger.error(e)
return 0
def toggle_status(cls, t_id, status, user_id):
"""
改变实验状态
"""
try:
data = test_col.update({"_id": t_id}, {
"$set": {
"t_status": int(status),
"update_time": Util.timeFormat()
}
})
if data["ok"] == 1:
test_status = "启用" if int(status) == 1 else "停用"
log_str = "实验ID：{}；状态：{}".format(t_id, test_status)
log_result = LogModel.add_log("启用/停用实验", log_str, user_id,
"update")
return 1
else:
return 0
except Exception as e:
current_app.logger.error(e)
return 0
def insert_stragegy(cls, t_id, s_name, s_desc, section_min, section_max,
user_id):
"""
新增策略
"""
try:
# 分桶区间逻辑判
# 查询实验的所有策略
data = stragegy_col.aggregate([{
"$match": {
"t_id": t_id
}
}, {
"$lookup": {
"from": "buckets",
"localField": "s_id",
"foreignField": "s_id",
"as": "buckets"
}
}, {
"$unwind": {
"path": "$buckets",
"preserveNullAndEmptyArrays": True
}
}])
# 如果有策略则进行判断（区间是否有重合）
data_list = list(data)
data_len = len(data_list)
s_id = ""
if data_len > 0:
for item in range(data_len):
temp_min = int(
data_list[item].get("buckets").get("section_min"))
temp_max = int(
data_list[item].get("buckets").get("section_max"))
if max(temp_min, int(section_min)) <= min(
temp_max, int(section_max)):
return -3002
if s_name == data_list[item].get("s_name"):
s_id = data_list[item].get("_id")
if s_id == "":
# 如果没有策略则直接插入
_id = str(uuid.uuid1())
s_id = stragegy_col.insert({
"_id": _id,
"s_id": _id,
"t_id": t_id,
"s_name": s_name,
"create_time": Util.timeFormat()
})
bucket_id = str(uuid.uuid1())
bucket_col.insert({
"_id": bucket_id,
"s_id": s_id,
"t_id": t_id,
"s_desc": s_desc,
"section_min": section_min,
"section_max": section_max,
"create_time": Util.timeFormat(),
"creator_id": user_id
})
log_str = "策略名称：{}；策略ID：{}；策略描述：{}；分桶ID：{}；分桶区间：{}".format(
s_name, s_id, s_desc, bucket_id,
str(section_min) + " ~ " + str(section_max))
log_result = LogModel.add_log("创建策略", log_str, user_id, "insert")
return 1
except Exception as e:
current_app.logger.error(e)
return 0
def update_stragegy(cls, t_id, s_id, b_id, s_name, s_desc, section_min,
section_max, user_id):
"""
更新策略
"""
try:
# 更新策略描述
stragegy_col.update({
"s_id": s_id,
"s_desc": {
"$ne": s_desc
}
}, {"$set": {
"s_desc": s_desc,
"update_time": Util.timeFormat()
}})
# 更新分桶
bucket_data = bucket_col.find({"t_id": t_id, "_id": {"$ne": b_id}})
for item in bucket_data:
temp_min = int(item.get("section_min"))
temp_max = int(item.get("section_max"))
if max(temp_min, int(section_min)) <= min(
temp_max, section_max):
return -3002
data = bucket_col.update(
{
"s_id":
s_id,
"_id":
b_id,
"$or": [{
"section_min": {
"$ne": section_min
}
}, {
"section_max": {
"$ne": section_max
}
}, {
"s_desc": {
"$ne": s_desc
}
}]
}, {
"$set": {
"s_desc": s_desc,
"section_min": section_min,
"section_max": section_max,
"update_time": Util.timeFormat()
}
})
if data.get("ok") == 1:
log_str = "策略ID：{}；策略描述：{}；分桶ID：{}；分桶区间：{}".format(
s_id, s_desc, b_id,
str(section_min) + " ~ " + str(section_max))
log_result = LogModel.add_log("修改策略", log_str, user_id,
"update")
return 1
else:
return 0
except Exception as e:
current_app.logger.error(e)
return 0