Experimetal Woblink.

[redakcja.git] / src / depot / templatetags / depot.py

diff --git a/src/depot/templatetags/depot.py b/src/depot/templatetags/depot.py
new file mode 100644 (file)
index 0000000..d1194f0
--- /dev/null
+++ b/src/depot/templatetags/depot.py
@@ -0,0 +1,20 @@
+from django.template import Library
+from depot.models import Shop
+
+
+register = Library()
+
+
+@register.simple_tag(takes_context=True)
+def depot_shops(context, book):
+    shops = []
+    for shop in Shop.objects.all():
+        d = {
+            'shop_id': shop.id,
+            'name': shop.name,
+        }
+        d.update(shop.can_publish(book))
+        d['last'] = shop.get_last(book)
+        d['id'] = getattr(book, shop.shop + '_id')
+        shops.append(d)
+    return shops