Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-05-16 03:03:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added opton to update_index to allow user to select which backend to update

Browse source
This commit is contained in:
Karl Hobley 2014-08-28 16:28:24 +01:00
parent deb2f35f1a
commit 64245e9c1a
1 changed files with 14 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
wagtail/wagtailsearch/management/commands/update_index.py
View file

@ -1,3 +1,5 @@
from optparse import make_option
from django.core.management.base import BaseCommand
from django.db import models
from django.conf import settings
@ -77,6 +79,15 @@ class Command(BaseCommand):
self.stdout.write(backend_name + ": Refreshing index")
backend.refresh_index()
option_list = BaseCommand.option_list + (
make_option('--backend',
action='store',
dest='backend_name',
default=False,
help="Specify a backend to update",
),
)
def handle(self, **options):
# Get object list
models, object_list = self.get_object_list()
@ -84,6 +95,9 @@ class Command(BaseCommand):
# Update backends
if 'backend' in options:
self.update_backend(options['backend'], models, object_list)
elif 'backend_name' in options:
backend = dict(get_search_backends())[options['backend_name']]
self.update_backend(backend, models, object_list, backend_name=options['backend_name'])
else:
for backend_name, backend in get_search_backends():
self.update_backend(backend, models, object_list, backend_name=backend_name)