Simple persistent black list DB based on a DBM database.

Author: rohe

src/oidcendpoint/list_db.py

@@ -0,0 +1,24 @@
+"""
+Persistent storage of a list of values.
+"""
+import dbm
+
+
+class ListDB:
+    def __init__(self, filename):
+        self.db = dbm.open(filename, "c")
+
+    def append(self, val):
+        self.db[val] = ""  # value is immaterial since it's never used
+
+    def remove(self, val):
+        del self.db[val]
+
+    def __contains__(self, item):
+        return item in self.db
+
+    def list(self):
+        return list(self.db.keys())
+
+    def close(self):
+        self.db.close()

tests/test_50_list_db.py

@@ -0,0 +1,40 @@
+import os
+
+from oidcendpoint.list_db import ListDB
+from oidcendpoint.token_handler import DefaultToken
+
+
+def full_path(local_file):
+    _dirname = os.path.dirname(os.path.abspath(__file__))
+    return os.path.join(_dirname, local_file)
+
+
+def test():
+    lst = ListDB("temp")
+    lst.append("xyz")
+    lst.append("foo")
+    lst.append("bar")
+
+    assert "foo" in lst
+    lst.close()
+
+
+def test_blacklist_in_handler():
+    password = "The longer the better. Is this close to enough ?"
+    grant_expires_in = 600
+
+    code_handler = DefaultToken(password, typ="A", lifetime=grant_expires_in,
+                                black_list=ListDB(full_path('code_bl')))
+
+    code_handler.black_list('mamba')
+
+    assert code_handler.is_black_listed("mamba")
+
+    # Close the DBM database
+    code_handler.blist.close()
+
+    # reopen
+    code_handler.blist = ListDB(full_path('code_bl'))
+
+    # Should still be there
+    assert code_handler.is_black_listed("mamba")