suyiiyii/nonebot-bison
1
0
Fork 0
mirror of https://github.com/suyiiyii/nonebot-bison.git synced 2026-06-25 23:24:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add pre-commit and format files

Browse Source
This commit is contained in:
hemengyang
2022-02-09 21:05:14 +08:00
parent 51c31637d5
commit 547be7608f
43 changed files with 2242 additions and 1365 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/plugins/nonebot_bison/admin_page/token_manager.py
+9 -7
View File
@@ -1,24 +1,26 @@
from typing import Optional
from expiringdict import ExpiringDict
import random
import string
import string
from typing import Optional
from expiringdict import ExpiringDict
class TokenManager:
def __init__(self):
self.token_manager = ExpiringDict(max_len=100, max_age_seconds=60*10)
self.token_manager = ExpiringDict(max_len=100, max_age_seconds=60 * 10)
def get_user(self, token: str) -> Optional[tuple]:
res = self.token_manager.get(token)
assert(res is None or isinstance(res, tuple))
assert res is None or isinstance(res, tuple)
return res
def save_user(self, token: str, qq: tuple) -> None:
self.token_manager[token] = qq
def get_user_token(self, qq: tuple) -> str:
token = ''.join(random.choices(string.ascii_letters + string.digits, k=16))
token = "".join(random.choices(string.ascii_letters + string.digits, k=16))
self.save_user(token, qq)
return token
token_manager = TokenManager()