from ipsilon.util.user import UserSession
from ipsilon.util.plugin import PluginLoader, PluginObject
from ipsilon.util.plugin import PluginInstaller
+from ipsilon.info.common import Info
+from ipsilon.util.cookies import SecureCookie
import cherrypy
+USERNAME_COOKIE = 'ipsilon_default_username'
+
+
class LoginManagerBase(PluginObject, Log):
def __init__(self):
super(LoginManagerBase, self).__init__()
self.path = '/'
self.next_login = None
+ self.info = None
def redirect_to_path(self, path):
base = cherrypy.config.get('base.mount', "")
raise cherrypy.HTTPRedirect('%s/login/%s' % (base, path))
- def auth_successful(self, username, userdata=None):
+ def auth_successful(self, username, auth_type=None, userdata=None):
# save ref before calling UserSession login() as it
# may regenerate the session
session = UserSession()
if not ref:
ref = cherrypy.config.get('base.mount', "") + '/'
+ if self.info:
+ userattrs = self.info.get_user_attrs(username)
+ if userdata:
+ userdata.update(userattrs or {})
+ else:
+ userdata = userattrs
+ self.debug("User %s attributes: %s" % (username, repr(userdata)))
+
+ if auth_type:
+ if userdata:
+ userdata.update({'auth_type': auth_type})
+ else:
+ userdata = {'auth_type': auth_type}
+
session.login(username, userdata)
+ # save username into a cookie if parent was form base auth
+ if auth_type == 'password':
+ cookie = SecureCookie(USERNAME_COOKIE, username)
+ # 15 days
+ cookie.maxage = 1296000
+ cookie.send()
+
raise cherrypy.HTTPRedirect(ref)
def auth_failed(self):
plugins['enabled'].append(self)
self._debug('Login plugin enabled: %s' % self.name)
+ # Get handle of the info plugin
+ self.info = root.info
+
def disable(self, site):
plugins = site[FACILITY]
if self not in plugins['enabled']:
if self.lm.next_login is not None:
next_url = self.lm.next_login.path
+ cookie = SecureCookie(USERNAME_COOKIE)
+ cookie.receive()
+ username = cookie.value
+ if username is None:
+ username = ''
+
context = {
"title": 'Login',
"action": '%s/%s' % (self.basepath, self.formpage),
"password_text": self.lm.password_text,
"description": self.lm.help_text,
"next_url": next_url,
+ "username": username,
}
context.update(kwargs)
return context
def __init__(self, *args, **kwargs):
super(Login, self).__init__(*args, **kwargs)
self.first_login = None
+ self.info = Info(self._site)
loader = PluginLoader(Login, FACILITY, 'LoginManager')
self._site[FACILITY] = loader.get_plugin_data()