Wrap a shell command in python, for examplem standup, setup and teardown of a fixture for a test.

with_context.py

#!/usr/bin/env python3

from itertools import (
    starmap,
)

import sys
import subprocess
import shlex


def main(
    env_command,
    aws,
):
    print("---")
    result = subprocess.run(
        env_command,
        shell=True,
    )
    print("---")

    sys.exit(result.returncode)


if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("command_word", nargs=argparse.ZERO_OR_MORE)

    args = parser.parse_args()
    env = {
        "test": "value",
    }
    env_command = "/usr/bin/env {environment} sh -c {command}".format(
        environment=" ".join(starmap(
            lambda name, value: "{}={}".format(name.upper(), value),
            env.items()
        )),
        command=shlex.quote(" ".join(args.command_word)),
    )
    print(env_command)

    main(
        env_command=env_command,
    )

Usage example: ./with_context.py 'echo hello && echo world && echo $TEST'

Just found that you could skip the environment part in the above script and instead just inject it to subprocess.open(.., env=<>, ..). A much cleaner solution