From c327645a032ae9e4fca84909d4708ca00230b1bc Mon Sep 17 00:00:00 2001 From: Rob Crittenden Date: Thu, 29 Jan 2015 10:24:02 -0500 Subject: [PATCH] Add helper to store provider specific data Signed-off-by: Simo Sorce Reviewed-by: Rob Crittenden --- ipsilon/util/user.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) diff --git a/ipsilon/util/user.py b/ipsilon/util/user.py index 5a8f159..fd557a0 100644 --- a/ipsilon/util/user.py +++ b/ipsilon/util/user.py @@ -158,6 +158,22 @@ class UserSession(Log): self._debug('Saved user attrs') self.userattrs = userattrs + def _get_provider_attr_name(self, provider): + return '%s_data' % provider + + def get_provider_data(self, provider): + attr = self._get_provider_attr_name(provider) + data = None + if attr in cherrypy.session: + data = cherrypy.session[attr] + return data + + def save_provider_data(self, provider, data): + attr = self._get_provider_attr_name(provider) + cherrypy.session[attr] = data + cherrypy.session.save() + self._debug('Saved %s provider data' % provider) + def save_data(self, facility, name, data): """ Save named data in the session so it can be retrieved later """ if facility not in cherrypy.session: -- 2.20.1