from django.core.management import BaseCommand
from whoosh.writing import AsyncWriter

import documents.index as index
from documents.mixins import Renderable
from documents.models import Document


class Command(Renderable, BaseCommand):

    help = "Recreates the document index"

    def __init__(self, *args, **kwargs):
        self.verbosity = 0
        BaseCommand.__init__(self, *args, **kwargs)

    def handle(self, *args, **options):

        self.verbosity = options["verbosity"]

        documents = Document.objects.all()

        ix = index.open_index(recreate=True)

        with AsyncWriter(ix) as writer:
            for document in documents:
                index.update_document(writer, document)