76 lines
3.3 KiB
Python
76 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
|
|
from django.core.management.base import BaseCommand
|
|
from django.db import connection, transaction
|
|
from django.conf import settings
|
|
|
|
settings.RELOAD_CONFIG = False
|
|
import app.monkey_patch
|
|
from ... import models
|
|
|
|
|
|
class Command(BaseCommand):
|
|
"""
|
|
print sql statement to add trigram
|
|
"""
|
|
help = 'sql create statements for find tables to use trigram index'
|
|
args = ''
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument('--debug', action='store_true', dest='debug',
|
|
default=False, help='print sql commans')
|
|
|
|
def handle(self, **options):
|
|
cursor = connection.cursor()
|
|
|
|
def create_index(index, table, key):
|
|
sql = 'CREATE INDEX "%s" ON "%s" USING gin ("%s" gin_trgm_ops)' % (index, table, key)
|
|
if options['debug']:
|
|
print(sql)
|
|
cursor.execute(sql)
|
|
|
|
if settings.DB_GIN_TRGM:
|
|
import entity.models
|
|
import document.models
|
|
for table, column in (
|
|
(models.ItemFind._meta.db_table, 'value'), # Item Find
|
|
(models.Clip._meta.db_table, 'findvalue'), # Clip Find
|
|
(models.Annotation._meta.db_table, 'findvalue'), # Annotation Find
|
|
(entity.models.Find._meta.db_table, 'value'), # Entity Find
|
|
(document.models.Find._meta.db_table, 'value'), # Document Find
|
|
):
|
|
cursor = connection.cursor()
|
|
contraints = connection.introspection.get_constraints(cursor, table)
|
|
indexes = {
|
|
','.join(c['columns']): {'primary_key': c['primary_key'], 'unique': c['unique']}
|
|
for k, c in contraints.items()
|
|
if c['index'] or c['primary_key'] or c['unique']
|
|
}
|
|
drop = []
|
|
if column in indexes:
|
|
for sql in (
|
|
"SELECT indexname, indexdef FROM pg_catalog.pg_indexes " + \
|
|
"WHERE indexdef LIKE '%ON {table}%' AND indexdef LIKE '%{column}%'".format(table=table, column=column),
|
|
"SELECT indexname, indexdef FROM pg_catalog.pg_indexes " + \
|
|
"WHERE indexdef LIKE '%ON public.{table}%' AND indexdef LIKE '%{column}%'".format(table=table, column=column),
|
|
):
|
|
cursor.execute(sql)
|
|
for r in cursor:
|
|
if 'USING gin' not in r[1]:
|
|
drop.append(r[0])
|
|
if drop:
|
|
for idx in drop:
|
|
sql = 'DROP INDEX ' + idx
|
|
if options['debug']:
|
|
print(sql)
|
|
cursor.execute(sql)
|
|
contraints = connection.introspection.get_constraints(cursor, table)
|
|
indexes = {
|
|
','.join(c['columns']): {'primary_key': c['primary_key'], 'unique': c['unique']}
|
|
for k, c in contraints.items()
|
|
if c['index'] or c['primary_key'] or c['unique']
|
|
}
|
|
if column not in indexes:
|
|
create_index("%s_%s_idx" % (table, column), table, column)
|
|
transaction.commit()
|