From eba03e6b9b62524a7b740affab4ac51ee8dbcb19 Mon Sep 17 00:00:00 2001 From: beerpsi Date: Sat, 16 Nov 2024 09:13:23 +0700 Subject: [PATCH] fix deprecated get_event_loop on 3.10+ --- dbutils.py | 18 ++++++++++-------- 1 file changed, 10 insertions(+), 8 deletions(-) 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())