fix in librarian
[wolnelektury.git] / apps / reporting / templatetags / reporting_stats.py
index 6f20c08..7bbe23d 100755 (executable)
@@ -2,13 +2,9 @@
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
-import feedparser
 from functools import wraps
-import datetime
-
 from django import template
-
-from catalogue.models import Book, BookMedia
+from catalogue.models import Book
 
 
 register = template.Library()
@@ -50,11 +46,11 @@ def count_books_all():
 
 @register_counter
 def count_books():
-    return Book.objects.exclude(html_file='').count()
+    return Book.objects.filter(children=None).count()
 
 @register_counter
 def count_books_parent():
-    return Book.objects.filter(html_file='').count()
+    return Book.objects.exclude(children=None).count()
 
 @register_counter
 def count_books_root():