You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
31 lines
843 B
31 lines
843 B
3 years ago
|
from datetime import datetime, timedelta
|
||
|
|
from types import TracebackType
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
USED_TOKENS = set()
|
||
|
|
LOGGED_IN_USERS: dict[str, (int, datetime)] = {}
|
||
|
|
|
||
|
|
def login_user(user_id: int) -> str:
|
||
|
|
'''
|
||
|
|
Creates token for user
|
||
|
|
'''
|
||
|
|
token = str(uuid4())
|
||
|
|
while token in USED_TOKENS:
|
||
|
|
token = str(uuid4())
|
||
|
|
if len(USED_TOKENS) > 10_000_000:
|
||
|
|
USED_TOKENS.clear()
|
||
|
|
USED_TOKENS.add(token)
|
||
|
|
LOGGED_IN_USERS[token] = user_id, datetime.now()
|
||
|
|
return token
|
||
|
|
|
||
|
|
def get_user(token: str) -> int | None:
|
||
|
|
if token not in LOGGED_IN_USERS:
|
||
|
|
return None
|
||
|
|
|
||
|
|
user_id, login_date = LOGGED_IN_USERS[token]
|
||
|
|
time_since_login: timedelta = datetime.now() - login_date
|
||
|
|
if time_since_login.total_seconds() > (60 * 30): # 30 mins
|
||
|
|
del LOGGED_IN_USERS[token]
|
||
|
|
return None
|
||
|
|
return user_id
|