DX improvements

Author: JonasScholl

.vscode/launch.json

@@ -0,0 +1,22 @@
+{
+    "version": "0.2.0",
+    "configurations": [
+      {
+        "name": "Python: Debug Tests",
+        "type": "python",
+        "request": "launch",
+        "program": "${file}",
+        "purpose": [
+          "debug-test"
+        ],
+        "console": "internalConsole",
+        "justMyCode": false,
+        "presentation": {
+          "hidden": true
+        },
+        "env": {
+          "PYTEST_ADDOPTS": "--no-cov"
+        }
+      }
+    ]
+}

database_setup_tools/setup.py

@@ -3,7 +3,7 @@
 
 import sqlalchemy_utils
 from sqlalchemy import MetaData, Table
-from sqlmodel import SQLModel
+from sqlmodel.main import SQLModel, SQLModelMetaclass
 
 from database_setup_tools.session_manager import SessionManager
 
@@ -85,7 +85,7 @@ def create_database(self) -> bool:
             return True
         return False
 
-    def truncate(self, tables: Optional[List[SQLModel]] = None):
+    def truncate(self, tables: Optional[List[SQLModel | SQLModelMetaclass]] = None):
         """Truncate all tables in the database"""
         tables_to_truncate: List[Table] = self.model_metadata.sorted_tables
         if tables is not None: