X-Git-Url: https://git.mdrn.pl/redakcja.git/blobdiff_plain/fc7a60feb9496c250cd1e8ca3db8c74267790047..8132fc186eb0c5fd02c86828c3a4735754296d02:/apps/toolbar/templatetags/toolbar_tags.py

diff --git a/apps/toolbar/templatetags/toolbar_tags.py b/apps/toolbar/templatetags/toolbar_tags.py
index 9957a939..0766677e 100644
--- a/apps/toolbar/templatetags/toolbar_tags.py
+++ b/apps/toolbar/templatetags/toolbar_tags.py
@@ -1,11 +1,18 @@
+# -*- coding: utf-8 -*-
+#
+# This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
+#
 from django import template
-# from toolbar import models
+from toolbar import models
 
 register = template.Library()
 
+
 @register.inclusion_tag('toolbar/toolbar.html')
-def toolbar(groups, extra):
-    return {'toolbar_groups': groups, 'toolbar_extra_group': extra}
+def toolbar():
+    return {'toolbar_groups': models.ButtonGroup.objects.all().select_related()}
+
 
 @register.inclusion_tag('toolbar/button.html')
 def toolbar_button(b):