fixed issue with rowcount = 0 when using count queries

worldveil · Jan 8, 2020 · b88f6b6 · b88f6b6
1 parent acca344
commit b88f6b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dejavu/base_classes/common_database.py b/dejavu/base_classes/common_database.py
@@ -59,7 +59,7 @@ def get_num_songs(self) -> int:
 
         :return: the amount of songs in the database.
         """
-        with self.cursor() as cur:
+        with self.cursor(buffered=True) as cur:
             cur.execute(self.SELECT_UNIQUE_SONG_IDS)
             count = cur.fetchone()[0] if cur.rowcount != 0 else 0
 
@@ -71,7 +71,7 @@ def get_num_fingerprints(self) -> int:
 
         :return: the number of fingerprints in the database.
         """
-        with self.cursor() as cur:
+        with self.cursor(buffered=True) as cur:
             cur.execute(self.SELECT_NUM_FINGERPRINTS)
             count = cur.fetchone()[0] if cur.rowcount != 0 else 0