20 lines
583 B
Python
20 lines
583 B
Python
from pomodoro_bot.models import UserState
|
|
from pomodoro_bot.redis_client import 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() |