def login(cls, username, password, seconds=60 * 60 * 24):
user = UserInfoDao.get_by_username(username)
if not user:
raise Exception('用户不存在')
if password != user["password"]:
raise Exception('密码错误')
return cls.create_token(user, seconds)
def register(username, password, seconds=60 * 60 * 24):
user = UserInfoDao.get_by_username(username)
if user:
raise Exception('用户已存在')
UserInfoDao.create_user(username, password)
user = UserInfoDao.get_by_username(username)
uid = user['uid']
token = uuid.uuid1().hex
expire_time = datetime.datetime.now() + datetime.timedelta(
seconds=seconds)
UserTokenDao.create_token(uid, token, expire_time)
return {
'uid': uid,
'username': username,
'token': token,
'expire_time': expire_time,
}
def get_user_info(token):
token_info = UserTokenDao.get_token_info(token)
if not token_info:
raise Exception('token 不存在')
if token_info['expire_time'] < datetime.datetime.now():
raise Exception('token 已失效')
user = UserInfoDao.get_by_uid(token_info['uid'])
return {
'uid': user['uid'],
'username': user['username'],
'token': token,
'expire_time': token_info['expire_time'],
}
def login(username, password, seconds=60 * 60 * 24):
user = UserInfoDao.get_by_username(username)
if not user:
raise Exception('用户不存在，请先注册')
uid = user['uid']
print(uid)
username = user['username']
if password != user['password']:
raise Exception('密码错误')
token = uuid.uuid1().hex
expire_time = datetime.datetime.now() + datetime.timedelta(
seconds=seconds)
UserTokenDao.create_token(uid, token, expire_time)
return {
'uid': uid,
'username': username,
'token': token,
'expire_time': expire_time,
}