接近完善的代码，待生产环境测试。
This commit is contained in:
panjunlan
@@ -1,101 +1,80 @@
|
||||
import pandas as pd
|
||||
import yaml
|
||||
from pyVmomi import vim
|
||||
import pandas as pd
|
||||
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, YAML_OUTPUT_PATH
|
||||
from config.settings import EXCEL_OUTPUT_PATH, YAML_OUTPUT_PATH, SNAPSHOT_RETENTION_DAYS
|
||||
from core.get_vm_snapshots import get_all_vms
|
||||
|
||||
# 输出数据到 Excel 文件
|
||||
def create_excel_report(vms):
|
||||
vm_data, snapshot_data, old_snapshots = [], [], []
|
||||
for vm in vms:
|
||||
vm_data.append({
|
||||
'NodeHost': vm['NodeHost'],
|
||||
'VMName': vm['name'],
|
||||
'MOID': vm['moId'],
|
||||
'PowerState': vm['powerState'],
|
||||
'System': vm['system'],
|
||||
'IPAddress': vm['ipAddress'],
|
||||
'HostName': vm['hostName'],
|
||||
'CurrentSnapshotID': vm.get('currentSnapshotId'),
|
||||
'DiskSpace/GB': vm['diskSpaceGB'],
|
||||
'createDate': vm['createDate'], # 虚拟机的创建时间
|
||||
'bootTime': vm['bootTime'], # 虚拟机上次启动的时间
|
||||
'Host': vm['Host'],
|
||||
'VMPath': vm['vmPath']
|
||||
})
|
||||
|
||||
if vm['snapshots']:
|
||||
for snapshot in vm['snapshots']:
|
||||
collect_snapshot_data(snapshot, vm, snapshot_data, old_snapshots)
|
||||
|
||||
vm_df = pd.DataFrame(vm_data)
|
||||
snapshot_df = pd.DataFrame(snapshot_data)
|
||||
logger.info(f"总共有 {len(snapshot_data)} 个快照")
|
||||
with pd.ExcelWriter(EXCEL_OUTPUT_PATH, 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
|
||||
style_sheet(workbook['VMs'])
|
||||
# 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())
|
||||
|
||||
logger.debug(f"Excel 文件已生成: {EXCEL_OUTPUT_PATH}")
|
||||
# 返回旧快照的数据，用于生成 Yaml 数据文件
|
||||
return old_snapshots
|
||||
|
||||
|
||||
def get_all_vms():
|
||||
""" 主函数，负责流程控制"""
|
||||
vm_list = []
|
||||
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))
|
||||
|
||||
for node in MANAGEMENT_NODES:
|
||||
try:
|
||||
si = SmartConnect(
|
||||
host=node['host'],
|
||||
user=node['user'],
|
||||
pwd=node['password'],
|
||||
disableSslCertValidation=True
|
||||
)
|
||||
|
||||
content = si.RetrieveContent()
|
||||
vm_view = content.viewManager.CreateContainerView(
|
||||
content.rootFolder, [vim.VirtualMachine], True
|
||||
)
|
||||
|
||||
for vm in vm_view.view:
|
||||
vm_info = build_vm_info(vm, node['host'])
|
||||
vm_list.append(vm_info)
|
||||
|
||||
except vim.fault.InvalidLogin as e:
|
||||
logger.info(f"登录 {node['host']} 失败，请检查用户名和密码：{e.msg}")
|
||||
except Exception as e:
|
||||
logger.error(f"无法连接到 {node['host']}：{e}")
|
||||
finally:
|
||||
if 'si' in locals(): # 确保 si 是定义过的
|
||||
Disconnect(si) # 确保连接被断开
|
||||
|
||||
print(f"获取到 {len(vm_list)} 台VM")
|
||||
return vm_list
|
||||
|
||||
|
||||
def build_vm_info(vm, node_host):
|
||||
"""构造虚拟机信息字典"""
|
||||
vm_info = {
|
||||
'NodeHost': node_host,
|
||||
'name': vm.name,
|
||||
'moId': vm._moId,
|
||||
'powerState': vm.runtime.powerState,
|
||||
'system': vm.config.guestFullName,
|
||||
'ipAddress': vm.guest.ipAddress,
|
||||
'hostName': vm.guest.hostName,
|
||||
'diskSpaceGB': get_virtual_disk_size(vm),
|
||||
'createDate': (vm.config.createDate + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S') if vm.runtime.bootTime else None, # 虚拟机的创建时间
|
||||
'bootTime': (vm.runtime.bootTime + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S') if vm.runtime.bootTime else None, # 虚拟机上次启动的时间 'createDate': (vm.config.createDate + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S'), # 虚拟机的创建时间
|
||||
# 'createDate': vm.config.createDate, # 虚拟机的创建时间
|
||||
# 'bootTime': vm.runtime.bootTime, # 虚拟机上次启动的时间
|
||||
'snapshots': [],
|
||||
'Host': vm.runtime.host.name,
|
||||
'vmPath': vm.config.files.vmPathName
|
||||
}
|
||||
# 判断虚拟机是否有快照，如果有就构造快照结构并记录当前快照ID；如果没有就设为 None。
|
||||
if vm.snapshot:
|
||||
current_snapshot = vm.snapshot.currentSnapshot
|
||||
root_snapshots = vm.snapshot.rootSnapshotList
|
||||
|
||||
for snapshot in root_snapshots:
|
||||
vm_info['snapshots'].extend(build_snapshot_dict(snapshot))
|
||||
|
||||
vm_info['currentSnapshotId'] = (
|
||||
current_snapshot._moId if current_snapshot else None
|
||||
)
|
||||
else:
|
||||
vm_info['snapshots'] = None
|
||||
vm_info['currentSnapshotId'] = None
|
||||
|
||||
return vm_info
|
||||
|
||||
def build_snapshot_dict(snapshot):
|
||||
"""递归构造快照结构"""
|
||||
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,
|
||||
'quiesced': getattr(snapshot, 'quiesced', None),
|
||||
'children': []
|
||||
'NodeHost': vm['NodeHost'],
|
||||
'VMName': vm['name'],
|
||||
'Snapshot Name': snapshot['name'],
|
||||
'Description': snapshot['description'],
|
||||
'CreateTime': snapshot['createTime'],
|
||||
'State': snapshot['state'],
|
||||
'ID': snapshot['id'],
|
||||
'MOID': snapshot['moId'],
|
||||
'Quiesced': snapshot['quiesced'],
|
||||
'is_old': is_old
|
||||
}
|
||||
|
||||
if snapshot.childSnapshotList:
|
||||
for child in snapshot.childSnapshotList:
|
||||
snapshot_info['children'].extend(build_snapshot_dict(child))
|
||||
# 如果是旧快照，添加到旧快照列表
|
||||
if is_old:
|
||||
old_snapshots.append(snapshot_info)
|
||||
|
||||
return [snapshot_info]
|
||||
snapshot_data.append(snapshot_info)
|
||||
|
||||
for child in snapshot['children']:
|
||||
collect_snapshot_data(child, vm, snapshot_data, old_snapshots)
|
||||
|
||||
|
||||
"""设置表格样式"""
|
||||
@@ -135,104 +114,17 @@ 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 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_info = {
|
||||
'NodeHost': vm['NodeHost'],
|
||||
'VMName': vm['name'],
|
||||
'Snapshot Name': snapshot['name'],
|
||||
'Description': snapshot['description'],
|
||||
'CreateTime': snapshot['createTime'],
|
||||
'State': snapshot['state'],
|
||||
'ID': snapshot['id'],
|
||||
'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, old_snapshots)
|
||||
|
||||
|
||||
# 输出数据到 Excel 文件
|
||||
def create_excel_report(vms):
|
||||
vm_data = []
|
||||
snapshot_data = []
|
||||
old_snapshots = [] # 用于存储旧快照的信息
|
||||
|
||||
for vm in vms:
|
||||
vm_data.append({
|
||||
'NodeHost': vm['NodeHost'],
|
||||
'VMName': vm['name'],
|
||||
'MOID': vm['moId'],
|
||||
'PowerState': vm['powerState'],
|
||||
'System': vm['system'],
|
||||
'IPAddress': vm['ipAddress'],
|
||||
'HostName': vm['hostName'],
|
||||
'CurrentSnapshotID': vm.get('currentSnapshotId'),
|
||||
'DiskSpace/GB': vm['diskSpaceGB'],
|
||||
'createDate': vm['createDate'], # 虚拟机的创建时间
|
||||
'bootTime': vm['bootTime'], # 虚拟机上次启动的时间
|
||||
'Host': vm['Host'],
|
||||
'VMPath': vm['vmPath']
|
||||
})
|
||||
|
||||
if vm['snapshots']:
|
||||
for snapshot in vm['snapshots']:
|
||||
collect_snapshot_data(snapshot, vm, snapshot_data, old_snapshots)
|
||||
|
||||
vm_df = pd.DataFrame(vm_data)
|
||||
snapshot_df = pd.DataFrame(snapshot_data)
|
||||
|
||||
with pd.ExcelWriter(EXCEL_OUTPUT_PATH, 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
|
||||
style_sheet(workbook['VMs'])
|
||||
# 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())
|
||||
|
||||
logger.debug(f"Excel 文件已生成: {EXCEL_OUTPUT_PATH}")
|
||||
# 返回旧快照的数据
|
||||
return old_snapshots
|
||||
|
||||
|
||||
# 输出待删除的旧快照到 YAML 文件
|
||||
def export_yaml(old_snapshots):
|
||||
print(old_snapshots)
|
||||
|
||||
logger.info(f"可删除的快照有 {len(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)
|
||||
old_snapshots = create_excel_report(vms) # 生成Excel报告并获取旧快照
|
||||
vms = get_all_vms() # 导出 excel 和 yaml 文件，需要先获取虚拟机信息
|
||||
old_snapshots = create_excel_report(vms) # 生成 Excel 报告并获取旧快照
|
||||
export_yaml(old_snapshots)
|
||||
# print(old_snapshots)
|
||||
# print(old_snapshots)
|
||||
100
core/get_vm_snapshots.py
Normal file
100
core/get_vm_snapshots.py
Normal file
@@ -0,0 +1,100 @@
|
||||
from pyVmomi import vim
|
||||
from datetime import timedelta
|
||||
from pyVim.connect import Disconnect
|
||||
from utils.logger import logger
|
||||
from core.vm_connector import connect_vcenter
|
||||
from config.settings import MANAGEMENT_NODES
|
||||
|
||||
|
||||
def get_all_vms():
|
||||
"""主函数，负责流程控制 """
|
||||
vm_list = []
|
||||
for node in MANAGEMENT_NODES:
|
||||
si, vm_view = None, None # 初始两个变量值
|
||||
try:
|
||||
# 1. 连接节点（核心逻辑不变）
|
||||
si = connect_vcenter(node['host'])
|
||||
content = si.RetrieveContent()
|
||||
vm_view = content.viewManager.CreateContainerView(
|
||||
content.rootFolder, [vim.VirtualMachine], True
|
||||
)
|
||||
# 2. 收集VM信息
|
||||
for vm in vm_view.view:
|
||||
vm_list.append(build_vm_info(vm, node['host']))
|
||||
except Exception as e: # 记录错误，跳过当前节点
|
||||
logger.error(f"处理节点 {node['host']} 失败：{e}")
|
||||
finally:
|
||||
# 无论成败，销毁视图+断开连接
|
||||
if vm_view:
|
||||
vm_view.Destroy()
|
||||
if si:
|
||||
Disconnect(si)
|
||||
logger.info(f"获取到 {len(vm_list)} 台虚拟机")
|
||||
return vm_list
|
||||
|
||||
|
||||
def build_vm_info(vm, node_host):
|
||||
"""构造虚拟机信息字典"""
|
||||
vm_info = {
|
||||
'NodeHost': node_host,
|
||||
'name': vm.name,
|
||||
'moId': vm._moId,
|
||||
'powerState': vm.runtime.powerState,
|
||||
'system': vm.config.guestFullName,
|
||||
'ipAddress': vm.guest.ipAddress,
|
||||
'hostName': vm.guest.hostName,
|
||||
'diskSpaceGB': get_virtual_disk_size(vm),
|
||||
'createDate': (vm.config.createDate + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S') if vm.runtime.bootTime else None, # 虚拟机的创建时间
|
||||
'bootTime': (vm.runtime.bootTime + timedelta(hours=8)).strftime('%Y-%m-%d %H:%M:%S') if vm.runtime.bootTime else None, # 虚拟机上次启动的时间
|
||||
'snapshots': [],
|
||||
'Host': vm.runtime.host.name,
|
||||
'vmPath': vm.config.files.vmPathName
|
||||
}
|
||||
# 判断虚拟机是否有快照，如果有就构造快照结构并记录当前快照ID；如果没有就设为 None。
|
||||
if vm.snapshot:
|
||||
current_snapshot = vm.snapshot.currentSnapshot
|
||||
root_snapshots = vm.snapshot.rootSnapshotList
|
||||
|
||||
for snapshot in root_snapshots:
|
||||
vm_info['snapshots'].extend(build_snapshot_dict(snapshot))
|
||||
|
||||
vm_info['currentSnapshotId'] = (
|
||||
current_snapshot._moId if current_snapshot else None
|
||||
)
|
||||
else:
|
||||
vm_info['snapshots'] = None
|
||||
vm_info['currentSnapshotId'] = None
|
||||
return vm_info
|
||||
|
||||
|
||||
def build_snapshot_dict(snapshot):
|
||||
"""递归构造快照结构"""
|
||||
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,
|
||||
'quiesced': getattr(snapshot, 'quiesced', None),
|
||||
'children': []
|
||||
}
|
||||
if snapshot.childSnapshotList:
|
||||
for child in snapshot.childSnapshotList:
|
||||
snapshot_info['children'].extend(build_snapshot_dict(child))
|
||||
return [snapshot_info]
|
||||
|
||||
|
||||
"""获取虚拟机的总磁盘大小（仅是虚拟磁盘的分配空间）"""
|
||||
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
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
vms = get_all_vms() # 主函数入口，获取虚拟机信息
|
||||
# print(vms)
|
||||
@@ -1,25 +1,22 @@
|
||||
import yaml
|
||||
import os, time
|
||||
import os, time, yaml
|
||||
from pyVmomi import vim
|
||||
from pyVim.connect import SmartConnect, Disconnect
|
||||
from config.settings import MANAGEMENT_NODES,YAML_OUTPUT_PATH
|
||||
from pyVim.connect import Disconnect
|
||||
from core.vm_connector import connect_vcenter
|
||||
from config.settings import YAML_OUTPUT_PATH
|
||||
from utils.logger import logger
|
||||
|
||||
|
||||
def connect_vcenter(host):
|
||||
"""根据 NodeHost 连接对应 vCenter"""
|
||||
for node in MANAGEMENT_NODES:
|
||||
if node['host'] == host:
|
||||
return SmartConnect(
|
||||
host=node['host'],
|
||||
user=node['user'],
|
||||
pwd=node['password'],
|
||||
disableSslCertValidation=True
|
||||
)
|
||||
raise Exception(f"未找到 {host} 的连接信息")
|
||||
def load_old_snapshots(file_path):
|
||||
"""从 YAML 文件中加载旧快照"""
|
||||
if os.path.exists(file_path):
|
||||
with open(file_path, 'r', encoding='utf-8') as yaml_file:
|
||||
return yaml.safe_load(yaml_file)
|
||||
else:
|
||||
logger.error(f"{file_path} 文件不存在.")
|
||||
return []
|
||||
|
||||
|
||||
def main(dele_snapshots):
|
||||
def remove_snapshot(dele_snapshots):
|
||||
"""根据 YAML 信息删除旧快照"""
|
||||
# print(dele_snapshots)
|
||||
if not dele_snapshots:
|
||||
@@ -31,52 +28,55 @@ def main(dele_snapshots):
|
||||
for snapshot in dele_snapshots:
|
||||
if snapshot['is_old']:
|
||||
grouped.setdefault(snapshot['NodeHost'], []).append(snapshot)
|
||||
|
||||
# print(grouped)
|
||||
for host, snapshots in grouped.items():
|
||||
logger.info(f"连接到 {host} 删除快照...")
|
||||
si = connect_vcenter(host)
|
||||
|
||||
logger.info(f"连接 {host} 进行删除快照...")
|
||||
try:
|
||||
for snapshot in snapshots:
|
||||
delete_snapshot(si, snapshot) # 调用删除快照函数
|
||||
finally:
|
||||
Disconnect(si)
|
||||
si = connect_vcenter(host)
|
||||
try:
|
||||
for snapshot in snapshots:
|
||||
# print(snapshot)
|
||||
delete_snapshot(si, snapshot) # 调用删除快照函数
|
||||
finally:
|
||||
Disconnect(si)
|
||||
except Exception as e:
|
||||
logger.error(f"快照删除失败，因为连接 {host} 失败：{e}" )
|
||||
|
||||
|
||||
# 执行快照删除的（核心）函数
|
||||
def delete_snapshot(si, snapshot_info):
|
||||
"""执行快照删除"""
|
||||
""" 执行快照删除 """
|
||||
content = si.RetrieveContent()
|
||||
snap_name = f"{snapshot_info['VMName']}-{snapshot_info['Snapshot Name']}-({snapshot_info['MOID']})"
|
||||
|
||||
vm = find_vm_by_name(content, snapshot_info['VMName']) # VMName即根据获取到的虚拟机名称查找虚拟机是否存放
|
||||
vm = find_vm_by_name(content, snapshot_info['VMName']) # 根据快照名，查找出相应的虚拟机
|
||||
# print(snap_name,vm)
|
||||
|
||||
if not vm:
|
||||
logger.info(f"未找到 VM: {snapshot_info['VMName']}")
|
||||
return
|
||||
# 检查该虚拟机是否有快照
|
||||
if not vm.snapshot:
|
||||
logger.warning(snap_name,"：快照不存在")
|
||||
logger.warning(f"{snap_name}：快照不存在")
|
||||
return
|
||||
|
||||
# 虚拟机的快照列表中找到具有指定 MOID 的快照对象
|
||||
# 调用函数获取虚拟机快照的 MOID 信息
|
||||
snapshot_obj = find_snapshot_by_moid(
|
||||
vm.snapshot.rootSnapshotList,
|
||||
snapshot_info['MOID']
|
||||
)
|
||||
# print(snapshot_obj)
|
||||
if not snapshot_obj:
|
||||
logger.warning(snap_name,"：未找到")
|
||||
return
|
||||
|
||||
logger.info(f"正在删除 Snapshot: {snap_name}")
|
||||
try:
|
||||
|
||||
"""删除快照核心代码，调用快照对象的 RemoveSnapshot_Task 方法执行。removeChildren = False：表示删除该快照时不删除其子快照。"""
|
||||
task = snapshot_obj.RemoveSnapshot_Task(removeChildren=False)
|
||||
logger.info(f"正在删除 Snapshot: {snap_name}")
|
||||
|
||||
task = snapshot_obj.RemoveSnapshot_Task(removeChildren=False)
|
||||
# 改进的等待逻辑
|
||||
while task.info.state in [vim.TaskInfo.State.running, vim.TaskInfo.State.queued]:
|
||||
time.sleep(1) # 避免 CPU 空转，每秒检查一次
|
||||
time.sleep(1) # 每秒检查一次
|
||||
|
||||
# 更完整的状态判断
|
||||
if task.info.state == vim.TaskInfo.State.success:
|
||||
@@ -94,8 +94,8 @@ def delete_snapshot(si, snapshot_info):
|
||||
raise
|
||||
|
||||
|
||||
"""根据 VM 名称查找虚拟机"""
|
||||
def find_vm_by_name(content, vm_name):
|
||||
"""根据 VM 名称查找虚拟机对象"""
|
||||
container = content.viewManager.CreateContainerView(
|
||||
content.rootFolder, [vim.VirtualMachine], True
|
||||
)
|
||||
@@ -109,8 +109,8 @@ def find_vm_by_name(content, vm_name):
|
||||
return None
|
||||
|
||||
|
||||
""" 递归查找 snapshot 对象 """
|
||||
def find_snapshot_by_moid(snapshot_tree, moid):
|
||||
"""递归查找 snapshot 对象"""
|
||||
for snapshot in snapshot_tree:
|
||||
if snapshot.snapshot._moId == moid:
|
||||
return snapshot.snapshot
|
||||
@@ -122,16 +122,6 @@ def find_snapshot_by_moid(snapshot_tree, moid):
|
||||
return None
|
||||
|
||||
|
||||
def load_old_snapshots(file_path):
|
||||
"""从 YAML 文件中加载旧快照"""
|
||||
if os.path.exists(file_path):
|
||||
with open(file_path, 'r', encoding='utf-8') as yaml_file:
|
||||
return yaml.safe_load(yaml_file)
|
||||
else:
|
||||
logger.error(f"{file_path} 文件不存在.")
|
||||
return []
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
old_snapshots = load_old_snapshots(YAML_OUTPUT_PATH)
|
||||
main(old_snapshots)
|
||||
old_snapshots = load_old_snapshots(YAML_OUTPUT_PATH) # 获取待删除的快照信息
|
||||
remove_snapshot(old_snapshots)
|
||||
36
core/vm_connector.py
Normal file
36
core/vm_connector.py
Normal file
@@ -0,0 +1,36 @@
|
||||
import ssl
|
||||
from pyVim.connect import SmartConnect, Disconnect
|
||||
from config.settings import MANAGEMENT_NODES
|
||||
from utils.logger import logger # 复用你之前的日志配置
|
||||
|
||||
|
||||
def connect_vcenter(host):
|
||||
"""
|
||||
根据NodeHost连接对应vCenter/ESXi
|
||||
:param host: 节点IP/主机名
|
||||
:return: ServiceInstance对象
|
||||
:raise: Exception（连接失败/未找到配置）
|
||||
"""
|
||||
# 遍历全局配置的MANAGEMENT_NODES
|
||||
for node in MANAGEMENT_NODES:
|
||||
if node['host'] == host:
|
||||
context = ssl._create_unverified_context() # 禁用SSL验证
|
||||
si = SmartConnect(
|
||||
host=node['host'],
|
||||
user=node['user'],
|
||||
pwd=node['password'],
|
||||
sslContext=context # 规范的SSL配置
|
||||
)
|
||||
if not si:
|
||||
raise Exception(f"{host}连接成功但未返回ServiceInstance")
|
||||
logger.info(f"成功连接到节点: {host}")
|
||||
return si
|
||||
|
||||
# 匹配不到 host 主动抛异常
|
||||
logger.error(f"未找到节点 {host} 的连接信息")
|
||||
raise Exception(f"未找到 {host} 的连接信息")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
si = connect_vcenter(MANAGEMENT_NODES[0]['host'])
|
||||
print(si)
|
||||
Reference in New Issue
Block a user