evaluate sqlmodel

scripts/sqlalchemy/import.py

@@ -0,0 +1,43 @@
+"""
+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_database_cursors, create_tables, get_logger, get_scripts, get_meta_data
+
+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')