manyi/apps/web/test/auth.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

import datetime
import time

from apps.web.test.models import Tester


def user_to_payload(user):
    exp = datetime.datetime.now() + datetime.timedelta(seconds = 3600 * 24 * 30)
    return {
        'user_id': str(user.id),
        'iat': int(time.time()),
        'exp': exp
    }


def payload_to_user(payload):
    if not payload:
        return None

    user_id = payload.get('user_id')
    try:
        user = Tester.objects.get(id = user_id)
    except:
        user = None

    return user