Add full context to the create_referral tag

pinax · Aug 6, 2012 · a1cf304 · a1cf304
1 parent 774e558
commit a1cf304
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/anafero/templatetags/anafero_tags.py b/anafero/templatetags/anafero_tags.py
@@ -8,12 +8,17 @@
 register = template.Library()
 
 
-@register.inclusion_tag("anafero/_create_referral_form.html")
+@register.inclusion_tag("anafero/_create_referral_form.html", takes_context=True)
-def create_referral(url, obj=None):
+def create_referral(context, url, obj=None):
     if obj:
-        return {"url": url, "obj": obj, "obj_ct": ContentType.objects.get_for_model(obj)}
+        context.update(
+            {"url": url, "obj": obj, "obj_ct": ContentType.objects.get_for_model(obj)}
+        )
     else:
-        return {"url": url, "obj": "", "obj_ct": ""}
+        context.update(
+            {"url": url, "obj": "", "obj_ct": ""}
+        )
+    return context
 
 
 @register.assignment_tag