43 lines
2.0 KiB
Python
43 lines
2.0 KiB
Python
from aiogram import Bot, Dispatcher, types
|
|
from aiogram.types import Message, InlineKeyboardMarkup, InlineKeyboardButton
|
|
from aiogram.utils import executor
|
|
from aiogram.dispatcher.filters import Text
|
|
from pomodoro_bot.user_manager import user_manager
|
|
from pomodoro_bot.timer_manager import timer_manager
|
|
import os
|
|
|
|
API_TOKEN = os.getenv("BOT_TOKEN")
|
|
|
|
bot = Bot(token=API_TOKEN)
|
|
dp = Dispatcher(bot)
|
|
|
|
menu_kb = InlineKeyboardMarkup(row_width=2)
|
|
menu_kb.add(
|
|
InlineKeyboardButton("▶️ Pomodoro", callback_data="pomodoro"),
|
|
InlineKeyboardButton("☕ Short Break", callback_data="shortbreak"),
|
|
InlineKeyboardButton("😌 Long Break", callback_data="longbreak"),
|
|
InlineKeyboardButton("⏹ Stop", callback_data="stop")
|
|
)
|
|
|
|
@dp.message_handler(commands=['start', 'help'])
|
|
async def send_welcome(message: Message):
|
|
await message.reply("Привет! Я твой личный Pomodoro-бот. Выбирай действие:", reply_markup=menu_kb)
|
|
|
|
@dp.callback_query_handler(Text(equals=["pomodoro", "shortbreak", "longbreak", "stop"]))
|
|
async def handle_callback(call: types.CallbackQuery):
|
|
if call.data == "pomodoro":
|
|
await timer_manager.start_timer(call.from_user.id, 25*60, call.message.chat.id, 'Pomodoro')
|
|
await call.message.answer("Начался 25-минутный Pomodoro! 🔥")
|
|
elif call.data == "shortbreak":
|
|
await timer_manager.start_timer(call.from_user.id, 5*60, call.message.chat.id, 'Short Break')
|
|
await call.message.answer("Начался 5-минутный перерыв ☕")
|
|
elif call.data == "longbreak":
|
|
await timer_manager.start_timer(call.from_user.id, 15*60, call.message.chat.id, 'Long Break')
|
|
await call.message.answer("Начался длинный перерыв 😌")
|
|
elif call.data == "stop":
|
|
await timer_manager.stop_timer(call.from_user.id)
|
|
await call.message.answer("Таймер остановлен ⏹")
|
|
await call.answer()
|
|
|
|
def run_bot():
|
|
executor.start_polling(dp, skip_updates=True) |