django-content-plugins/content_plugins/plugins/mixins.py

import os

from django.db import models
from django.utils.translation import ugettext_lazy as _


class StyleField(models.CharField):
    """
    Allows overriding of STYLE_CHOICES in subclasses.
    """

    def contribute_to_class(self, cls, name, **kwargs):
        if hasattr(cls, 'STYLE_CHOICES'):
            self.choices = cls.STYLE_CHOICES
        super().contribute_to_class(cls, name, **kwargs)


class StyleMixin(models.Model):
    STYLE_CHOICES = tuple()
    style = StyleField(_("style"), max_length=50, null=True, blank=True)

    class Meta:
        abstract = True

    def get_style_slug(self):
        return getattr(self, 'style', None) or 'default'

    # # Compatibiliy with super classes not having a prefixed_path method
    # def prefixed_path(self, path):
    #     if hasattr(super(), 'prefixed_path'):
    #         return super().prefixed_path(path)
    #     else:
    #         return path

    def add_styled_template_names(self, template_names):
        """
        if super().get_template_names():
            [
                "article/_richtext.html",
                "_richtext.html",
            ]
        then add_styled_template_names returns
            [
                "article/_richtext/_<style>.html",
                "article/_richtext.html",
                "_richtext/_<style>.html",
                "_richtext.html",
            ]
        """
        extended_template_names = []
        for t in template_names:
            name, ext = os.path.splitext(t)
            extended_template_names += [
                "{name}/_{style}.{ext}".format(
                    name=name, style=self.style, ext=ext),
                t
            ]
        return extended_template_names

    def get_template_names(self):
        if hasattr(super(), 'get_template_names'):
            template_names = super().get_template_names()
            if self.style:
                template_names = self.add_styled_template_names(template_names)
            return template_names
        else:
            return []

    def get_plugin_context(self, context=None, **kwargs):
        if hasattr(super(), 'get_plugin_context'):
            context = super().get_plugin_context(**kwargs)
        else:
            context = {}
        context['style'] = self.get_style_slug()
        return context