forked from Hay1tsme/artemis
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
#!/usr/bin/env python3
|
|
import argparse
|
|
import logging
|
|
from os import mkdir, path, access, W_OK
|
|
import yaml
|
|
import asyncio
|
|
|
|
from core.data import Data
|
|
from core.config import CoreConfig
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Database utilities")
|
|
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")
|
|
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()
|
|
|
|
elif args.action == "upgrade":
|
|
data.schema_upgrade(args.version)
|
|
|
|
elif args.action == "create-owner":
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(data.create_owner(args.email, args.access_code))
|
|
data.schema_upgrade(args.version)
|
|
|
|
elif args.action == "migrate":
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(data.migrate())
|
|
|
|
else:
|
|
logging.getLogger("database").info(f"Unknown action {args.action}")
|