refactor: add __eq__ on Table

piccolo-orm · Oct 14, 2024 · 14c3cc9 · 14c3cc9
1 parent f4bf25a
commit 14c3cc9
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/piccolo/table.py b/piccolo/table.py
@@ -851,6 +851,13 @@ def __repr__(self) -> str:
         )
         return f"<{self.__class__.__name__}: {pk}>"
 
+    def __eq__(self, other: Table):
+        return (
+            isinstance(other, self.__class__)
+            and self.__class__._meta.primary_key
+            == other.__class__._meta.primary_key
+        )
+
     ###########################################################################
     # Classmethods