Fix: prevent errors when trying to get a PK from a not yet saved model

hyperflask · Sep 2, 2024 · ffb5b43 · ffb5b43
1 parent 3eca767
commit ffb5b43
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sqlorm/mapper.py b/sqlorm/mapper.py
@@ -136,6 +136,8 @@ def get_primary_key(self, obj):
         pk = []
         for col in self.columns:
             if col.primary_key:
+                if not hasattr(obj, col.attribute):
+                    return
                 pk.append(getattr(obj, col.attribute))
         if len(pk) > 1:
             return pk