Do not try to handle messages until fortnite client is ready

1 files changed, 16 insertions, 12 deletions

diff --git a/telegram_bot/commands.py b/telegram_bot/commands.py
index cba5d5f..1e3295b 100644
--- a/telegram_bot/commands.py
+++ b/telegram_bot/commands.py
@@ -44,8 +44,9 @@ class GetFriendsCommand(CommandHandler):
         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))
+        if self.__fortnite_client.is_initialized():
+            friends = await self.__fortnite_client.get_friends()
+            await self.__telegram_bot.reply(message, format_users(friends))
 
 
 class GetStatsCommand(CommandHandler):
@@ -58,9 +59,10 @@ class GetStatsCommand(CommandHandler):
         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))
+        if self.__fortnite_client.is_initialized():
+            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):
 
@@ -74,10 +76,11 @@ class GetTodayStatsCommand(CommandHandler):
         self.__stats_repository = stats_repository
 
     async def handle(self, message: telebot.types.Message):
-        persisted_stats = self.__stats_repository.get_stats()
-        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))
+        if self.__fortnite_client.is_initialized():
+            persisted_stats = self.__stats_repository.get_stats()
+            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):
 
@@ -89,6 +92,7 @@ class RecordStatsCommand(CommandHandler):
         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.put_stats(friend)
\ No newline at end of file
+        if self.__fortnite_client.is_initialized():
+            friends = await self.__fortnite_client.get_friends()
+            for friend in friends:
+                await self.__stats_repository.put_stats(friend)
\ No newline at end of file