ASJServer/Object/UidTokenObject.py

#!/usr/bin/env python3  
# -*- coding: utf-8 -*-  
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD018
@NAME: AnsjerFormal
@software: PyCharm
@DATE: 2018/12/5 11:52
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: UidTokenObject.py
@Contact: chanjunkai@163.com
"""
from Ansjer.config import UID_TOKEN_KEY
import jwt


class UidTokenObject:

    def __init__(self, token=None):
        self.token = token
        self.UID = ''
        self.flag = self.valid()

    def valid(self):
        token = self.token
        if self.token is None:
            return False
        res = jwt.decode(token, UID_TOKEN_KEY, algorithms='HS256')
        UID = res.get('uid', None)
        if UID is None:
            return False
        self.UID = UID

    def generate(self, data={}):
        token = jwt.encode(data, UID_TOKEN_KEY, algorithm='HS256').decode('utf-8')
        return token