from yams.constraints import UniqueConstraint

from cubicweb.schema import PURE_VIRTUAL_RTYPES

from cubicweb.server.checkintegrity import expected_indexes, database_indexes

source = repo.system_source

for rschema in schema.relations():

    if rschema.rule or rschema in PURE_VIRTUAL_RTYPES:

        continue

    if rschema.final or rschema.inlined:

        for rdef in rschema.rdefs.values():

            table = 'cw_{0}'.format(rdef.subject)

            column = 'cw_{0}'.format(rdef.rtype)

            if any(isinstance(cstr, UniqueConstraint) for cstr in rdef.constraints):

                source.create_index(cnx, table, column, unique=True)

                commit(ask_confirm=False)

            if rschema.inlined or rdef.indexed:

                source.create_index(cnx, table, column)

                commit(ask_confirm=False)

schema_indices = expected_indexes(cnx)

db_indices = database_indexes(cnx)

for additional_index in (db_indices - set(schema_indices)):

    try:

        sql('DROP INDEX %s' % additional_index)

        commit()

    except:

        # ignore if this is not an index but a constraint

        pass

if source.dbhelper == 'postgres' and 'appears_words_idx' not in db_indices:

    sql('CREATE INDEX appears_words_idx ON appears USING gin(words)')

    db_indices.add('appears_words_idx')

for missing_index in (set(schema_indices) - db_indices):

    print('WARNING: missing index', missing_index)