X-Git-Url: https://git.mxchange.org/?p=quix0rs-apt-p2p.git;a=blobdiff_plain;f=apt_p2p_Khashmir%2Fdb.py;h=bd777b60e78f65126859645a29152b49aeab80bd;hp=4ee9cb38068d69978828ca8c328b36d053fe15cc;hb=e525e8eb8cc4536857146a1579f79d2d7793522d;hpb=0ebb5e0abe0740bbd24675e4ac83c48f48c00b0c diff --git a/apt_p2p_Khashmir/db.py b/apt_p2p_Khashmir/db.py index 4ee9cb3..bd777b6 100644 --- a/apt_p2p_Khashmir/db.py +++ b/apt_p2p_Khashmir/db.py @@ -154,11 +154,16 @@ class DB: @return: the number of distinct keys and total values in the database """ c = self.conn.cursor() - c.execute("SELECT COUNT(DISTINCT key) as num_keys, COUNT(value) as num_values FROM kv") - keys, values = 0, 0 + c.execute("SELECT COUNT(value) as num_values FROM kv") + values = 0 row = c.fetchone() if row: - keys, values = row[0], row[1] + values = row[0] + c.execute("SELECT COUNT(key) as num_keys FROM (SELECT DISTINCT key FROM kv)") + keys = 0 + row = c.fetchone() + if row: + keys = row[0] return keys, values class TestDB(unittest.TestCase):