remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-09-18 14:34:24 +02:00
Code Issues Projects Releases Packages Wiki Activity

Add an option to create triggers for fts table.

Browse source
This commit is contained in:
Amjith Ramanujam 2019-09-01 20:09:02 -07:00
commit 68e63ad07f
1 changed files with 24 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

26
sqlite_utils/db.py
View file

@ -710,8 +710,8 @@ class Table(Queryable):
)
self.db.add_foreign_keys([(self.name, column, other_table, other_column)])
def enable_fts(self, columns, fts_version="FTS5"):
"Enables FTS on the specified columns"
def enable_fts(self, columns, fts_version="FTS5", create_update_triggers=False):
"Enables FTS on the specified columns."
sql = """
CREATE VIRTUAL TABLE "{table}_fts" USING {fts_version} (
{columns},
@ -724,6 +724,28 @@ class Table(Queryable):
)
self.db.conn.executescript(sql)
self.populate_fts(columns)
if create_update_triggers:
old_cols = ", ".join("old.[{}]".format(c) for c in columns)
new_cols = ", ".join("new.[{}]".format(c) for c in columns)
triggers = """
CREATE TRIGGER "{table}_ai" AFTER INSERT ON "{table}" BEGIN
INSERT INTO "{table}_fts" (rowid, {columns}) VALUES (new.rowid, {new_cols});
END;
CREATE TRIGGER "{table}_ad" AFTER DELETE ON "{table}" BEGIN
INSERT INTO "{table}_fts" ("{table}_fts", rowid, {columns}) VALUES('delete', old.rowid, {old_cols});
END;
CREATE TRIGGER "{table}_au" AFTER UPDATE ON "{table}" BEGIN
INSERT INTO "{table}_fts" ("{table}_fts", rowid, {columns}) VALUES('delete', old.rowid, {old_cols});
INSERT INTO "{table}_fts" (rowid, {columns}) VALUES (new.rowid, {new_cols});
END;
""".format(
table=self.name,
columns=", ".join("[{}]".format(c) for c in columns),
old_cols=old_cols,
new_cols=new_cols,
)
self.db.conn.executescript(triggers)
return self
def populate_fts(self, columns):