diff --git a/admin_interface/templatetags/admin_interface_tags.py b/admin_interface/templatetags/admin_interface_tags.py index 4957d91..9a09d22 100644 --- a/admin_interface/templatetags/admin_interface_tags.py +++ b/admin_interface/templatetags/admin_interface_tags.py @@ -1,21 +1,20 @@ # -*- coding: utf-8 -*- -from django import template +from django import template, VERSION from admin_interface.models import Theme from admin_interface.version import __version__ register = template.Library() -try: - assignment_tag = register.assignment_tag -except AttributeError: - assignment_tag = register.simple_tag +if VERSION < (1, 9): + simple_tag = register.assignment_tag +else: + simple_tag = register.simple_tag -@assignment_tag(takes_context=True) +@simple_tag(takes_context=True) def get_admin_interface_theme(context): - theme = None request = context.get('request', None) @@ -31,6 +30,6 @@ def get_admin_interface_theme(context): return theme -@assignment_tag(takes_context=False) +@simple_tag(takes_context=False) def get_admin_interface_version(): return __version__