Added compatibility with tinydb 4

hivesolutions · May 5, 2020 · 265881f · 265881f
1 parent 11bc2b1
commit 265881f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/quorum/data.py b/src/quorum/data.py
@@ -189,7 +189,8 @@ def get_db(self):
     def drop_db(self, *args, **kwargs):
         if self._db == None: return
         db = self.get_db()
-        db.purge_tables()
+        if hasattr(db, "drop_tables"): db.drop_tables()
+        else: db.purge_tables()
         db.close()
         self._db = None
         method = getattr(self, "_drop_db_%s" % self.storage)