Context manager to emphasize Django translated strings

debug_trans.py

import contextlib
from django.utils import translation
from django.utils.safestring import SafeData, SafeText


def prefix_decorator(function):
    def prefix_wrapper(*args, **kwargs):
        result = function(*args, **kwargs)
        wrapped_result = "[[ " + result + " ]]"
        # if isinstance(result, SafeData):
        #     wrapped_result = SafeText(wrapped_result)
        return wrapped_result

    return prefix_wrapper


class DebugTrans:
    methods = [
        "gettext",
        "ugettext",
        "ngettext",
        "ungettext",
        "pgettext",
        "npgettext",
    ]

    def __init__(self, trans):
        self.trans = trans

    def __getattr__(self, real_name):
        method = getattr(self.trans, real_name)
        if real_name in self.methods:
            method = prefix_decorator(method)
        return method


@contextlib.contextmanager
def debug_trans():
    orig_trans = translation._trans
    translation._trans = DebugTrans(orig_trans)
    yield
    translation._trans = orig_trans