6. Environment Variables

Explicitly set environment variables

examples/env_variables.py

from cosmos.api import Cosmos, py_call

environment_variables_dict = {"Cosmos": "jobs", "are": "very", "cool": "!"}


def command_with_env_variables():
    lines = ["import os"] + [
        f"assert os.getenv({k}) == {v}"
        # this makes assert variable == value for each env variable
        for k, v in environment_variables_dict.items()
    ]
    command = f"python -c \"{';'.join(lines)}\""
    return command


def main():
    cosmos = Cosmos()
    cosmos.initdb()
    workflow = cosmos.start("env_variables", skip_confirm=True)
    workflow.add_task(
        func=command_with_env_variables, environment_variables=environment_variables_dict, uid="special"
    )
    workflow.run(cmd_wrapper=py_call)


if __name__ == "__main__":
    main()