cnr-ibba / IMAGE-InjectTool

class DeleteMaterialMixin(OwnerMixin):

    """A common DeleteMixin for Material classes (Sample/Animal)"""

    def dispatch(self, request, *args, **kwargs):

        handler = super(DeleteMaterialMixin, self).dispatch(

                request, *args, **kwargs)

        # here I've done get_queryset. Check for submission status

        if hasattr(self, "object") and not self.object.can_edit():

            message = "Cannot delete %s: submission status is: %s" % (

                    self.object, self.object.submission.get_status_display())

            logger.warning(message)

            messages.warning(

                request=self.request,

                message=message,

                extra_tags="alert alert-dismissible alert-warning")

            return redirect(self.object.get_absolute_url())

        return handler

    def get_success_url(self):

        return reverse(

    # override this attribute with a real validation class

    validationresult_class = None

    def dispatch(self, request, *args, **kwargs):

        handler = super(UpdateMaterialMixin, self).dispatch(

                request, *args, **kwargs)

        # here I've done get_queryset. Check for submission status

        if hasattr(self, "object") and not self.object.can_edit():

            message = "Cannot edit %s: submission status is: %s" % (

                    self.object, self.object.submission.get_status_display())

            logger.warning(message)

            messages.warning(

                request=self.request,

                message=message,

                extra_tags="alert alert-dismissible alert-warning")

            return redirect(self.object.get_absolute_url())

        return handler

    # add the request to the kwargs

    # https://chriskief.com/2012/12/18/django-modelform-formview-and-the-request-object/

    template_name = "submissions/submission_confirm_delete.html"

    success_url = reverse_lazy('image_app:dashboard')

    def dispatch(self, request, *args, **kwargs):

        handler = super(DeleteSubmissionView, self).dispatch(

                request, *args, **kwargs)

        # here I've done get_queryset. Check for submission status

        if hasattr(self, "object") and not self.object.can_edit():

            message = "Cannot delete %s: submission status is: %s" % (

                    self.object, self.object.get_status_display())

            logger.warning(message)

            messages.warning(

                request=self.request,

                message=message,

                extra_tags="alert alert-dismissible alert-warning")

            return redirect(self.object.get_absolute_url())

        return handler

    # https://stackoverflow.com/a/39533619/4385116

    def get_context_data(self, **kwargs):

                Q(animal__isnull=False) | Q(sample__isnull=False))

            ).order_by('id')

    def dispatch(self, request, *args, **kwargs):

        handler = super(EditSubmissionView, self).dispatch(

                request, *args, **kwargs)

        # here I've done get_queryset. Check for submission status

        if hasattr(self, "submission") and not self.submission.can_edit():

            message = "Cannot edit submission: current status is: %s" % (

                    self.submission.get_status_display())

            logger.warning(message)

            messages.warning(

                request=self.request,

                message=message,

                extra_tags="alert alert-dismissible alert-warning")

            return redirect(self.submission.get_absolute_url())

        return handler

    def get_context_data(self, **kwargs):

        # Call the base implementation first to get a context