mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-12 13:31:56 +00:00
perf: Add task handler for ops job with creator assignment
This commit is contained in:
@@ -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)
|
||||
|
||||
|
Reference in New Issue
Block a user