Source Bash file using a Python function

envutils.py

import os
from subprocess import Popen, PIPE
import pickle


PYTHON_DUMP_ENVIRON = """\
import sys
import os
import pickle

data = pickle.dumps(os.environ)
stdout = os.fdopen(sys.stdout.fileno(), "wb")
stdout.write(data)
"""


def source_bash_file(path):
    bash_cmds = [
        "source '%s'" % path,
        "python -c '%s'" % PYTHON_DUMP_ENVIRON,
    ]
    p = Popen(['bash', '-c', '&&'.join(bash_cmds)], stdout=PIPE)
    stdout, _ = p.communicate()
    if stdout:
        environ = pickle.loads(stdout)
        for k, v in environ.items():
            os.environ[k] = v

@DevStrikerTech My script requires bash shell, as its title says: "Source Bash file using a Python function". It won't work on Windows where there is no bash by default. It might be possible to get it via cygwin, but there might be better alternatives to getting env vars on Windows.

Thanks for this gist! I had to modify line 11 in a more recent version of python to:

data = pickle.dumps(dict(os.environ))

in order to make it work. Apparently os.environ is now wrapped in a class that cannot be pickled.