3
2
forked from Dniel97/artemis
artemis/dbutils.py

61 lines
1.9 KiB
Python
Raw Normal View History

2024-01-09 08:07:04 +00:00
#!/usr/bin/env python3
import argparse
import logging
2024-01-09 08:07:04 +00:00
from os import mkdir, path, access, W_OK
import yaml
import asyncio
2024-01-09 18:57:59 +00:00
from core.data import Data
from core.config import CoreConfig
2023-03-09 16:38:58 +00:00
if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Database utilities")
2023-03-09 16:38:58 +00:00
parser.add_argument(
"--config", "-c", type=str, help="Config folder to use", default="config"
)
parser.add_argument(
"--version",
"-v",
type=str,
help="Version of the database to upgrade/rollback to",
)
parser.add_argument("--email", "-e", type=str, help="Email for the new user")
2024-01-09 08:07:04 +00:00
parser.add_argument("--access_code", "-a", type=str, help="Access code for new/transfer user", default="00000000000000000000")
parser.add_argument("action", type=str, help="create, upgrade, create-owner")
args = parser.parse_args()
cfg = CoreConfig()
if path.exists(f"{args.config}/core.yaml"):
cfg_dict = yaml.safe_load(open(f"{args.config}/core.yaml"))
cfg_dict.get("database", {})["loglevel"] = "info"
cfg.update(cfg_dict)
if not path.exists(cfg.server.log_dir):
mkdir(cfg.server.log_dir)
if not access(cfg.server.log_dir, W_OK):
print(
f"Log directory {cfg.server.log_dir} NOT writable, please check permissions"
)
exit(1)
data = Data(cfg)
if args.action == "create":
data.create_database()
2024-01-09 08:07:04 +00:00
elif args.action == "upgrade":
data.schema_upgrade(args.version)
2023-03-18 06:12:58 +00:00
elif args.action == "create-owner":
2024-01-09 08:07:04 +00:00
loop = asyncio.get_event_loop()
loop.run_until_complete(data.create_owner(args.email, args.access_code))
2024-01-09 18:57:59 +00:00
data.schema_upgrade(args.version)
elif args.action == "migrate":
loop = asyncio.get_event_loop()
loop.run_until_complete(data.migrate())
2024-01-09 08:07:04 +00:00
else:
logging.getLogger("database").info(f"Unknown action {args.action}")