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             email = token['username']
  27             username = email.split('@')[-1]
  28             password = token['password']
  29             request = token['request']
  30
  31             auth = {'AuthMethod': 'password', 'Username': email, 'AuthString': password}
  32             api = ManifoldAPI(config.manifold_url(), auth)
  33             sessions_result = api.forward(Query.create('local:session').to_dict())
  34             sessions = sessions_result.ok_value()
  35             if not sessions:
  36                 logger.error("GetSession failed: {}".format(sessions_result.error()))
  37                 return None
  38             session = sessions[0]
  39             logger.debug("SESSION : {}".format(session.keys()))
  40
  41             # Change to session authentication
  42             api.auth = {'AuthMethod': 'session', 'session': session['session']}
  43             #api.auth = session_auth
  44             self.api = api
  45
  46             # Get account details
  47             # the new API would expect Get('local:user') instead
  48             persons_result = api.forward(Query.get('local:user').to_dict())
  49             persons = persons_result.ok_value()
  50             if not persons:
  51                 logger.error("GetPersons failed: {}".format(persons_result.error()))
  52                 return None
  53             person = persons[0]
  54             logger.debug("PERSON : {}".format(person))
  55
  56             request.session['manifold'] = {'auth': api.auth, 'person': person, 'expires': session['expires']}
  57
  58             #logger.info("{} {} <{}> logged in"\
  59             #    .format(person['config']['first_name'], person['config']['last_name'], person['config']['email']))
  60
  61             #SessionCache().store_auth(request, session_auth)
  62
  63         except ManifoldException as e:
  64             logger.error("ManifoldException in Auth Backend: {}".format(e.manifold_result))
  65         except Exception as e:
  66             logger.error("Exception in Manifold Auth Backend: {}".format(e))
  67             import traceback
  68             logger.error(traceback.format_exc())
  69             return None
  70
  71         try:
  72             # Check if the user exists in Django's local database
  73             user = User.objects.get(email=email)
  74         except User.DoesNotExist:
  75             # Create a user in Django's local database
  76             user = User.objects.create_user(username, email, 'passworddoesntmatter')
  77             user.email = person['email']
  78
  79         if 'firstname' in person:
  80             user.first_name = person['firstname']
  81         if 'lastname' in person:
  82             user.last_name = person['lastname']
  83
  84         user.pi = authority_check_pis (request, user.email)
  85         request.session['user'] = {'email':user.email,'pi':user.pi,'firstname':user.first_name,'lastname':user.last_name}
  86         return user
  87
  88     # Required for your backend to work properly - unchanged in most scenarios
  89     def get_user(self, user_id):
  90         try:
  91             return User.objects.get(pk=user_id)
  92         except User.DoesNotExist:
  93             return None
  94
  95