Fixed SSEChat message error when no badges

- If user does not have any badges, SSEChat messages .badges failed
thelabcat · Jun 7, 2024 · ae31030 · ae31030
1 parent 62c1474
commit ae31030
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cocorum"
-version = "1.4.1"
+version = "1.4.2"
 keywords = ["rumble", "api", "wrapper", "livestream"]
 authors = [
   { name="Wilbur Jaywright", email="[email protected]" },

diff --git a/src/cocorum/ssechat.py b/src/cocorum/ssechat.py
@@ -82,7 +82,12 @@ def color(self):
     @property
     def badges(self):
         """Badges the user has"""
-        return [self.chat.badges[badge_slug] for badge_slug in self["badges"]]
+        try:
+            return [self.chat.badges[badge_slug] for badge_slug in self["badges"]]
+
+        #User has no badges
+        except KeyError:
+            return []
 
 class SSEChatChannel(SSEChatter):
     """A channel in the SSE chat"""