From 87ad6cd049c3783c7d95d5677f475b733262709d Mon Sep 17 00:00:00 2001 From: "Karl O. Pinc" Date: Sun, 9 May 2021 16:49:17 -0500 Subject: [PATCH] Use session to track whether login credentials are good --- src/pgwui_core/core.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/pgwui_core/core.py b/src/pgwui_core/core.py index 8bc424d..4b92571 100644 --- a/src/pgwui_core/core.py +++ b/src/pgwui_core/core.py @@ -320,7 +320,9 @@ class CredsLoadedForm(LoadedForm): Produces the dict pyramid will use to render the form. ''' response = super().write(result, errors) - if response.get('havecreds', False): + havecreds = self.uh.session.get('havecreds', False) + response.update({'havecreds': havecreds}) + if havecreds: response['user'] = self['user'] else: # We don't know if the credentials are good or @@ -1411,7 +1413,7 @@ class DBConnector(object): Reserved keys: havecreds Boolean. Supplied credentials can connect to db. ''' - return {'havecreds': False} + return {'havecreds': self.uh.session.get('havecreds', False)} def nodberror_factory(self): return core_ex.NoDBError('No database name supplied') @@ -1535,8 +1537,7 @@ class DBConnector(object): havecreds = True errors = self.call_alter_db(func, conn) - response.update({'havecreds': havecreds - }) + self.uh.session.update({'havecreds': havecreds}) return (errors, response) def read_uh(self): -- 2.34.1