From ffb5b432877a9be5162f1d231e2930dee0b07138 Mon Sep 17 00:00:00 2001
From: Maxime Bouroumeau-Fuseau <maxime.bouroumeau@gmail.com>
Date: Mon, 2 Sep 2024 22:57:48 +0200
Subject: [PATCH] Fix: prevent errors when trying to get a PK from a not yet
 saved model

---
 sqlorm/mapper.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sqlorm/mapper.py b/sqlorm/mapper.py
index 91b1c7e..f810296 100644
--- 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