check existing items with data from file

kontor-scripts/import.py

@@ -3,8 +3,10 @@ import data from json file to MariaDB
 """
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
 
-from typing import Generator
+from typing import Any, AnyStr, Dict, List
 import os
+import json
+from pathlib import Path
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker, Session
 from db.models.base import Base
@@ -13,27 +15,48 @@ from config import get_logger
 
 parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
 parser.add_argument('--verbose', '-v', action='count', default=0)
-parser.add_argument('--recreate-db', action='store_true')
+parser.add_argument('--dry-run', '-m', action='store_true')
+parser.add_argument('--cleanup', '-c', action='store_true')
 parser.add_argument('--file', '-f', default='~/data.json')
 args = parser.parse_args()
 
 DB_USER: str = os.getenv("DB_USER", "kontor")
 DB_PASSWORD: str = os.getenv("DB_PASSWORD", "kontor")
 DB_SERVER: str = os.getenv("DB_SERVER", "127.0.0.1")
-DB_PORT: int = os.getenv("DB_PORT", 5432)
+DB_PORT: int = int(os.getenv("DB_PORT", 5432))
 DB_DBNAME: str = os.getenv("DB_DBNAME", "kontor")
 DATABASE_URL: str = f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_SERVER}:{DB_PORT}/{DB_DBNAME}"
 
 
-def cleanup_database(db: Session, log):
+def cleanup_database(db: Session, log, dry_run: bool):
+    log.info("cleanup_database")
     # get tables from registry
     for table in registry:
         log.info(f"{table}")
         model = registry[table]
         entries = db.query(model).all()
         for entry in entries:
-            db.delete(entry)
-            db.commit()
+            if not dry_run:
+                db.delete(entry)
+                db.commit()
+
+def load_data(filename: str, log) -> Dict[AnyStr, Dict[AnyStr, Any]]:
+    log.info("load_data")
+    import_file = Path(filename)
+    if not import_file.exists():
+        log.info(f"File {filename} does not exist. Do nothing.")
+        raise FileNotFoundError()
+    log.info("read json file")
+    with open(filename, 'r') as json_file:
+        json_load = json.load(json_file)
+    return json_load
+
+def get_ids(items: List[Any]) -> List[AnyStr]:
+    result: List[AnyStr] = []
+    for item in items:
+        result.append(item.id)
+    return result
+
 
 if __name__ == '__main__':
     logger = get_logger(args.verbose, "kontor")
@@ -42,6 +65,20 @@ if __name__ == '__main__':
     Base.metadata.create_all(bind=engine, checkfirst=True)
     SessionLocal = sessionmaker(bind=engine)
     with SessionLocal() as db:
-        cleanup_database(db, logger)
+        if args.cleanup:
+            cleanup_database(db, logger, args.dry_run)
+        data = load_data(args.file, logger)
+        for tablename in data:
+            model = registry[tablename]
+            existing_items = db.query(model).all()
+            existing_ids = get_ids(existing_items)
+            logger.debug(f"found {len(existing_items)} for table {tablename}")
+            import_items = data[tablename]
+            for import_item in import_items:
+                if import_item['id'] in existing_ids:
+                    logger.info(f"update {import_item['id']}")
+                else:
+                    logger.info(f"import {import_item['id']}")
+
     logger.info('kontor.import finished')