Esempio n. 1
0
class ResultJob(Job):
    """
    A job that stores the result of the execution in the DB
    """

    def __init__(self):
        self.properly_setup = False
        self.repeated_task = False

    def execute(self, *args, **kwargs):
        self.pre_run(*args, **kwargs)

        if not self.repeated_task:
            self.result = self.run(*args, **kwargs)
            self.post_run(*args, **kwargs)

    def pre_run(self, *args, **kwargs):
        super(ResultJob, self).pre_run(*args, **kwargs)

        # Take as id the left part from the hash symbol
        task_id = self.id.split('#')[0]
        try:
            self.job_status = JobStatus(
                job_id=task_id,
                job_name=self.__repr__()
            )
            self.job_status.save(force_insert=True)
            self.properly_setup = True
        except IntegrityError:
            self.job_status = JobStatus.objects.get(job_id=task_id)
            if self.job_status.status == JobStatus.FAILURE:
                self.job_status.status = JobStatus.PENDING
                self.job_status.save()
                self.properly_setup = True
            else:
                self.repeated_task = True

    def post_run(self, *args, **kwargs):
        self.job_status.date_done = datetime.now()
        self.job_status.result = json.dumps(self.result)
        self.job_status.save(force_update=True)
        super(ResultJob, self).post_run(*args, **kwargs)

    def on_success(self, *args, **kwargs):
        if not self.repeated_task:
            self.job_status.status = JobStatus.SUCCESS
            self.job_status.save(force_update=True)
            super(ResultJob, self).on_success(*args, **kwargs)

    def on_failure(self):
        if not self.repeated_task:
            if self.properly_setup:
                self.job_status.status = JobStatus.FAILURE
                self.job_status.save(force_update=True)
            super(ResultJob, self).on_failure()
Esempio n. 2
0
class ResultJob(Job):

    def __init__(self):
        self.properly_setup = False
        self.repeated_task = False

    def execute(self, *args, **kwargs):
        self.set_up(*args, **kwargs)
        if not self.repeated_task:
            self.result = self.run(*args, **kwargs)
            self.tear_down(*args, **kwargs)

    def set_up(self, *args, **kwargs):
        super(ResultJob, self).set_up(*args, **kwargs)
        # Take as id the left part from the hash symbol
        task_id = self.id.split('#')[0]
        try:
            self.job_status = JobStatus(
                job_id=task_id,
                job_name=self.__repr__()
            )
            self.job_status.save(force_insert=True)
            self.properly_setup = True
        except IntegrityError:
            self.job_status = JobStatus.objects.get(job_id=task_id)
            if self.job_status.status == JobStatus.FAILURE:
                self.job_status.status = JobStatus.PENDING
                self.job_status.save()
                self.properly_setup = True
            else:
                self.repeated_task = True

    @abstractmethod
    def run(self, *args, **kwargs):
        raise NotImplementedError

    def tear_down(self, *args, **kwargs):
        self.job_status.date_done = datetime.now()
        self.job_status.result = json.dumps(self.result)
        self.job_status.save(force_update=True)
        super(ResultJob, self).tear_down(*args, **kwargs)

    def on_success(self, *args, **kwargs):
        if not self.repeated_task:
            self.job_status.status = JobStatus.SUCCESS
            self.job_status.save(force_update=True)
            super(ResultJob, self).on_success(*args, **kwargs)

    def on_failure(self, *args, **kwargs):
        if not self.repeated_task:
            if self.properly_setup:
                self.job_status.status = JobStatus.FAILURE
                self.job_status.save(force_update=True)
            super(ResultJob, self).on_failure(*args, **kwargs)
Esempio n. 3
0
class ResultJob(Job):
    """
    A job that stores the result of the execution in the DB
    """
    def __init__(self):
        self.properly_setup = False
        self.repeated_task = False

    def execute(self, *args, **kwargs):
        self.pre_run(*args, **kwargs)

        if not self.repeated_task:
            self.result = self.run(*args, **kwargs)
            self.post_run(*args, **kwargs)

    def pre_run(self, *args, **kwargs):
        super(ResultJob, self).pre_run(*args, **kwargs)

        # Take as id the left part from the hash symbol
        task_id = self.id.split('#')[0]
        try:
            self.job_status = JobStatus(job_id=task_id,
                                        job_name=self.__repr__())
            self.job_status.save(force_insert=True)
            self.properly_setup = True
        except IntegrityError:
            self.job_status = JobStatus.objects.get(job_id=task_id)
            if self.job_status.status == JobStatus.FAILURE:
                self.job_status.status = JobStatus.PENDING
                self.job_status.save()
                self.properly_setup = True
            else:
                self.repeated_task = True

    def post_run(self, *args, **kwargs):
        self.job_status.date_done = datetime.now()
        self.job_status.result = json.dumps(self.result)
        self.job_status.save(force_update=True)
        super(ResultJob, self).post_run(*args, **kwargs)

    def on_success(self, *args, **kwargs):
        if not self.repeated_task:
            self.job_status.status = JobStatus.SUCCESS
            self.job_status.save(force_update=True)
            super(ResultJob, self).on_success(*args, **kwargs)

    def on_failure(self):
        if not self.repeated_task:
            if self.properly_setup:
                self.job_status.status = JobStatus.FAILURE
                self.job_status.save(force_update=True)
            super(ResultJob, self).on_failure()
Esempio n. 4
0
    def pre_run(self, *args, **kwargs):
        super(ResultJob, self).pre_run(*args, **kwargs)

        # Take as id the left part from the hash symbol
        task_id = self.id.split('#')[0]
        try:
            self.job_status = JobStatus(job_id=task_id,
                                        job_name=self.__repr__())
            self.job_status.save(force_insert=True)
            self.properly_setup = True
        except IntegrityError:
            self.job_status = JobStatus.objects.get(job_id=task_id)
            if self.job_status.status == JobStatus.FAILURE:
                self.job_status.status = JobStatus.PENDING
                self.job_status.save()
                self.properly_setup = True
            else:
                self.repeated_task = True
Esempio n. 5
0
 def set_up(self, *args, **kwargs):
     super(ResultJob, self).set_up(*args, **kwargs)
     # Take as id the left part from the hash symbol
     task_id = self.id.split('#')[0]
     try:
         self.job_status = JobStatus(
             job_id=task_id,
             job_name=self.__repr__()
         )
         self.job_status.save(force_insert=True)
         self.properly_setup = True
     except IntegrityError:
         self.job_status = JobStatus.objects.get(job_id=task_id)
         if self.job_status.status == JobStatus.FAILURE:
             self.job_status.status = JobStatus.PENDING
             self.job_status.save()
             self.properly_setup = True
         else:
             self.repeated_task = True