83 lines
2.3 KiB
Python
83 lines
2.3 KiB
Python
|
#!/usr/bin/env python
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
"""
|
|||
|
|
任务实例ID修复脚本
|
|||
|
|
为所有现有的任务实例记录生成并设置instance_id
|
|||
|
|
|
|||
|
|
用法:
|
|||
|
|
python scripts/fix_instance_ids.py
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import logging
|
|||
|
|
import uuid
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
|
|||
|
|
# 添加项目根目录到系统路径
|
|||
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|||
|
|
|
|||
|
|
# 导入项目模块
|
|||
|
|
from data.session import get_db
|
|||
|
|
from data.models.task_instance import TaskInstance
|
|||
|
|
|
|||
|
|
# 设置日志
|
|||
|
|
logging.basicConfig(
|
|||
|
|
level=logging.INFO,
|
|||
|
|
format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
|
|||
|
|
handlers=[
|
|||
|
|
logging.StreamHandler(sys.stdout), # 输出到控制台
|
|||
|
|
logging.FileHandler("fix_instance_ids.log") # 输出到文件
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
logger = logging.getLogger("fix_instance_ids")
|
|||
|
|
|
|||
|
|
def fix_instance_ids():
|
|||
|
|
"""为所有现有的任务实例记录生成并设置instance_id"""
|
|||
|
|
try:
|
|||
|
|
# 获取数据库会话
|
|||
|
|
db = next(get_db())
|
|||
|
|
|
|||
|
|
# 查询所有没有instance_id的任务实例记录
|
|||
|
|
instances = db.query(TaskInstance).filter(
|
|||
|
|
(TaskInstance.instance_id == None) |
|
|||
|
|
(TaskInstance.instance_id == '')
|
|||
|
|
).all()
|
|||
|
|
|
|||
|
|
logger.info(f"发现 {len(instances)} 条没有instance_id的任务实例记录")
|
|||
|
|
|
|||
|
|
# 为每条记录生成并设置instance_id
|
|||
|
|
for instance in instances:
|
|||
|
|
instance_id = str(uuid.uuid4())
|
|||
|
|
instance.instance_id = instance_id
|
|||
|
|
logger.info(f"为任务实例 ID={instance.id} 设置instance_id={instance_id}")
|
|||
|
|
|
|||
|
|
# 提交更改
|
|||
|
|
db.commit()
|
|||
|
|
logger.info("所有任务实例记录的instance_id已更新")
|
|||
|
|
|
|||
|
|
return True
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"修复任务实例ID失败: {str(e)}")
|
|||
|
|
if 'db' in locals():
|
|||
|
|
db.rollback()
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主函数"""
|
|||
|
|
logger.info("开始修复任务实例ID...")
|
|||
|
|
|
|||
|
|
# 修复任务实例ID
|
|||
|
|
success = fix_instance_ids()
|
|||
|
|
|
|||
|
|
if success:
|
|||
|
|
logger.info("任务实例ID修复成功！")
|
|||
|
|
else:
|
|||
|
|
logger.error("任务实例ID修复失败！")
|
|||
|
|
|
|||
|
|
# 根据修复结果设置退出码
|
|||
|
|
sys.exit(0 if success else 1)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|