From 2b90267fa827439a29cb016b8548aec23f3b07c2 Mon Sep 17 00:00:00 2001 From: Andrew Godwin Date: Sat, 8 Jan 2011 13:40:23 +0000 Subject: [1.2.X] Fixed #9029 -- Allow use of fieldname_id with get_or_create. Thanks to aaron and mrmachine. Backport of [15156] from trunk git-svn-id: http://code.djangoproject.com/svn/django/branches/releases/1.2.X@15157 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- django/db/models/query.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) (limited to 'django') diff --git a/django/db/models/query.py b/django/db/models/query.py index 362114b535..d1e18a80c0 100644 --- a/django/db/models/query.py +++ b/django/db/models/query.py @@ -367,9 +367,13 @@ class QuerySet(object): assert kwargs, \ 'get_or_create() must be passed at least one keyword argument' defaults = kwargs.pop('defaults', {}) + lookup = kwargs.copy() + for f in self.model._meta.fields: + if f.column in lookup: + lookup[f.name] = lookup.pop(f.column) try: self._for_write = True - return self.get(**kwargs), False + return self.get(**lookup), False except self.model.DoesNotExist: try: params = dict([(k, v) for k, v in kwargs.items() if '__' not in k]) @@ -382,7 +386,7 @@ class QuerySet(object): except IntegrityError, e: transaction.savepoint_rollback(sid, using=self.db) try: - return self.get(**kwargs), False + return self.get(**lookup), False except self.model.DoesNotExist: raise e -- cgit v1.3