Python Entrouvert

from django.core.urlresolvers import RegexURLPattern, RegexURLResolver

'''Decorators to apply a decorator to all views inside an urlpattern'''

__all__ = [ 'decorated_includes' ]

class DecoratedURLPattern(RegexURLPattern):

    def resolve(self, *args, **kwargs):

        result = super(DecoratedURLPattern, self).resolve(*args, **kwargs)

        if result:

            result.func = self._decorate_with(result.func)

        return result

class DecoratedRegexURLResolver(RegexURLResolver):

    def resolve(self, *args, **kwargs):

        result = super(DecoratedRegexURLResolver, self).resolve(*args, **kwargs)

        if result:

            result.func = self._decorate_with(result.func)

        return result

def decorated_includes(func, includes, *args, **kwargs):

    urlconf_module, app_name, namespace = includes

    for item in urlconf_module:

        if isinstance(item, RegexURLPattern):

            item.__class__ = DecoratedURLPattern

            item._decorate_with = func

        elif isinstance(item, RegexURLResolver):

            item.__class__ = DecoratedRegexURLResolver

            item._decorate_with = func

    return urlconf_module, app_name, namespace