diff --git a/lib/nntp/tiny/db.py b/lib/nntp/tiny/db.py index 5e953b9..f747595 100644 --- a/lib/nntp/tiny/db.py +++ b/lib/nntp/tiny/db.py @@ -114,3 +114,13 @@ class Database(): def get(self, table, values: dict=dict()): return self.query(table, values).fetchone() + + def count(self, table, values: dict=dict()): + sql = "select count(id) as num from {table.name}" + + if len(values) > 0: + sql += " and ".join([f"{k} = ?" for k in values]) + + result = self.db.execute(sql, values.values()) + + return result[0][0]