X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/d90edc4826d6c3d3a40f52fa5c88cf4b1741ab63..c925a324ddc30bc183555c4e3edff89a560f6323:/apps/search/management/commands/reindex.py?ds=sidebyside

diff --git a/apps/search/management/commands/reindex.py b/apps/search/management/commands/reindex.py
index 9639a5ae7..890110238 100755
--- a/apps/search/management/commands/reindex.py
+++ b/apps/search/management/commands/reindex.py
@@ -1,16 +1,33 @@
 from django.core.management.base import BaseCommand
 
+from optparse import make_option
 class Command(BaseCommand):
     help = 'Reindex everything.'
     args = ''
-
+    
+    option_list = BaseCommand.option_list + (
+        make_option('-n', '--book-id', action='store_true', dest='book_id', default=False,
+            help='book id'),
+    )
     def handle(self, *args, **opts):
         from catalogue.models import Book
         import search
         idx = search.ReusableIndex()
         idx.open()
-        for b in Book.objects.all():
+
+        if args:
+            books = []
+            for a in args:
+                if opts['book_id']:
+                    books += Book.objects.filter(id=int(a)).all()
+                else:
+                    books += Book.objects.filter(slug=a).all()
+        else:
+            books = Book.objects.all()
+            
+        for b in books:
             print b.title
-            idx.index_book(b, None)
+            idx.index_book(b)
         print 'Reindexing tags.'
         idx.index_tags()
+        idx.close()