Hopiu/django-model-utils
1
0
Fork 0
mirror of https://github.com/Hopiu/django-model-utils.git synced 2026-03-17 04:10:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update implementation of TimeStampedModel.save to handle various ways of specifying 'update_fields'

Browse source
This commit is contained in:
Arseniy Panfilov 2020-09-30 21:31:24 -04:00 committed by Asif Saif Uddin
parent fdd3af34d9
commit e9c45c71d1
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
model_utils/models.py
View file

@ -34,8 +34,12 @@ class TimeStampedModel(models.Model):
modified field is updated even if it is not given as
a parameter to the update field argument.
"""
if 'update_fields' in kwargs and 'modified' not in kwargs['update_fields']:
kwargs['update_fields'] += ['modified']
update_fields = kwargs.get('update_fields', None)
if update_fields is not None:
update_fields = set(update_fields)
if update_fields:
kwargs['update_fields'] = update_fields.union({'modified'})
super().save(*args, **kwargs)
class Meta: