1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import sys
from django.core.management.base import BaseCommand
import django.apps
class Command(BaseCommand):
help = "./manage.py update_search_vectors\n\n"\
"Update full texte search vectors."
def add_arguments(self, parser):
parser.add_argument(
'--model', type=str, default='', dest='model',
help='Specific model to update')
parser.add_argument(
'--quiet', dest='quiet', action='store_true',
help='Quiet output')
def handle(self, *args, **options):
quiet = options['quiet']
nb_total = 0
for model in django.apps.apps.get_models():
if options['model'] and model.__name__ != options['model']:
continue
if hasattr(model, "update_search_vector") and \
(getattr(model, "BASE_SEARCH_VECTORS", None) or
getattr(model, "INT_SEARCH_VECTORS", None) or
getattr(model, "M2M_SEARCH_VECTORS", None) or
getattr(model, "PARENT_SEARCH_VECTORS", None)):
if not quiet:
self.stdout.write("\n* update {}".format(model))
total = model.objects.count()
idx = 0
for idx, item in enumerate(model.objects.all()):
if not quiet:
sys.stdout.write("\r{}/{} ".format(idx + 1, total))
sys.stdout.flush()
try:
item.update_search_vector()
except:
pass
nb_total += idx
if not quiet:
self.stdout.write("\n")
self.stdout.write("{} items updated\n".format(nb_total))
|
