auth.py 547 B

1234567891011121314151617181920212223242526272829
  1. # -*- coding: utf-8 -*-
  2. # !/usr/bin/env python
  3. import datetime
  4. import time
  5. from apps.web.test.models import Tester
  6. def user_to_payload(user):
  7. exp = datetime.datetime.now() + datetime.timedelta(seconds = 3600 * 24 * 30)
  8. return {
  9. 'user_id': str(user.id),
  10. 'iat': int(time.time()),
  11. 'exp': exp
  12. }
  13. def payload_to_user(payload):
  14. if not payload:
  15. return None
  16. user_id = payload.get('user_id')
  17. try:
  18. user = Tester.objects.get(id = user_id)
  19. except:
  20. user = None
  21. return user