- Fix rate limit error throwing

- Fix SignAPIError message formatting
- Allow failed parses to print an error to the console instead of crashing the client
- Fix shadowing Enum `MessageType` which causes a crash

TikTokLive/client/client.py

@@ -1,5 +1,6 @@
 import asyncio
 import logging
+import traceback
 import urllib.parse
 from asyncio import AbstractEventLoop, Task, CancelledError
 from logging import Logger
@@ -319,7 +320,12 @@ def _parse_webcast_response(self, response: Optional[WebcastResponseMessage]) ->
             return [response_event, UnknownEvent().from_pydict(response.to_dict())]
 
         # Get the underlying events
-        proto_event: ProtoEvent = event_type().parse(response.payload)
+        try:
+            proto_event: ProtoEvent = event_type().parse(response.payload)
+        except Exception:
+            self._logger.error(traceback.format_exc())
+            return [response_event]
+
         parsed_events: List[Event] = [response_event, proto_event]
         custom_event: Optional[Event] = self._parse_custom_event(response, proto_event)
 

TikTokLive/client/web/routes/fetch_sign.py

@@ -44,7 +44,7 @@ def __init__(
         self.reason = reason
         args = list(args)
         args.insert(0, f"[{reason.name}]")
-        super().__init__(*args)
+        super().__init__(" ".join(args))
 
 
 class SignatureRateLimitError(SignAPIError):
@@ -70,7 +70,7 @@ def __init__(self, retry_after: int, reset_time: int, *args):
         _args = list(args)
         _args[0] = str(args[0]) % str(self.retry_after)
 
-        super().__init__(reason=SignAPIError.ErrorReason.RATE_LIMIT, *args)
+        super().__init__(SignAPIError.ErrorReason.RATE_LIMIT, *_args)
 
     @property
     def retry_after(self) -> int:

TikTokLive/proto/custom_proto.py

@@ -6,10 +6,13 @@
 
 from TikTokLive.proto import User, GiftStruct
 
-MessageType: Type = TypeVar('MessageType', bound=betterproto.Message)
+# "MessageType" is a proto enum field.
+# This underscore is the difference between life & death, because if you shadow the proto field,
+# everything crashes & burns in a fiery hell.
+_MessageType: Type = TypeVar('_MessageType', bound=betterproto.Message)
 
 
-def proto_extension(cls: MessageType):
+def proto_extension(cls: _MessageType):
     """
     Betterproto doesn't properly handle inheriting existing messages.
     This method takes the superclass proto metadata and assigns that to this one.