Implement method for counting database records

This commit is contained in:
XANTRONIX Development 2024-11-09 10:39:04 -05:00
parent adaff5638d
commit 27f6b1c692

View file

@ -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]