Get Vms Snapshots and Export Excel correct info.

core/get_vms.py

@@ -1,125 +1,13 @@
 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):
     """设置工作表样式：列宽、边框、加粗标题、冻结首行和设置背景颜色"""
     # 定义边框样式
@@ -156,49 +44,156 @@ def style_sheet(sheet, is_old_data=None):
                     cell.fill = PatternFill(start_color='ADD8E6', end_color='ADD8E6', fill_type='solid')
 
 
+"""获取虚拟机的总磁盘大小（仅是虚拟磁盘的分配空间）"""
+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():
+    vm_list = []
+
+    for node in MANAGEMENT_NODES:
+        # 连接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
+        )
+
+        for vm in vm_view.view:
+            # print(dir(vm))  # 输出所有可用的属性和方法
+            # print(vm.runtime)
+            # print(vm.summary)
+            # print(vm.snapshot)
+            # 初始化VM信息字典
+            # print(vars(vm.summary))
+            # print(vm.summary)
+            vm_info = {
+                'NodeHost': node['host'],
+                '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 主机名，如果是从 Esxi 主机直接获取的可能拿不到正确信息
+                'snapshots': [],  # 添加快照信息
+                'diskSpaceGB': get_virtual_disk_size(vm)  # 添加虚拟机占用的磁盘空间
+            }
+            # print(vm_info)
+            # 获取快照信息
+            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)
+            # print(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'),  # + timedelta(hours=8) 是时间运算，转换为北京时间
+        '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
+
+
 """将虚拟机和快照信息写入Excel文件，并标记创建时间在15天前的快照"""
 def create_excel_report(vms):
     vm_data = []
     snapshot_data = []
 
-    def add_snapshots_to_report(snapshot, vm_name):
-        """递归将快照信息加入报告"""
+    def add_snapshots(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,
+            'NodeHost': vm['NodeHost'],
+            'VMName': vm_name,
             'Snapshot Name': snapshot['name'],
             'Description': snapshot['description'],
-            'Create Time': snapshot['createTime'],
+            'CreateTime': snapshot['createTime'],
             'State': snapshot['state'],
             'ID': snapshot['id'],
-            'MO ID': snapshot['moId'],
-            'Size (MB)': snapshot['sizeMB'],
+            'MOID': 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)
+            add_snapshots(child, vm_name)
 
     for vm in vms:
+        # print(vm['NodeHost'])
         vm_data.append({
-            'VM Name': vm['name'],
-            'MO ID': vm['moId'],
-            'Power State': vm['powerState'],
+            'NodeHost': vm['NodeHost'],
+            'VMName': vm['name'],
+            'MOID': vm['moId'],
+            'PowerState': vm['powerState'],
             'System': vm['system'],
-            'IP Address': vm['ipAddress'],
-            'Host Name': vm['hostName'],
-            'VM Path': vm['vmPath'],
+            'IPAddress': vm['ipAddress'],
+            'HostName': vm['hostName'],
+            'CurrentSnapshotID': vm.get('currentSnapshotId', None),
+            'DiskSpace/GB': vm['diskSpaceGB'],
             'Host': vm['Host'],
-            'Current Snapshot ID': vm.get('currentSnapshotId', None),
-            'DiskSpace/GB': vm['diskSpaceGB']
+            'VMPath': vm['vmPath']
         })
-
         if vm['snapshots']:
+            # print(vm['snapshots'])
             for snapshot in vm['snapshots']:
-                add_snapshots_to_report(snapshot, vm['name'])
+                add_snapshots(snapshot, vm['name'])
 
     vm_df = pd.DataFrame(vm_data)
     snapshot_df = pd.DataFrame(snapshot_data)
@@ -222,16 +217,7 @@ def create_excel_report(vms):
 
 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("无快照")
+    # print(vms)
 
     # 生成Excel报告
     create_excel_report(vms)