"""
import data from json file to MariaDB
"""
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter

import yaml
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from platformdirs import PlatformDirs
from pathlib import Path

from schema import Base, KontorDB
from config import get_logger

parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
parser.add_argument('--recreate-db', action='store_true')
parser.add_argument('--verbose', '-v', action='count', default=0)
parser.add_argument('--file', '-f', default='~/data.json')
args = parser.parse_args()


if __name__ == '__main__':
    logger = get_logger(args.verbose)
    logger.info('kontor.import started')
    dirs = PlatformDirs("kontor")
    database_config = Path(dirs.user_config_dir, 'database-config.yaml')
    with open(database_config, 'rt') as f:
        db_config = yaml.safe_load(f.read())
    connect_string = ('mariadb+mariadbconnector://{}:{}@{}:{}/{}'.format(
        db_config['mariadb']['user'],
        db_config['mariadb']['password'],
        db_config['mariadb']['host'],
        db_config['mariadb']['port'],
        db_config['mariadb']['database']
    ))
    engine = create_engine(connect_string)
    Base.metadata.create_all(bind=engine, checkfirst=True)
    __session__ = sessionmaker(bind=engine)
    kontor_db = KontorDB(engine, logger)
    if args.recreate_db:
        kontor_db.delete_entries()
    kontor_db.import_db(args.file)
    logger.info('kontor.import finished')