From 014c9d812612197398bfbab9909305a26de16205 Mon Sep 17 00:00:00 2001 From: peppelinux Date: Fri, 14 May 2021 14:51:48 +0200 Subject: [PATCH] feat: additional check on session db consistency --- src/oidcop/session/database.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/oidcop/session/database.py b/src/oidcop/session/database.py index 8dbdac74..4e8e3e1a 100644 --- a/src/oidcop/session/database.py +++ b/src/oidcop/session/database.py @@ -28,6 +28,10 @@ class NoSuchGrant(KeyError): pass +class InconsistentDatabase(TypeError): + pass + + class Database(ImpExp): parameter = { "db": DLDict, @@ -91,6 +95,8 @@ def get(self, path: List[str]) -> Union[SessionInfo, Grant]: user_info = self.db[uid] except KeyError: raise KeyError('No such UserID') + except TypeError: + raise InconsistentDatabase('Missing session db') else: if user_info is None: raise KeyError('No such UserID')