如何用python禁用u盘

用Python禁用U盘的方法包括：修改注册表、使用系统命令、利用第三方库等。其中，修改注册表的方法最为常见和有效。下面我们将详细介绍其中一种方法——通过修改注册表来禁用U盘。

一、修改注册表禁用U盘

1、Windows注册表简介

Windows注册表是一个层次结构的数据库，存储了系统和应用程序的设置和选项。通过修改注册表，可以控制系统的许多行为，包括硬件设备的启用和禁用。

2、修改注册表的原理

禁用U盘的关键在于修改与USB存储设备相关的注册表项。通常，我们需要修改以下注册表项：

HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\USBSTOR

通过将该注册表项下的Start值从默认的3更改为4，可以禁用U盘。

3、使用Python修改注册表

Python可以通过winreg模块（在Windows下使用）来访问和修改注册表。下面是一个示例代码：

import winreg as reg
def disable_usb():
    try:
        # 打开注册表项
        key = reg.OpenKey(reg.HKEY_LOCAL_MACHINE, r"SYSTEM\CurrentControlSet\Services\USBSTOR", 0, reg.KEY_SET_VALUE)
        # 设置值为4，禁用U盘
        reg.SetValueEx(key, "Start", 0, reg.REG_DWORD, 4)
        reg.CloseKey(key)
        print("U盘已禁用")
    except Exception as e:
        print(f"修改注册表时出现错误: {e}")
def enable_usb():
    try:
        # 打开注册表项
        key = reg.OpenKey(reg.HKEY_LOCAL_MACHINE, r"SYSTEM\CurrentControlSet\Services\USBSTOR", 0, reg.KEY_SET_VALUE)
        # 设置值为3，启用U盘
        reg.SetValueEx(key, "Start", 0, reg.REG_DWORD, 3)
        reg.CloseKey(key)
        print("U盘已启用")
    except Exception as e:
        print(f"修改注册表时出现错误: {e}")
if __name__ == "__main__":
    disable_usb()  # 禁用U盘
    # enable_usb()  # 启用U盘

二、使用系统命令禁用U盘

1、利用`diskpart`命令禁用U盘

diskpart是Windows系统自带的磁盘管理工具，可以通过脚本方式来禁用U盘。下面是一个示例代码：

import subprocess
def disable_usb():
    try:
        # 创建diskpart脚本
        script = """
        list volume
        select volume X  # X是U盘的卷号
        remove letter=X
        """
        with open("disable_usb_script.txt", "w") as file:
            file.write(script)
        # 执行diskpart脚本
        subprocess.run(["diskpart", "/s", "disable_usb_script.txt"], check=True)
        print("U盘已禁用")
    except Exception as e:
        print(f"执行diskpart命令时出现错误: {e}")
if __name__ == "__main__":
    disable_usb()  # 禁用U盘

三、利用第三方库禁用U盘

1、使用`pywin32`库

pywin32库提供了对Windows API的访问，可以用于管理系统设备。下面是一个示例代码：

import win32api
import win32con
def disable_usb():
    try:
        # 获取设备管理器
        dev_mgr = win32api.OpenSCManager(None, None, win32con.SC_MANAGER_ALL_ACCESS)
        # 打开USB设备服务
        service = win32api.OpenService(dev_mgr, "USBSTOR", win32con.SERVICE_ALL_ACCESS)
        # 停止服务，禁用U盘
        win32api.ControlService(service, win32con.SERVICE_CONTROL_STOP)
        win32api.ChangeServiceConfig2(service, win32con.SERVICE_CONFIG_DESCRIPTION, "USB storage device disabled")
        win32api.CloseServiceHandle(service)
        win32api.CloseServiceHandle(dev_mgr)
        print("U盘已禁用")
    except Exception as e:
        print(f"使用pywin32库禁用U盘时出现错误: {e}")
if __name__ == "__main__":
    disable_usb()  # 禁用U盘

2、使用`pywinauto`库

pywinauto库是一个用于自动化Windows GUI的Python库，可以用于操作设备管理器来禁用U盘。下面是一个示例代码：

from pywinauto.application import Application
def disable_usb():
    try:
        # 启动设备管理器
        app = Application().start("devmgmt.msc")
        # 等待设备管理器窗口出现
        window = app.window(title_re="Device Manager")
        window.wait("visible")
        # 展开通用串行总线控制器
        window.tree_view.GetItem([u"Computer", u"Universal Serial Bus controllers"]).click()
        # 找到USB存储设备
        usb_device = window.tree_view.GetItem([u"Computer", u"Universal Serial Bus controllers", u"USB Mass Storage Device"])
        usb_device.click()
        # 禁用设备
        window[u"Action"].click_input()
        window[u"Disable"].click_input()
        print("U盘已禁用")
    except Exception as e:
        print(f"使用pywinauto库禁用U盘时出现错误: {e}")
if __name__ == "__main__":
    disable_usb()  # 禁用U盘