Refaktor przyciskow.

[redakcja.git] / apps / toolbar / templatetags / toolbar_tags.py

diff --git a/apps/toolbar/templatetags/toolbar_tags.py b/apps/toolbar/templatetags/toolbar_tags.py
index a38ed01..9957a93 100644 (file)
--- a/apps/toolbar/templatetags/toolbar_tags.py
+++ b/apps/toolbar/templatetags/toolbar_tags.py
@@ -1,9 +1,12 @@
 from django import template
-from toolbar import models
+# from toolbar import models
 
 register = template.Library()
 
 @register.inclusion_tag('toolbar/toolbar.html')
-def toolbar():
-    groups = models.ButtonGroup.objects.all()
-    return {'groups': groups}
+def toolbar(groups, extra):
+    return {'toolbar_groups': groups, 'toolbar_extra_group': extra}
+
+@register.inclusion_tag('toolbar/button.html')
+def toolbar_button(b):
+    return {'button': b}