60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
import time
|
||
from apscheduler.schedulers.background import BackgroundScheduler
|
||
from utils.logger import logger
|
||
from config.settings import YAML_OUTPUT_PATH
|
||
from core.get_vm_snapshots import get_all_vms
|
||
from core.data_exporter import create_excel_report, export_yaml
|
||
from core.remove_snapshots import load_old_snapshots, remove_snapshot
|
||
|
||
|
||
def export_files():
|
||
"""导出Excel和Yaml文件的函数"""
|
||
logger.info("🔍 开始收集VM和快照信息...")
|
||
vms = get_all_vms() # 主函数入口，获取虚拟机信息
|
||
|
||
# 导出Excel报表
|
||
logger.info("📝 开始导出Excel报表...")
|
||
old_snapshots = create_excel_report(vms) # 生成Excel报告并获取旧快照
|
||
|
||
# 导出Yaml文件
|
||
logger.info("📝 开始导出 Yaml 文件...")
|
||
export_yaml(old_snapshots)
|
||
logger.info("========== Excel和Yaml文件导出完成 ==========")
|
||
|
||
|
||
def delete_old_snapshots():
|
||
"""删除旧快照的函数"""
|
||
logger.info("🗑️ 开始删除过旧快照...")
|
||
old_snapshots = load_old_snapshots(YAML_OUTPUT_PATH)
|
||
remove_snapshot(old_snapshots)
|
||
logger.info("========== VM快照清理任务执行完成 ==========")
|
||
|
||
|
||
def main():
|
||
"""主执行函数"""
|
||
|
||
# 设置定时任务
|
||
scheduler = BackgroundScheduler()
|
||
|
||
# 每周六凌晨4点导出Excel和Yaml文件
|
||
scheduler.add_job(export_files, 'cron', day_of_week='sat', hour=4, minute=0)
|
||
|
||
# 每周六晚上7点执行删除快照任务
|
||
scheduler.add_job(delete_old_snapshots, 'cron', day_of_week='sat', hour=19, minute=0)
|
||
|
||
scheduler.start()
|
||
|
||
logger.info("定时任务已设置：每周六凌晨4点导出文件，晚上7点删除快照")
|
||
|
||
try:
|
||
# 保持主程序运行，以便调度器能正常工作
|
||
while True:
|
||
time.sleep(1)
|
||
|
||
except (KeyboardInterrupt, SystemExit):
|
||
scheduler.shutdown()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|