采购文件以从python脚本中设置环境变量

Question

Running on Ubuntu, I have a python script, written by someone else, which simply runs like this:

python cool_script.py command line args

However, this script needs some environment variables set before it can run. So I created exports.sh located in the same dir as the python script, which looks something like:

export ENV1='foo'
export ENV2='bar'
export ENV3='baz'

so running:

$source exports.sh
$python cool_script.py command line args

works great. However, my case is a bit more complex. I need to run this on several environments, where the values of ENV1,ENV2,ENV3 will have to be different. I also want to do some stuff to the outputs of cool_script.py . So I wrote cool_script_wrapper.py , which looks like this:

# set environment variables
import os
exports_file = os.path.dirname(os.path.realpath(__file__)) + '/exports.sh'
os.system('source %s' % exports_file)
# run cool_script.py
os.system('python cool_script.py command line args')
# do some stuff to the output
...

I was planning to have different exports.sh scripts for each environment I need to run on, always keeping them in the same dir as the main script. Only problem is that this 'source' approach doesn't work. The environment variables are simply not available to cool_script.py .
Searching around Stack Overflow, I understand now that it's not supposed to work, but I didn't find any solution that suits my needs. I don't want to use os.environ with hard-coded values, and I don't want to parse the exports file. I guess I could make my wrapper a bash script rather than a python, but I hope there is a better solution. Any ideas?

Answer 1

The problem is in the source command that attempts to call a bash builtin. This should work:

import os
os.environ['ENV1'] = "foo"
os.system("python cool_script.py command line args")

compare with this answer: Calling the "source" command from subprocess.Popen