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

.reset_counts() method and reset-counts command, closes #219

Browse source
This commit is contained in:
Simon Willison 2021-01-03 12:59:31 -08:00
commit 0d2a47eab9
4 changed files with 56 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

11
sqlite_utils/db.py
View file

@ -320,6 +320,17 @@ class Database:
except OperationalError:
return {}
def reset_counts(self):
tables = [table for table in self.tables if table.has_counts_triggers]
with self.conn:
self._ensure_counts_table()
counts_table = self[self._counts_table_name]
counts_table.delete_where()
counts_table.insert_all(
{"table": table.name, "count": table.execute_count()}
for table in tables
)
def execute_returning_dicts(self, sql, params=None):
cursor = self.execute(sql, params or tuple())
keys = [d[0] for d in cursor.description]