splinext/users/__init__.py

   1 from pkg_resources import resource_filename
   2
   3 from pylons import c, session
   4
   5 from spline.lib.plugin import PluginBase
   6 from spline.lib.plugin import PluginBase, PluginLink, Priority
   7 import spline.model.meta as meta
   8
   9 import splinext.users.controllers.accounts
  10 import splinext.users.controllers.users
  11 from splinext.users import model as users_model
  12
  13 def add_routes_hook(map, *args, **kwargs):
  14     """Hook to inject some of our behavior into the routes configuration."""
  15     # Login, logout
  16     map.connect('/accounts/login', controller='accounts', action='login')
  17     map.connect('/accounts/login_begin', controller='accounts', action='login_begin')
  18     map.connect('/accounts/login_finish', controller='accounts', action='login_finish')
  19     map.connect('/accounts/logout', controller='accounts', action='logout')
  20
  21     # Self-admin
  22     map.connect('/users/{id};{name}/edit', controller='users', action='profile_edit')
  23
  24     # Public user pages
  25     map.connect('/users', controller='users', action='list')
  26     map.connect('/users/{id};{name}', controller='users', action='profile')
  27     map.connect('/users/{id}', controller='users', action='profile')
  28
  29 def check_userid_hook(action, **params):
  30     """Hook to see if a user is logged in and, if so, stick a user object in
  31     c.
  32     """
  33
  34     c.user = None
  35
  36     if not 'user_id' in session:
  37         return
  38
  39     user = meta.Session.query(users_model.User).get(session['user_id'])
  40     if not user:
  41         # Bogus id in the session somehow.  Clear it
  42         del session['user_id']
  43         session.save()
  44         return
  45
  46     c.user = user
  47
  48
  49 class UsersPlugin(PluginBase):
  50     def controllers(self):
  51         return dict(
  52             accounts = splinext.users.controllers.accounts.AccountsController,
  53             users = splinext.users.controllers.users.UsersController,
  54         )
  55
  56     def template_dirs(self):
  57         return [
  58             (resource_filename(__name__, 'templates'), Priority.NORMAL)
  59         ]
  60
  61     def hooks(self):
  62         return [
  63             ('routes_mapping',    Priority.NORMAL,      add_routes_hook),
  64             ('before_controller', Priority.VERY_FIRST,  check_userid_hook),
  65         ]
  66
  67     def widgets(self):
  68         return [
  69             ('page_header', Priority.NORMAL, 'widgets/user_state.mako'),
  70         ]