diff options
author | ue86388 <dmitrii.morozov@sbb.ch> | 2024-04-09 13:44:23 +0200 |
---|---|---|
committer | ue86388 <dmitrii.morozov@sbb.ch> | 2024-04-09 13:44:23 +0200 |
commit | 199b7c0f26a39fb644ebb69a6d2536a43819e915 (patch) | |
tree | 10c96e7bfc23899eb35848ea3ee97258a8bfff74 | |
parent | 997fe0d4aaaeb5be81583b8ad4d82935d6118969 (diff) |
Test
-rwxr-xr-x | FortniteClient.py | 3 | ||||
-rw-r--r-- | FortniteStatusFormatter.py | 11 | ||||
-rwxr-xr-x | tgbot.py | 14 |
3 files changed, 28 insertions, 0 deletions
diff --git a/FortniteClient.py b/FortniteClient.py index f426fa7..7c5858c 100755 --- a/FortniteClient.py +++ b/FortniteClient.py @@ -22,6 +22,9 @@ class FortniteClient(fortnitepy.Client): def attach(self, observer: any): self.observers.append(observer) + + def get_friends(self): + return self.friends def __auth_authorization_code(self): code = input("Enter authorization code (https://www.epicgames.com/id/api/redirect?clientId=3446cd72694c4a4485d81b77adbb2141&responseType=code):") diff --git a/FortniteStatusFormatter.py b/FortniteStatusFormatter.py index 9595304..be87f0c 100644 --- a/FortniteStatusFormatter.py +++ b/FortniteStatusFormatter.py @@ -18,4 +18,15 @@ def formatFortniteStatus(fortniteStatus): formatting.mbold("Fortnite status"), "", '\n'.join(statuses), + separator='\n') + +def __formatFriend(friend): + return friend.display_name + +def formatFriends(friends): + friends_formatted = [__formatFriend(friend) for friend in friends] + return formatting.format_text( + formatting.mbold("Registered friends:"), + "", + '\n'.join(friends_formatted), separator='\n')
\ No newline at end of file @@ -23,6 +23,20 @@ def startCommand(message): userRepository.putUser(message.chat.id) bot.reply_to(message, "This chat successfully registered to receive Fortnite updates!") +@bot.message_handler(commands = ['status']) +def getStatus(message): + bot.reply_to( + message, + formatFortniteStatus(fortniteStatus.getStatus()), + parse_mode='MarkdownV2') + +@bot.message_handler(commands = ['friends']) +def getFriends(message): + bot.reply_to( + message, + formatFriends(fortniteClient.get_friends()), + parse_mode='MarkdownV2') + class FortniteStatusObserver(Observer): def update(self, fortniteStatus) -> None: for user in userRepository.getAllUsers(): |