From 5eb14d1c1f2e76d67fb70128a61ad22a7b38cd6a Mon Sep 17 00:00:00 2001 From: Simon Willison Date: Thu, 24 Sep 2020 07:51:36 -0700 Subject: [PATCH] Added several missing 'return self' to support chaining --- sqlite_utils/db.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/sqlite_utils/db.py b/sqlite_utils/db.py index e9e6ffa..a00e7ba 100644 --- a/sqlite_utils/db.py +++ b/sqlite_utils/db.py @@ -1167,6 +1167,7 @@ class Table(Queryable): with self.db.conn: for trigger_name in trigger_names: self.db.execute("DROP TRIGGER IF EXISTS [{}]".format(trigger_name)) + return self def rebuild_fts(self): fts_table = self.detect_fts() @@ -1178,6 +1179,7 @@ class Table(Queryable): table=fts_table ) ) + return self def detect_fts(self): "Detect if table has a corresponding FTS virtual table and return it" @@ -1245,6 +1247,7 @@ class Table(Queryable): ) with self.db.conn: self.db.execute(sql, pk_values) + return self def delete_where(self, where=None, where_args=None): if not self.exists(): @@ -1253,6 +1256,7 @@ class Table(Queryable): if where is not None: sql += " where " + where self.db.execute(sql, where_args or []) + return self def update( self, @@ -1696,6 +1700,7 @@ class Table(Queryable): for col_name, col_type in needed_columns.items(): if col_name not in current_columns: self.add_column(col_name, col_type) + return self def lookup(self, column_values): # lookups is a dictionary - all columns will be used for a unique index