From f3cf1a17f32154055bcddace1c6be973e68e4a5c Mon Sep 17 00:00:00 2001
From: Thomas Peetz <thomas.peetz@cimt-ag.de>
Date: Wed, 15 Jan 2025 14:16:35 +0100
Subject: [PATCH] add import functionality

---
 python/kontor/controllers/database.py |  6 +--
 python/kontor/database/__init__.py    | 62 ++++++++++++++++++++++++++-
 2 files changed, 62 insertions(+), 6 deletions(-)

diff --git a/python/kontor/controllers/database.py b/python/kontor/controllers/database.py
index 2f3fde6..02a565e 100644
--- a/python/kontor/controllers/database.py
+++ b/python/kontor/controllers/database.py
@@ -49,7 +49,5 @@ class Database(Controller):
         if self.app.pargs.db_file is not None:
             data['db_file'] = self.app.pargs.db_file
         kontor_db = KontorDB(self.app.session, self.app.log)
-        if self.app.pargs.dry_run:
-            self.app.render(data, 'import.jinja2')
-        else:
-            kontor_db.import_db(data['db_file'])
+        self.app.render(data, 'import.jinja2')
+        kontor_db.import_db(data['db_file'], self.app.pargs.dry_run)
diff --git a/python/kontor/database/__init__.py b/python/kontor/database/__init__.py
index 31fc94a..f62097b 100644
--- a/python/kontor/database/__init__.py
+++ b/python/kontor/database/__init__.py
@@ -154,5 +154,63 @@ class KontorDB:
         if export_file.exists():
             self.log.debug(f"{export_file} exists")
 
-    def import_db(self, import_file_name: str):
-        pass
+    def import_db(self, import_file_name: str, dry_run: bool):
+        import_file = Path(import_file_name)
+        if not import_file.exists():
+            print(f"File {import_file_name} does not exist. Do nothing.")
+            return
+        self.log.debug(f"evaluate type from file extension: {import_file.suffix}")
+        match import_file.suffix:
+            case '.json':
+                print("read json file")
+                with open(import_file_name, 'r') as json_file:
+                    json_load = json.load(json_file)
+                for table in json_load:
+                    print(f"{table}: {len(json_load[table])}")
+                    self.import_table(table, json_load[table], dry_run)
+            case '.yml':
+                print("read yaml file")
+            case '.yaml':
+                print("read yaml file")
+            case '.db':
+                print("read sqlite file")
+
+    def import_table(self, table_name, items, dry_run: bool):
+        existing_ids = self.get_ids(table_name)
+        for item in items:
+            self.log.debug(f"{item}")
+            current_id = item['id']
+            found_item = self.session.query(self.registry[table_name]).get(current_id)
+            self.log.debug(f"found: {found_item}")
+            if found_item is not None:
+                changed = self.update_entry(found_item, item, dry_run)
+                if changed:
+                    print(f"{current_id} has changed")
+                existing_ids.remove(current_id)
+        if len(existing_ids) > 0:
+            print("remaining items")
+        self.session.commit()
+
+    def get_ids(self, table_name: str) -> list:
+        existing_ids = []
+        items = self.session.query(self.registry[table_name]).all()
+        for item in items:
+            existing_ids.append(getattr(item, 'id'))
+        return existing_ids
+
+    def update_entry(self, existing_item, update_item: dict, dry_run: bool) -> bool:
+        changed = False
+        for key in update_item.keys():
+            update_value = update_item[key]
+            existing_value = getattr(existing_item, key)
+            if type(existing_value) is not type(update_value):
+                self.log.debug(f"compare {type(existing_value)} with {type(update_value)}")
+                existing_value = str(existing_value)
+            if existing_value != update_value:
+                print(f"{key} has changed: {existing_value} != {update_value}")
+                if not dry_run:
+                    setattr(existing_item, key, update_value)
+                    # existing_item[key] = update_value
+                    changed = True
+                self.log.info(f"update {key} with {update_value}")
+        return changed