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]