From 696de54c8d4732a3dc1c7bfb167e3810e13a83c8 Mon Sep 17 00:00:00 2001 From: XANTRONIX Development Date: Sat, 9 Nov 2024 11:38:46 -0500 Subject: [PATCH] Fix Database.count() method --- lib/nntp/tiny/db.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/lib/nntp/tiny/db.py b/lib/nntp/tiny/db.py index 1593570..8744064 100644 --- a/lib/nntp/tiny/db.py +++ b/lib/nntp/tiny/db.py @@ -118,11 +118,13 @@ class Database(): return self.query(table, values).fetchone() def count(self, table, values: dict=dict()): - sql = "select count(id) as num from {table.name}" + sql = f"select count(id) as num from {table.name}" if len(values) > 0: + sql += " where " sql += " and ".join([f"{k} = ?" for k in values]) - result = self.db.execute(sql, values.values()) + cr = self.db.execute(sql, list(values.values())) + row = cr.fetchone() - return result[0][0] + return row[0]