From 16a8b158e04bb23eec501599bc8ca38e945181c6 Mon Sep 17 00:00:00 2001 From: Justin Littman Date: Tue, 5 Jan 2016 11:59:43 -0500 Subject: [PATCH] Change get_for_objects to distinguish between number and text keys. --- src/auditlog/models.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/auditlog/models.py b/src/auditlog/models.py index 085dc67..43e8904 100644 --- a/src/auditlog/models.py +++ b/src/auditlog/models.py @@ -92,7 +92,11 @@ class LogEntryManager(models.Manager): content_type = ContentType.objects.get_for_model(queryset.model) primary_keys = queryset.values_list(queryset.model._meta.pk.name, flat=True) - return self.filter(content_type=content_type).filter(Q(object_id__in=primary_keys) | Q(object_pk__in=primary_keys)).distinct() + if isinstance(primary_keys[0], integer_types): + return self.filter(content_type=content_type).filter(Q(object_id__in=primary_keys)).distinct() + else: + return self.filter(content_type=content_type).filter(Q(object_pk__in=primary_keys)).distinct() + def get_for_model(self, model): """