How to create a custom mixin in django?

Question

I have a decorator which prevents the user from accessing URLs if the product with id=1 is not activated.

I want to create a mixin similar to this.

This is my decorator:

from django.core.exceptions import PermissionDenied
from ecommerce_integration.models import Product 


def product_1_activation(function):
    def wrap(request, *args, **kwargs):
        products = Product.objects.filter(pk=1, activate=True)
        if products:
            return function(request, *args, **kwargs)
        else:
            raise PermissionDenied
    wrap.__doc__ = function.__doc__
    wrap.__name__ = function.__name__
    return wrap

Any idea how to create a custom mixin similar to the above decorator?

Actually I want to create mixins for my Class Based View.

Can anyone help me out in this?

Answer 1

We can make a Mixin that just overrides the dispatch method, like:

class ProductExistsRequiredMixin:

    def dispatch(self, request, *args, **kwargs):
        if Product.objects.filter(pk=1, activate=True):
            return super().dispatch(request, *args, **kwargs)
        else:
            raise PermissionDenied

and then use it in a view like:

class MyCustomView(ProductExistsRequiredMixin, View):

    # ...