f3-data-models 0.3.7__tar.gz → 0.3.8__tar.gz

{f3_data_models-0.3.7 → f3_data_models-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: f3-data-models
-Version: 0.3.7
+Version: 0.3.8
 Summary: The data schema and models for F3 Nation applications.
 License: MIT
 Author: Evan Petzoldt

{f3_data_models-0.3.7 → f3_data_models-0.3.8}/f3_data_models/models.py

@@ -17,6 +17,7 @@ from sqlalchemy import (
     UniqueConstraint,
     Enum,
     Uuid,
+    inspect,
 )
 from typing_extensions import Annotated
 from sqlalchemy.orm import (
@@ -25,6 +26,7 @@ from sqlalchemy.orm import (
     Mapped,
     relationship,
 )
+from sqlalchemy.orm.attributes import InstrumentedAttribute
 import enum
 
 # Custom Annotations
@@ -150,6 +152,26 @@ class Base(DeclarativeBase):
             if c.key not in ["created", "updated"]
         }
 
+    def to_update_dict(self) -> Dict[InstrumentedAttribute, Any]:
+        update_dict = {}
+        mapper = inspect(self).mapper
+
+        # Add simple attributes
+        for attr in mapper.column_attrs:
+            if attr.key not in ["created", "updated", "id"]:
+                update_dict[attr] = getattr(self, attr.key)
+
+        # Add relationships
+        for rel in mapper.relationships:
+            related_value = getattr(self, rel.key)
+            if related_value is not None:
+                if rel.uselist:
+                    update_dict[rel] = [item for item in related_value]
+                    print(rel, update_dict[rel])
+                else:
+                    update_dict[rel] = related_value
+        return update_dict
+
     def __repr__(self):
         """
         Get a string representation of the model instance.

{f3_data_models-0.3.7 → f3_data_models-0.3.8}/f3_data_models/testing.py

@@ -15,7 +15,7 @@ def test_update_event():
         start_time="0500",
         end_time="0600",
         event_x_event_types=[
-            EventType_x_Event(event_type_id=4),
+            EventType_x_Event(event_type_id=3),
         ],
         recurrence_pattern=Event_Cadence.weekly,
         day_of_week=Day_Of_Week.monday,
@@ -23,17 +23,10 @@ def test_update_event():
         index_within_interval=1,
         name="Test Event",
     )
-    update_dict = event.__dict__
+    update_dict = event.to_update_dict()
     DbManager.update_records(Event, [Event.id == 3], update_dict)
 
-    # with get_session() as session:
-    #     # delete event_x_event_types
-    #     session.query(EventType_x_Event).filter(
-    #         EventType_x_Event.event_id == 3
-    #     ).delete()
-    #     # add event_x_event_types
-    #     session.add(EventType_x_Event(event_id=3, event_type_id=4))
-    #     session.commit()
+    event = DbManager.get(Event, 3)
 
 
 if __name__ == "__main__":

{f3_data_models-0.3.7 → f3_data_models-0.3.8}/f3_data_models/utils.py

@@ -1,6 +1,6 @@
 import os
 from dataclasses import dataclass
-from typing import List, Optional, Tuple, TypeVar
+from typing import List, Optional, Tuple, TypeVar, Type, Generic  # noqa
 
 import sqlalchemy
 from sqlalchemy import Select, and_, select, inspect
@@ -89,7 +89,7 @@ def _joinedloads(cls: T, query: Select, joinedloads: list | str) -> Select:
 
 
 class DbManager:
-    def get(cls: T, id: int, joinedloads: list | str = []) -> T:
+    def get(cls: Type[T], id: int, joinedloads: list | str = []) -> T:
         session = get_session()
         try:
             query = select(cls).filter(cls.id == id)
@@ -175,9 +175,9 @@ class DbManager:
 
             mapper = class_mapper(cls)
             relationships = mapper.relationships.keys()
-            for key, value in fields.items():
+            for attr, value in fields.items():
+                key = attr.key
                 if hasattr(cls, key) and key not in relationships:
-                    attr = getattr(cls, key)
                     if isinstance(attr, InstrumentedAttribute):
                         setattr(record, key, value)
                 elif key in relationships:
@@ -232,14 +232,16 @@ class DbManager:
 
             for obj in objects:
                 # Update simple fields
-                for key, value in fields.items():
+                for attr, value in fields.items():
+                    key = attr.key
                     if key in valid_attributes and not isinstance(
                         value, InstrumentedList
                     ):
                         setattr(obj, key, value)
 
                 # Update relationships separately
-                for key, value in fields.items():
+                for attr, value in fields.items():
+                    key = attr.key
                     if key in valid_relationships:
                         # Handle relationships separately
                         relationship = inspect(cls).mapper.relationships[key]

{f3_data_models-0.3.7 → f3_data_models-0.3.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "f3-data-models"
-version = "0.3.7"
+version = "0.3.8"
 description = "The data schema and models for F3 Nation applications."
 authors = ["Evan Petzoldt <evan.petzoldt@protonmail.com>"]
 readme = "README.md"