Python执行Linux命令捕获输出及root权限校验问题排查

你之前的代码有三个核心问题：

1. os.system()的返回值从来不是命令的打印内容，而是命令的退出状态码：命令执行成功返回0，执行失败返回对应非零错误码，这就是你打印user得到0的根本原因。
2. os.system()会直接把执行命令的输出绑定到当前终端，输出会直接打印到屏幕，本身就不支持把输出内容存到变量里。
3. 额外语法错误：elif关键字后必须跟判断条件，你写的裸elif:不符合Python语法，兜底分支应该用else:。

最优方案：原生检测root身份（无外部命令调用，无多余输出） #

检测当前运行用户根本不需要调用whoami这类外部命令，Python标准库os直接提供了获取进程用户ID的接口，root用户的有效UID固定为0，直接判断即可，无任何终端输出，兼容性和性能都远好于调用外部命令，Arch Linux环境下完全可用：

import os

if os.geteuid() != 0:
    print("please run as root")
    exit()

# 后续需要root权限的业务逻辑写在这里

注：os.geteuid()获取的是进程的有效用户ID，即使用户通过sudo提权运行脚本也能正确识别，不会出现判断错误。

通用方案：调用Linux命令时隐藏终端输出、捕获执行结果 #

如果后续你有调用其他Linux命令的需求，需要隐藏输出同时拿到执行结果，用Python官方推荐的subprocess模块替代os.system()即可：

import subprocess

# 执行whoami命令，捕获输出，不在终端打印
cmd_result = subprocess.run(
    ["whoami"],
    capture_output=True,
    text=True
)
# 命令的标准输出内容存在stdout属性中，用strip()去掉末尾的换行符
current_user = cmd_result.stdout.strip()
# 命令退出状态码存在returncode属性中
# 标准错误输出存在stderr属性中

if current_user != "root":
    print("please run as root")
    exit()

如果需要完全静默执行命令，连错误输出都不打印，可以把输出流定向到空设备：

import subprocess

subprocess.run(
    ["your_command", "arg1", "arg2"],
    stdout=subprocess.DEVNULL,
    stderr=subprocess.DEVNULL
)

内容的提问来源于stack exchange，提问作者rdex999