pandora/pandora/item/management/commands/sqlfindindex.py

# -*- coding: utf-8 -*-
# vi:si:et:sw=4:sts=4:ts=4
from __future__ import print_function

from optparse import make_option

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 = ''
    option_list = BaseCommand.option_list + (
        make_option('--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
            for table_name, name in (
                (models.ItemFind._meta.db_table, 'value'),   # Item Find
                (models.Clip._meta.db_table, 'findvalue'),   # Clip Find
                (entity.models.Find._meta.db_table, 'value'),# Entity Find
            ):
                cursor = connection.cursor()
                indexes = connection.introspection.get_indexes(cursor, table_name)
                if name not in indexes:
                    create_index("%s_%s_idx"%(table_name, name), table_name, name)
            transaction.commit_unless_managed()