python 管理员权限执行cmd命令

原创

mob64ca12eee07b 2024-09-23 06:07:55 ©著作权

文章标签 管理员权限 Python 返回结果 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12eee07b的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何使用 Python 实现管理员权限执行 CMD 命令

在某些情况下，我们的 Python 脚本可能需要执行一些需要管理员权限的操作。例如，在 Windows 上，一些命令需要提升权限才能执行。本教程将指导你如何以管理员身份运行 CMD 命令，并操作 Python 脚本来实现这一目标。

流程概述

以下是实现 Python 管理员权限执行 CMD 命令的基本步骤：

步骤	描述
1	导入必要的库
2	创建管理员权限的函数
3	调用 CMD 命令并获取输出
4	处理返回结果

1. 导入必要的库

在开始之前，我们需要导入两个必要的库：subprocess 和 ctypes。subprocess 用于执行系统命令，而 ctypes 用于修改应用程序权限。

import subprocess  # 用来执行子进程和命令行
import ctypes       # 用来处理权限

代码说明

subprocess：提供了一种与系统交互的方法，可以用它来运行 shell 命令。
ctypes：允许我们调用 C 语言的 API，其中包括处理 Windows 权限的功能。

2. 创建管理员权限的函数

接下来，我们需要定义一个函数，以管理员身份执行 CMD 命令。在 Windows 系统上，我们可以使用 ctypes.windll.shell32.ShellExecute 来实现。

def run_as_admin(command):
    """以管理员身份执行给定的命令。"""
    try:
        # 将命令转换为列表
        command_list = command.split()
        
        # 使用 ShellExecute 以管理员身份运行命令
        ctypes.windll.shell32.ShellExecuteW(None, "runas", command_list[0], ' '.join(command_list[1:]), None, 1)
    except Exception as e:
        print(f"出错了：{e}")

代码说明

run_as_admin(command)：接受一个字符串参数 command，即需要执行的 CMD 命令。
command.split()：将命令字符串分割成命令和参数的列表。
ShellExecuteW(..., "runas", ...)：以管理员身份执行命令。

3. 调用 CMD 命令并获取输出

为了可以从 CMD 获取执行结果，我们可以使用 subprocess.run() 方法。

def execute_command(command):
    """执行不需要管理员权限的命令并返回输出。"""
    try:
        # 执行命令
        result = subprocess.run(command, capture_output=True, text=True, shell=True)
        
        # 输出返回结果
        if result.returncode == 0:  # 判断是否执行成功
            return result.stdout
        else:
            return result.stderr
    except Exception as e:
        print(f"出错了：{e}")

代码说明

execute_command(command)：接受一个字符串参数 command，执行该命令并返回执行结果。
subprocess.run(...)：使用系统命令的方式执行 CMD 命令，capture_output=True 用于捕获输出，text=True 指定输出为字符串格式。
result.returncode：检查命令的返回代码，如果为 0 则表示成功执行，其他值表示出错。

4. 处理返回结果

最后，我们需要将上述两个函数结合起来，来实际运行一个 CMD 命令，并获取返回结果。比如，我们可以执行一个需要管理员权限的命令，然后再执行一个普通命令。

if __name__ == "__main__":
    # 需要管理员权限执行的命令
    admin_command = "net user"
    run_as_admin(admin_command)  # 执行需要管理员权限的命令

    # 执行不需要管理员权限的命令
    normal_command = "echo Hello World"
    output = execute_command(normal_command)
    print(f"输出: {output}")

代码说明

__name__ == "__main__"：确保代码在直接运行时执行，不在导入时执行。
admin_command 和 normal_command：定义需执行的命令，将它们传递到相应的函数中。

类图

我们可以用以下类图来展示整个实现过程。

classDiagram
    class CommandExecutor {
        +run_as_admin(command: str)
        +execute_command(command: str): str
    }
    CommandExecutor --|> AdminCommand
    CommandExecutor --|> NormalCommand