VMware 与 Hyper-V 基础扫盲：差异、迁移与脚本实战

import os
import subprocess
import ctypes
import sys
import re
from pathlib import Path

def is_admin():
    try:
        return ctypes.windll.shell32.IsUserAnAdmin()
    except:
        return False

def check_dependencies():
    print("[*] 正在检查依赖环境...")
    # 检查 qemu-img
    try:
        subprocess.run(['qemu-img', '--version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, check=True)
        print("[+] qemu-img 已安装，状态正常。")
    except FileNotFoundError:
        print("[-] 错误：找不到 'qemu-img' 命令。")
        print("    请确保您已经安装了 QEMU 并将其添加到了系统的 PATH 环境变量中。")
        print("    下载地址: https://qemu.weilnetz.de/w64/")
        sys.exit(1)
    
    # 检查管理员权限
    if not is_admin():
        print("[!] 警告：当前未以管理员权限运行！")
        print("    执行 `fsutil sparse setflag` 命令可能需要管理员权限。")
        print("    建议您以管理员身份重新运行此终端/控制台，或者忽略此警告继续。")
        choice = input("    是否继续运行？(y/n) [n]: ").strip().lower()
        if choice != 'y':
            sys.exit(0)
    else:
        print("[+] 管理员权限检查通过。")

def find_vmdk_files():
    print("\n[*] 正在全盘深度扫描存在的虚拟机配置文件 (.vmx) ... (这可能需要几分钟，请耐心等待)")
    drives = [chr(x) + ':\\' for x in range(ord('C'), ord('Z')+1) if os.path.exists(chr(x) + ':\\')]
    
    vmdk_targets = []
    
    # 我们首选寻找 .vmx 配置文件，从而定位主磁盘，这样可以避免被数百个拆分文件淹没
    for drive in drives:
        print(f"    正在扫描 {drive} 驱动器...")
        # 排除一些系统和缓存目录加速搜索
        exclude_dirs = ['$Recycle.Bin', 'Windows', 'Program Files', 'Program Files (x86)', 'ProgramData', 'AppData']
        
        for root, dirs, files in os.walk(drive):
            dirs[:] = [d for d in dirs if d not in exclude_dirs]
            
            for file in files:
                if file.endswith('.vmx'):
                    vmx_path = os.path.join(root, file)
                    try:
                        with open(vmx_path, 'r', encoding='utf-8', errors='ignore') as f:
                            content = f.read()
                            # 尝试解析 scsi0:0.fileName 等常见的磁盘挂载点
                            match = re.search(r'fileName\\s*=\\s*"([^"]+\\.vmdk)"', content)
                            if match:
                                vmdk_name = match.group(1)
                                full_vmdk_path = os.path.join(root, vmdk_name)
                                full_vmdk_path = os.path.normpath(full_vmdk_path)
                                if os.path.exists(full_vmdk_path):
                                    vmdk_targets.append(full_vmdk_path)
                    except Exception as e:
                        pass
                        
                # 同时也寻找一下独立的 vmdk，但是要排除雷电模拟器和分块文件 (-s001.vmdk)
                if file.endswith('.vmdk'):
                    # 过滤掉常见的拆分文件特征 (-s001.vmdk, -f001.vmdk, -flat.vmdk 等)
                    if re.search(r'-(s\\d{3}|f\\d{3}|flat|\\d+)\\.vmdk$', file.lower()):
                        continue
                    if 'ldplayer' in root.lower() or 'leidian' in root.lower():
                        continue
                        
                    full_vmdk_path = os.path.join(root, file)
                    # 去重，如果前面通过 vmx 已经找到了这个，就不再添加
                    if full_vmdk_path not in vmdk_targets:
                         vmdk_targets.append(full_vmdk_path)

    # 去重
    vmdk_targets = list(set(vmdk_targets))
    return vmdk_targets

def convert_vmdk(vmdk_path):
    print(f"\n[{'='*50}]")
    print(f"[*] 开始处理: {vmdk_path}")
    
    vmdk_path_obj = Path(vmdk_path)
    base_dir = vmdk_path_obj.parent
    base_name = vmdk_path_obj.stem
    
    vhdx_name = f"{base_name}.vhdx"
    vhdx_path = base_dir / vhdx_name
    
    # 冲突处理：如果 vhdx 已存在，添加自动编号
    counter = 1
    while vhdx_path.exists():
        vhdx_name = f"{base_name}_{counter}.vhdx"
        vhdx_path = base_dir / vhdx_name
        counter += 1
        
    print(f"[*] 目标输出: {vhdx_path}")
    
    # 第一步：转换磁盘格式 (qemu-img)
    cmd_convert = ['qemu-img', 'convert', '-p', '-f', 'vmdk', '-O', 'vhdx', str(vmdk_path), str(vhdx_path)]
    print(f"[*] [1/2] 正在进行动态扩展格式转换 (这可能需要较长时间，请不要关闭窗口)...")
    print(f"    > {' '.join(cmd_convert)}")
    
    try:
        # 使用 stdout=None 可以直接将 qemu-img 的进度输出到控制台
        subprocess.run(cmd_convert, check=True)
        print("[+] 格式转换完成！")
    except subprocess.CalledProcessError as e:
        print(f"[-] qemu-img 转换失败。错误码: {e.returncode}")
        # 如果转换失败了，没必要继续执行 fsutil
        return False
    except FileNotFoundError:
        print("[-] 系统中未找到 qemu-img 命令。")
        return False
        
    # 第二步：去除稀疏属性 (fsutil sparse setflag)
    cmd_fsutil = ['fsutil', 'sparse', 'setflag', str(vhdx_path), '0']
    print(f"[*] [2/2] 正在优化磁盘属性 (移除稀疏标记)...")
    print(f"    > {' '.join(cmd_fsutil)}")
    
    try:
        result = subprocess.run(cmd_fsutil, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
        if result.returncode == 0:
            print("[+] 优化成功！已成功去除稀疏属性标记。")
        else:
             print(f"[-] fsutil 执行遇到问题 (可能是由于权限不足或不支持该特性):")
             print(f"    输出信息: {result.stderr.strip()}")
    except FileNotFoundError:
        print("[-] 未在系统中找到 fsutil 命令，跳过属性修改环节。")

    print(f"[+] '{base_name}' 的全部迁移步骤完成！")
    print(f"    准备就绪的 Hyper-V 磁盘文件位于: {vhdx_path}")
    print(f"[{'='*50}]")
    return True


def main():
    print("=====================================================")
    print("      VMware (VMDK) 到 Hyper-V (VHDX) 智能迁移工具      ")
    print("=====================================================\n")
    
    check_dependencies()
    
    print("\n请选择您的操作模式：")
    print(" 1. 自动全盘扫描您本地的 VMware 虚拟机 (推荐)")
    print(" 2. 手动指定一个 .vmdk 文件路径")
    
    try:
        mode = input("\n请输入您的选择 [1]: ").strip()
    except KeyboardInterrupt:
        sys.exit(0)
        
    if mode == '2':
        manual_path = input("请输入 .vmdk 文件的绝对路径（可以包含引号）: ").strip()
        manual_path = manual_path.strip('"').strip("'")
        if os.path.exists(manual_path) and manual_path.lower().endswith('.vmdk'):
            convert_vmdk(manual_path)
        else:
            print("[-] 错误：文件不存在或不是 .vmdk 文件。")
            
    else:
        vmdks = find_vmdk_files()
        
        if not vmdks:
            print("\n[-] 没有在您的磁盘上找到可识别的、主要的 .vmdk 文件。")
            print("    您可以尝试直接使用模式 2 手动输入路径来进行转换。")
            sys.exit(0)
            
        print("\n\n[*] 扫描完成！找到以下可转换的虚拟磁盘（已排除碎片文件）：")
        print("-" * 70)
        for i, vmdk in enumerate(vmdks, 1):
            file_size_gb = os.path.getsize(vmdk) / (1024**3)
            print(f" [{i}] {os.path.basename(vmdk)} (主索引大小/结构大小: {file_size_gb:.4f} GB)")
            print(f"     路径: {vmdk}")
            print("-" * 70)
            
        print("\n请输入您要转换的磁盘序号。")
        print(" (支持多选，用逗号分隔，例如：1, 3, 5。 输入 'all' 转换全部，输入 'q' 退出)")
        
        try:
            choice = input("\n您的选择: ").strip().lower()
        except KeyboardInterrupt:
            sys.exit(0)
            
        if choice == 'q' or not choice:
            print("已取消退出。")
            sys.exit(0)
            
        selected_indexes = []
        if choice == 'all':
            selected_indexes = list(range(1, len(vmdks) + 1))
        else:
            try:
                # 解析用户输入的数字列表，如 '1,2,3' -> [1, 2, 3]
                parts = choice.split(',')
                for p in parts:
                    idx = int(p.strip())
                    if 1 <= idx <= len(vmdks):
                        selected_indexes.append(idx)
            except ValueError:
                print("[-] 您的输入格式有误，请重新运行脚本。")
                sys.exit(1)
                
        if not selected_indexes:
            print("[-] 未选择任何有效的序号。程序退出。")
            sys.exit(1)
            
        print(f"\n[*] 您选择了 {len(selected_indexes)} 个虚拟机进行排队转换。")
        for idx in selected_indexes:
            target_vmdk = vmdks[idx - 1]
            convert_vmdk(target_vmdk)
            
    print("\n[+] 所有指定的任务已运行完毕。")