cnr-ibba / IMAGE-InjectTool

        return 'success'

class BatchUpdateSamples(MyTask, BatchUpdateMixin):

    name = "Batch update samples"

    description = """Batch update of field in samples"""

    item_cls = Sample

    submission_cls = Submission

    # Ovverride default on failure method

    # This is not a failed validation for a wrong value, this is an

    # error in task that mean an error in coding

    def on_failure(self, exc, task_id, args, kwargs, einfo):

        logger.error('{0!r} failed: {1!r}'.format(task_id, exc))

        # get submission object

        submission_obj = Submission.objects.get(pk=args[0])

        # mark submission with ERROR

        submission_obj.status = ERROR

        submission_obj.message = ("Error in batch update for samples: %s"

                                  % (str(exc)))

        submission_obj.save()

        send_message(submission_obj)

        # send a mail to the user with the stacktrace (einfo)

        submission_obj.owner.email_user(

            "Error in batch update for samples: %s" % (args[0]),

            ("Something goes wrong in batch update for samples. Please report "

             "this to InjectTool team\n\n %s" % str(einfo)),

        )

        # TODO: submit mail to admin

    def run(self, submission_id, sample_ids, attribute):

        """Function for batch update attribute in samples

        Args:

            submission_id (int): id of submission

            sample_ids (dict): dict with id and values to update

            attribute (str): attribute to update

        """

        logger.info("Start batch update for samples")

        super(BatchUpdateSamples, self).batch_update(submission_id, sample_ids,

                                                     attribute)

        return 'success'

# register explicitly tasks

        return 'success'

class BatchUpdateAnimals(MyTask, BatchUpdateMixin):

    name = "Batch update animals"

    description = """Batch update of field in animals"""

    item_cls = Animal

    submission_cls = Submission

    # Ovverride default on failure method

    # This is not a failed validation for a wrong value, this is an

    # error in task that mean an error in coding

    def on_failure(self, exc, task_id, args, kwargs, einfo):

        logger.error('{0!r} failed: {1!r}'.format(task_id, exc))

        # get submission object

        submission_obj = Submission.objects.get(pk=args[0])

        # mark submission with ERROR

        submission_obj.status = ERROR

        submission_obj.message = ("Error in batch update for animals: %s"

                                  % (str(exc)))

        submission_obj.save()

        send_message(submission_obj)

        # send a mail to the user with the stacktrace (einfo)

        submission_obj.owner.email_user(

            "Error in batch update for animals: %s" % (args[0]),

            ("Something goes wrong  in batch update for animals. Please "

             "report this to InjectTool team\n\n %s" % str(einfo)),

        )

        # TODO: submit mail to admin

    def run(self, submission_id, animal_ids, attribute):

        """Function for batch update attribute in animals

        Args:

            submission_id (int): id of submission

            animal_ids (dict): dict with id and values to update

            attribute (str): attribute to update

        """

        logger.info("Start batch update for animals")

        super(BatchUpdateAnimals, self).batch_update(submission_id, animal_ids,

                                                     attribute)

        return 'success'

# register explicitly tasks