fixed bug for opgg command

diff --git a/skgyorugo/commands/opgg.py b/skgyorugo/commands/opgg.py
index 89b31305a8f35d3e1edeb180f1060c82ad023cd3..1dd2cf17b7d514af94c7c32311796f894ed85997 100644 (file)
--- a/skgyorugo/commands/opgg.py
+++ b/skgyorugo/commands/opgg.py
@@ -29,15 +29,12 @@ GLOBAL_COOLDOWN = 15
 PATH = os.path.dirname(os.path.realpath(__file__))
 PATH = os.path.join(PATH, "..")
 
-def get_streamer_data():
-    raise NotImplementedError()
-
-def get_twitch_user_data(twitch_id: int):
-    raise NotImplementedError()
-
 def main(bot: Bot, message: Message):
+    index_skip = 0
+    if message.tags.get("reply-parent-user-id", None):
+        index_skip += 1
     try:
-        twitch_user = message.value.split(' ')[1]
+        twitch_user = message.value.split(' ')[1 + index_skip]
     except IndexError:
         twitch_user = message.tags.get("reply-parent-display-name", message.channel)
         twitch_id = message.tags.get("reply-parent-user-id", ttv_api.users.get_users(user_logins=[message.channel]))

diff --git a/skgyorugo/commands/t.py b/skgyorugo/commands/t.py
index 30f2c5962c1f163e4250f99f17bfcd4710a4fb19..337f1fa6e7e807533e0e88dbdd8c30c993a0e209 100644 (file)
--- a/skgyorugo/commands/t.py
+++ b/skgyorugo/commands/t.py
@@ -10,10 +10,7 @@ GLOBAL_COOLDOWN = 5
 
 
 def main(bot: Bot, message: Message):
-    try:
-        replied_message = message.tags["reply-parent-msg-body"]
-    except KeyError:
-        replied_message = None
+    replied_message = message.tags.get("reply-parent-msg-body", None)
     if replied_message:
         msg = replied_message
         replied_msg_id = message.tags["reply-parent-msg-id"]