From f5c57d8e73f33ca1d7374a2662fbc7a4592eb7cd Mon Sep 17 00:00:00 2001
From: Dmitrii Morozov <snoopdesigns@gmail.com>
Date: Tue, 7 May 2024 16:34:14 +0200
Subject:  Python code style

---
 commands.py | 94 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 commands.py

(limited to 'commands.py')

diff --git a/commands.py b/commands.py
new file mode 100644
index 0000000..4cf330b
--- /dev/null
+++ b/commands.py
@@ -0,0 +1,94 @@
+import telebot
+from TelegramBot import *
+from Formatter import *
+from persistence import *
+from FortniteClient import FortniteClient
+from pythonFortniteStatus.FortniteStatus import *
+
+class StartCommand(CommandHandler):
+
+    __telegram_bot: TelegramBot
+    __user_repository: UserRepository
+
+    def __init__(self, telegram_bot: TelegramBot, user_repository: UserRepository):
+        self.__telegram_bot = telegram_bot
+        self.__user_repository = user_repository
+
+    async def handle(self, message: telebot.types.Message):
+        if message.chat.type == 'private':
+            alias = message.chat.username
+        else:
+            alias = message.chat.title
+        self.__user_repository.putUser(message.chat.id, alias)
+        await self.__telegram_bot.reply(message, 'This chat successfully registered to receive Fortnite updates')
+
+class GetStatusCommand(CommandHandler):
+
+    __telegram_bot: TelegramBot
+    __fortnite_status: FortniteStatus
+
+    def __init__(self, telegram_bot: TelegramBot):
+        self.__telegram_bot = telegram_bot
+        self.__fortnite_status = FortniteStatus()
+
+    async def handle(self, message: telebot.types.Message):
+        await self.__telegram_bot.reply(message, format_fortnite_status(self.__fortnite_status.getStatus()))
+
+class GetFriendsCommand(CommandHandler):
+
+    __telegram_bot: TelegramBot
+    __fortnite_client: FortniteClient
+
+    def __init__(self, telegram_bot: TelegramBot, fortnite_client: FortniteClient):
+        self.__telegram_bot = telegram_bot
+        self.__fortnite_client = fortnite_client
+
+    async def handle(self, message: telebot.types.Message):
+        friends = await self.__fortnite_client.get_friends()
+        await self.__telegram_bot.reply(message, format_users(friends))
+
+
+class GetStatsCommand(CommandHandler):
+
+    __telegram_bot: TelegramBot
+    __fortnite_client: FortniteClient
+
+    def __init__(self, telegram_bot: TelegramBot, fortnite_client: FortniteClient):
+        self.__telegram_bot = telegram_bot
+        self.__fortnite_client = fortnite_client
+
+    async def handle(self, message: telebot.types.Message):
+        friends = await self.__fortnite_client.get_friends()
+        stats = [await friend.fetch_stats() for friend in friends]
+        await self.__telegram_bot.reply(message, format_user_stats_list(stats))
+
+class GetTodayStatsCommand(CommandHandler):
+
+    __telegram_bot: TelegramBot
+    __fortnite_client: FortniteClient
+    __stats_repository: StatsRepository
+
+    def __init__(self, telegram_bot: TelegramBot, fortnite_client: FortniteClient, stats_repository: StatsRepository):
+        self.__telegram_bot = telegram_bot
+        self.__fortnite_client = fortnite_client
+        self.__stats_repository = stats_repository
+
+    async def handle(self, message: telebot.types.Message):
+        persisted_stats = self.__stats_repository.getStats()
+        friends = await self.__fortnite_client.get_friends()
+        current_stats = [await friend.fetch_stats() for friend in friends]
+        await self.__telegram_bot.reply(message, format_user_stats_difference(persisted_stats, current_stats))
+
+class RecordStatsCommand(CommandHandler):
+
+    __fortnite_client: FortniteClient
+    __stats_repository: StatsRepository
+
+    def __init__(self, fortnite_client: FortniteClient, stats_repository: StatsRepository):
+        self.__fortnite_client = fortnite_client
+        self.__stats_repository = stats_repository
+
+    async def handle(self, message: telebot.types.Message):
+        friends = await self.__fortnite_client.get_friends()
+        for friend in friends:
+            await self.__stats_repository.putStats(friend)
\ No newline at end of file
-- 
cgit v1.2.3