localauth/manifoldbackend.py

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