pjdev-sqlmodel 4.1.0__tar.gz → 4.1.0a2__tar.gz

{pjdev_sqlmodel-4.1.0 → pjdev_sqlmodel-4.1.0a2}/PKG-INFO

@@ -1,11 +1,12 @@
-Metadata-Version: 2.3
+Metadata-Version: 2.4
 Name: pjdev-sqlmodel
-Version: 4.1.0
+Version: 4.1.0a2
 Project-URL: Documentation, https://gitlab.purplejay.net/keystone/python
 Project-URL: Issues, https://gitlab.purplejay.net/keystone/python/issues
 Project-URL: Source, https://gitlab.purplejay.net/keystone/python
 Author-email: Purple Jay LLC <apps+keystone-python-221-issue-@purplejay.io>
-License: MIT
+License-Expression: MIT
+License-File: LICENSE.txt
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3.12

{pjdev_sqlmodel-4.1.0 → pjdev_sqlmodel-4.1.0a2}/src/pjdev_sqlmodel/__about__.py

@@ -1,4 +1,4 @@
 # SPDX-FileCopyrightText: 2024-present Chris O'Neill <chris@purplejay.io>
 #
 # SPDX-License-Identifier: MIT
-__version__ = "4.1.0"
+__version__ = "4.1.0a2"

{pjdev_sqlmodel-4.1.0 → pjdev_sqlmodel-4.1.0a2}/src/pjdev_sqlmodel/utilities.py

@@ -8,6 +8,7 @@ import pandas as pd
 from openpyxl.reader.excel import load_workbook
 from loguru import logger
 from sqlalchemy import Engine
+from sqlmodel import select
 
 from pjdev_sqlmodel import session_context, sqlmodel_service
 from pjdev_sqlmodel.models import ModelBase
@@ -15,10 +16,14 @@ from pjdev_sqlmodel.models import ModelBase
 T = TypeVar("T", bound=ModelBase)
 
 
-def get_files_in_directory(directory: Path) -> List[Path]:
-    return [f for f in directory.glob("**/*.xlsx") if not f.name.startswith("~$")] + [
-        f for f in directory.glob("**/*.csv")
+def get_files_in_directory(directory: Path, file_name='*', force_presence=False) -> List[Path]:
+    file_obj = [f for f in directory.glob(f"**/{file_name}.xlsx") if not f.name.startswith("~$")] + [
+        f for f in directory.glob(f"**/{file_name}.csv")
     ]
+    if force_presence and len(file_obj) == 0:
+        logger.error(f'Failed to find files matching the {directory}/{file_name} pattern and presence is required. Exiting...')
+        exit(1)
+    return file_obj
 
 
 def get_csv_columns(file_path):
@@ -167,6 +172,36 @@ def load_raw_excel_data(
     )
 
 
+def convert_table_to_df(InputTable: type[BaseModel]):
+    with session_context() as session:
+        statement = select(InputTable)
+        results = session.exec(statement).all()
+
+        headers = []
+        field_names = []
+        for name, annotation in InputTable.__annotations__.items():
+            field = InputTable.__fields__[name]
+            alias = field.alias if field.alias else name
+            headers.append(alias)
+            field_names.append(name)
+
+        data = []
+        for res in results:
+            new_obj = {}
+            for i in range(len(headers)):
+                header = headers[i]
+                field_name = field_names[i]
+                new_obj[header] = getattr(res, field_name)
+            data.append(new_obj)
+        df = pd.DataFrame(data)
+        return df
+
+
+def convert_table_to_csv(InputTable: type[BaseModel], file: Path):
+    df = convert_table_to_df(InputTable)
+    df.to_csv(file, index=False)
+
+
 def __read_csv(
     file: Path | str,
     cols: Optional[List[str]] = None,
@@ -208,4 +243,4 @@ def convert_to_csv(
     df = pd.DataFrame(dict_data)
 
     # Export DataFrame to CSV
-    df.to_csv(filename, index=index, columns=cols)
+    df.to_csv(filename, index=index, columns=cols)