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)