fix: support relationships as string (#29)

Author: ViAchKoN

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
 

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"

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")