diff --git a/dbutils.py b/dbutils.py index 9080afc..154df0a 100644 --- a/dbutils.py +++ b/dbutils.py @@ -9,7 +9,7 @@ import yaml from core.config import CoreConfig from core.data import Data -if __name__ == "__main__": +async def main(): parser = argparse.ArgumentParser(description="Database utilities") parser.add_argument( "--config", "-c", type=str, help="Config folder to use", default="config" @@ -44,10 +44,8 @@ if __name__ == "__main__": data = Data(cfg) - loop = asyncio.get_event_loop() - if args.action == "create": - loop.run_until_complete(data.create_database()) + await data.create_database() elif args.action == "upgrade": data.schema_upgrade(args.version) @@ -59,16 +57,20 @@ if __name__ == "__main__": data.schema_downgrade(args.version) elif args.action == "create-owner": - loop.run_until_complete(data.create_owner(args.email, args.access_code)) + await data.create_owner(args.email, args.access_code) elif args.action == "migrate": - loop.run_until_complete(data.migrate()) + await data.migrate() elif args.action == "create-revision": - loop.run_until_complete(data.create_revision(args.message)) + await data.create_revision(args.message) elif args.action == "create-autorevision": - loop.run_until_complete(data.create_revision_auto(args.message)) + await data.create_revision_auto(args.message) else: logging.getLogger("database").info(f"Unknown action {args.action}") + + +if __name__ == "__main__": + asyncio.run(main())