2024-01-09 08:07:04 +00:00
|
|
|
#!/usr/bin/env python3
|
2023-02-16 05:06:42 +00:00
|
|
|
import argparse
|
2023-04-15 04:13:04 +00:00
|
|
|
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-02-16 05:06:42 +00:00
|
|
|
|
2023-03-09 16:38:58 +00:00
|
|
|
if __name__ == "__main__":
|
2023-02-17 06:02:21 +00:00
|
|
|
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",
|
|
|
|
)
|
2023-03-04 05:04:47 +00:00
|
|
|
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")
|
2024-01-12 01:48:27 +00:00
|
|
|
parser.add_argument("--message", "-m", type=str, help="Revision message")
|
2024-01-09 08:07:04 +00:00
|
|
|
parser.add_argument("action", type=str, help="create, upgrade, create-owner")
|
2023-02-17 06:02:21 +00:00
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
|
|
cfg = CoreConfig()
|
2023-03-05 02:58:51 +00:00
|
|
|
if path.exists(f"{args.config}/core.yaml"):
|
2023-04-15 07:06:11 +00:00
|
|
|
cfg_dict = yaml.safe_load(open(f"{args.config}/core.yaml"))
|
2023-04-24 01:04:52 +00:00
|
|
|
cfg_dict.get("database", {})["loglevel"] = "info"
|
2023-04-15 07:06:11 +00:00
|
|
|
cfg.update(cfg_dict)
|
2023-04-24 01:04:52 +00:00
|
|
|
|
2023-03-17 01:36:42 +00:00
|
|
|
if not path.exists(cfg.server.log_dir):
|
|
|
|
mkdir(cfg.server.log_dir)
|
2023-04-24 01:04:52 +00:00
|
|
|
|
2023-03-17 01:36:42 +00:00
|
|
|
if not access(cfg.server.log_dir, W_OK):
|
|
|
|
print(
|
|
|
|
f"Log directory {cfg.server.log_dir} NOT writable, please check permissions"
|
|
|
|
)
|
|
|
|
exit(1)
|
|
|
|
|
2023-02-17 06:02:21 +00:00
|
|
|
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
|
|
|
|
2023-03-04 05:04:47 +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())
|
2023-03-04 05:04:47 +00:00
|
|
|
|
2024-01-12 01:48:27 +00:00
|
|
|
elif args.action == "create-revision":
|
|
|
|
loop = asyncio.get_event_loop()
|
|
|
|
loop.run_until_complete(data.create_revision(args.message))
|
|
|
|
|
2024-01-09 08:07:04 +00:00
|
|
|
else:
|
|
|
|
logging.getLogger("database").info(f"Unknown action {args.action}")
|