Definitely hardcoding to 'id' column as pk should be replaced to 'pk'…

… attribute
ionelmc · May 30, 2021 · 90abc6d · 90abc6d
1 parent 4906d0e
commit 90abc6d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/prefetch.py b/src/prefetch.py
@@ -194,7 +194,7 @@ def decorator(user, user_group_associations=()):
 
     * mapper(object):
 
-        Optional (defaults to ``lambda obj: obj.id``).
+        Optional (defaults to ``lambda obj: obj.pk``).
 
         A function that returns the key for a given object in your query set.
 
@@ -232,7 +232,7 @@ def __init__(self, filter=None, reverse_mapper=None, decorator=None, mapper=None
 
     @staticmethod
     def mapper(obj):
-        return obj.id
+        return obj.pk
 
     def fetch(self, dataset, name, model, forwarders, db):
         collect = self.collect or forwarders