From cc2e489f756a48e4b5cb5af05a3bc406b3a75add Mon Sep 17 00:00:00 2001 From: "John R. Tipton" Date: Thu, 19 Feb 2015 09:03:39 -0500 Subject: [PATCH] - Check for model using isinstance - Check for model using isinstance and pk using _get_pk_value --- src/auditlog/models.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/auditlog/models.py b/src/auditlog/models.py index 12f8f5f..64051ee 100644 --- a/src/auditlog/models.py +++ b/src/auditlog/models.py @@ -65,9 +65,9 @@ class LogEntryManager(models.Manager): pk_field = instance._meta.pk.name pk = getattr(instance, pk_field, None) - # Check to make sure that we got an pk not a class object. - if hasattr(pk, '_meta'): - pk = getattr(instance, '%s_id' % pk_field, None) + # Check to make sure that we got an pk not a model object. + if isinstance(pk, models.Model): + pk = self._get_pk_value(pk) return pk