fix: support relationships as string

ViAchKoN · Nov 26, 2024 · ab24f60 · ab24f60
1 parent 4d541dc
commit ab24f60
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/dataclass_sqlalchemy_mixins/base/mixins.py b/dataclass_sqlalchemy_mixins/base/mixins.py
@@ -47,7 +47,7 @@ def get_foreign_key_filtered_column(
 
                 if related_model:
                     # Updating original model to continue search
-                    model = related_model.argument
+                    model = related_model.entity.class_
                     models.append(model)
                     continue
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dataclass-sqlalchemy-mixins"
-version = "0.3.0"
+version = "0.3.1"
 description = "Allows to convert dataclasses to sqlalchemy filters and orderings."
 authors = ["ViAchKoN"]
 readme = "README.md"

diff --git a/tests/models.py b/tests/models.py
@@ -66,7 +66,7 @@ class Group(BaseModel):
             Owner.id,
         ),
     )
-    owner = relationship(Owner)
+    owner = relationship("Owner")
 
     items = relationship("Item", back_populates="group", lazy="dynamic")