Fix Database.count() method

This commit is contained in:
XANTRONIX Development 2024-11-09 11:38:46 -05:00
parent ced21d3b98
commit 696de54c8d

View file

@ -118,11 +118,13 @@ class Database():
return self.query(table, values).fetchone() return self.query(table, values).fetchone()
def count(self, table, values: dict=dict()): 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: if len(values) > 0:
sql += " where "
sql += " and ".join([f"{k} = ?" for k in values]) 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]