1 # Copyright (C) 2013 Ipsilon project Contributors, for license see COPYING
3 from ipsilon.util.data import UserStore
4 from ipsilon.util.log import Log
10 def __init__(self, value):
11 # implement lookup of sites id for link/name
17 def __init__(self, username):
20 self._userdata = dict()
22 self._userdata = self._get_user_data(username)
25 def _get_user_data(self, username):
27 return store.load_user_preferences(username)
31 self._userdata = dict()
34 def is_anonymous(self):
41 if 'is_admin' in self._userdata:
42 if str(self._userdata['is_admin']) == '1':
47 def is_admin(self, value):
49 self._userdata['is_admin'] = '1'
51 self._userdata['is_admin'] = '0'
55 if 'fullname' in self._userdata:
56 return self._userdata['fullname']
61 def fullname(self, value):
62 self._userdata['fullname'] = value
66 if 'email' in self._userdata:
67 return self._userdata['email']
73 if 'sites' in self._userdata:
75 for site in self._userdata['sites']:
82 # TODO: implement setting sites via the user object ?
85 def save_plugin_data(self, plugin, data):
87 store.save_plugin_data(plugin, self.name, data)
89 def load_plugin_data(self, plugin):
91 return store.load_plugin_data(plugin, self.name)
94 class UserSession(Log):
96 self.user = self.get_data('user', 'name')
97 self.userattrs = self.get_user_attrs()
100 return User(self.user)
102 def remote_login(self):
103 if cherrypy.request.login:
104 self.login(cherrypy.request.login)
106 self.nuke_data('user')
108 def login(self, username, userattrs=None):
109 if self.user == username:
110 if userattrs and not self.get_user_attrs():
111 self.save_user_attrs(userattrs)
114 # REMOTE_USER changed, replace user
115 self.nuke_data('user')
116 self.save_data('user', 'name', username)
119 # Save additional data provided by the login manager
120 self.nuke_data('userattrs')
122 self.save_user_attrs(userattrs)
124 cherrypy.log('LOGIN SUCCESSFUL: %s' % username)
126 def logout(self, user):
128 if not isinstance(user, User):
130 # Completely reset user data
131 cherrypy.log.error('%s %s' % (user.name, user.fullname),
132 severity=logging.INFO)
135 # Destroy current session in all cases
136 cherrypy.lib.sessions.expire()
138 def get_user_attrs(self):
140 if 'userattrs' in cherrypy.session:
141 userattrs = cherrypy.session['userattrs']
144 def save_user_attrs(self, userattrs):
145 cherrypy.session['userattrs'] = userattrs
146 self.debug('Saved user attrs')
147 self.userattrs = userattrs
149 def _get_provider_attr_name(self, provider):
150 return '%s_data' % provider
152 def get_provider_data(self, provider):
153 attr = self._get_provider_attr_name(provider)
155 if attr in cherrypy.session:
156 data = cherrypy.session[attr]
159 def save_provider_data(self, provider, data):
160 attr = self._get_provider_attr_name(provider)
161 cherrypy.session[attr] = data
162 self.debug('Saved %s provider data' % provider)
164 def save_data(self, facility, name, data):
165 """ Save named data in the session so it can be retrieved later """
166 if facility not in cherrypy.session:
167 cherrypy.session[facility] = dict()
168 cherrypy.session[facility][name] = data
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,))