+
+class UserSession(Log):
+ def __init__(self):
+ self.user = self.get_data('user', 'name')
+ self.userattrs = self.get_user_attrs()
+
+ def get_user(self):
+ return User(self.user)
+
+ def remote_login(self):
+ if cherrypy.request.login:
+ self.login(cherrypy.request.login)
+ else:
+ self.nuke_data('user')
+
+ def login(self, username, userattrs=None):
+ if self.user == username:
+ return
+
+ # REMOTE_USER changed, replace user
+ self.nuke_data('user')
+ self.save_data('user', 'name', username)
+ self.user = username
+
+ # Save additional data provided by the login manager
+ self.nuke_data('userattrs')
+ if userattrs:
+ self.save_user_attrs(userattrs)
+
+ cherrypy.log('LOGIN SUCCESSFUL: %s' % username)
+
+ def logout(self, user):
+ if user is not None:
+ if not type(user) is User:
+ raise TypeError
+ # Completely reset user data
+ cherrypy.log.error('%s %s' % (user.name, user.fullname))
+ user.reset()
+
+ # Destroy current session in all cases
+ cherrypy.lib.sessions.expire()
+
+ def get_user_attrs(self):
+ userattrs = dict()
+ if 'userattrs' in cherrypy.session:
+ userattrs = cherrypy.session['userattrs']
+ return userattrs
+
+ def save_user_attrs(self, userattrs):
+ cherrypy.session['userattrs'] = userattrs
+ cherrypy.session.save()
+ self._debug('Saved user attrs')
+ self.userattrs = userattrs
+
+ 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:
+ cherrypy.session[facility] = dict()
+ cherrypy.session[facility][name] = data
+ cherrypy.session.save()
+ self._debug('Saved session data named [%s:%s]' % (facility, name))
+
+ def get_data(self, facility, name):
+ """ Get named data in the session if available """
+ if facility not in cherrypy.session:
+ return None
+ if name not in cherrypy.session[facility]:
+ return None
+ return cherrypy.session[facility][name]
+
+ def nuke_data(self, facility, name=None):
+ if facility not in cherrypy.session:
+ return
+ if name:
+ if name not in cherrypy.session[facility]:
+ return
+ cherrypy.session[facility][name] = None
+ del cherrypy.session[facility][name]
+ self._debug('Nuked session data named [%s:%s]' % (facility, name))
+ else:
+ del cherrypy.session[facility]
+ self._debug('Nuked session facility [%s]' % (facility,))
+ cherrypy.session.save()