+import sys
from django.core.management.base import BaseCommand
+from optparse import make_option
+
+def query_yes_no(question, default="yes"):
+ """Ask a yes/no question via raw_input() and return their answer.
+
+ "question" is a string that is presented to the user.
+ "default" is the presumed answer if the user just hits <Enter>.
+ It must be "yes" (the default), "no" or None (meaning
+ an answer is required of the user).
+
+ The "answer" return value is one of "yes" or "no".
+ """
+ valid = {"yes":True, "y":True, "ye":True,
+ "no":False, "n":False}
+ if default == None:
+ prompt = " [y/n] "
+ elif default == "yes":
+ prompt = " [Y/n] "
+ elif default == "no":
+ prompt = " [y/N] "
+ else:
+ raise ValueError("invalid default answer: '%s'" % default)
+
+ while True:
+ sys.stdout.write(question + prompt)
+ choice = raw_input().lower()
+ if default is not None and choice == '':
+ return valid[default]
+ elif choice in valid:
+ return valid[choice]
+ else:
+ sys.stdout.write("Please respond with 'yes' or 'no' "\
+ "(or 'y' or 'n').\n")
+
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 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
import search
- idx = search.ReusableIndex()
- idx.open()
- for b in Book.objects.all():
- print b.title
- idx.index_book(b, None)
+ idx = search.Index()
+
+ 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 = list(Book.objects.all())
+
+ while books:
+ try:
+ b = books[0]
+ print b.title
+ idx.index_book(b)
+ idx.index.commit()
+ books.pop(0)
+ except Exception, e:
+ print "Error occured: %s" % e
+ try:
+ # we might not be able to rollback
+ idx.index.rollback()
+ except:
+ pass
+ retry = query_yes_no("Retry?")
+ if not retry:
+ break
+
print 'Reindexing tags.'
idx.index_tags()
+ idx.index.commit()