Add drop table

hexlib/db.py

@@ -283,6 +283,13 @@ class PersistentState:
     def __getitem__(self, table):
         return self._table_factory(self, table)
 
+    def __delitem__(self, key):
+        with sqlite3.connect(self.dbfile, **self.dbargs) as conn:
+            try:
+                conn.execute(f"DROP TABLE {key}")
+            except:
+                pass
+
 
 def pg_fetch_cursor_all(cur, name, batch_size=1000):
     while True:

setup.py

@@ -2,7 +2,7 @@ from setuptools import setup
 
 setup(
     name="hexlib",
-    version="1.82",
+    version="1.83",
     description="Misc utility methods",
     author="simon987",
     author_email="me@simon987.net",

test/test_PersistentState.py

@@ -131,3 +131,13 @@ class TestPersistentState(TestCase):
         s["a"][0] = {"x": b'abc'}
 
         self.assertEqual(list(s["a"])[0]["x"], b'abc')
+
+    def test_drop_table(self):
+        s = PersistentState()
+
+        s["a"][0] = {"x": 1}
+        s["a"][1] = {"x": 2}
+        self.assertEqual(len(list(s["a"])), 2)
+
+        del s["a"]
+        self.assertEqual(len(list(s["a"])), 0)