Class based template registry

class_based_template_tag.py

import re
from django import template


class BaseSimpleTag:
    def __init__(self, context: dict):
        self.context = context
        self.request = context["request"]

    def __str__(self):
        return str(self.get())

    def get(self):
        raise NotImplementedError


class TemplateLibraryWithClasses(template.Library):
    def class_simple_tag(self, cls: BaseSimpleTag.__class__):
        # Convert from CamelCase to snake_case
        name = re.sub("(?!^)([A-Z]+)", r"_\1", cls.__name__).lower()

        def wrapper(context):
            return cls(context).get()

        self.simple_tag(wrapper, takes_context=True, name=name)
        return cls