From 27f6b1c692c661114c6907217b573f116c73bae5 Mon Sep 17 00:00:00 2001 From: XANTRONIX Development Date: Sat, 9 Nov 2024 10:39:04 -0500 Subject: [PATCH] Implement method for counting database records --- lib/nntp/tiny/db.py | 10 ++++++++++ 1 file changed, 10 insertions(+) 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]