diff --git a/dev.py b/dev.py index 587c403..5023903 100755 --- a/dev.py +++ b/dev.py @@ -4,7 +4,6 @@ from subprocess import run from argparse import ArgumentParser import os - if __name__ == '__main__': parser = ArgumentParser() parser.add_argument( @@ -20,8 +19,10 @@ if __name__ == '__main__': default='postgres://psql:example@localhost:5432' ) args = parser.parse_args() - env = os.environ + os.environ['DB_CONNECTION_STRING'] = args.db_url if args.init_db: - run('cargo run --bin admin-cli -- --setup'.split(), env=env) - else: - print('Nothing to do') + run( + 'cargo run --bin admin-cli -- --setup', + env=os.environ, + shell=True + )