Hopiu/django-celery-monitor
1
0
Fork 0
mirror of https://github.com/jazzband/django-celery-monitor.git synced 2026-03-16 22:00:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Improve code path for updating task state, use update_fields parameter.

Browse source
This commit is contained in:
Jannis Leidel 2017-05-10 15:36:06 +02:00
parent b11aa0e380
commit cfcc54bc82
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
django_celery_monitor/managers.py
View file

@ -98,14 +98,13 @@ class TaskStateQuerySet(ExtendedQuerySet):
)
if created:
return obj
if states.state(state) < states.state(obj.state):
keep = Task.merge_rules[states.RECEIVED]
else:
if states.state(state) < states.state(obj.state):
keep = Task.merge_rules[states.RECEIVED]
defaults = dict(
(k, v) for k, v in defaults.items()
if k not in keep
)
for k, v in defaults.items():
setattr(obj, k, v)
obj.save()
return obj
keep = {}
for key, value in defaults.items():
if key not in keep:
setattr(obj, key, value)
obj.save(update_fields=tuple(defaults.keys()))
return obj