Change format of multithread parser to class

pytchat/processors/compatible/parser.py

@@ -4,40 +4,41 @@ from .renderer.paidmessage import LiveChatPaidMessageRenderer
 from .renderer.paidsticker import LiveChatPaidStickerRenderer
 from .renderer.legacypaid import LiveChatLegacyPaidMessageRenderer
 
-def parse(sitem):
+class Parser:
+    def parse(self, sitem):
 
-    action = sitem.get("addChatItemAction")
+        action = sitem.get("addChatItemAction")
-    if action:
+        if action:
-        item = action.get("item")
+            item = action.get("item")
-    if item is None: return None
+        if item is None: return None
-    rd={}
+        rd={}
-    try:
+        try:
-        renderer = get_renderer(item)
+            renderer = self.get_renderer(item)
-        if renderer == None:
+            if renderer == None:
+                return None
+
+            rd["kind"] = "youtube#liveChatMessage"
+            rd["etag"] = ""
+            rd["id"] = 'LCC.' + renderer.get_id()
+            rd["snippet"]       = renderer.get_snippet()
+            rd["authorDetails"] = renderer.get_authordetails()
+        except (KeyError,TypeError,AttributeError) as e:
+            print(f"------{str(type(e))}-{str(e)}----------")
+            print(sitem)
             return None
         
-        rd["kind"] = "youtube#liveChatMessage"
+        return rd        
-        rd["etag"] = ""
-        rd["id"] = 'LCC.' + renderer.get_id()
-        rd["snippet"]       = renderer.get_snippet()
-        rd["authorDetails"] = renderer.get_authordetails()
-    except (KeyError,TypeError,AttributeError) as e:
-        print(f"------{str(type(e))}-{str(e)}----------")
-        print(sitem)
-        return None
 
-    return rd        
+    def get_renderer(self, item):
-
+        if item.get("liveChatTextMessageRenderer"):
-def get_renderer(item):
+            renderer = LiveChatTextMessageRenderer(item)
-    if item.get("liveChatTextMessageRenderer"):
+        elif item.get("liveChatPaidMessageRenderer"):
-        renderer = LiveChatTextMessageRenderer(item)
+            renderer = LiveChatPaidMessageRenderer(item)
-    elif item.get("liveChatPaidMessageRenderer"):
+        elif item.get( "liveChatPaidStickerRenderer"):
-        renderer = LiveChatPaidMessageRenderer(item)
+            renderer = LiveChatPaidStickerRenderer(item)
-    elif item.get( "liveChatPaidStickerRenderer"):
+        elif item.get("liveChatLegacyPaidMessageRenderer"):
-        renderer = LiveChatPaidStickerRenderer(item)
+            renderer = LiveChatLegacyPaidMessageRenderer(item)
-    elif item.get("liveChatLegacyPaidMessageRenderer"):
+        else:
-        renderer = LiveChatLegacyPaidMessageRenderer(item)
+            renderer = None
-    else:
+        return renderer
-        renderer = None
-    return renderer
 

pytchat/processors/compatible/processor.py

@@ -1,10 +1,13 @@
-from . import parser
+from . parser import Parser
 import json
 import os
 import traceback
 import datetime
 import time
-class CompatibleProcessor():
+
+class CompatibleProcessor:
+    def __init__(self):
+        self.parser = Parser()
 
     def process(self, chat_components: list):
 
@@ -26,7 +29,7 @@ class CompatibleProcessor():
                     if action.get('addChatItemAction') is None: continue
                     if action['addChatItemAction'].get('item') is None: continue
 
-                    chat = parser.parse(action)
+                    chat = self.parser.parse(action)
                     if chat:
                         chatlist.append(chat)
         ret["pollingIntervalMillis"] = int(timeout*1000)