3 # Copyright (C) 2013 Simo Sorce <simo@redhat.com>
5 # see file 'COPYING' for use and warranty information
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
20 from ipsilon.util.data import UserStore
21 from ipsilon.util.log import Log
26 def __init__(self, value):
27 # implement lookup of sites id for link/name
33 def __init__(self, username):
36 self._userdata = dict()
38 self._userdata = self._get_user_data(username)
41 def _get_user_data(self, username):
43 return store.load_user_preferences(username)
47 self._userdata = dict()
50 def is_anonymous(self):
57 if 'is_admin' in self._userdata:
58 if str(self._userdata['is_admin']) == '1':
63 def is_admin(self, value):
65 self._userdata['is_admin'] = '1'
67 self._userdata['is_admin'] = '0'
71 if 'fullname' in self._userdata:
72 return self._userdata['fullname']
77 def fullname(self, value):
78 self._userdata['fullname'] = value
82 if 'email' in self._userdata:
83 return self._userdata['email']
89 if 'sites' in self._userdata:
91 for site in self._userdata['sites']:
98 # TODO: implement setting sites via the user object ?
101 def save_plugin_data(self, plugin, data):
103 store.save_plugin_data(plugin, self.name, data)
105 def load_plugin_data(self, plugin):
107 return store.load_plugin_data(plugin, self.name)
110 class UserSession(Log):
112 self.user = self.get_data('user', 'name')
113 self.userattrs = self.get_user_attrs()
116 return User(self.user)
118 def remote_login(self):
119 if cherrypy.request.login:
120 self.login(cherrypy.request.login)
122 self.nuke_data('user')
124 def login(self, username, userattrs=None):
125 if self.user == username:
128 # REMOTE_USER changed, replace user
129 self.nuke_data('user')
130 self.save_data('user', 'name', username)
133 # Save additional data provided by the login manager
134 self.nuke_data('userattrs')
136 self.save_user_attrs(userattrs)
138 cherrypy.log('LOGIN SUCCESSFUL: %s' % username)
140 def logout(self, user):
142 if not type(user) is User:
144 # Completely reset user data
145 cherrypy.log.error('%s %s' % (user.name, user.fullname))
148 # Destroy current session in all cases
149 cherrypy.lib.sessions.expire()
151 def get_user_attrs(self):
153 if 'userattrs' in cherrypy.session:
154 userattrs = cherrypy.session['userattrs']
157 def save_user_attrs(self, userattrs):
158 cherrypy.session['userattrs'] = userattrs
159 cherrypy.session.save()
160 self._debug('Saved user attrs')
161 self.userattrs = userattrs
163 def save_data(self, facility, name, data):
164 """ Save named data in the session so it can be retrieved later """
165 if facility not in cherrypy.session:
166 cherrypy.session[facility] = dict()
167 cherrypy.session[facility][name] = data
168 cherrypy.session.save()
169 self._debug('Saved session data named [%s:%s]' % (facility, name))
171 def get_data(self, facility, name):
172 """ Get named data in the session if available """
173 if facility not in cherrypy.session:
175 if name not in cherrypy.session[facility]:
177 return cherrypy.session[facility][name]
179 def nuke_data(self, facility, name=None):
180 if facility not in cherrypy.session:
183 if name not in cherrypy.session[facility]:
185 cherrypy.session[facility][name] = None
186 del cherrypy.session[facility][name]
187 self._debug('Nuked session data named [%s:%s]' % (facility, name))
189 del cherrypy.session[facility]
190 self._debug('Nuked session facility [%s]' % (facility,))
191 cherrypy.session.save()