merge
[unfold.git] / 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