From aaef526c6c37ed1b3d4016092a2937433cd4d7d7 Mon Sep 17 00:00:00 2001 From: Dmitrii Morozov Date: Tue, 7 May 2024 04:08:27 +0200 Subject: Minor refactoring, extract commands to separate classes --- Commands.py | 96 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 96 insertions(+) create mode 100644 Commands.py (limited to 'Commands.py') diff --git a/Commands.py b/Commands.py new file mode 100644 index 0000000..926b25b --- /dev/null +++ b/Commands.py @@ -0,0 +1,96 @@ +import telebot +from TelegramBot import * +from Formatter import * +from persistence import * +from FortniteClient import FortniteClient +from pythonFortniteStatus.FortniteStatus import * + +class StartCommand(CommandHandler): + + __telegramBot: TelegramBot + __userRepository: UserRepository + + def __init__(self, telegramBot: TelegramBot, userRepository: UserRepository): + self.__telegramBot = telegramBot + self.__userRepository = userRepository + + async def handle(self, message: telebot.types.Message): + if message.chat.type == 'private': + alias = message.chat.username + else: + alias = message.chat.title + self.__userRepository.putUser(message.chat.id, alias) + await self.__telegramBot.reply(message, 'This chat successfully registered to receive Fortnite updates') + +class GetStatusCommand(CommandHandler): + + __telegramBot: TelegramBot + __fortniteStatus: FortniteStatus + + def __init__(self, telegramBot: TelegramBot): + self.__telegramBot = telegramBot + self.__fortniteStatus = FortniteStatus() + + async def handle(self, message: telebot.types.Message): + await self.__telegramBot.reply(message, formatFortniteStatus(self.__fortniteStatus.getStatus())) + +class GetFriendsCommand(CommandHandler): + + __telegramBot: TelegramBot + __fortniteClient: FortniteClient + + def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient): + self.__telegramBot = telegramBot + self.__fortniteClient = fortniteClient + + async def handle(self, message: telebot.types.Message): + friends = await self.__fortniteClient.get_friends() + await self.__telegramBot.reply(message, formatUsers(friends)) + + +class GetStatsCommand(CommandHandler): + + __telegramBot: TelegramBot + __fortniteClient: FortniteClient + + def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient): + self.__telegramBot = telegramBot + self.__fortniteClient = fortniteClient + + async def handle(self, message: telebot.types.Message): + friends = await self.__fortniteClient.get_friends() + stats = [await friend.fetch_stats() for friend in friends] + await self.__telegramBot.reply(message, formatUserStatsList(stats)) + +class GetTodayStatsCommand(CommandHandler): + + __telegramBot: TelegramBot + __fortniteClient: FortniteClient + __statsRepository: StatsRepository + + def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient, statsRepository: StatsRepository): + self.__telegramBot = telegramBot + self.__fortniteClient = fortniteClient + self.__statsRepository = statsRepository + + async def handle(self, message: telebot.types.Message): + persisted_stats = self.__statsRepository.getStats() + friends = await self.__fortniteClient.get_friends() + current_stats = [await friend.fetch_stats() for friend in friends] + await self.__telegramBot.reply(message, formatUserStatsDifference(persisted_stats, current_stats)) + +class RecordStatsCommand(CommandHandler): + + __telegramBot: TelegramBot + __fortniteClient: FortniteClient + __statsRepository: StatsRepository + + def __init__(self, telegramBot: TelegramBot, fortniteClient: FortniteClient, statsRepository: StatsRepository): + self.__telegramBot = telegramBot + self.__fortniteClient = fortniteClient + self.__statsRepository = statsRepository + + async def handle(self, message: telebot.types.Message): + friends = await self.__fortniteClient.get_friends() + for friend in friends: + await self.__statsRepository.putStats(friend) \ No newline at end of file -- cgit v1.2.3