Skip to content

Commit

Permalink
Fixed #20272 - Moved update_fields existence check into Model._do_upd…
Browse files Browse the repository at this point in the history
…ate.

Thanks Gavin Wahl.
  • Loading branch information
Tim Graham committed May 30, 2013
1 parent 5923581 commit 616f3c4
Showing 1 changed file with 11 additions and 11 deletions.
22 changes: 11 additions & 11 deletions django/db/models/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -633,15 +633,7 @@ def _save_table(self, raw=False, cls=None, force_insert=False,
base_qs = cls._base_manager.using(using)
values = [(f, None, (getattr(self, f.attname) if raw else f.pre_save(self, False)))
for f in non_pks]
if not values:
# We can end up here when saving a model in inheritance chain where
# update_fields doesn't target any field in current model. In that
# case we just say the update succeeded. Another case ending up here
# is a model with just PK - in that case check that the PK still
# exists.
updated = update_fields is not None or base_qs.filter(pk=pk_val).exists()
else:
updated = self._do_update(base_qs, using, pk_val, values)
updated = self._do_update(base_qs, using, pk_val, values, update_fields)
if force_update and not updated:
raise DatabaseError("Forced update did not affect any rows.")
if update_fields and not updated:
Expand All @@ -665,13 +657,21 @@ def _save_table(self, raw=False, cls=None, force_insert=False,
setattr(self, meta.pk.attname, result)
return updated

def _do_update(self, base_qs, using, pk_val, values):
def _do_update(self, base_qs, using, pk_val, values, update_fields):
"""
This method will try to update the model. If the model was updated (in
the sense that an update query was done and a matching row was found
from the DB) the method will return True.
"""
return base_qs.filter(pk=pk_val)._update(values) > 0
if not values:
# We can end up here when saving a model in inheritance chain where
# update_fields doesn't target any field in current model. In that
# case we just say the update succeeded. Another case ending up here
# is a model with just PK - in that case check that the PK still
# exists.
return update_fields is not None or base_qs.filter(pk=pk_val).exists()
else:
return base_qs.filter(pk=pk_val)._update(values) > 0

def _do_insert(self, manager, using, fields, update_pk, raw):
"""
Expand Down

0 comments on commit 616f3c4

Please sign in to comment.