python - 如何在Python中访问批处理脚本变量？

Question

我有一个 bat 文件，它为变量设置值，我需要在 python 脚本中使用这些变量。

env.bat:

SET USERNAME=username
SET PASSWORD=password

我需要在 python 脚本中使用上述变量。我已尝试以下操作，但输出为 None。

登录.py

import os
import subprocess
subprocess.call(['env.bat'])
print(os.getenv('USERNAME'))

您能帮我解决我的错误吗？或者有其他方法可以实现这一目标吗？

提前致谢。

Answer 1

import os
import subprocess
p = subprocess.Popen(['cmd', '/v:on', '/q', '/c', 'env.bat', '&echo(!USERNAME!&echo(!PASSWORD!'], stdout=subprocess.PIPE, universal_newlines=True)
stdout, stderr = p.communicate()
username, password = stdout.splitlines()
print(username)
print(password)

由于env.bat不使用setlocal，环境变量运行 env.bat 后，将在 cmd 的同一进程中可用。这允许将 !USERNAME! 和 !PASSWORD! 的值回显到标准输出流。

cmd的选项:

• /v:on 启用延迟扩展。
• /q 设置回显关闭。
• /c 运行以下命令字符串。

注意:

• 在Python 2和Python 3中进行测试。 Python 2 缺少对上下文管理器带有 subprocess，因此未使用 with 语句。

• USERNAME 环境变量是一个内置名称。如果没有充分的理由进行更改可能会产生意想不到的问题。

• 由于启用了延迟扩展，在 env.bat 中使用 ! 可能需要进行转义，否则 cmd 解析文件时可能会省略 !。在命令行中call echo可以避免启用延迟如果需要的话进行扩展。