Fixed #9029 -- Allow use of fieldname_id with get_or_create. Thanks to aaron and mrmachine.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@15156 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Andrew Godwin 2011-01-08 13:27:30 +00:00
parent 32a344dfe2
commit 81e05a418d
2 changed files with 16 additions and 3 deletions

View file

@ -369,9 +369,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])
@ -384,7 +388,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