PLC/Methods/GetSession.py

   1 import time
   2
   3 from PLC.Method import Method
   4 from PLC.Parameter import Parameter, Mixed
   5 from PLC.Auth import Auth
   6 from PLC.Sessions import Session, Sessions
   7 from PLC.Nodes import Node, Nodes
   8 from PLC.Persons import Person, Persons
   9
  10 class GetSession(Method):
  11     """
  12     Returns a new session key if a user or node authenticated
  13     successfully, faults otherwise.
  14     """
  15
  16     roles = ['admin', 'pi', 'user', 'tech', 'node']
  17     accepts = [Auth()]
  18     returns = Session.fields['session_id']
  19
  20     event_type = 'Get'
  21     object_type = 'Session'
  22
  23     def call(self, auth):
  24         # Authenticated with a session key, just return it
  25         if auth.has_key('session'):
  26             return auth['session']
  27
  28         session = Session(self.api)
  29
  30         if isinstance(self.caller, Person):
  31             # XXX Make this configurable
  32             session['expires'] = int(time.time()) + (24 * 60 * 60)
  33
  34         session.sync(commit = False)
  35
  36         if isinstance(self.caller, Node):
  37             session.add_node(self.caller, commit = True)
  38         elif isinstance(self.caller, Person):
  39             session.add_person(self.caller, commit = True)
  40
  41         return session['session_id']