template fix

[wolnelektury.git] / apps / search / management / commands / reindex.py

diff --git a/apps/search/management/commands/reindex.py b/apps/search/management/commands/reindex.py
index 9639a5a..bce4708 100755 (executable)
--- a/apps/search/management/commands/reindex.py
+++ b/apps/search/management/commands/reindex.py
@@ -9,8 +9,17 @@ class Command(BaseCommand):
         import search
         idx = search.ReusableIndex()
         idx.open()
-        for b in Book.objects.all():
+
+        if args:
+            books = []
+            for a in args:
+                books += Book.objects.filter(slug=a).all()
+        else:
+            books = Book.objects.all()
+            
+        for b in books:
             print b.title
             idx.index_book(b, None)
         print 'Reindexing tags.'
         idx.index_tags()
+        idx.close()