perf: Add task handler for ops job with creator assignment

This commit is contained in:
wangruidong
2024-11-07 18:39:39 +08:00
committed by Bryan
parent 5b27acf4ef
commit dd5a272cdf
2 changed files with 13 additions and 1 deletions

View File

@@ -105,6 +105,17 @@ class CeleryTaskExecution(models.Model):
def is_success(self):
return self.state == 'SUCCESS'
def set_creator_if_need(self):
from ops.models import Job
if self.creator:
return
if self.name == 'ops.tasks.run_ops_job' and self.args:
job_id = self.args[0]
job = Job.objects.filter(id=job_id).first()
if job:
self.creator = job.creator
self.save()
def __str__(self):
return "{}: {}".format(self.name, self.id)