actaeon/scripts/db-import.py

31 lines
968 B
Python
Raw Normal View History

2024-03-12 10:48:40 +00:00
import argparse
from dotenv import load_dotenv
from urllib.parse import urlparse
import mariadb
import os
from importers import get_importers
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--env', '-e', default='../.env.local', help='env file to load DATABASE_URL from')
subparsers = parser.add_subparsers(dest='game', help='game importer to use', required=True)
importers = get_importers()
for name, importer in importers.items():
importer.register(subparsers.add_parser(name))
args = parser.parse_args()
load_dotenv(args.env)
parsed = urlparse(os.getenv('DATABASE_URL'))
conn = mariadb.connect(
user=parsed.username,
password=parsed.password,
host=parsed.hostname,
port=parsed.port or 3306,
database=parsed.path[1:]
)
importer = importers[args.game](conn=conn, **vars(args))
importer.do_import()
conn.commit()
conn.close()