From: Rob Crittenden <rcritten@redhat.com>
Date: Thu, 29 Jan 2015 15:24:02 +0000 (-0500)
Subject: Add helper to store provider specific data
X-Git-Tag: v0.4.0~32
X-Git-Url: http://git.cascardo.eti.br/?p=cascardo%2Fipsilon.git;a=commitdiff_plain;h=c327645a032ae9e4fca84909d4708ca00230b1bc

Add helper to store provider specific data

Signed-off-by: Simo Sorce <simo@redhat.com>

Reviewed-by: Rob Crittenden <rcritten@redhat.com>
---

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: