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.config import CoreConfig
|
||||||
from core.data import Data
|
from core.data import Data
|
||||||
|
|
||||||
if __name__ == "__main__":
|
async def main():
|
||||||
parser = argparse.ArgumentParser(description="Database utilities")
|
parser = argparse.ArgumentParser(description="Database utilities")
|
||||||
parser.add_argument(
|
parser.add_argument(
|
||||||
"--config", "-c", type=str, help="Config folder to use", default="config"
|
"--config", "-c", type=str, help="Config folder to use", default="config"
|
||||||
@ -44,10 +44,8 @@ if __name__ == "__main__":
|
|||||||
|
|
||||||
data = Data(cfg)
|
data = Data(cfg)
|
||||||
|
|
||||||
loop = asyncio.get_event_loop()
|
|
||||||
|
|
||||||
if args.action == "create":
|
if args.action == "create":
|
||||||
loop.run_until_complete(data.create_database())
|
await data.create_database()
|
||||||
|
|
||||||
elif args.action == "upgrade":
|
elif args.action == "upgrade":
|
||||||
data.schema_upgrade(args.version)
|
data.schema_upgrade(args.version)
|
||||||
@ -59,16 +57,20 @@ if __name__ == "__main__":
|
|||||||
data.schema_downgrade(args.version)
|
data.schema_downgrade(args.version)
|
||||||
|
|
||||||
elif args.action == "create-owner":
|
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":
|
elif args.action == "migrate":
|
||||||
loop.run_until_complete(data.migrate())
|
await data.migrate()
|
||||||
|
|
||||||
elif args.action == "create-revision":
|
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":
|
elif args.action == "create-autorevision":
|
||||||
loop.run_until_complete(data.create_revision_auto(args.message))
|
await data.create_revision_auto(args.message)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
logging.getLogger("database").info(f"Unknown action {args.action}")
|
logging.getLogger("database").info(f"Unknown action {args.action}")
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
asyncio.run(main())
|
||||||
|
Reference in New Issue
Block a user