diff --git a/bonobo/commands/run.py b/bonobo/commands/run.py index 2204a3b..a37282c 100644 --- a/bonobo/commands/run.py +++ b/bonobo/commands/run.py @@ -106,6 +106,7 @@ def register_generic_run_arguments(parser, required=True): source_group = parser.add_mutually_exclusive_group(required=required) source_group.add_argument('filename', nargs='?', type=str) source_group.add_argument('--module', '-m', type=str) + parser.add_argument('--env', '-e', action='append') return parser @@ -115,5 +116,4 @@ def register(parser): verbosity_group.add_argument('--quiet', '-q', action='store_true') verbosity_group.add_argument('--verbose', '-v', action='store_true') parser.add_argument('--install', '-I', action='store_true') - parser.add_argument('--env', '-e', action='append') return execute diff --git a/tests/util/get_passed_env.py b/tests/util/get_passed_env.py deleted file mode 100644 index d9c4ba6..0000000 --- a/tests/util/get_passed_env.py +++ /dev/null @@ -1,22 +0,0 @@ -import os - -from bonobo import Graph - - -def extract(): - env_test_user = os.getenv('ENV_TEST_USER') - env_test_number = os.getenv('ENV_TEST_NUMBER') - env_test_string = os.getenv('ENV_TEST_STRING') - return env_test_user, env_test_number, env_test_string - - -def load(s: str): - print(s) - - -graph = Graph(extract, load) - -if __name__ == '__main__': - from bonobo import run - - run(graph)