localauth/manifoldbackend.py

   1 import time
   2
   3 from django.contrib.auth.models import User
   4
   5 from manifoldapi.manifoldapi    import ManifoldAPI, ManifoldException, ManifoldResult
   6 from manifold.core.query        import Query
   7
   8 from myslice.settings import config, logger, DEBUG
   9
  10 from portal.actions import authority_check_pis
  11
  12 # from unfold.sessioncache import SessionCache
  13
  14 # Name my backend 'ManifoldBackend'
  15 class ManifoldBackend:
  16
  17     # Create an authentication method
  18     # This is called by the standard Django login procedure
  19     def authenticate(self, token=None):
  20         if not token:
  21             return None
  22
  23         person = {}
  24
  25         try:
  26             username = token['username']
  27             password = token['password']
  28             request = token['request']
  29
  30             auth = {'AuthMethod': 'password', 'Username': username, 'AuthString': password}
  31             api = ManifoldAPI(auth)
  32             sessions_result = api.forward(Query.create('local:session').to_dict())
  33             sessions = sessions_result.ok_value()
  34             if not sessions:
  35                 logger.error("GetSession failed: {}".format(sessions_result.error()))
  36                 return None
  37             session = sessions[0]
  38             logger.debug("SESSION : {}".format(session.keys()))
  39
  40             # Change to session authentication
  41             api.auth = {'AuthMethod': 'session', 'session': session['session']}
  42             #api.auth = session_auth
  43             self.api = api
  44
  45             # Get account details
  46             # the new API would expect Get('local:user') instead
  47             persons_result = api.forward(Query.get('local:user').to_dict())
  48             persons = persons_result.ok_value()
  49             if not persons:
  50                 logger.error("GetPersons failed: {}".format(persons_result.error()))
  51                 return None
  52             person = persons[0]
  53             logger.debug("PERSON : {}".format(person))
  54
  55             request.session['manifold'] = {'auth': api.auth, 'person': person, 'expires': session['expires']}
  56
  57             #logger.info("{} {} <{}> logged in"\
  58             #    .format(person['config']['first_name'], person['config']['last_name'], person['config']['email']))
  59
  60             #SessionCache().store_auth(request, session_auth)
  61
  62         except ManifoldException as e:
  63             logger.error("ManifoldException in Auth Backend: {}".format(e.manifold_result))
  64         except Exception as e:
  65             logger.error("Exception in Manifold Auth Backend: {}".format(e))
  66             import traceback
  67             logger.error(traceback.format_exc())
  68             return None
  69
  70         try:
  71             # Check if the user exists in Django's local database
  72             user = User.objects.get(username=username)
  73         except User.DoesNotExist:
  74             # Create a user in Django's local database
  75             user = User.objects.create_user(username, username, 'passworddoesntmatter')
  76             user.email = person['email']
  77
  78         if 'firstname' in person:
  79             user.first_name = person['firstname']
  80         if 'lastname' in person:
  81             user.last_name = person['lastname']
  82
  83         user.pi = authority_check_pis (request, user.email)
  84         request.session['user'] = {'email':user.email,'pi':user.pi,'firstname':user.first_name,'lastname':user.last_name}
  85         return user
  86
  87     # Required for your backend to work properly - unchanged in most scenarios
  88     def get_user(self, user_id):
  89         try:
  90             return User.objects.get(pk=user_id)
  91         except User.DoesNotExist:
  92             return None
  93
  94