File size: 4,783 Bytes
a757bd3 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
import os,sys,json
os.chdir('/www/server/panel')
if not 'class/' in sys.path:
sys.path.insert(0,'class/')
import public
# 标记文件路径
CHECK_MOD_PUSH_FILE = "/www/server/panel/data/mod_push_data/check_mod_push_file.pl"
OLD_PUSH_DATA_PATH = "/www/server/panel/class/push/push.json"
NEW_PUSH_DATA_PATH = "/www/server/panel/data/mod_push_data/task.json"
def check_and_delete_old_tasks(old_push_data: dict, new_push_data: dict):
for task_group_key, task_group_value in old_push_data.items():
if not task_group_value:
continue
tasks_to_delete = []
for old_task_key, old_task_value in task_group_value.items():
# old_task_type = old_task_value.get("type")
# old_task_title = old_task_value.get("title")
# print(f"处理具体任务: {old_task_key}, type: {old_task_type}, title: {old_task_title}")
# if old_task_type is None or old_task_title is None:
# print(f"具体任务 {old_task_key} 的 type 或 title 字段为 None，跳过该任务")
# continue
for new_task in new_push_data:
if compare_tasks(old_task_value, new_task):
tasks_to_delete.append(old_task_key)
print(f"迁移成功，删除具体任务: {old_task_key}")
break
# 打印要删除的任务列表
print("tasks_to_delete:", tasks_to_delete)
print("task_group_value before deletion:", task_group_value)
# 删除任务
for task_key in tasks_to_delete:
if task_key in task_group_value:
del task_group_value[task_key]
print(f"实际删除任务: {task_key}")
old_push_data[task_group_key] = task_group_value
# 更新旧任务数据文件
public.writeFile(OLD_PUSH_DATA_PATH, json.dumps(old_push_data, ensure_ascii=False, indent=4))
def compare_tasks(old_task, new_task):
try:
"""
比较旧任务和新任务是否相同
第一种情况：只比较类型，但类型名改变了
第二种情况：比较类型和项目名，但类型改变了
第三种情况：类型名不变，比较类型
"""
old_task_type = old_task['type']
old_project = old_task.get('project', '')
if 'task_data' not in new_task:
return False
new_task_type = new_task['task_data'].get('type', '')
new_project = new_task['task_data'].get('project', '')
# 第一种情况：只比较类型，类型名改变了
type_mapping = {
'site_endtime': 'site_end_time',
'panel_pwd_endtime': 'panel_pwd_end_time',
'disk': 'system_disk',
'cpu': 'system_cpu',
'load': 'system_load',
'mem': 'system_mem',
'ssl': 'site_ssl',
'mysql_pwd_endtime': 'mysql_pwd_end'
}
if old_task_type in type_mapping and new_task_type == type_mapping[old_task_type]:
return True
# print(new_project)
# print(old_project)
# 第二种情况：比较类型和项目名，类型改变了
if old_task_type == 'ssl' and new_task.get('source') == 'site_ssl' and old_project == new_project:
return True
if old_task_type == 'project_status' and new_task_type == 'project_status' and old_project == new_project:
return True
if old_task_type == 'services' and new_task_type == 'services' and old_project == new_project:
return True
# 第三种情况：类型名不变，直接比较类型
if old_task_type == new_task_type:
return True
# 处理特殊情况，比如 "panel_update、 ssh_login" 的关键字匹配
if new_task.get('source') == old_task_type:
return True
if new_task.get('keyword') == type_mapping[old_task_type]:
return True
if new_task.get('source') == type_mapping[old_task_type]:
return True
return False
except Exception as e:
print(e)
def main():
try:
old_push_data=json.loads(public.readFile(OLD_PUSH_DATA_PATH))
# print("old_push_data:",old_push_data)
except:
return
try:
new_push_data=json.loads(public.readFile(NEW_PUSH_DATA_PATH))
# print("new_push_data:",new_push_data)
except:
return
if old_push_data and new_push_data:
try:
check_and_delete_old_tasks(old_push_data,new_push_data)
except:
pass
# 写入标记文件，表明处理完成
public.writeFile(CHECK_MOD_PUSH_FILE,"")
if __name__=="__main__":
main()
|