Fix Database.count() method
This commit is contained in:
parent
ced21d3b98
commit
696de54c8d
1 changed files with 5 additions and 3 deletions
|
@ -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]
|
||||||
|
|
Loading…
Add table
Reference in a new issue