Implement method for counting database records
This commit is contained in:
parent
adaff5638d
commit
27f6b1c692
1 changed files with 10 additions and 0 deletions
|
@ -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]
|
||||
|
|
Loading…
Add table
Reference in a new issue