diff --git a/nwxraybot/handlers/hello.py b/nwxraybot/handlers/hello.py index fcf9a4f..05c29bc 100644 --- a/nwxraybot/handlers/hello.py +++ b/nwxraybot/handlers/hello.py @@ -1,13 +1,33 @@ -from aiogram import types +import json +from datetime import datetime + +from aiogram import F, types +from aiogram.enums import ParseMode from aiogram.filters import Command +from aiogram.types import (CallbackQuery, InlineKeyboardButton, + InlineKeyboardMarkup) from nwxraybot.meta import Handler +from nwxraybot.models import User class HelloHandler(Handler): - def __init__(self, bot) -> None: - super().__init__(bot) + def __non_admin_main_menu(self): + markup = [[ + InlineKeyboardButton(text="Обновить данные", + callback_data='update') + ]] + return InlineKeyboardMarkup( + inline_keyboard=markup + ) + + def __init__(self) -> None: + super().__init__() @self.router.message(Command("start")) async def hello(message: types.Message): - await message.reply("Приветствуем в боте NwXray! Здесь вы сможете получить информацию о своем подключении к NwXray") + await message.reply("Приветствуем в боте NwXray! Здесь вы сможете получить информацию о своем подключении к NwXray", reply_markup=self.__non_admin_main_menu()) + + @self.router.callback_query(F.data == 'update') + async def update_data(callback: CallbackQuery): + await callback.message.delete()