stats: Check keys instead of bool value of db.

ArchiveTeam · Apr 3, 2015 · 5ded09d · 5ded09d
1 parent a20fb78
commit 5ded09d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/wpull/stats.py b/wpull/stats.py
@@ -90,8 +90,12 @@ def is_quota_exceeded(self):
         if self._required_urls_db is None:
             return False
 
-        if self._required_urls_db:
-            return False
+        try:
+            # bool(db) is not guaranteed to be supported, esp on PyPy
+            if next(iter(self._required_urls_db.keys())):
+                return False
+        except StopIteration:
+            pass
 
         if self.quota:
             return self.size >= self.quota