添加了删除快照的功能

core/getVmsSnapshots.py

@@ -1,10 +1,11 @@
 import pandas as pd
+import yaml
 from pyVmomi import vim
 from datetime import datetime, timedelta
 from pyVim.connect import SmartConnect, Disconnect
 from openpyxl.styles import Border, Side, Font, PatternFill
 from utils.logger import logger
-from config.settings import MANAGEMENT_NODES, SNAPSHOT_RETENTION_DAYS, EXCEL_OUTPUT_PATH
+from config.settings import MANAGEMENT_NODES, SNAPSHOT_RETENTION_DAYS, EXCEL_OUTPUT_PATH, YAML_OUTPUT_PATH
 
 
 def get_all_vms():
@@ -30,10 +31,8 @@ def get_all_vms():
                 vm_list.append(vm_info)
 
         except vim.fault.InvalidLogin as e:
-            print(f"登录 {node['host']} 失败，请检查用户名和密码：{e.msg}")
             logger.info(f"登录 {node['host']} 失败，请检查用户名和密码：{e.msg}")
         except Exception as e:
-            print(f"无法连接到 {node['host']}：{e}")
             logger.error(f"无法连接到 {node['host']}：{e}")
         finally:
             if 'si' in locals():  # 确保 si 是定义过的
@@ -146,12 +145,12 @@ def get_virtual_disk_size(vm):
     return total_size
 
 
-def collect_snapshot_data(snapshot, vm, snapshot_data):
-    """递归扁平化快照数据用于Excel"""
+def collect_snapshot_data(snapshot, vm, snapshot_data, old_snapshots):
+    """递归扁平化快照数据用于Excel，并收集旧快照"""
     create_time = datetime.strptime(snapshot['createTime'], '%Y-%m-%d %H:%M:%S')
     is_old = create_time < (datetime.now() - timedelta(days=SNAPSHOT_RETENTION_DAYS))
 
-    snapshot_data.append({
+    snapshot_info = {
         'NodeHost': vm['NodeHost'],
         'VMName': vm['name'],
         'Snapshot Name': snapshot['name'],
@@ -162,16 +161,23 @@ def collect_snapshot_data(snapshot, vm, snapshot_data):
         'MOID': snapshot['moId'],
         'Quiesced': snapshot['quiesced'],
         'is_old': is_old
-    })
+    }
+
+    # 如果是旧快照，添加到旧快照列表
+    if is_old:
+        old_snapshots.append(snapshot_info)
+
+    snapshot_data.append(snapshot_info)
 
     for child in snapshot['children']:
-        collect_snapshot_data(child, vm, snapshot_data)
+        collect_snapshot_data(child, vm, snapshot_data, old_snapshots)
 
 
 # 输出数据到 Excel 文件
 def create_excel_report(vms):
     vm_data = []
     snapshot_data = []
+    old_snapshots = []  # 用于存储旧快照的信息
 
     for vm in vms:
         vm_data.append({
@@ -192,7 +198,7 @@ def create_excel_report(vms):
 
         if vm['snapshots']:
             for snapshot in vm['snapshots']:
-                collect_snapshot_data(snapshot, vm, snapshot_data)
+                collect_snapshot_data(snapshot, vm, snapshot_data, old_snapshots)
 
     vm_df = pd.DataFrame(vm_data)
     snapshot_df = pd.DataFrame(snapshot_data)
@@ -203,13 +209,30 @@ def create_excel_report(vms):
 
         workbook = writer.book
         style_sheet(workbook['VMs'])
-        style_sheet(workbook['Snapshots'], snapshot_df['is_old'].tolist())
+        # style_sheet(workbook['Snapshots'], snapshot_df['is_old'].tolist())
+        # 判断 DataFrame 是否有数据，没有数据则跳过，有数据则写入 excel 文件
+        if 'is_old' in snapshot_df.columns and snapshot_df['is_old'].tolist():
+            style_sheet(workbook['Snapshots'], snapshot_df['is_old'].tolist())
 
-    print("报告已生成:", EXCEL_OUTPUT_PATH)
+    logger.debug(f"Excel 文件已生成: {EXCEL_OUTPUT_PATH}")
+    # 返回旧快照的数据
+    return old_snapshots
+
+
+# 输出待删除的旧快照到 YAML 文件
+def export_yaml(old_snapshots):
+    print(old_snapshots)
+
+    # 将旧快照信息存储到 YAML 文件
+    with open(YAML_OUTPUT_PATH, 'w', encoding='utf-8') as yaml_file:
+        # allow_unicode：输出 Unicode 字符（中文等），allow_unicode：使用块样式（多行缩进），sort_keys：不按键名排序，保留原始插入顺序
+        yaml.dump(old_snapshots, yaml_file, allow_unicode=True, default_flow_style=False, sort_keys=False)
+    logger.debug(f"YAML 文件已生成: {YAML_OUTPUT_PATH}")
 
 
 if __name__ == '__main__':
     vms = get_all_vms()  # 主函数入口，获取虚拟机信息
     # print(vms)
-    create_excel_report(vms)    # 生成Excel报告
-
+    old_snapshots = create_excel_report(vms)  # 生成Excel报告并获取旧快照
+    export_yaml(old_snapshots)
+    # print(old_snapshots)