This commit is contained in:
20
pomodoro_bot/user_manager.py
Normal file
20
pomodoro_bot/user_manager.py
Normal file
@@ -0,0 +1,20 @@
|
||||
from .models import UserState
|
||||
from .redis_client import r as redis
|
||||
|
||||
class UserManager:
|
||||
def __init__(self):
|
||||
self.users = {}
|
||||
|
||||
def get_user(self, user_id):
|
||||
if user_id not in self.users:
|
||||
self.users[user_id] = UserState()
|
||||
return self.users[user_id]
|
||||
|
||||
async def increment_pomodoros(self, user_id):
|
||||
await redis.incr(f"user:{user_id}:pomodoros")
|
||||
|
||||
async def get_pomodoros(self, user_id):
|
||||
val = await redis.get(f"user:{user_id}:pomodoros")
|
||||
return int(val) if val else 0
|
||||
|
||||
user_manager = UserManager()
|
||||
Reference in New Issue
Block a user