iamninja/pomodoro_tg_bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Sergey Bahmatov
2025-03-29 09:12:42 +05:00
parent e4011db2fe
commit a42fef3442
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pomodoro_bot/bot.py
View File

@@ -25,7 +25,7 @@ from pomodoro_bot.redis_client import init_redis
async def on_startup(dispatcher): async def on_startup(dispatcher):
await init_redis() await init_redis()
dispatcher['timer_manager'] = TimerManager(bot, dispatcher.loop) dispatcher['timer_manager'] = TimerManager(bot)
@dispatcher.message_handler(commands=['start', 'help']) @dispatcher.message_handler(commands=['start', 'help'])
async def send_welcome(message: Message): async def send_welcome(message: Message):

6
pomodoro_bot/timer_manager.py
View File

@@ -3,9 +3,8 @@ from .user_manager import user_manager
from .models import UserState from .models import UserState
class TimerManager: class TimerManager:
def __init__(self, bot, loop): def __init__(self, bot):
self.bot = bot self.bot = bot
self.loop = loop
self.tasks = {} self.tasks = {}
async def start_timer(self, user_id, duration, chat_id, label): async def start_timer(self, user_id, duration, chat_id, label):
@@ -17,7 +16,8 @@ class TimerManager:
await user_manager.increment_pomodoros(user_id) await user_manager.increment_pomodoros(user_id)
await self.bot.send_message(chat_id, f"⏰ {label} завершён!") await self.bot.send_message(chat_id, f"⏰ {label} завершён!")
task = self.loop.create_task(timer()) loop = asyncio.get_running_loop()
task = loop.create_task(timer())
user = user_manager.get_user(user_id) user = user_manager.get_user(user_id)
user.current_timer = label user.current_timer = label
user.task = task user.task = task