Get Vms Snapshots and Export Excel has completed.

.env

@@ -1,8 +1,8 @@
-# vCenter配置（支持多个）
-VCENTER_HOSTS=192.168.40.134
+# 多个vCenter地址（逗号分隔）
+VCENTER_HOSTS=192.168.40.134,vc2.example.com,vc3.example.com
+# 所有vCenter共用的账号密码
 VCENTER_USER=administrator@lan.com
 VCENTER_PASSWORD=Root@2025
-# 快照保留天数
 SNAPSHOT_RETENTION_DAYS=15
 # 每台vCenter同时删除的快照数量限制
 MAX_DELETE_CONCURRENT=4

README.md

@@ -1,7 +1,69 @@
-# 以下需求需要每个月执行一次，使用ansible实现，还是使用python代码实行好？
--[x] 获取所有vms
--[ ] 获取所有snapshots
--[ ] 筛选出15天（半个月）前的snapshots
--[ ] 以上内容以Excel表格的形式导出
--[ ] 最后删除15天前的snapshot，并同时记录删除的snapshot日志信息
--[ ] 需要控制每台vCenter不可以同时删除超过4个快照
+# RemoveWeeklySnapshot
+
+> 以下需求需要每周执行一次
+
+- [x] 连接vCenter/Esxi/Hyper-V
+- [x] 获取所有 vms
+- [x] 获取所有 snapshots
+- [x] 筛选出15天（半个月）前的snapshots
+- [x] 以上内容以Excel表格的形式导出，超出15天的快照蓝底标识
+- [ ] 最后删除 15 天前的 snapshot，并同时记录删除的 snapshot 日志信息
+- [ ] 需要控制每台 vCenter 不可以同时删除超过4个快照
+- [ ] 增加排除不能删除的快照
+
+
+
+
+
+| 你想获取     | 代码                                | 示例输出                                       |
+| ------------ | ----------------------------------- | ---------------------------------------------- |
+| **名称**     | `vm.name`                           | `"WebServer-01"`                               |
+| **MOID**     | `vm._moId`                          | `"vm-12"`                                      |
+| **电源状态** | `vm.runtime.powerState`             | `poweredOn` / `poweredOff`                     |
+| **开机时间** | `vm.runtime.bootTime`               | `datetime` 对象                                |
+| **CPU数**    | `vm.config.hardware.numCPU`         | `4`                                            |
+| **内存(MB)** | `vm.config.hardware.memoryMB`       | `8192`                                         |
+| **操作系统** | `vm.config.guestFullName`           | `"CentOS 7 (64-bit)"`                          |
+| **IP地址**   | `vm.guest.ipAddress`                | `"192.168.1.100"`                              |
+| **主机名**   | `vm.guest.hostName`                 | `"webserver01.local"`                          |
+| **存储路径** | `vm.config.files.vmPathName`        | `"[Datastore1] WebServer-01/WebServer-01.vmx"` |
+| **快照数量** | `len(vm.snapshot.rootSnapshotList)` | `3`                                            |
+
+>vm
+>├── 基础标识
+>│   ├── name          VM名称
+>│   └── _moId         内部ID (vm-12)
+>│
+>├── runtime           【运行状态】
+>│   ├── powerState    poweredOn/Off/Suspended
+>│   ├── bootTime      开机时间
+>│   └── host          所在物理机
+>│
+>├── config            【硬件配置】
+>│   ├── hardware      CPU/内存/硬盘
+>│   ├── guestFullName 操作系统
+>│   └── files         VMX文件路径
+>│
+>├── guest             【客户机内部信息】
+>│   ├── hostName      主机名
+>│   ├── ipAddress     IP地址
+>│   └── toolsStatus   VMware Tools状态
+>│
+>├── snapshot          【快照】
+>│   └── rootSnapshotList  快照树
+>│
+>├── storage           【存储】
+>│   └── perDatastoreUsage 各数据存储用量
+>│
+>├── network           【网络】
+>│   └── [Network]     连接的端口组
+>│
+>└── summary           【快速汇总】
+>├── overallStatus 整体健康状态
+>└── quickStats    实时性能数据
+
+
+
+[{'name': 'snap-02', 'description': 'test-Ansible snapshot', 'createTime': '2026-02-17 08:57:39', 'state': 'poweredOff', 'id': 2, 'moId': 'snapshot-27', 'sizeMB': None, 'quiesced': False, 'children': []}]
+[{'name': 'snap-01', 'description': 'Ansible snapshot', 'createTime': '2026-02-17 03:26:08', 'state': 'poweredOff', 'id': 1, 'moId': 'snapshot-26', 'sizeMB': None, 'quiesced': False, 'children': [{'name': 'snap-02', 'description': 'test-Ansible snapshot', 'createTime': '2026-02-17 08:57:39', 'state': 'poweredOff', 'id': 2, 'moId': 'snapshot-27', 'sizeMB': None, 'quiesced': False, 'children': []}]}]
+获取到 2 台VM

config/config.yaml

@@ -0,0 +1,35 @@
+# 管理节点配置（包含vCenter和ESXi）
+management_nodes:
+  # vCenter节点
+  - type: vcenter          # 标记类型为vcenter
+    name: vc1              # 节点名称（用于日志）
+    host: 192.168.40.134  # 地址
+    user: administrator@lan.com
+    password: Root@2025
+    max_delete_concurrent: 4  # 该节点最大并发删除数
+
+  # ESXi节点（未接入 vCenter 的 Esxi 主机）
+  - type: esxi             # 标记类型为esxi
+    name: esxi1
+    host: 192.168.40.133
+    user: root             # ESXi默认用root
+    password: Root@2025
+    max_delete_concurrent: 2  # ESXi性能较弱，并发数可设小些
+
+  # 另一个ESXi节点
+  - type: esxi
+    name: esxi2
+    host: esxi2.example.com
+    user: root
+    password: esxi2_password
+    max_delete_concurrent: 2
+
+# 全局策略配置
+global:
+  snapshot_retention_days: 15
+  excel_output_path: ./vm_snapshots_report.xlsx
+  # excel_output_path: /tmp/vm_snapshots_report.xlsx
+  # log_file_path: /var/log/vm_snapshot_cleanup.log
+  log_file_path: ./vm_snapshot_cleanup.log
+  # ESXi连接特殊配置（禁用SSL验证，ESXi默认自签证书）
+  disable_ssl_verify: true

config/settings.py

@@ -1,33 +1,88 @@
+import yaml
 import os
 from datetime import datetime, timedelta
-from dotenv import load_dotenv
+# from utils.logger import logger
 
-# 加载.env文件
-load_dotenv()
+# 配置文件路径
+CONFIG_PATH = os.path.join(os.path.dirname(__file__), 'config.yaml')
 
-# ========== 基础配置 ==========
-# vCenter配置
-VCENTER_HOSTS = os.getenv('VCENTER_HOSTS', '').split(',')
-VCENTER_USER = os.getenv('VCENTER_USER', '')
-VCENTER_PASSWORD = os.getenv('VCENTER_PASSWORD', '')
 
-# 快照策略配置
-SNAPSHOT_RETENTION_DAYS = int(os.getenv('SNAPSHOT_RETENTION_DAYS', 15))
-MAX_DELETE_CONCURRENT = int(os.getenv('MAX_DELETE_CONCURRENT', 4))
+def load_config():
+    """加载YAML配置，区分vCenter和ESXi"""
+    try:
+        with open(CONFIG_PATH, 'r', encoding='utf-8') as f:
+            raw_config = yaml.safe_load(f)
 
-# 输出路径配置
-EXCEL_OUTPUT_PATH = os.getenv('EXCEL_OUTPUT_PATH', '/tmp/vm_snapshots_report.xlsx')
-LOG_FILE_PATH = os.getenv('LOG_FILE_PATH', '/var/log/vm_snapshot_cleanup.log')
+        # 全局配置
+        global_config = raw_config.get('global', {})
+        config = {
+            # vCenter/ESXi节点列表
+            "MANAGEMENT_NODES": raw_config.get('management_nodes', []),
+            # vCenter/ESXi节点列表
+            "SNAPSHOT_RETENTION_DAYS": int(global_config.get('snapshot_retention_days', 15)),
+            "EXCEL_OUTPUT_PATH": global_config.get('excel_output_path', '/tmp/vm_snapshots_report.xlsx'),
+            "LOG_FILE_PATH": global_config.get('log_file_path', '/var/log/vm_snapshot_cleanup.log'),
+            "DISABLE_SSL_VERIFY": global_config.get('disable_ssl_verify', True),
+            # 算出的过期时间点
+            "EXPIRE_DATE": datetime.now() - timedelta(days=int(global_config.get('snapshot_retention_days', 15)))
+        }
 
-# 计算快照过期时间（全局变量）
-EXPIRE_DATE = datetime.now() - timedelta(days=SNAPSHOT_RETENTION_DAYS)
+        # 验证配置
+        if not config["MANAGEMENT_NODES"]:
+            raise ValueError("未配置任何管理节点（vCenter/ESXi）")
 
-# 验证必要配置
+        # 检查每个节点的必填字段
+        required_fields = ['type', 'name', 'host', 'user', 'password', 'max_delete_concurrent']
+        for node in config["MANAGEMENT_NODES"]:
+            missing = [f for f in required_fields if f not in node]
+            if missing:
+                raise ValueError(f"节点 {node.get('name', '未知')} 缺少配置字段: {missing}")
+            # 验证类型合法性
+            if node['type'] not in ['vcenter', 'esxi']:
+                raise ValueError(f"节点 {node['name']} 类型错误（仅支持vcenter/esxi）")
+
+        #logger.info(f"✅ 成功加载配置，共 {len(config['MANAGEMENT_NODES'])} 个管理节点")
+        return config
+
+    except Exception as e:
+        #logger.error(f"❌ 加载配置失败: {str(e)}")
+        raise
+
+
+# 加载配置并导出全局变量
+config = load_config()  # 模块导入时立即执行
+MANAGEMENT_NODES = config["MANAGEMENT_NODES"]
+SNAPSHOT_RETENTION_DAYS = config["SNAPSHOT_RETENTION_DAYS"]
+EXCEL_OUTPUT_PATH = config["EXCEL_OUTPUT_PATH"]
+LOG_FILE_PATH = config["LOG_FILE_PATH"]
+DISABLE_SSL_VERIFY = config["DISABLE_SSL_VERIFY"]
+EXPIRE_DATE = config["EXPIRE_DATE"]
+
+
+# 验证配置函数
 def validate_config():
-    """验证配置是否完整"""
-    required = [
-        VCENTER_HOSTS, VCENTER_USER, VCENTER_PASSWORD,
-        SNAPSHOT_RETENTION_DAYS, MAX_DELETE_CONCURRENT
-    ]
-    if not all(required) or '' in VCENTER_HOSTS:
-        raise ValueError("配置不完整，请检查.env文件中的vCenter信息和策略配置")
+    pass  # 加载时已验证，此处留空
+
+
+if __name__ == "__main__":
+    # 打印全局配置
+    print("\n【全局配置】")
+    print(f"  快照保留天数: {config['SNAPSHOT_RETENTION_DAYS']}")
+    print(f"  过期日期: {config['EXPIRE_DATE']}")
+    print(f"  Excel输出路径: {config['EXCEL_OUTPUT_PATH']}")
+    print(f"  日志文件路径: {config['LOG_FILE_PATH']}")
+    print(f"  禁用SSL验证: {config['DISABLE_SSL_VERIFY']}")
+
+    # 打印管理节点
+    nodes = config['MANAGEMENT_NODES']
+    print(f"\n【管理节点】共 {len(nodes)} 个")
+
+    for i, node in enumerate(nodes, 1):
+        print(f"\n  节点[{i}]:")
+        print(f"    类型: {node.get('type')}")
+        print(f"    名称: {node.get('name')}")
+        print(f"    地址: {node.get('host')}")
+        print(f"    用户: {node.get('user')}")
+        print(f"    密码: {'*' * len(node.get('password', ''))}")
+        # print(f"    密码: {node.get('password', '')}")  # 直接打印出密码
+        print(f"    最大并发删除: {node.get('max_delete_concurrent')}")

core.py

@@ -0,0 +1,2 @@
+def vcenter_connector():
+    return None

core/get_vms.py

@@ -0,0 +1,237 @@
+from pyVmomi import vim
+from pyVim.connect import SmartConnect, Disconnect
+from config.settings import MANAGEMENT_NODES
+from datetime import timedelta
+import pandas as pd
+import openpyxl
+from openpyxl.styles import Border, Side, Font, PatternFill
+from datetime import datetime, timedelta
+
+
+"""计算虚拟机的总磁盘大小（仅是虚拟磁盘的分配空间）"""
+def get_virtual_disk_size(vm):
+    total_size = 0
+    for device in vm.config.hardware.device:
+        if isinstance(device, vim.vm.device.VirtualDisk):
+            # 获取每个虚拟磁盘的容量
+            total_size += device.capacityInKB / (1024 * 1024)  # 转换为GB
+    return total_size
+
+
+"""获取所有虚拟机"""
+def get_all_vms():
+    # 取第一个节点
+    node = MANAGEMENT_NODES[0]
+
+    # 连接vCenter
+    si = SmartConnect(
+        host=node['host'],
+        user=node['user'],
+        pwd=node['password'],
+        disableSslCertValidation=True
+    )
+
+    # 获取所有VM
+    content = si.RetrieveContent()
+    vm_view = content.viewManager.CreateContainerView(
+        content.rootFolder, [vim.VirtualMachine], True
+    )
+
+    vm_list = []
+    for vm in vm_view.view:
+        # 初始化VM信息字典
+        vm_info = {
+            'name': vm.name,
+            'moId': vm._moId,
+            'powerState': str(vm.runtime.powerState),
+            'system': vm.config.guestFullName,
+            'ipAddress': vm.guest.ipAddress,
+            'hostName': vm.guest.hostName,
+            'vmPath': vm.config.files.vmPathName,
+            'Host': vm.runtime.host.name,  # 拿到 Host 主机名
+            'snapshots': [],  # 添加快照信息
+            'diskSpaceGB': get_virtual_disk_size(vm)  # 添加虚拟机占用的磁盘空间
+        }
+
+        # 获取快照信息
+        if vm.snapshot is not None:
+            current_snapshot = vm.snapshot.currentSnapshot
+            root_snapshots = vm.snapshot.rootSnapshotList
+
+            # 处理根快照列表
+            for snapshot in root_snapshots:
+                snapshot_info = get_snapshot_info(snapshot)
+                vm_info['snapshots'].extend(snapshot_info)
+
+            # 添加当前快照ID（如果有）
+            if current_snapshot:
+                vm_info['currentSnapshotId'] = current_snapshot._moId
+        else:
+            vm_info['snapshots'] = None
+            vm_info['currentSnapshotId'] = None
+
+        vm_list.append(vm_info)
+
+    vm_view.Destroy()
+    Disconnect(si)
+
+    print(f"获取到 {len(vm_list)} 台VM")
+    return vm_list
+
+
+"""递归获取快照信息"""
+def get_snapshot_info(snapshot):
+
+    snapshot_list = []
+    # 当前快照信息
+    snapshot_info = {
+        'name': snapshot.name,
+        'description': snapshot.description,
+        'createTime': (snapshot.createTime + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S'),
+        'state': str(snapshot.state),
+        'id': snapshot.id,
+        'moId': snapshot.snapshot._moId,  # 快照的Managed Object ID
+        'sizeMB': snapshot.diskSizeMB if hasattr(snapshot, 'diskSizeMB') else None,  # 快照大小
+        'quiesced': snapshot.quiesced if hasattr(snapshot, 'quiesced') else None,  # 是否静默快照
+        'children': []  # 子快照
+    }
+
+    # 递归处理子快照
+    if snapshot.childSnapshotList:
+        for child in snapshot.childSnapshotList:
+            snapshot_info['children'].extend(get_snapshot_info(child))
+
+    snapshot_list.append(snapshot_info)
+    return snapshot_list
+
+
+def print_snapshot_tree(snapshot_info, level=0):
+    """递归打印快照树（辅助函数）"""
+    indent = "  " * level
+    print(f"{indent}├─ {snapshot_info['name']}")
+    print(f"{indent}│  ├─ 创建时间: {snapshot_info['createTime']}")
+    print(f"{indent}│  ├─ 描述: {snapshot_info['description']}")
+    print(f"{indent}│  ├─ 状态: {snapshot_info['state']}")
+    if snapshot_info['sizeMB']:
+        print(f"{indent}│  ├─ 大小: {snapshot_info['sizeMB']} MB")
+
+    for child in snapshot_info['children']:
+        print_snapshot_tree(child, level + 1)
+
+
+"""表格样式"""
+def style_sheet(sheet, is_old_data=None):
+    """设置工作表样式：列宽、边框、加粗标题、冻结首行和设置背景颜色"""
+    # 定义边框样式
+    thin = Side(border_style="thin", color="000000")
+    border = Border(left=thin, right=thin, top=thin, bottom=thin)
+
+    # 设置列宽和边框
+    for column in sheet.columns:
+        max_length = 0
+        column_letter = column[0].column_letter
+
+        for cell in column:
+            try:
+                if len(str(cell.value)) > max_length:
+                    max_length = len(str(cell.value))
+                cell.border = border  # 为每个单元格添加边框
+            except Exception as e:
+                pass
+
+        adjusted_width = (max_length + 2)
+        sheet.column_dimensions[column_letter].width = adjusted_width
+
+    # 加粗标题并冻结首行
+    for cell in sheet[1]:  # 加粗标题
+        cell.font = Font(bold=True)
+    sheet.freeze_panes = 'A2'  # 冻结首行
+
+    # 设置背景颜色
+    if is_old_data is not None:
+        for row in range(2, len(is_old_data) + 2):
+            if is_old_data[row - 2]:
+                for col in range(1, sheet.max_column + 1):  # 使用 sheet.max_column
+                    cell = sheet.cell(row=row, column=col)
+                    cell.fill = PatternFill(start_color='ADD8E6', end_color='ADD8E6', fill_type='solid')
+
+
+"""将虚拟机和快照信息写入Excel文件，并标记创建时间在15天前的快照"""
+def create_excel_report(vms):
+    vm_data = []
+    snapshot_data = []
+
+    def add_snapshots_to_report(snapshot, vm_name):
+        """递归将快照信息加入报告"""
+        create_time = datetime.strptime(snapshot['createTime'], '%Y-%m-%d %H:%M:%S')
+        is_old = create_time < (datetime.now() - timedelta(days=1))
+
+        snapshot_data.append({
+            'VM Name': vm_name,
+            'Snapshot Name': snapshot['name'],
+            'Description': snapshot['description'],
+            'Create Time': snapshot['createTime'],
+            'State': snapshot['state'],
+            'ID': snapshot['id'],
+            'MO ID': snapshot['moId'],
+            'Size (MB)': snapshot['sizeMB'],
+            'Quiesced': snapshot['quiesced'],
+            'is_old': is_old
+        })
+
+        for child in snapshot['children']:
+            add_snapshots_to_report(child, vm_name)
+
+    for vm in vms:
+        vm_data.append({
+            'VM Name': vm['name'],
+            'MO ID': vm['moId'],
+            'Power State': vm['powerState'],
+            'System': vm['system'],
+            'IP Address': vm['ipAddress'],
+            'Host Name': vm['hostName'],
+            'VM Path': vm['vmPath'],
+            'Host': vm['Host'],
+            'Current Snapshot ID': vm.get('currentSnapshotId', None),
+            'DiskSpace/GB': vm['diskSpaceGB']
+        })
+
+        if vm['snapshots']:
+            for snapshot in vm['snapshots']:
+                add_snapshots_to_report(snapshot, vm['name'])
+
+    vm_df = pd.DataFrame(vm_data)
+    snapshot_df = pd.DataFrame(snapshot_data)
+
+    with pd.ExcelWriter('vm_and_snapshots_report.xlsx', engine='openpyxl') as writer:
+        vm_df.to_excel(writer, sheet_name='VMs', index=False)
+        snapshot_df.to_excel(writer, sheet_name='Snapshots', index=False)
+
+        # 获取写入的工作表
+        workbook = writer.book
+        vm_sheet = workbook['VMs']
+        snapshot_sheet = workbook['Snapshots']
+
+        # 调用样式设置函数，传入 is_old_data
+        style_sheet(vm_sheet)  # 设置 VMs 工作表样式
+        style_sheet(snapshot_sheet, snapshot_df['is_old'].tolist())  # 设置 Snapshots 工作表样式并传入旧数据标识列表
+
+    print("报告已生成: vm_and_snapshots_report.xlsx")
+
+
+
+if __name__ == '__main__':
+    vms = get_all_vms()
+
+    # # 打印示例：显示每个VM的快照信息
+    # for vm in vms:
+    #     print(f"\nVM: {vm['name']}")
+    #     if vm['snapshots']:
+    #         print(f"快照数量: {len(vm['snapshots'])}")
+    #         for snapshot in vm['snapshots']:
+    #             print_snapshot_tree(snapshot)
+    #     else:
+    #         print("无快照")
+
+    # 生成Excel报告
+    create_excel_report(vms)

core/snapshot_collector.py

@@ -0,0 +1,105 @@
+from pyVmomi import vim
+from config.settings import MANAGEMENT_NODES
+from core.vcenter_connector import VCenterManager  # 使用管理器类
+from utils.logger import logger
+from typing import List, Dict
+
+
+def get_all_vms_from_node(si, node_name: str) -> List[Dict]:
+    """
+    从单个vCenter/ESXi获取所有虚拟机
+    :param si: ServiceInstance
+    :param node_name: 节点名称（用于标识来源）
+    :return: VM列表
+    """
+    try:
+        content = si.RetrieveContent()
+        vm_view = content.viewManager.CreateContainerView(
+            content.rootFolder, [vim.VirtualMachine], True
+        )
+        vms = vm_view.view
+        vm_view.Destroy()
+
+        vm_list = []
+        for vm in vms:
+            # 安全获取快照数量
+            num_snapshots = 0
+            if vm.snapshot and vm.snapshot.rootSnapshotList:
+                num_snapshots = len(vm.snapshot.rootSnapshotList)
+
+            vm_info = {
+                'vcenter_node': node_name,  # 来自哪个vCenter/ESXi
+                'vcenter_host': si._stub.host,  # vCenter主机地址
+                'vm_name': vm.name,
+                'vm_moid': vm._moId,
+                'power_state': str(vm.runtime.powerState),  # 转字符串
+                'num_snapshots': num_snapshots,
+                'guest_os': vm.config.guestFullName if vm.config else 'Unknown'
+            }
+            vm_list.append(vm_info)
+
+        logger.info(f"📥 [{node_name}] 获取到 {len(vm_list)} 台VM")
+        return vm_list
+
+    except Exception as e:
+        logger.error(f"❌ [{node_name}] 获取VM列表失败: {str(e)}")
+        return []
+
+
+def get_all_vms_from_all_nodes() -> List[Dict]:
+    """
+    从所有配置的vCenter/ESXi节点获取虚拟机
+    :return: 合并后的VM列表
+    """
+    all_vms = []
+
+    # 使用VCenterManager连接所有节点
+    with VCenterManager() as manager:
+        connections = manager.connect_all()
+
+        if not connections:
+            logger.error("没有可用的vCenter/ESXi连接")
+            return []
+
+        # 遍历每个连接获取VM
+        for conn in connections:
+            vms = get_all_vms_from_node(conn.si, conn.name)
+            all_vms.extend(vms)  # 合并列表
+
+    logger.info(f"📊 总计获取 {len(all_vms)} 台VM（来自 {len(connections)} 个节点）")
+    return all_vms
+
+
+def print_vm_table(vm_list: List[Dict]):
+    """打印VM列表表格（调试用）"""
+    if not vm_list:
+        print("没有获取到VM数据")
+        return
+
+    print(f"\n{'=' * 100}")
+    print(f"{'节点':<15} {'VM名称':<30} {'状态':<12} {'快照数':<8} {'操作系统':<20}")
+    print(f"{'-' * 100}")
+
+    for vm in vm_list:
+        print(f"{vm['vcenter_node']:<15} {vm['vm_name']:<30} "
+              f"{vm['power_state']:<12} {vm['num_snapshots']:<8} "
+              f"{vm['guest_os'][:20]:<20}")
+
+    print(f"{'=' * 100}")
+    print(f"总计: {len(vm_list)} 台VM")
+
+
+# ==================== 主程序入口 ====================
+
+if __name__ == '__main__':
+    # 获取所有VM
+    vm_list = get_all_vms_from_all_nodes()
+
+    # 打印结果
+    print_vm_table(vm_list)
+
+    # 示例：筛选有快照的VM
+    vms_with_snapshots = [vm for vm in vm_list if vm['num_snapshots'] > 0]
+    print(f"\n有快照的VM: {len(vms_with_snapshots)} 台")
+    for vm in vms_with_snapshots:
+        print(f"  - {vm['vm_name']}: {vm['num_snapshots']} 个快照")

core/vcenter_connector.py

@@ -0,0 +1,184 @@
+from pyVim.connect import SmartConnect, Disconnect
+from pyVmomi import vim
+from config.settings import MANAGEMENT_NODES
+from utils.logger import logger
+from dataclasses import dataclass
+from typing import List, Optional, Dict
+
+
+@dataclass
+class Connection:
+    """包装连接对象，包含节点信息和ServiceInstance"""
+    name: str  # 节点名称
+    host: str  # 主机地址
+    node_type: str  # vcenter 或 esxi
+    si: any  # ServiceInstance
+    max_concurrent: int  # 该节点的最大并发数
+
+
+class VCenterManager:
+    """管理多个vCenter/ESXi连接"""
+
+    def __init__(self):
+        # 构造方法，创建对象时自动执行
+        self.connections: List[Connection] = []  # 属性1：成功连接列表
+        self.failed_nodes: List[Dict] = []  # 属性2：失败节点列表
+
+    def connect_all(self) -> List[Connection]:
+        """
+        连接所有配置的管理节点
+        :return: 成功连接的列表
+        """
+        logger.info(f"开始连接 {len(MANAGEMENT_NODES)} 个管理节点...")
+
+        for node in MANAGEMENT_NODES:
+            name = node['name']
+            host = node['host']
+            user = node['user']
+            password = node['password']
+            node_type = node['type']
+            max_concurrent = node.get('max_delete_concurrent', 4)
+
+            logger.info(f"正在连接节点: {name} ({host}) [{node_type}]")
+
+            si = self._connect_single(host, user, password, node_type)
+
+            if si:
+                conn = Connection(
+                    name=name,
+                    host=host,
+                    node_type=node_type,
+                    si=si,
+                    max_concurrent=max_concurrent
+                )
+                self.connections.append(conn)
+                logger.info(f"✅ {name} 连接成功")
+            else:
+                self.failed_nodes.append({
+                    'name': name,
+                    'host': host,
+                    'type': node_type
+                })
+                logger.error(f"❌ {name} 连接失败")
+
+        logger.info(f"连接完成: 成功 {len(self.connections)}/{len(MANAGEMENT_NODES)}")
+        return self.connections
+
+    def _connect_single(self, host: str, user: str, password: str, node_type: str):
+        """连接单个节点"""
+        try:
+            si = SmartConnect(
+                host=host,
+                user=user,
+                pwd=password,
+                disableSslCertValidation=True
+            )
+            return si
+        except Exception as e:
+            logger.error(f"连接 {host} 失败: {str(e)}")
+            return None
+
+    def get_connection(self, name: str = None) -> Optional[Connection]:
+        """
+        获取指定名称的连接，不指定则返回第一个
+        """
+        if not self.connections:
+            return None
+
+        if name:
+            for conn in self.connections:
+                if conn.name == name:
+                    return conn
+            return None
+
+        return self.connections[0]
+
+    def get_all_connections(self) -> List[Connection]:
+        """获取所有成功连接"""
+        return self.connections
+
+    def get_vcenter_connections(self) -> List[Connection]:
+        """仅获取 vcenter 类型的连接"""
+        return [c for c in self.connections if c.node_type == 'vcenter']
+
+    def get_esxi_connections(self) -> List[Connection]:
+        """仅获取 esxi 类型的连接"""
+        return [c for c in self.connections if c.node_type == 'esxi']
+
+    def disconnect_all(self):
+        """关闭所有连接"""
+        for conn in self.connections:
+            try:
+                Disconnect(conn.si)
+                logger.info(f"已断开 {conn.name}")
+            except Exception as e:
+                logger.warning(f"断开 {conn.name} 时出错: {e}")
+        self.connections = []
+
+    def __enter__(self):
+        """上下文管理器支持"""
+        self.connect_all()
+        return self
+
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        """确保退出时断开连接"""
+        self.disconnect_all()
+
+
+# ==================== 使用示例 ====================
+
+def main():
+    # 方式1：手动管理连接
+    manager = VCenterManager()
+    connections = manager.connect_all()
+
+    # 遍历所有连接执行操作
+    for conn in connections:
+        print(f"\n处理节点: {conn.name} ({conn.host})")
+        print(f"最大并发: {conn.max_concurrent}")
+
+        # 获取内容视图
+        content = conn.si.RetrieveContent()
+
+        # 获取所有VM
+        from pyVmomi import vim
+        obj_view = content.viewManager.CreateContainerView(
+            content.rootFolder, [vim.VirtualMachine], True
+        )
+        vms = obj_view.view
+        print(f"该节点有 {len(vms)} 台虚拟机")
+        obj_view.Destroy()
+
+    # 断开所有连接
+    manager.disconnect_all()
+
+
+def main_with_context():
+    # 方式2：使用上下文管理器（推荐，自动断开）
+    with VCenterManager() as manager:
+        # 获取所有vcenter连接
+        vcenters = manager.get_vcenter_connections()
+
+        for conn in vcenters:
+            print(f"处理vCenter: {conn.name}")
+            # 执行操作...
+
+        # 退出时自动断开所有连接
+
+
+def main_single_operation():
+    # 方式3：只连接特定节点
+    manager = VCenterManager()
+    manager.connect_all()
+
+    # 获取指定节点
+    prod_vc = manager.get_connection('vcenter-prod')
+    if prod_vc:
+        print(f"连接到生产环境: {prod_vc.host}")
+        # 执行操作...
+
+    manager.disconnect_all()
+
+
+if __name__ == '__main__':
+    main()

utils/logger.py

@@ -0,0 +1,34 @@
+import logging
+from config.settings import LOG_FILE_PATH
+
+
+def setup_logger():
+    """配置日志系统，返回logger实例"""
+    # 创建logger
+    logger = logging.getLogger('vm_snapshot_cleanup')
+    logger.setLevel(logging.INFO)
+
+    # 避免重复添加处理器
+    if logger.handlers:
+        return logger
+
+    # 定义日志格式
+    formatter = logging.Formatter('%(asctime)s - %(levelname)s - %(message)s')
+
+    # 文件处理器（写入日志文件）
+    file_handler = logging.FileHandler(LOG_FILE_PATH, encoding='utf-8')
+    file_handler.setFormatter(formatter)
+
+    # 控制台处理器（输出到终端）
+    console_handler = logging.StreamHandler()
+    console_handler.setFormatter(formatter)
+
+    # 添加处理器
+    logger.addHandler(file_handler)
+    logger.addHandler(console_handler)
+
+    return logger
+
+
+# 全局logger实例
+logger = setup_logger()

vcenter_connector.py

@@ -1,40 +0,0 @@
-from pyVim.connect import SmartConnect, Disconnect
-from pyVmomi import vim
-from config.settings import VCENTER_USER, VCENTER_PASSWORD
-from utils.logger import logger
-
-
-def connect_to_vcenter(vcenter_host):
-    """
-    连接到指定的vCenter服务器
-    :param vcenter_host: vCenter主机名/IP
-    :return: 成功返回ServiceInstance，失败返回None
-    """
-    try:
-        # 禁用SSL证书验证（生产环境建议启用证书验证）
-        si = SmartConnect(
-            host=vcenter_host,
-            user=VCENTER_USER,
-            pwd=VCENTER_PASSWORD,
-            disableSslCertValidation=True
-        )
-        if not si:
-            logger.error(f"❌ 无法连接到vCenter: {vcenter_host}（无返回实例）")
-            return None
-
-        logger.info(f"✅ 成功连接到vCenter: {vcenter_host}")
-        return si
-
-    except Exception as e:
-        logger.error(f"❌ 连接vCenter {vcenter_host} 失败: {str(e)}")
-        return None
-
-
-def disconnect_from_vcenter(si):
-    """
-    关闭vCenter连接
-    :param si: ServiceInstance实例
-    """
-    if si:
-        Disconnect(si)
-        logger.debug("已关闭vCenter连接")