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 Store
25 def __init__(self, value):
26 # implement lookup of sites id for link/name
32 def __init__(self, username):
35 self._userdata = dict()
37 self._userdata = self._get_user_data(username)
40 def _get_user_data(self, username):
42 return store.get_user_preferences(username)
46 self._userdata = dict()
50 if 'is_admin' in self._userdata:
51 if self._userdata['is_admin'] == '1':
56 def is_admin(self, value):
58 self._userdata['is_admin'] = '1'
60 self._userdata['is_admin'] = '0'
64 if 'fullname' in self._userdata:
65 return self._userdata['fullname']
70 def fullname(self, value):
71 self._userdata['fullname'] = value
75 if 'sites' in self._userdata:
77 for site in self._userdata['sites']:
84 #TODO: implement setting sites via the user object ?
88 class UserSession(object):
90 self.user = cherrypy.session.get('user', None)
92 def _debug(self, fact):
93 if cherrypy.config.get('debug', False):
97 return User(self.user)
99 def remote_login(self):
100 if cherrypy.request.login:
101 return self.login(cherrypy.request.login)
103 def login(self, username):
104 if self.user == username:
107 # REMOTE_USER changed, replace user
108 cherrypy.session['user'] = username
109 cherrypy.session.save()
111 cherrypy.log('LOGIN SUCCESSFUL: %s', username)
113 def logout(self, user):
115 if not type(user) is User:
117 # Completely reset user data
118 cherrypy.log.error('%s %s' % (user.name, user.fullname))
121 # Destroy current session in all cases
122 cherrypy.lib.sessions.expire()
124 def save_data(self, facility, name, data):
125 """ Save named data in the session so it can be retrieved later """
126 if facility not in cherrypy.session:
127 cherrypy.session[facility] = dict()
128 cherrypy.session[facility][name] = data
129 cherrypy.session.save()
130 self._debug('Saved session data named [%s:%s]' % (facility, name))
132 def get_data(self, facility, name):
133 """ Get named data in the session if available """
134 if facility not in cherrypy.session:
136 if name not in cherrypy.session[facility]:
138 return cherrypy.session[facility][name]
140 def nuke_data(self, facility, name):
141 if facility not in cherrypy.session:
143 if name not in cherrypy.session[facility]:
145 cherrypy.session[facility][name] = None
146 del cherrypy.session[facility][name]
147 cherrypy.session.save()
148 self._debug('Nuked session data named [%s:%s]' % (facility, name))