New Model

gravy-jones-locker · gravy-jones-locker · commit 3cdde1226e7f · 2023-07-06T13:01:37.000+01:00
diff --git a/src/strawberry_sqlalchemy_mapper/mapper.py b/src/strawberry_sqlalchemy_mapper/mapper.py
@@ -405,9 +405,9 @@ async def resolve(self, info: Info):
                 relationship_key = tuple(
                     [
                         [
-                            v.value
-                            for v in instance_state.attrs.values()
-                            if local in getattr(v, "columns", [])
+                            getattr(self, k)
+                            for k, column in self.__mapper__.c.items()
+                            if local.key == column.key
                         ][0]
                         for local, _ in relationship.local_remote_pairs
                     ]

Original file line number	Diff line number	Diff line change
`@@ -405,9 +405,9 @@ async def resolve(self, info: Info):`
`405`	`405`	`relationship_key = tuple(`
`406`	`406`	`[`
`407`	`407`	`[`
`408`		`- v.value`
`409`		`- for v in instance_state.attrs.values()`
`410`		`- if local in getattr(v, "columns", [])`
	`408`	`+ getattr(self, k)`
	`409`	`+ for k, column in self.__mapper__.c.items()`
	`410`	`+ if local.key == column.key`
`411`	`411`	`][0]`
`412`	`412`	`for local, _ in relationship.local_remote_pairs`
`413`	`413`	`]`