Added db.index_foreign_keys() method plus test

This commit is contained in:
Simon Willison 2019-06-30 16:38:14 -07:00
commit 61164d98cf
3 changed files with 21 additions and 7 deletions

View file

@ -228,13 +228,7 @@ def index_foreign_keys(path):
Ensure every foreign key column has an index on it.
"""
db = sqlite_utils.Database(path)
for table_name in db.table_names():
table = db[table_name]
existing_indexes = {i.columns[0] for i in table.indexes if len(i.columns) == 1}
for fk in table.foreign_keys:
if fk.column not in existing_indexes:
print("Creating index on {}.{}".format(table_name, fk.column))
table.create_index([fk.column])
db.index_foreign_keys()
@cli.command(name="create-index")