Skip to content
This repository was archived by the owner on Jun 12, 2021. It is now read-only.

Commit 815b0b3

Browse files
committed
Using shelve based db for sso and session db.
1 parent c3a2977 commit 815b0b3

File tree

1 file changed

+30
-0
lines changed

1 file changed

+30
-0
lines changed

src/oidcendpoint/shelve_db.py

Lines changed: 30 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,30 @@
1+
import shelve
2+
3+
4+
class ShelveDataBase(object):
5+
def __init__(self, filename, flag='c', protocol=None, writeback=False):
6+
self.db = shelve.open(filename=filename, flag=flag, protocol=protocol, writeback=writeback)
7+
8+
def __contains__(self, key):
9+
if key in self.db:
10+
return True
11+
else:
12+
return False
13+
14+
def set(self, key, value):
15+
self.db[key] = value
16+
17+
def get(self, key, default=None):
18+
try:
19+
return self.db[key]
20+
except KeyError:
21+
return default
22+
23+
def delete(self, key):
24+
try:
25+
del self.db[key]
26+
except KeyError:
27+
return
28+
29+
def keys(self):
30+
return self.db.keys()

0 commit comments

Comments
 (0)