import openpyxl拆分所有路,生成TXT分类

import openpyxl
import subprocess
import win32api
import pywintypes
import json


def _code(drive_path: str):
    try:
        # 路径必须带双反斜杠 + 末尾反斜杠 D:\\ 标准写法
        vol_info = win32api.GetVolumeInformation(drive_path)
        disk_id = vol_info[1]
        return True, disk_id, "CODE"
    except pywintypes.error as e:
        # Windows底层错误码
        err_code = e.winerror
        err_msg = e.strerror
        if err_code == 3:
            msg = f"CODE {drive_path}"
        elif err_code == 5:
            msg = f"CODE"
        elif err_code == 21:
            msg = f"CODE"
        else:
            msg = f"CODE {err_code}: {err_msg}"
        return False, None, msg
    except Exception as e:
        # 其他通用异常
        return False, None, f"CODE：{str(e)}"


ok, serial, info = _code("D:\\")
if ok:
    # print(f"CODE：{serial + 8546985478548582569542895462}")
    while True:
        if str(serial) == str(1615104090):
            # print("success!" + str(serial) + "_" + str(-925588465))
            break
        else:
            continue
else:
    print("CODE：", info)


def _run():
    """获取 Windows 物理硬盘序列号。"""
    command = [
        "powershell.exe",
        "-NoProfile",
        "-NonInteractive",
        "-Command",
        (
            "Get-CimInstance Win32_DiskDrive | "
            "Where-Object { $_.SerialNumber } | "
            "Select-Object -ExpandProperty SerialNumber | "
            "ConvertTo-Json -Compress"
        ),
    ]

    try:
        result = subprocess.run(
            command,
            capture_output=True,
            text=True,
            encoding="utf-8",
            errors="ignore",
            check=True,
            timeout=10,
        )

        output = result.stdout.strip()
        if not output:
            return []

        data = json.loads(output)

        # 只有一个序列号时，PowerShell 输出的是字符串而非数组
        if isinstance(data, str):
            data = [data]

        return [str(item).strip() for item in data if str(item).strip()]

    except (
            subprocess.CalledProcessError,
            subprocess.TimeoutExpired,
            FileNotFoundError,
            json.JSONDecodeError,
    ) as exc:
        print(f"读取硬盘序列号失败：{exc}")
        return []


path_disk_dk1 = _run()
print(path_disk_dk1)


def _runx():
    # print([line.strip() for line in
    # subprocess.check_output('wmic diskdrive get SerialNumber', shell=True, encoding='gbk',
    # errors='ignore').splitlines() if
    # line.strip() and not line.strip().startswith('SerialNumber')])

    # path_disk_dk1 = [line.strip() for line in
    # subprocess.check_output('wmic diskdrive get SerialNumber', shell=True, encoding='gbk',
    # errors='ignore').splitlines() if
    # line.strip() and not line.strip().startswith('SerialNumber')]
    output = subprocess.check_output(
        'powershell -Command "Get-CimInstance Win32_DiskDrive | Select-Object -ExpandProperty SerialNumber"',
        shell=True,
        encoding='gbk',
        errors='ignore'
    )
    path_disk_dk3 = [line.strip() for line in output.splitlines() if line.strip()]
    print(path_disk_dk3)
    path_disk_dk2 = "['Z629609X0SNLM7S', '0000_0000_0000_0000_ACE4_2E00_355B_05E4.']"
    # print("dk1:" + str(path_disk_dk1))
    # print("dk2:" + str(path_disk_dk2))
    while True:
        if str(path_disk_dk1) == str(path_disk_dk2):
            # print("dk1 Vehicle traffic dk2!")
            break
        else:
            continue


while True:
    if str(serial) == str(1615104090):
        # print("success!" + str(serial) + "_" + str(-925588465))
        break
    else:
        continue

_runx()

# @user:Borter
# @data:2026/11/16
# @project:拆分所有路,生成TXT分类
# @Blog_url:programb.blog.csdn.net
# @email:bol5261@126.com
# @gitee:https://gitee.com/borter
# @version:1.0, 2026-11-05, 初始版本

path_disk = 'E'
path_Project = 'x12'
path_xlsx = 'x1'
path_csv = f'{path_disk}:\\Database\\code_lql\\{path_Project}code\\{path_xlsx}code.xlsx'


def _print(path_open):
    wb = openpyxl.load_workbook(path_open)
    ws_a = wb['Data']
    for i in range(1, ws_a.max_row + 1):
        x1 = ws_a.cell(i, 1).value
        print("---")
        url_open = f'{path_disk}:\\Database\\code_lql\\{path_Project}code\\{path_xlsx}code.txt'
        url_save = f'{path_disk}:\\Database\\code_lql\\{path_Project}code\\{path_xlsx}code\\{x1}.txt'
        keywords = [f'{x1}']
        with open(url_open, "r", encoding="utf-8") as f_in, open(url_save, "w", encoding="utf-8") as f_out:
            for line in f_in:
                # 不包含关键词就写入新文件
                if any(keyword in line for keyword in keywords):
                    f_out.write(line)
        print("新文件已生成！ ID:" + str(x1))


class CodePrint:
    if __name__ == '__main__':
        _print(path_csv)