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

Commit 4213321

Browse files
committed
Configuration specified DBs for client DB and JWT jtis.
1 parent 68c0375 commit 4213321

File tree

1 file changed

+15
-2
lines changed

1 file changed

+15
-2
lines changed

src/oidcendpoint/endpoint_context.py

Lines changed: 15 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def __init__(
149149
self.th_args = get_token_handlers(conf)
150150

151151
# client database
152-
self.cdb = client_db or {}
152+
self.set_client_db()
153153

154154
# session db
155155
self._sub_func = {}
@@ -260,7 +260,20 @@ def set_session_db(self, sso_db=None, db=None):
260260
logger.debug("Session DB: {}".format(self.sdb.__dict__))
261261

262262
def set_jti_db(self, db=None):
263-
self.jti_db = db or InMemoryDataBase()
263+
if db is None and self.conf.get("jti_db"):
264+
_spec = self.conf.get("jti_db")
265+
_kwargs = _spec.get("kwargs", {})
266+
self.jti_db = importer(_spec["class"])(**_kwargs)
267+
else:
268+
self.jti_db = db or InMemoryDataBase()
269+
270+
def set_client_db(self, db=None):
271+
if db is None and self.conf.get("client_db"):
272+
_spec = self.conf.get("client_db")
273+
_kwargs = _spec.get("kwargs", {})
274+
self.cdb = importer(_spec["class"])(**_kwargs)
275+
else:
276+
self.cdb = {}
264277

265278
def do_add_on(self):
266279
if self.conf.get("add_on"):

0 commit comments

Comments
 (0)