forked from Hay1tsme/artemis
fix deprecated get_event_loop on 3.10+
This commit is contained in:
18
dbutils.py
18
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())
|
||||
|
Reference in New Issue
Block a user