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 # Name my backend 'ManifoldBackend'
  11 class ManifoldBackend:
  12
  13     # Create an authentication method
  14     # This is called by the standard Django login procedure
  15     def authenticate(self, token=None):
  16         if not token:
  17             return None
  18
  19         person = {}
  20
  21         try:
  22             username = token['username']
  23             password = token['password']
  24             request = token['request']
  25
  26             auth = {'AuthMethod': 'password', 'Username': username, 'AuthString': password}
  27             api = ManifoldAPI(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
  33             session = sessions[0]
  34             logger.debug("SESSION : {}".format(session))
  35
  36             # Change to session authentication
  37             api.auth = {'AuthMethod': 'session', 'session': session['session']}
  38             self.api = api
  39
  40             # Get account details
  41             # the new API would expect Get('local:user') instead
  42             persons_result = api.forward(Query.get('local:user').to_dict())
  43             persons = persons_result.ok_value()
  44             if not persons:
  45                 logger.error("GetPersons failed: {}".format(persons_result.error()))
  46                 return
  47             person = persons[0]
  48             logger.debug("PERSON : {}".format(person))
  49             #logger.info("{} {} <{}> logged in"\
  50             #    .format(person['config']['first_name'], person['config']['last_name'], person['config']['email']))
  51
  52             request.session['manifold'] = {'auth': api.auth, 'person': person, 'expires': session['expires']}
  53         except ManifoldException as e:
  54             logger.error("ManifoldException in Auth Backend: {}".format(e.manifold_result))
  55         except Exception as e:
  56             logger.error("Exception in Manifold Auth Backend: {}".format(e))
  57             import traceback
  58             traceback.print_exc()
  59             return None
  60
  61         try:
  62             # Check if the user exists in Django's local database
  63             user = User.objects.get(username=username)
  64         except User.DoesNotExist:
  65             # Create a user in Django's local database
  66             user = User.objects.create_user(username, username, 'passworddoesntmatter')
  67             user.email = person['email']
  68
  69         if 'firstname' in person:
  70             user.first_name = person['firstname']
  71         if 'lastname' in person:
  72             user.last_name = person['lastname']
  73
  74         return user
  75
  76     # Required for your backend to work properly - unchanged in most scenarios
  77     def get_user(self, user_id):
  78         try:
  79             return User.objects.get(pk=user_id)
  80         except User.DoesNotExist:
  81             return None
  82
  83