remote/sqlite-utils
1
0
Fork 0
mirror of https://github.com/simonw/sqlite-utils.git synced 2026-07-23 01:14:31 +02:00
Code Issues Projects Releases Packages Wiki Activity

.enable_fts(..., replace=True) argument, closes #160

Browse source
This commit is contained in:
Simon Willison 2020-09-20 15:05:46 -07:00
commit ecb50c8f76
4 changed files with 97 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

31
sqlite_utils/db.py
View file

@ -814,16 +814,21 @@ class Table(Queryable):
self.db.add_foreign_keys([(self.name, column, other_table, other_column)])
def enable_fts(
self, columns, fts_version="FTS5", create_triggers=False, tokenize=None
self,
columns,
fts_version="FTS5",
create_triggers=False,
tokenize=None,
replace=False,
):
"Enables FTS on the specified columns."
sql = (
create_fts_sql = (
textwrap.dedent(
"""
CREATE VIRTUAL TABLE [{table}_fts] USING {fts_version} (
{columns},{tokenize}
content=[{table}]
);
)
"""
)
.strip()
@ -834,7 +839,25 @@ class Table(Queryable):
tokenize="\n tokenize='{}',".format(tokenize) if tokenize else "",
)
)
self.db.executescript(sql)
should_recreate = False
if replace and self.db["{}_fts".format(self.name)].exists():
# Does the table need to be recreated?
fts_schema = self.db["{}_fts".format(self.name)].schema
if fts_schema != create_fts_sql:
should_recreate = True
expected_triggers = {self.name + suffix for suffix in ("_ai", "_ad", "_au")}
existing_triggers = {t.name for t in self.triggers}
has_triggers = existing_triggers.issuperset(expected_triggers)
if has_triggers != create_triggers:
should_recreate = True
if not should_recreate:
# Table with correct configuration already exists
return self
if should_recreate:
self.disable_fts()
self.db.executescript(create_fts_sql)
self.populate_fts(columns)
if create_triggers: