将bash脚本中的环境变量分配给Python中的当前会话

  • Z时代
  • 2024-01-10
  • 分类：问答

我有许多bash脚本可帮助设置我当前的会话环境变量。我需要设置env变量，以便可以使用子进程模块在python脚本中运行命令。这是我执行bash脚本的方式：

. ./file1.sh

以下是bash脚本的开头：

echo "Setting Environment Variable..."
export HORCMINST=99
echo $HORCMINST
...

有没有办法从python脚本调用这些bash脚本或在python脚本中执行类似的操作？

回答：

使用shell=True与您现有的脚本

首先，就 最简单的事情而言 -如果您使用shell=True，则可以告诉外壳程序开始运行未经修改的现有脚本内容。

也就是说-如果您最初是这样做的：

subprocess.Popen(['your-command', 'arg1', 'arg2'])

…然后您可以执行以下操作，以几乎相同的安全性保证执行同一命令（唯一的其他漏洞，只要它们的内容file1.sh受信任，就是带壳问题，例如shellshock）：

# this has the security of passing explicit out-of-band args
# but sources your script before the out-of-process command
subprocess.Popen(['. "$1"; shift; exec "$@"', "_", "./file1.sh",
    "your-command", "arg1", "arg2"], shell=True)

使用/proc/self/environ在NUL分隔的流出口环境变量

理想的做法是以明确的形式导出环境变量-NUL分隔的流是理想的-然后在Python中解析该流（格式非常明确）。

假设使用Linux，则可以按如下所示导出完整的环境变量集：

# copy all our environment variables, in a NUL-delimited stream, to myvars.environ
cat </proc/self/environ >myvars.environ

…或者您可以手动导出一组特定的变量：

for varname in HORCMINST PATH; do
  printf '%s=%s\0' "$varname" "${!varname}"
done >myvars.environ

在Python中读取和解析NUL分隔的流

然后，您只需要阅读并解析它们：

#!/usr/bin/env python
env = {}
for var_def in open('myvars.environ', 'r').read().split('\0'):
  (key, value) = var_def.split('=', 1)
  env[key] = value
import subprocess
subprocess.Popen(['your-command', 'arg1', 'arg2'], env=env)

您也可以通过运行立即应用这些变量os.environ[key]=value。

在bash中读取和解析NUL分隔的流

顺便说一下，同样的格式也很容易在bash中解析：

while IFS= read -r -d '' var_def; do
  key=${var_def%%=*}
  value=${var_def#*=}
  printf -v "$key" '%s' "$value"
  export "$key"
done <myvars.environ
# ...put the rest of your bash script here

现在， 为什么要 使用NUL分隔的流？由于环境变量是C字符串-与Python字符串不同，它们不能包含NUL。因此，NUL是可以安全地用于定界的唯一字符。

例如，一个尝试使用换行符的人可能会被一个 包含 文字换行符的环境变量所困扰-

并且，例如，如果有人将一个简短的Python脚本嵌入到一个环境变量中，那是一个非常合理的事件！

以上是 将bash脚本中的环境变量分配给Python中的当前会话 的全部内容， 来源链接： utcz.com/qa/416327.html

回到顶部