add CRUD for WorkType

kontor-scripts/export.py

@@ -2,17 +2,17 @@
 import data from json file to MariaDB
 """
 from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
-
+from datetime import datetime
+import json
 import yaml
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 from platformdirs import PlatformDirs
 from pathlib import Path
-
-from schema.base import Base 
-from schema.database import KontorDB
+from db.models import registry
+from db.models.base import Base
 from config import get_logger
-from schema.database import ExportType
+from db.repository.metadata import get_tables, get_columns_for_table
 
 parser = ArgumentParser(formatter_class=ArgumentDefaultsHelpFormatter)
 parser.add_argument('--verbose', '-v', action='count', default=0)
@@ -38,6 +38,34 @@ if __name__ == '__main__':
     engine = create_engine(connect_string)
     Base.metadata.create_all(bind=engine, checkfirst=True)
     __session__ = sessionmaker(bind=engine)
-    kontor_db = KontorDB(engine, logger)
-    kontor_db.export_db(ExportType.JSON, args.file)
+    with __session__() as db:
+        data = {}
+        tables = get_tables(db)
+        for table in tables:
+            # logger.info(f"Table {table.name} with {table.id}")
+            columns = get_columns_for_table(db, table)
+            model = registry[table.name]
+            rows = db.query(model).all()
+            entries = []
+            for row in rows:
+                entry = {}
+                for column in columns:
+                    # logger.info(f"    Column {column.order} {column.name} with {column.id}")
+                    try:
+                        value = getattr(row, column.name)
+                        if isinstance(value, datetime):
+                            entry[column.name] = str(value)
+                        else:
+                            entry[column.name] = value
+                    except AttributeError as error:
+                        logger.info(f"{error}")
+                entries.append(entry)
+            data[table.name] = entries
+            logger.info(f"{table.name}: {len(entries)} exported")
+        json_dump = json.dumps(data, indent=4)
+        with open(args.file, "w") as dump_file:
+            dump_file.write(json_dump)
+        logger.info(f"{len(data)} tables exported")
+    #kontor_db = KontorDB(engine, logger)
+    #kontor_db.export_db(ExportType.JSON, args.file)
     logger.info('kontor.export finished')