Turned the cart template tag into a classytag, because being classy i…

…s good ;-)
mabo77 · Apr 5, 2011 · a943452 · a943452
1 parent 42082bb
commit a943452
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -16,6 +16,7 @@
     classifiers=CLASSIFIERS,
     install_requires=[
         'Django>=1.2',
+        'django-classy-tags>=0.3.3',
     ],
     packages=find_packages(exclude=["example", "example.*"]),
     zip_safe = False

diff --git a/shop/templatetags/shop_tags.py b/shop/templatetags/shop_tags.py
@@ -1,17 +1,22 @@
 # -*- coding: utf-8 -*-
 from django import template
+from classytags.helpers import InclusionTag
 
 from shop.util.cart import get_or_create_cart
 
+
 register = template.Library()
 
 
-@register.inclusion_tag('shop/templatetags/_cart.html', takes_context=True)
-def cart(context):
+class Cart(InclusionTag):
     '''
     Inclusion tag for displaying cart summary.
     '''
-    cart = get_or_create_cart(context['request'])
-    return {
-        'cart': cart,
-        }
+    template = 'shop/templatetags/_cart.html'
+
+    def get_context(self, context):
+        cart = get_or_create_cart(context['request'])
+        return {
+            'cart': cart
+        } 
+register.tag(Cart)