From 0f76fabe3eb1842b6515d2571844bd63e894d07c Mon Sep 17 00:00:00 2001
From: Marcin Koziej <marcin@lolownia.org>
Date: Thu, 2 Feb 2012 12:52:23 +0100
Subject: [PATCH] add -t to reindex command to just reindex tags

---
 apps/search/index.py                       |  1 -
 apps/search/management/commands/reindex.py | 31 ++++++++++++----------
 2 files changed, 17 insertions(+), 15 deletions(-)

diff --git a/apps/search/index.py b/apps/search/index.py
index 1a36993d6..ead10b5dd 100644
--- a/apps/search/index.py
+++ b/apps/search/index.py
@@ -234,7 +234,6 @@ class Index(BaseIndex):
         for pdtag in PDCounterBook.objects.all():
             doc = Document()
             doc.add(NumericField("tag_id", Field.Store.YES, True).setIntValue(int(pdtag.id)))
-            print pdtag.title
             doc.add(Field("tag_name", pdtag.title, Field.Store.NO, Field.Index.ANALYZED))
             doc.add(Field("tag_name_pl", pdtag.title, Field.Store.NO, Field.Index.ANALYZED))
             doc.add(Field("tag_category", 'pd_book', Field.Store.YES, Field.Index.NOT_ANALYZED))
diff --git a/apps/search/management/commands/reindex.py b/apps/search/management/commands/reindex.py
index 890110238..3b6517f2e 100755
--- a/apps/search/management/commands/reindex.py
+++ b/apps/search/management/commands/reindex.py
@@ -7,7 +7,9 @@ class Command(BaseCommand):
     
     option_list = BaseCommand.option_list + (
         make_option('-n', '--book-id', action='store_true', dest='book_id', default=False,
-            help='book id'),
+            help='book id instead of slugs'),
+        make_option('-t', '--just-tags', action='store_true', dest='just_tags', default=False,
+            help='just reindex tags'),
     )
     def handle(self, *args, **opts):
         from catalogue.models import Book
@@ -15,19 +17,20 @@ class Command(BaseCommand):
         idx = search.ReusableIndex()
         idx.open()
 
-        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)
+        if not opts['just_tags']:
+            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)
         print 'Reindexing tags.'
         idx.index_tags()
         idx.close()
-- 
2.20.1