Merge pull request #4 from hecrogon/bannertags

banner_tags as a inclusion tag
Arpaso · Jul 21, 2014 · a938825 · a938825
2 parents 4bd095c + a2bdf96
commit a938825
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/src/banner/templatetags/banner_tags.py b/src/banner/templatetags/banner_tags.py
@@ -1,16 +1,16 @@
 ### -*- coding: utf-8 -*- ####################################################
 
 from django.conf import settings
-from native_tags.decorators import function
+from django import template
+from banner.models import Banner
 
-from ..models import Banner
+register = template.Library()
 
+@register.inclusion_tag('banner_source.html')
 def get_banner(slug, width, height, queryset=Banner.objects.all()):
     banner = queryset.get_or_create(slug=slug.encode('utf-8'))[0]
-    return "banner_source.html", {
-                'banner': banner, 'size': "%ix%i" % (width, height), 
-                'width': width, 'height': height, 
-                'STATIC_URL': settings.STATIC_URL,
+    return {
+        'banner': banner, 'size': "%ix%i" % (width, height), 
+        'width': width, 'height': height, 
+        'STATIC_URL': settings.STATIC_URL,
     }
-
-get_banner = function(get_banner, inclusion=True, cache=60)