def get_token(user):
s = JwtSerializer(secret_key="I dont know", expires_in=10800)
return s.dumps({
"userid": user.user_id,
"username": user.user_name,
"loginname": user.user_login_name,
"issued_at": time()
})
def cls2token(self,
client_secret,
exprires=Config.ACCESS_TOKEN_DEFAULT_EXPIRES):
s = JwtSerializer(secret_key=client_secret, expires_in=exprires)
return s.dumps({
"client_id": self.client_id,
"user": self.user,
"issued_at": time()
})
def token2cls(cls, client_secret, token):
s = JwtSerializer(client_secret)
try:
data = s.loads(token)
if "client_id" in data and "user" in data:
return cls(data["client_id"], data["user"])
else:
return None
except:
return None
def info_to_token(self, user_object):
roles = user_object.job.role.all()
self.info = {
"id": user_object.id,
"email": user_object.email,
"name": user_object.name,
"loginName": user_object.email.split("@")[0],
"role": [role.name for role in roles] if roles else []
}
s = JwtSerializer(secret_key=self.secret, expires_in=self.exprires)
return s.dumps({"user": self.info, "issued_at": time()})
def token2cls(cls, token):
if token:
s = JwtSerializer(secret_key=Config.ID_TOKEN_SECRET_KEY,
salt=Config.ID_TOKEN_SALT)
try:
data = s.loads(token)
if data.get("user_id"):
return cls(data.get("user_id"))
return None
except:
return None
else:
return None
def get_dict_by_token(token):
try:
s = JwtSerializer(secret_key="I dont know")
return s.loads(token)
except:
return None
def cls2Token(self, expires=Config.ID_TOKEN_DEFAULT_EXPIRES):
s = JwtSerializer(secret_key=Config.ID_TOKEN_SECRET_KEY,
salt=Config.ID_TOKEN_SALT,
expires_in=expires)
return s.dumps({"user_id": self.user_id, "issued_at": time()})