X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/2847eae5fd8d22c713e88fc8dbb37eb06a191e37..d2199704c75d29ba98e0f92d27cb63743183e906:/apps/catalogue/templatetags/catalogue_tags.py

diff --git a/apps/catalogue/templatetags/catalogue_tags.py b/apps/catalogue/templatetags/catalogue_tags.py
index d6a795fb4..2020d9097 100644
--- a/apps/catalogue/templatetags/catalogue_tags.py
+++ b/apps/catalogue/templatetags/catalogue_tags.py
@@ -361,8 +361,9 @@ def book_mini(book):
 @register.inclusion_tag('catalogue/work-list.html', takes_context=True)
 def work_list(context, object_list):
     request = context.get('request')
-    if object_list:
-        object_type = type(object_list[0]).__name__
+    for obj in object_list:
+        obj.object_type = type(object_list[0]).__name__
+
     return locals()
 
 
@@ -468,3 +469,9 @@ def license_icon(license_url):
 @register.simple_tag
 def related_tag_name(tag, lang=None):
     return _related_tag_name(tag, lang)
+
+
+@register.simple_tag
+def class_name(obj):
+    return obj.__class__.__name__
+