Merge branch 'release/v0.0.9'

v0.0.8

README.md

@@ -27,7 +27,7 @@ pip install pytchat
 ### CLI
 
 One-liner command.
-Save chat data to html.
+Save chat data to html, with embedded custom emojis.
 
 ```bash
 $ pytchat -v ZJ6Q4U_Vg6s -o "c:/temp/"
@@ -148,6 +148,20 @@ def main():
 if __name__ == '__main__':
   main()
 ```
+### Extract archived chat data as [HTML](https://github.com/taizan-hokuto/pytchat/wiki/HTMLArchiver) or [tab separated values](https://github.com/taizan-hokuto/pytchat/wiki/TSVArchiver).
+```python
+from pytchat import HTMLArchiver, Extractor
+
+video_id = "*******"
+ex = Extractor(
+    video_id,
+    div=10,
+    processor=HTMLArchiver("c:/test.html")
+)
+
+ex.extract()
+print("finished.")
+```
 
 ## Structure of Default Processor
 Each item can be got with `items` function.
@@ -175,7 +189,7 @@ Each item can be got with `items` function.
   <tr>
     <td>messageEx</td>
     <td>str</td>
-    <td>list of message texts and emoji URLs.</td>
+    <td>list of message texts and emoji dicts(id, txt, url).</td>
   </tr>
   <tr>
     <td>timestamp</td>

pytchat/__init__.py

@@ -1,8 +1,8 @@
 """
-pytchat is a python library for fetching youtube live chat without using yt api, Selenium, or BeautifulSoup.
+pytchat is a lightweight python library to browse youtube livechat without Selenium or BeautifulSoup.
 """
 __copyright__    = 'Copyright (C) 2019 taizan-hokuto'
-__version__      = '0.0.6.6'
+__version__      = '0.0.9'
 __license__      = 'MIT'
 __author__       = 'taizan-hokuto'
 __author_email__ = '55448286+taizan-hokuto@users.noreply.github.com'
@@ -28,3 +28,5 @@ from .api import (
     SuperchatCalculator,
     VideoInfo
 )
+
+# flake8: noqa

pytchat/api.py

@@ -14,3 +14,5 @@ from .processors.speed.calculator import SpeedCalculator
 from .processors.superchat.calculator import SuperchatCalculator
 from .tool.extract.extractor import Extractor
 from .tool.videoinfo import VideoInfo
+
+# flake8: noqa

pytchat/cli/__init__.py

@@ -1,11 +1,7 @@
 import argparse
-import os
 from pathlib import Path
-from typing import List, Callable
 from .arguments import Arguments
-
-from .. exceptions import InvalidVideoIdException, NoContentsException
-from .. processors.tsv_archiver import TSVArchiver
+from .. exceptions import InvalidVideoIdException, NoContents
 from .. processors.html_archiver import HTMLArchiver
 from .. tool.extract.extractor import Extractor
 from .. tool.videoinfo import VideoInfo
@@ -18,6 +14,8 @@ https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 
 '''
+
+
 def main():
     # Arguments
     parser = argparse.ArgumentParser(description=f'pytchat v{__version__}')
@@ -47,14 +45,16 @@ def main():
                 path = Path(Arguments().output + video_id + '.html')
                 print(f"output path: {path.resolve()}")
                 Extractor(video_id,
-                  processor = HTMLArchiver(Arguments().output+video_id+'.html'),
+                          processor=HTMLArchiver(
+                              Arguments().output + video_id + '.html'),
                           callback=_disp_progress
                           ).extract()
                 print("\nExtraction end.\n")
-            except (InvalidVideoIdException, NoContentsException) as e:
+            except (InvalidVideoIdException, NoContents) as e:
                 print(e)
         return
     parser.print_help()
 
+
 def _disp_progress(a, b):
     print('.', end="", flush=True)

pytchat/cli/arguments.py

@@ -8,6 +8,7 @@ https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 '''
 
+
 class Arguments(metaclass=Singleton):
     """
     Arguments singleton

pytchat/cli/singleton.py

@@ -4,6 +4,8 @@ Petter Kraabøl's Twitch-Chat-Downloader
 https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 '''
+
+
 class Singleton(type):
     """
     Abstract class for singletons

pytchat/config/__init__.py

@@ -1,11 +1,9 @@
 import logging
 from . import mylogger
-
 headers = {
-        'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/77.0.3865.120 Safari/537.36'}
+    'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/83.0.4103.106 Safari/537.36'}
+
 
 def logger(module_name: str, loglevel=None):
     module_logger = mylogger.get_logger(module_name, loglevel=loglevel)
     return module_logger
-
-

pytchat/config/mylogger.py

@@ -1,11 +1,11 @@
-from logging import NullHandler, getLogger, StreamHandler, FileHandler, Formatter
+from logging import NullHandler, getLogger, StreamHandler, FileHandler
 import logging
 from datetime import datetime
 
 
 def get_logger(modname, loglevel=logging.DEBUG):
     logger = getLogger(modname)
-    if loglevel == None:
+    if loglevel is None:
         logger.addHandler(NullHandler())
         return logger
     logger.setLevel(loglevel)
@@ -22,10 +22,10 @@ def get_logger(modname,loglevel=logging.DEBUG):
         handler2.setLevel(logging.ERROR)
         handler2.setFormatter(my_formatter)
 
-
         logger.addHandler(handler2)
     return logger
 
+
 class MyFormatter(logging.Formatter):
     def format(self, record):
         timestamp = (

pytchat/core_async/buffer.py

@@ -1,5 +1,7 @@
 
 import asyncio
+
+
 class Buffer(asyncio.Queue):
     '''
     チャットデータを格納するバッファの役割を持つFIFOキュー
@@ -10,6 +12,7 @@ class Buffer(asyncio.Queue):
         格納するチャットブロックの最大個数。0の場合は無限。
         最大値を超える場合は古いチャットブロックから破棄される。
     '''
+
     def __init__(self, maxsize=0):
         super().__init__(maxsize)
 

pytchat/core_async/livechat.py

@@ -1,7 +1,6 @@
-import aiohttp, asyncio
-import datetime
+import aiohttp
+import asyncio
 import json
-import random
 import signal
 import time
 import traceback
@@ -12,7 +11,7 @@ from asyncio import Queue
 from .buffer import Buffer
 from ..parser.live import Parser
 from .. import config
-from ..exceptions  import ChatParseException,IllegalFunctionCall
+from .. import exceptions
 from ..paramgen import liveparam, arcparam
 from ..processors.default.processor import DefaultProcessor
 from ..processors.combinator import Combinator
@@ -75,7 +74,7 @@ class LiveChatAsync:
     _setup_finished = False
 
     def __init__(self, video_id,
-                seektime = 0,
+                 seektime=-1,
                  processor=DefaultProcessor(),
                  buffer=None,
                  interruptable=True,
@@ -87,7 +86,7 @@ class LiveChatAsync:
                  topchat_only=False,
                  logger=config.logger(__name__),
                  ):
-        self.video_id  = video_id
+        self._video_id = video_id
         self.seektime = seektime
         if isinstance(processor, tuple):
             self.processor = Combinator(processor)
@@ -103,28 +102,26 @@ class LiveChatAsync:
         self._parser = Parser(is_replay=self._is_replay)
         self._pauser = Queue()
         self._pauser.put_nowait(None)
-        self._setup()
         self._first_fetch = True
         self._fetch_url = "live_chat/get_live_chat?continuation="
         self._topchat_only = topchat_only
         self._logger = logger
+        self.exception = None
         LiveChatAsync._logger = logger
 
-        if not LiveChatAsync._setup_finished:
-            LiveChatAsync._setup_finished = True
         if exception_handler:
             self._set_exception_handler(exception_handler)
         if interruptable:
             signal.signal(signal.SIGINT,
                 (lambda a, b: asyncio.create_task(
-                    LiveChatAsync.shutdown(None,signal.SIGINT,b))
-                ))
+                 LiveChatAsync.shutdown(None, signal.SIGINT, b))))
+        self._setup()
 
     def _setup(self):
         # direct modeがTrueでcallback未設定の場合例外発生。
         if self._direct_mode:
             if self._callback is None:
-                raise IllegalFunctionCall(
+                raise exceptions.IllegalFunctionCall(
                     "When direct_mode=True, callback parameter is required.")
         else:
             # direct modeがFalseでbufferが未設定ならばデフォルトのbufferを作成
@@ -139,18 +136,18 @@ class LiveChatAsync:
                 loop.create_task(self._callback_loop(self._callback))
         # _listenループタスクの開始
         loop = asyncio.get_event_loop()
-        listen_task = loop.create_task(self._startlisten())
+        self.listen_task = loop.create_task(self._startlisten())
         # add_done_callbackの登録
         if self._done_callback is None:
-            listen_task.add_done_callback(self.finish)
+            self.listen_task.add_done_callback(self._finish)
         else:
-            listen_task.add_done_callback(self._done_callback)
+            self.listen_task.add_done_callback(self._done_callback)
 
     async def _startlisten(self):
         """Fetch first continuation parameter,
         create and start _listen loop.
         """
-        initial_continuation = liveparam.getparam(self.video_id,3)
+        initial_continuation = liveparam.getparam(self._video_id, 3)
         await self._listen(initial_continuation)
 
     async def _listen(self, continuation):
@@ -172,13 +169,14 @@ class LiveChatAsync:
 
                     timeout = metadata['timeoutMs'] / 1000
                     chat_component = {
-                        "video_id" : self.video_id,
+                        "video_id": self._video_id,
                         "timeout": timeout,
                         "chatdata": chatdata
                     }
                     time_mark = time.time()
                     if self._direct_mode:
-                        processed_chat = self.processor.process([chat_component])
+                        processed_chat = self.processor.process(
+                            [chat_component])
                         if isinstance(processed_chat, tuple):
                             await self._callback(*processed_chat)
                         else:
@@ -188,14 +186,15 @@ class LiveChatAsync:
                     diff_time = timeout - (time.time() - time_mark)
                     await asyncio.sleep(diff_time)
                     continuation = metadata.get('continuation')
-        except ChatParseException as e:
-            self._logger.debug(f"[{self.video_id}]{str(e)}")
-            return            
+        except exceptions.ChatParseException as e:
+            self._logger.debug(f"[{self._video_id}]{str(e)}")
+            raise
         except (TypeError, json.JSONDecodeError):
             self._logger.error(f"{traceback.format_exc(limit = -1)}")
-            return
+            raise
 
-        self._logger.debug(f"[{self.video_id}]finished fetching chat.")
+        self._logger.debug(f"[{self._video_id}]finished fetching chat.")
+        raise exceptions.ChatDataFinished
 
     async def _check_pause(self, continuation):
         if self._pauser.empty():
@@ -207,7 +206,7 @@ class LiveChatAsync:
             self._pauser.put_nowait(None)
             if not self._is_replay:
                 continuation = liveparam.getparam(
-                    self.video_id, 3, self._topchat_only)
+                    self._video_id, 3, self._topchat_only)
         return continuation
 
     async def _get_contents(self, continuation, session, headers):
@@ -219,9 +218,7 @@ class LiveChatAsync:
           -------
             'continuationContents' which includes metadata & chatdata.
         '''
-        livechat_json = (await 
-            self._get_livechat_json(continuation, session, headers)
-        )
+        livechat_json = await self._get_livechat_json(continuation, session, headers)
         contents = self._parser.get_contents(livechat_json)
         if self._first_fetch:
             if contents is None or self._is_replay:
@@ -229,7 +226,7 @@ class LiveChatAsync:
                 self._parser.is_replay = True
                 self._fetch_url = "live_chat_replay/get_live_chat_replay?continuation="
                 continuation = arcparam.getparam(
-                    self.video_id, self.seektime, self._topchat_only)
+                    self._video_id, self.seektime, self._topchat_only)
                 livechat_json = (await self._get_livechat_json(
                                  continuation, session, headers))
                 reload_continuation = self._parser.reload_continuation(
@@ -260,7 +257,7 @@ class LiveChatAsync:
                     await asyncio.sleep(1)
                     continue
         else:
-            self._logger.error(f"[{self.video_id}]"
+            self._logger.error(f"[{self._video_id}]"
                                f"Exceeded retry count. status_code={status_code}")
             return None
         return livechat_json
@@ -290,9 +287,12 @@ class LiveChatAsync:
              : Processorによって加工されたチャットデータ
         """
         if self._callback is None:
+            if self.is_alive():
                 items = await self._buffer.get()
                 return self.processor.process(items)
-        raise IllegalFunctionCall(
+            else:
+                return []
+        raise exceptions.IllegalFunctionCall(
             "既にcallbackを登録済みのため、get()は実行できません。")
 
     def is_replay(self):
@@ -313,22 +313,37 @@ class LiveChatAsync:
     def is_alive(self):
         return self._is_alive
 
-    def finish(self,sender):
+    def _finish(self, sender):
         '''Listener終了時のコールバック'''
         try:
-            self.terminate()
+            self._task_finished()
         except CancelledError:
-            self._logger.debug(f'[{self.video_id}]cancelled:{sender}')
+            self._logger.debug(f'[{self._video_id}]cancelled:{sender}')
 
     def terminate(self):
+        if self._pauser.empty():
+            self._pauser.put_nowait(None)
+        self._is_alive = False
+        self._buffer.put_nowait({})
+        self.processor.finalize()
+        
+    def _task_finished(self):
         '''
         Listenerを終了する。
         '''
-        self._is_alive = False
-        if self._direct_mode == False:
-            #bufferにダミーオブジェクトを入れてis_alive()を判定させる
-            self._buffer.put_nowait({'chatdata':'','timeout':0}) 
-        self._logger.info(f'[{self.video_id}]finished.')
+        if self.is_alive():
+            self.terminate()
+            try:
+                self.listen_task.result()
+            except Exception as e:
+                self.exception = e
+                if not isinstance(e, exceptions.ChatParseException):
+                    self._logger.error(f'Internal exception - {type(e)}{str(e)}')
+        self._logger.info(f'[{self._video_id}]終了しました')
+
+    def raise_for_status(self):
+        if self.exception is not None:
+            raise self.exception
 
     @classmethod
     def _set_exception_handler(cls, handler):
@@ -342,7 +357,7 @@ class LiveChatAsync:
                  asyncio.current_task()]
         [task.cancel() for task in tasks]
 
-        cls._logger.debug(f"complete remaining tasks...")
+        cls._logger.debug("complete remaining tasks...")
         await asyncio.gather(*tasks, return_exceptions=True)
         loop = asyncio.get_event_loop()
         loop.stop()

pytchat/core_multithread/buffer.py

@@ -1,6 +1,7 @@
 
 import queue
 
+
 class Buffer(queue.Queue):
     '''
     チャットデータを格納するバッファの役割を持つFIFOキュー
@@ -11,6 +12,7 @@ class Buffer(queue.Queue):
         格納するチャットブロックの最大個数。0の場合は無限。
         最大値を超える場合は古いチャットブロックから破棄される。
     '''
+
     def __init__(self, maxsize=0):
         super().__init__(maxsize=maxsize)
 

pytchat/core_multithread/livechat.py

@@ -1,17 +1,16 @@
 import requests
-import datetime
 import json
-import random
 import signal
 import time
 import traceback
 import urllib.parse
 from concurrent.futures import CancelledError, ThreadPoolExecutor
 from queue import Queue
+from threading import Event
 from .buffer import Buffer
 from ..parser.live import Parser
 from .. import config
-from ..exceptions  import ChatParseException,IllegalFunctionCall
+from .. import exceptions
 from ..paramgen import liveparam, arcparam
 from ..processors.default.processor import DefaultProcessor
 from ..processors.combinator import Combinator
@@ -72,11 +71,9 @@ class LiveChat:
     '''
 
     _setup_finished = False
-    #チャット監視中のListenerのリスト
-    _listeners = []
 
     def __init__(self, video_id,
-                seektime = 0,
+                 seektime=-1,
                  processor=DefaultProcessor(),
                  buffer=None,
                  interruptable=True,
@@ -87,7 +84,7 @@ class LiveChat:
                  topchat_only=False,
                  logger=config.logger(__name__)
                  ):
-        self.video_id  = video_id
+        self._video_id = video_id
         self.seektime = seektime
         if isinstance(processor, tuple):
             self.processor = Combinator(processor)
@@ -103,25 +100,21 @@ class LiveChat:
         self._parser = Parser(is_replay=self._is_replay)
         self._pauser = Queue()
         self._pauser.put_nowait(None)
-        self._setup()
         self._first_fetch = True
         self._fetch_url = "live_chat/get_live_chat?continuation="
         self._topchat_only = topchat_only
+        self._event = Event()
         self._logger = logger
-        LiveChat._logger = logger
-        if not LiveChat._setup_finished:
-            LiveChat._setup_finished = True
+        self.exception = None
         if interruptable:
-                signal.signal(signal.SIGINT,  (lambda a, b:  
-                (LiveChat.shutdown(None,signal.SIGINT,b))
-                ))
-        LiveChat._listeners.append(self)
+            signal.signal(signal.SIGINT, lambda a, b: self.terminate())
+        self._setup()
 
     def _setup(self):
         # direct modeがTrueでcallback未設定の場合例外発生。
         if self._direct_mode:
             if self._callback is None:
-                raise IllegalFunctionCall(
+                raise exceptions.IllegalFunctionCall(
                     "When direct_mode=True, callback parameter is required.")
         else:
             # direct modeがFalseでbufferが未設定ならばデフォルトのbufferを作成
@@ -134,19 +127,19 @@ class LiveChat:
                 # callbackを呼ぶループタスクの開始
                 self._executor.submit(self._callback_loop, self._callback)
         # _listenループタスクの開始
-        listen_task = self._executor.submit(self._startlisten)
+        self.listen_task = self._executor.submit(self._startlisten)
         # add_done_callbackの登録
         if self._done_callback is None:
-            listen_task.add_done_callback(self.finish)
+            self.listen_task.add_done_callback(self._finish)
         else:
-            listen_task.add_done_callback(self._done_callback)
+            self.listen_task.add_done_callback(self._done_callback)
 
     def _startlisten(self):
         time.sleep(0.1)  # sleep shortly to prohibit skipping fetching data
         """Fetch first continuation parameter,
         create and start _listen loop.
         """
-        initial_continuation = liveparam.getparam(self.video_id,3)
+        initial_continuation = liveparam.getparam(self._video_id, 3)
         self._listen(initial_continuation)
 
     def _listen(self, continuation):
@@ -162,19 +155,18 @@ class LiveChat:
             with requests.Session() as session:
                 while(continuation and self._is_alive):
                     continuation = self._check_pause(continuation)
-                    contents = self._get_contents(
-                        continuation, session, headers)
+                    contents = self._get_contents(continuation, session, headers)
                     metadata, chatdata = self._parser.parse(contents)
-
                     timeout = metadata['timeoutMs'] / 1000
                     chat_component = {
-                        "video_id" : self.video_id,
+                        "video_id": self._video_id,
                         "timeout": timeout,
                         "chatdata": chatdata
                     }
                     time_mark = time.time()
                     if self._direct_mode:
-                        processed_chat = self.processor.process([chat_component])
+                        processed_chat = self.processor.process(
+                            [chat_component])
                         if isinstance(processed_chat, tuple):
                             self._callback(*processed_chat)
                         else:
@@ -182,16 +174,17 @@ class LiveChat:
                     else:
                         self._buffer.put(chat_component)
                     diff_time = timeout - (time.time() - time_mark)
-                    time.sleep(diff_time if diff_time > 0 else 0)        
+                    self._event.wait(diff_time if diff_time > 0 else 0)
                     continuation = metadata.get('continuation')
-        except ChatParseException as e:
-            self._logger.debug(f"[{self.video_id}]{str(e)}")
-            return            
+        except exceptions.ChatParseException as e:
+            self._logger.debug(f"[{self._video_id}]{str(e)}")
+            raise
         except (TypeError, json.JSONDecodeError):
             self._logger.error(f"{traceback.format_exc(limit=-1)}")
-            return
+            raise
 
-        self._logger.debug(f"[{self.video_id}]finished fetching chat.")
+        self._logger.debug(f"[{self._video_id}]finished fetching chat.")
+        raise exceptions.ChatDataFinished
 
     def _check_pause(self, continuation):
         if self._pauser.empty():
@@ -202,7 +195,7 @@ class LiveChat:
             '''
             self._pauser.put_nowait(None)
             if not self._is_replay:
-                continuation = liveparam.getparam(self.video_id,3)
+                continuation = liveparam.getparam(self._video_id, 3)
         return continuation
 
     def _get_contents(self, continuation, session, headers):
@@ -224,9 +217,8 @@ class LiveChat:
                 self._parser.is_replay = True
                 self._fetch_url = "live_chat_replay/get_live_chat_replay?continuation="
                 continuation = arcparam.getparam(
-                    self.video_id, self.seektime, self._topchat_only)
-                livechat_json = ( self._get_livechat_json(
-                    continuation, session, headers))
+                    self._video_id, self.seektime, self._topchat_only)
+                livechat_json = (self._get_livechat_json(continuation, session, headers))
                 reload_continuation = self._parser.reload_continuation(
                     self._parser.get_contents(livechat_json))
                 if reload_continuation:
@@ -255,9 +247,9 @@ class LiveChat:
                     time.sleep(1)
                     continue
         else:
-            self._logger.error(f"[{self.video_id}]"
+            self._logger.error(f"[{self._video_id}]"
                                f"Exceeded retry count. status_code={status_code}")
-            return None
+            raise exceptions.RetryExceedMaxCount()
         return livechat_json
 
     def _callback_loop(self, callback):
@@ -285,9 +277,12 @@ class LiveChat:
              : Processorによって加工されたチャットデータ
         """
         if self._callback is None:
+            if self.is_alive():
                 items = self._buffer.get()
                 return self.processor.process(items)
-        raise IllegalFunctionCall(
+            else:
+                return []
+        raise exceptions.IllegalFunctionCall(
             "既にcallbackを登録済みのため、get()は実行できません。")
 
     def is_replay(self):
@@ -308,25 +303,35 @@ class LiveChat:
     def is_alive(self):
         return self._is_alive
 
-    def finish(self,sender):
+    def _finish(self, sender):
         '''Listener終了時のコールバック'''
         try:
-            self.terminate()
+            self._task_finished()
         except CancelledError:
-            self._logger.debug(f'[{self.video_id}]cancelled:{sender}')
+            self._logger.debug(f'[{self._video_id}]cancelled:{sender}')
 
     def terminate(self):
+        if self._pauser.empty():
+            self._pauser.put_nowait(None)
+        self._is_alive = False
+        self._buffer.put({})
+        self._event.set()
+        self.processor.finalize()
+
+    def _task_finished(self):
         '''
         Listenerを終了する。
         '''
-        self._is_alive = False
-        if self._direct_mode == False:
-            #bufferにダミーオブジェクトを入れてis_alive()を判定させる
-            self._buffer.put({'chatdata':'','timeout':0}) 
-        self._logger.info(f'[{self.video_id}]finished.')
+        if self.is_alive():
+            self.terminate()
+            try:
+                self.listen_task.result()
+            except Exception as e:
+                self.exception = e
+                if not isinstance(e, exceptions.ChatParseException):
+                    self._logger.error(f'Internal exception - {type(e)}{str(e)}')
+        self._logger.info(f'[{self._video_id}]終了しました')
 
-    @classmethod
-    def shutdown(cls, event, sig = None, handler=None):
-        cls._logger.debug("shutdown...")
-        for t in LiveChat._listeners:
-            t._is_alive = False
+    def raise_for_status(self):
+        if self.exception is not None:
+            raise self.exception

pytchat/exceptions.py

@@ -4,11 +4,6 @@ class ChatParseException(Exception):
     '''
     pass
 
-class NoYtinitialdataException(ChatParseException):
-    '''
-    Thrown when the video is not found.
-    '''
-    pass
 
 class ResponseContextError(ChatParseException):
     '''
@@ -16,25 +11,21 @@ class ResponseContextError(ChatParseException):
     '''
     pass
 
-class NoLivechatRendererException(ChatParseException):
-    '''
-    Thrown when livechatRenderer is missing in JSON.
-    '''
-    pass
 
-
-class NoContentsException(ChatParseException):
+class NoContents(ChatParseException):
     '''
     Thrown when ContinuationContents is missing in JSON.
     '''
     pass
 
-class NoContinuationsException(ChatParseException):
+
+class NoContinuation(ChatParseException):
     '''
     Thrown when continuation is missing in ContinuationContents.
     '''
     pass
 
+
 class IllegalFunctionCall(Exception):
     '''
     Thrown when get() is called even though
@@ -42,11 +33,32 @@ class IllegalFunctionCall(Exception):
     '''
     pass
 
+
 class InvalidVideoIdException(Exception):
     '''
     Thrown when the video_id is not exist (VideoInfo).
     '''
     pass
 
+
 class UnknownConnectionError(Exception):
     pass
+
+
+class RetryExceedMaxCount(Exception):
+    '''
+    thrown when the number of retries exceeds the maximum value.
+    '''
+    pass
+
+
+class ChatDataFinished(ChatParseException):
+    pass
+
+
+class ReceivedUnknownContinuation(ChatParseException):
+    pass
+
+
+class FailedExtractContinuation(ChatDataFinished):
+    pass

pytchat/paramgen/arcparam.py

@@ -1,107 +1,55 @@
-from base64 import urlsafe_b64encode as b64enc
-from functools import reduce
-import math
-import random
-import urllib.parse
+from .pb.header_pb2 import Header
+from .pb.replay_pb2 import Continuation
+from urllib.parse import quote
+import base64
 
 '''
 Generate continuation parameter of youtube replay chat.
 
-Author: taizan-hokuto (2019) @taizan205
+Author: taizan-hokuto
 
-ver 0.0.1 2019.10.05
+ver 0.0.1 2019.10.05 : Initial release.
+ver 0.0.2 2020.05.30 : Use Protocol Buffers.
 '''
 
-def _gen_vid(video_id):
-    """generate video_id parameter.
-    Parameter
-    ---------
-        video_id : str
 
-    Return
-    ---------
-        bytes : base64 encoded video_id parameter.
-    """
-    header_magic = b'\x0A\x0F\x1A\x0D\x0A'
-    header_id =  video_id.encode()
-    header_sep_1 = b'\x1A\x13\xEA\xA8\xDD\xB9\x01\x0D\x0A\x0B'
-    header_terminator = b'\x20\x01'
+def _gen_vid(video_id) -> str:
+    header = Header()
+    header.info.video.id = video_id
+    header.terminator = 1
+    return base64.urlsafe_b64encode(header.SerializeToString()).decode()
 
-    item = [
-        header_magic,
-        _nval(len(header_id)),
-        header_id,
-        header_sep_1,
-        header_id,
-        header_terminator
-    ]
 
-    return urllib.parse.quote(
-        b64enc(reduce(lambda x, y: x+y, item)).decode()
-    ).encode()
+def _build(video_id, seektime, topchat_only) -> str:
+    chattype = 1
+    timestamp = 0
+    if topchat_only:
+        chattype = 4
 
-def _nval(val):
-    """convert value to byte array"""
-    if val<0: raise ValueError
-    buf = b''
-    while val >> 7:
-        m = val & 0xFF | 0x80
-        buf += m.to_bytes(1,'big')
-        val >>= 7
-    buf += val.to_bytes(1,'big')
-    return buf
-
-def _build(video_id, seektime, topchat_only):
-    switch_01 = b'\x04' if topchat_only else b'\x01'
+    fetch_before_start = 3
     if seektime < 0:
-        times =_nval(0)
-        switch = b'\x04'       
+        fetch_before_start = 4
     elif seektime == 0:
-        times =_nval(1)
-        switch = b'\x03'       
+        timestamp = 1
     else:
-        times =_nval(int(seektime*1000000))
-        switch = b'\x03'
-    parity = b'\x00'
+        timestamp = int(seektime * 1000000)
+    continuation = Continuation()
+    entity = continuation.entity
+    entity.header = _gen_vid(video_id)
+    entity.timestamp = timestamp
+    entity.s6 = 0
+    entity.s7 = 0
+    entity.s8 = 0
+    entity.s9 = fetch_before_start
+    entity.s10 = ''
+    entity.s12 = chattype
+    entity.chattype.value = chattype
+    entity.s15 = 0
+    return quote(
+        base64.urlsafe_b64encode(continuation.SerializeToString()).decode())
 
-    header_magic= b'\xA2\x9D\xB0\xD3\x04'
-    sep_0       = b'\x1A'
-    vid         = _gen_vid(video_id)
-    time_tag    = b'\x28'
-    timestamp1  = times
-    sep_1       = b'\x30\x00\x38\x00\x40\x00\x48'
-    sep_2       = b'\x52\x1C\x08\x00\x10\x00\x18\x00\x20\x00'
-    chkstr      = b'\x2A\x0E\x73\x74\x61\x74\x69\x63\x63\x68\x65\x63\x6B\x73\x75\x6D\x40'
-    sep_3       = b'\x00\x58\x03\x60'
-    sep_4       = b'\x68' + parity + b'\x72\x04\x08'
-    sep_5       = b'\x10' + parity + b'\x78\x00'
-    body = [
-        sep_0,
-        _nval(len(vid)),
-        vid,
-        time_tag,
-        timestamp1,
-        sep_1,
-        switch,
-        sep_2,
-        chkstr,
-        sep_3,
-        switch_01,
-        sep_4,
-        switch_01,
-        sep_5
-    ]
 
-    body = reduce(lambda x, y: x+y, body)
-    
-    return  urllib.parse.quote(
-                b64enc( header_magic +
-                        _nval(len(body)) +
-                        body
-                ).decode()
-            )
-
-def getparam(video_id, seektime = 0, topchat_only = False):
+def getparam(video_id, seektime=-1, topchat_only=False) -> str:
     '''
     Parameter
     ---------

pytchat/paramgen/arcparam_mining.py

@@ -1,7 +1,5 @@
 from base64 import urlsafe_b64encode as b64enc
 from functools import reduce
-import math
-import random
 import urllib.parse
 
 '''
@@ -12,6 +10,7 @@ Author: taizan-hokuto (2019) @taizan205
 ver 0.0.1 2019.10.05
 '''
 
+
 def _gen_vid_long(video_id):
     """generate video_id parameter.
     Parameter
@@ -40,6 +39,7 @@ def _gen_vid_long(video_id):
         b64enc(reduce(lambda x, y: x + y, item)).decode()
     ).encode()
 
+
 def _gen_vid(video_id):
     """generate video_id parameter.
     Parameter
@@ -65,9 +65,11 @@ def _gen_vid(video_id):
         b64enc(reduce(lambda x, y: x + y, item)).decode()
     ).encode()
 
+
 def _nval(val):
     """convert value to byte array"""
-    if val<0: raise ValueError
+    if val < 0:
+        raise ValueError
     buf = b''
     while val >> 7:
         m = val & 0xFF | 0x80
@@ -76,6 +78,7 @@ def _nval(val):
     buf += val.to_bytes(1, 'big')
     return buf
 
+
 def _build(video_id, seektime, topchat_only):
     switch_01 = b'\x04' if topchat_only else b'\x01'
     if seektime < 0:
@@ -85,9 +88,7 @@ def _build(video_id, seektime, topchat_only):
     else:
         times = _nval(int(seektime * 1000))
     if seektime > 0:
-        _len_time = ( b'\x5A'
-                    + (len(times)+1).to_bytes(1,'big')
-                    + b'\x10')
+        _len_time = b'\x5A' + (len(times) + 1).to_bytes(1, 'big') + b'\x10'
     else:
         _len_time = b''
 
@@ -114,12 +115,11 @@ def _build(video_id, seektime, topchat_only):
     body = reduce(lambda x, y: x + y, body)
 
     return urllib.parse.quote(
-                b64enc( header_magic +
-                        _nval(len(body)) +
-                        body
+        b64enc(header_magic + _nval(len(body)) + body
                ).decode()
     )
 
+
 def getparam(video_id, seektime=0.0, topchat_only=False):
     '''
     Parameter

pytchat/paramgen/liveparam.py

@@ -1,17 +1,21 @@
-from base64 import urlsafe_b64encode as b64enc
-from functools import reduce
-import time
+from .pb.header_pb2 import Header
+from .pb.live_pb2 import Continuation
+from urllib.parse import quote
+import base64
 import random
-import urllib.parse
+import time
 
 '''
 Generate continuation parameter of youtube live chat.
 
-Author: taizan-hokuto (2019) @taizan205
+Author: taizan-hokuto
 
-ver 0.0.1 2019.10.05
+ver 0.0.1 2019.10.05 : Initial release.
+ver 0.0.2 2020.05.30 : Use Protocol Buffers.
 '''
-def _gen_vid(video_id):
+
+
+def _gen_vid(video_id) -> str:
     """generate video_id parameter.
     Parameter
     ---------
@@ -19,130 +23,51 @@ def _gen_vid(video_id):
 
     Return
     ---------
-    byte[] : base64 encoded video_id parameter.
+    str : base64 encoded video_id parameter.
     """
-    header_magic = b'\x0A\x0F\x0A\x0D\x0A'
-    header_id =  video_id.encode()
-    header_sep_1 = b'\x1A'
-    header_sep_2 = b'\x43\xAA\xB9\xC1\xBD\x01\x3D\x0A'
-    header_suburl = ('https://www.youtube.com/live_chat?v='
-                    f'{video_id}&is_popout=1').encode()
-    header_terminator = b'\x20\x02'
+    header = Header()
+    header.info.video.id = video_id
+    header.terminator = 1
+    return base64.urlsafe_b64encode(header.SerializeToString()).decode()
 
-    item = [
-        header_magic,
-        _nval(len(header_id)),
-        header_id,
-        header_sep_1,
-        header_sep_2,
-        _nval(len(header_suburl)),
-        header_suburl,
-        header_terminator
-    ]
 
-    return urllib.parse.quote(
-        b64enc(reduce(lambda x, y: x+y, item)).decode()
-    ).encode()
+def _build(video_id, ts1, ts2, ts3, ts4, ts5, topchat_only) -> str:
+    chattype = 1
+    if topchat_only:
+        chattype = 4
+    continuation = Continuation()
+    entity = continuation.entity
 
-def _tzparity(video_id,times):
-    t=0
-    for i,s in enumerate(video_id):
-        ss = ord(s)
-        if(ss % 2 == 0):
-            t += ss*(12-i)
-        else:
-            t ^= ss*i
+    entity.header = _gen_vid(video_id)
+    entity.timestamp1 = ts1
+    entity.s6 = 0
+    entity.s7 = 0
+    entity.s8 = 1
+    entity.body.b1 = 0
+    entity.body.b2 = 0
+    entity.body.b3 = 0
+    entity.body.b4 = 0
+    entity.body.b7 = ''
+    entity.body.b8 = 0
+    entity.body.b9 = ''
+    entity.body.timestamp2 = ts2
+    entity.body.b11 = 3
+    entity.body.b15 = 0
+    entity.timestamp3 = ts3
+    entity.timestamp4 = ts4
+    entity.s13 = chattype
+    entity.chattype.value = chattype
+    entity.s17 = 0
+    entity.str19.value = 0
+    entity.timestamp5 = ts5
 
-    return ((times^t) % 2).to_bytes(1,'big')
-
-def _nval(val):
-    """convert value to byte array"""
-    if val<0: raise ValueError
-    buf = b''
-    while val >> 7:
-        m = val & 0xFF | 0x80
-        buf += m.to_bytes(1,'big')
-        val >>= 7
-    buf += val.to_bytes(1,'big')
-    return buf
-
-def _build(video_id, _ts1, _ts2, _ts3, _ts4, _ts5, topchat_only):
-    #_short_type2
-    switch_01 = b'\x04' if topchat_only else b'\x01'
-    parity = _tzparity(video_id, _ts1^_ts2^_ts3^_ts4^_ts5)
-
-    header_magic= b'\xD2\x87\xCC\xC8\x03'
-    sep_0       = b'\x1A'
-    vid         = _gen_vid(video_id)
-    time_tag    = b'\x28'
-    timestamp1  = _nval(_ts1)
-    sep_1       = b'\x30\x00\x38\x00\x40\x02\x4A'
-    un_len      = b'\x2B'
-    sep_2       = b'\x08'+parity+b'\x10\x00\x18\x00\x20\x00'
-    chkstr      = b'\x2A\x0E\x73\x74\x61\x74\x69\x63\x63\x68\x65\x63\x6B\x73\x75\x6D'
-    sep_3       = b'\x3A\x00\x40\x00\x4A'
-    sep_4_len   = b'\x02'
-    sep_4       = b'\x08\x01'
-    ts_2_start  = b'\x50'
-    timestamp2  = _nval(_ts2)
-    ts_2_end    = b'\x58'
-    sep_5       = b'\x03'
-    ts_3_start  = b'\x50'
-    timestamp3  = _nval(_ts3)
-    ts_3_end    = b'\x58'
-    timestamp4  = _nval(_ts4)
-    sep_6       = b'\x68'
-    #switch
-    sep_7       = b'\x82\x01\x04\x08'
-    #switch
-    sep_8       = b'\x10\x00'
-    sep_9       = b'\x88\x01\x00\xA0\x01'
-    timestamp5  = _nval(_ts5)
-
-    body = [
-        sep_0,
-        _nval(len(vid)),
-        vid,
-        time_tag,
-        timestamp1,
-        sep_1,
-        un_len,
-        sep_2,
-        chkstr,
-        sep_3,
-        sep_4_len,
-        sep_4,
-        ts_2_start,
-        timestamp2,
-        ts_2_end,
-        sep_5,
-        ts_3_start,
-        timestamp3,
-        ts_3_end,
-        timestamp4,
-        sep_6,
-        switch_01,#
-        sep_7,
-        switch_01,#
-        sep_8,
-        sep_9,
-        timestamp5
-    ]
-
-    body = reduce(lambda x, y: x+y, body)
-    
-    return  urllib.parse.quote(
-                b64enc( header_magic +
-                        _nval(len(body)) +
-                        body
-                ).decode()
+    return quote(
+        base64.urlsafe_b64encode(continuation.SerializeToString()).decode()
     )
 
 
 def _times(past_sec):
-    
     n = int(time.time())
-    
     _ts1 = n - random.uniform(0, 1 * 3)
     _ts2 = n - random.uniform(0.01, 0.99)
     _ts3 = n - past_sec + random.uniform(0, 1)
@@ -151,7 +76,7 @@ def _times(past_sec):
     return list(map(lambda x: int(x * 1000000), [_ts1, _ts2, _ts3, _ts4, _ts5]))
 
 
-def getparam(video_id, past_sec = 0, topchat_only = False):
+def getparam(video_id, past_sec=0, topchat_only=False) -> str:
     '''
     Parameter
     ---------
@@ -161,4 +86,3 @@ def getparam(video_id, past_sec = 0, topchat_only = False):
         if True, fetch only 'top chat'
     '''
     return _build(video_id, *_times(past_sec), topchat_only)
-

pytchat/paramgen/pb/header_pb2.py

@@ -0,0 +1,159 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: header.proto
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+  name='header.proto',
+  package='',
+  syntax='proto3',
+  serialized_options=None,
+  create_key=_descriptor._internal_create_key,
+  serialized_pb=b'\n\x0cheader.proto\"\x13\n\x05Video\x12\n\n\x02id\x18\x01 \x01(\t\"#\n\nHeaderInfo\x12\x15\n\x05video\x18\x01 \x01(\x0b\x32\x06.Video\"7\n\x06Header\x12\x19\n\x04info\x18\x01 \x01(\x0b\x32\x0b.HeaderInfo\x12\x12\n\nterminator\x18\x04 \x01(\x05\x62\x06proto3'
+)
+
+
+
+
+_VIDEO = _descriptor.Descriptor(
+  name='Video',
+  full_name='Video',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='id', full_name='Video.id', index=0,
+      number=1, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=16,
+  serialized_end=35,
+)
+
+
+_HEADERINFO = _descriptor.Descriptor(
+  name='HeaderInfo',
+  full_name='HeaderInfo',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='video', full_name='HeaderInfo.video', index=0,
+      number=1, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=37,
+  serialized_end=72,
+)
+
+
+_HEADER = _descriptor.Descriptor(
+  name='Header',
+  full_name='Header',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='info', full_name='Header.info', index=0,
+      number=1, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='terminator', full_name='Header.terminator', index=1,
+      number=4, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=74,
+  serialized_end=129,
+)
+
+_HEADERINFO.fields_by_name['video'].message_type = _VIDEO
+_HEADER.fields_by_name['info'].message_type = _HEADERINFO
+DESCRIPTOR.message_types_by_name['Video'] = _VIDEO
+DESCRIPTOR.message_types_by_name['HeaderInfo'] = _HEADERINFO
+DESCRIPTOR.message_types_by_name['Header'] = _HEADER
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+Video = _reflection.GeneratedProtocolMessageType('Video', (_message.Message,), {
+  'DESCRIPTOR' : _VIDEO,
+  '__module__' : 'header_pb2'
+  # @@protoc_insertion_point(class_scope:Video)
+  })
+_sym_db.RegisterMessage(Video)
+
+HeaderInfo = _reflection.GeneratedProtocolMessageType('HeaderInfo', (_message.Message,), {
+  'DESCRIPTOR' : _HEADERINFO,
+  '__module__' : 'header_pb2'
+  # @@protoc_insertion_point(class_scope:HeaderInfo)
+  })
+_sym_db.RegisterMessage(HeaderInfo)
+
+Header = _reflection.GeneratedProtocolMessageType('Header', (_message.Message,), {
+  'DESCRIPTOR' : _HEADER,
+  '__module__' : 'header_pb2'
+  # @@protoc_insertion_point(class_scope:Header)
+  })
+_sym_db.RegisterMessage(Header)
+
+
+# @@protoc_insertion_point(module_scope)

pytchat/paramgen/pb/live_pb2.py

@@ -0,0 +1,381 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: live.proto
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+  name='live.proto',
+  package='live',
+  syntax='proto3',
+  serialized_options=None,
+  create_key=_descriptor._internal_create_key,
+  serialized_pb=b'\n\nlive.proto\x12\x04live\"\x88\x01\n\x04\x42ody\x12\n\n\x02\x62\x31\x18\x01 \x01(\x05\x12\n\n\x02\x62\x32\x18\x02 \x01(\x05\x12\n\n\x02\x62\x33\x18\x03 \x01(\x05\x12\n\n\x02\x62\x34\x18\x04 \x01(\x05\x12\n\n\x02\x62\x37\x18\x07 \x01(\t\x12\n\n\x02\x62\x38\x18\x08 \x01(\x05\x12\n\n\x02\x62\x39\x18\t \x01(\t\x12\x12\n\ntimestamp2\x18\n \x01(\x03\x12\x0b\n\x03\x62\x31\x31\x18\x0b \x01(\x05\x12\x0b\n\x03\x62\x31\x35\x18\x0f \x01(\x05\"\x19\n\x08\x43hatType\x12\r\n\x05value\x18\x01 \x01(\x05\"\x16\n\x05STR19\x12\r\n\x05value\x18\x01 \x01(\x05\"\x8a\x02\n\x12\x43ontinuationEntity\x12\x0e\n\x06header\x18\x03 \x01(\t\x12\x12\n\ntimestamp1\x18\x05 \x01(\x03\x12\n\n\x02s6\x18\x06 \x01(\x05\x12\n\n\x02s7\x18\x07 \x01(\x05\x12\n\n\x02s8\x18\x08 \x01(\x05\x12\x18\n\x04\x62ody\x18\t \x01(\x0b\x32\n.live.Body\x12\x12\n\ntimestamp3\x18\n \x01(\x03\x12\x12\n\ntimestamp4\x18\x0b \x01(\x03\x12\x0b\n\x03s13\x18\r \x01(\x05\x12 \n\x08\x63hattype\x18\x10 \x01(\x0b\x32\x0e.live.ChatType\x12\x0b\n\x03s17\x18\x11 \x01(\x05\x12\x1a\n\x05str19\x18\x13 \x01(\x0b\x32\x0b.live.STR19\x12\x12\n\ntimestamp5\x18\x14 \x01(\x03\";\n\x0c\x43ontinuation\x12+\n\x06\x65ntity\x18\xfa\xc0\x89\x39 \x01(\x0b\x32\x18.live.ContinuationEntityb\x06proto3'
+)
+
+
+
+
+_BODY = _descriptor.Descriptor(
+  name='Body',
+  full_name='live.Body',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='b1', full_name='live.Body.b1', index=0,
+      number=1, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b2', full_name='live.Body.b2', index=1,
+      number=2, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b3', full_name='live.Body.b3', index=2,
+      number=3, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b4', full_name='live.Body.b4', index=3,
+      number=4, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b7', full_name='live.Body.b7', index=4,
+      number=7, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b8', full_name='live.Body.b8', index=5,
+      number=8, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b9', full_name='live.Body.b9', index=6,
+      number=9, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp2', full_name='live.Body.timestamp2', index=7,
+      number=10, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b11', full_name='live.Body.b11', index=8,
+      number=11, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='b15', full_name='live.Body.b15', index=9,
+      number=15, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=21,
+  serialized_end=157,
+)
+
+
+_CHATTYPE = _descriptor.Descriptor(
+  name='ChatType',
+  full_name='live.ChatType',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='value', full_name='live.ChatType.value', index=0,
+      number=1, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=159,
+  serialized_end=184,
+)
+
+
+_STR19 = _descriptor.Descriptor(
+  name='STR19',
+  full_name='live.STR19',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='value', full_name='live.STR19.value', index=0,
+      number=1, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=186,
+  serialized_end=208,
+)
+
+
+_CONTINUATIONENTITY = _descriptor.Descriptor(
+  name='ContinuationEntity',
+  full_name='live.ContinuationEntity',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='header', full_name='live.ContinuationEntity.header', index=0,
+      number=3, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp1', full_name='live.ContinuationEntity.timestamp1', index=1,
+      number=5, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s6', full_name='live.ContinuationEntity.s6', index=2,
+      number=6, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s7', full_name='live.ContinuationEntity.s7', index=3,
+      number=7, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s8', full_name='live.ContinuationEntity.s8', index=4,
+      number=8, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='body', full_name='live.ContinuationEntity.body', index=5,
+      number=9, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp3', full_name='live.ContinuationEntity.timestamp3', index=6,
+      number=10, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp4', full_name='live.ContinuationEntity.timestamp4', index=7,
+      number=11, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s13', full_name='live.ContinuationEntity.s13', index=8,
+      number=13, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='chattype', full_name='live.ContinuationEntity.chattype', index=9,
+      number=16, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s17', full_name='live.ContinuationEntity.s17', index=10,
+      number=17, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='str19', full_name='live.ContinuationEntity.str19', index=11,
+      number=19, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp5', full_name='live.ContinuationEntity.timestamp5', index=12,
+      number=20, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=211,
+  serialized_end=477,
+)
+
+
+_CONTINUATION = _descriptor.Descriptor(
+  name='Continuation',
+  full_name='live.Continuation',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='entity', full_name='live.Continuation.entity', index=0,
+      number=119693434, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=479,
+  serialized_end=538,
+)
+
+_CONTINUATIONENTITY.fields_by_name['body'].message_type = _BODY
+_CONTINUATIONENTITY.fields_by_name['chattype'].message_type = _CHATTYPE
+_CONTINUATIONENTITY.fields_by_name['str19'].message_type = _STR19
+_CONTINUATION.fields_by_name['entity'].message_type = _CONTINUATIONENTITY
+DESCRIPTOR.message_types_by_name['Body'] = _BODY
+DESCRIPTOR.message_types_by_name['ChatType'] = _CHATTYPE
+DESCRIPTOR.message_types_by_name['STR19'] = _STR19
+DESCRIPTOR.message_types_by_name['ContinuationEntity'] = _CONTINUATIONENTITY
+DESCRIPTOR.message_types_by_name['Continuation'] = _CONTINUATION
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+Body = _reflection.GeneratedProtocolMessageType('Body', (_message.Message,), {
+  'DESCRIPTOR' : _BODY,
+  '__module__' : 'live_pb2'
+  # @@protoc_insertion_point(class_scope:live.Body)
+  })
+_sym_db.RegisterMessage(Body)
+
+ChatType = _reflection.GeneratedProtocolMessageType('ChatType', (_message.Message,), {
+  'DESCRIPTOR' : _CHATTYPE,
+  '__module__' : 'live_pb2'
+  # @@protoc_insertion_point(class_scope:live.ChatType)
+  })
+_sym_db.RegisterMessage(ChatType)
+
+STR19 = _reflection.GeneratedProtocolMessageType('STR19', (_message.Message,), {
+  'DESCRIPTOR' : _STR19,
+  '__module__' : 'live_pb2'
+  # @@protoc_insertion_point(class_scope:live.STR19)
+  })
+_sym_db.RegisterMessage(STR19)
+
+ContinuationEntity = _reflection.GeneratedProtocolMessageType('ContinuationEntity', (_message.Message,), {
+  'DESCRIPTOR' : _CONTINUATIONENTITY,
+  '__module__' : 'live_pb2'
+  # @@protoc_insertion_point(class_scope:live.ContinuationEntity)
+  })
+_sym_db.RegisterMessage(ContinuationEntity)
+
+Continuation = _reflection.GeneratedProtocolMessageType('Continuation', (_message.Message,), {
+  'DESCRIPTOR' : _CONTINUATION,
+  '__module__' : 'live_pb2'
+  # @@protoc_insertion_point(class_scope:live.Continuation)
+  })
+_sym_db.RegisterMessage(Continuation)
+
+
+# @@protoc_insertion_point(module_scope)

pytchat/paramgen/pb/replay_pb2.py

@@ -0,0 +1,215 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: replay.proto
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor.FileDescriptor(
+  name='replay.proto',
+  package='replay',
+  syntax='proto3',
+  serialized_options=None,
+  create_key=_descriptor._internal_create_key,
+  serialized_pb=b'\n\x0creplay.proto\x12\x06replay\"\x19\n\x08\x43hatType\x12\r\n\x05value\x18\x01 \x01(\x05\"\xb2\x01\n\x12\x43ontinuationEntity\x12\x0e\n\x06header\x18\x03 \x01(\t\x12\x11\n\ttimestamp\x18\x05 \x01(\x03\x12\n\n\x02s6\x18\x06 \x01(\x05\x12\n\n\x02s7\x18\x07 \x01(\x05\x12\n\n\x02s8\x18\x08 \x01(\x05\x12\n\n\x02s9\x18\t \x01(\x05\x12\x0b\n\x03s10\x18\n \x01(\t\x12\x0b\n\x03s12\x18\x0c \x01(\x05\x12\"\n\x08\x63hattype\x18\x0e \x01(\x0b\x32\x10.replay.ChatType\x12\x0b\n\x03s15\x18\x0f \x01(\x05\"=\n\x0c\x43ontinuation\x12-\n\x06\x65ntity\x18\xd4\x83\xb6J \x01(\x0b\x32\x1a.replay.ContinuationEntityb\x06proto3'
+)
+
+
+
+
+_CHATTYPE = _descriptor.Descriptor(
+  name='ChatType',
+  full_name='replay.ChatType',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='value', full_name='replay.ChatType.value', index=0,
+      number=1, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=24,
+  serialized_end=49,
+)
+
+
+_CONTINUATIONENTITY = _descriptor.Descriptor(
+  name='ContinuationEntity',
+  full_name='replay.ContinuationEntity',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='header', full_name='replay.ContinuationEntity.header', index=0,
+      number=3, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='timestamp', full_name='replay.ContinuationEntity.timestamp', index=1,
+      number=5, type=3, cpp_type=2, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s6', full_name='replay.ContinuationEntity.s6', index=2,
+      number=6, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s7', full_name='replay.ContinuationEntity.s7', index=3,
+      number=7, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s8', full_name='replay.ContinuationEntity.s8', index=4,
+      number=8, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s9', full_name='replay.ContinuationEntity.s9', index=5,
+      number=9, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s10', full_name='replay.ContinuationEntity.s10', index=6,
+      number=10, type=9, cpp_type=9, label=1,
+      has_default_value=False, default_value=b"".decode('utf-8'),
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s12', full_name='replay.ContinuationEntity.s12', index=7,
+      number=12, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='chattype', full_name='replay.ContinuationEntity.chattype', index=8,
+      number=14, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+    _descriptor.FieldDescriptor(
+      name='s15', full_name='replay.ContinuationEntity.s15', index=9,
+      number=15, type=5, cpp_type=1, label=1,
+      has_default_value=False, default_value=0,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=52,
+  serialized_end=230,
+)
+
+
+_CONTINUATION = _descriptor.Descriptor(
+  name='Continuation',
+  full_name='replay.Continuation',
+  filename=None,
+  file=DESCRIPTOR,
+  containing_type=None,
+  create_key=_descriptor._internal_create_key,
+  fields=[
+    _descriptor.FieldDescriptor(
+      name='entity', full_name='replay.Continuation.entity', index=0,
+      number=156074452, type=11, cpp_type=10, label=1,
+      has_default_value=False, default_value=None,
+      message_type=None, enum_type=None, containing_type=None,
+      is_extension=False, extension_scope=None,
+      serialized_options=None, file=DESCRIPTOR,  create_key=_descriptor._internal_create_key),
+  ],
+  extensions=[
+  ],
+  nested_types=[],
+  enum_types=[
+  ],
+  serialized_options=None,
+  is_extendable=False,
+  syntax='proto3',
+  extension_ranges=[],
+  oneofs=[
+  ],
+  serialized_start=232,
+  serialized_end=293,
+)
+
+_CONTINUATIONENTITY.fields_by_name['chattype'].message_type = _CHATTYPE
+_CONTINUATION.fields_by_name['entity'].message_type = _CONTINUATIONENTITY
+DESCRIPTOR.message_types_by_name['ChatType'] = _CHATTYPE
+DESCRIPTOR.message_types_by_name['ContinuationEntity'] = _CONTINUATIONENTITY
+DESCRIPTOR.message_types_by_name['Continuation'] = _CONTINUATION
+_sym_db.RegisterFileDescriptor(DESCRIPTOR)
+
+ChatType = _reflection.GeneratedProtocolMessageType('ChatType', (_message.Message,), {
+  'DESCRIPTOR' : _CHATTYPE,
+  '__module__' : 'replay_pb2'
+  # @@protoc_insertion_point(class_scope:replay.ChatType)
+  })
+_sym_db.RegisterMessage(ChatType)
+
+ContinuationEntity = _reflection.GeneratedProtocolMessageType('ContinuationEntity', (_message.Message,), {
+  'DESCRIPTOR' : _CONTINUATIONENTITY,
+  '__module__' : 'replay_pb2'
+  # @@protoc_insertion_point(class_scope:replay.ContinuationEntity)
+  })
+_sym_db.RegisterMessage(ContinuationEntity)
+
+Continuation = _reflection.GeneratedProtocolMessageType('Continuation', (_message.Message,), {
+  'DESCRIPTOR' : _CONTINUATION,
+  '__module__' : 'replay_pb2'
+  # @@protoc_insertion_point(class_scope:replay.Continuation)
+  })
+_sym_db.RegisterMessage(Continuation)
+
+
+# @@protoc_insertion_point(module_scope)

pytchat/paramgen/proto/header.proto

@@ -0,0 +1,14 @@
+syntax = "proto3";
+
+message Video {
+    string id = 1;
+}
+
+message HeaderInfo {
+    Video video = 1;
+}
+
+message Header {
+    HeaderInfo info = 1;
+    int32 terminator = 4;
+}

pytchat/paramgen/proto/live.proto

@@ -0,0 +1,45 @@
+syntax = "proto3";
+
+package live;
+
+message Body {
+    int32 b1 = 1;
+    int32 b2 = 2;
+    int32 b3 = 3;
+    int32 b4 = 4;
+    string b7 = 7;
+    int32 b8 = 8;
+    string b9 = 9;
+    int64 timestamp2 = 10;
+    int32 b11 = 11;
+    int32 b15 = 15;
+}
+
+message ChatType {
+    int32 value = 1;
+}
+
+message STR19 {
+    int32 value = 1;
+}
+
+message ContinuationEntity {
+    string header = 3;
+    int64 timestamp1 = 5;
+    int32 s6 = 6;
+    int32 s7 = 7;
+    int32 s8 = 8;
+    Body body = 9;
+    int64 timestamp3 = 10;
+    int64 timestamp4 = 11;
+    int32 s13 = 13;
+    ChatType chattype = 16;
+    int32 s17 = 17;
+    STR19 str19 = 19;
+    int64 timestamp5 = 20;
+}
+
+message Continuation {
+    ContinuationEntity entity = 119693434;
+}
+

pytchat/paramgen/proto/replay.proto

@@ -0,0 +1,24 @@
+syntax = "proto3";
+
+package replay;
+
+message ChatType {
+    int32 value = 1;
+}
+
+message ContinuationEntity {
+    string header = 3;
+    int64 timestamp = 5;
+    int32 s6 = 6;
+    int32 s7 = 7;
+    int32 s8 = 8;
+    int32 s9 = 9;
+    string s10 = 10;
+    int32 s12 = 12;
+    ChatType chattype = 14;
+    int32 s15 = 15;
+}
+
+message Continuation {
+    ContinuationEntity entity = 156074452;
+}  

pytchat/parser/live.py

@@ -4,12 +4,8 @@ pytchat.parser.live
 Parser of live chat JSON.
 """
 
-import json
-from .. exceptions import ( 
-    ResponseContextError, 
-    NoContentsException, 
-    NoContinuationsException,
-    ChatParseException )
+from .. import exceptions
+
 
 class Parser:
 
@@ -20,10 +16,10 @@ class Parser:
 
     def get_contents(self, jsn):
         if jsn is None:
-            raise ChatParseException('Called with none JSON object.')
+            raise exceptions.IllegalFunctionCall('Called with none JSON object.')
         if jsn['response']['responseContext'].get('errors'):
-            raise ResponseContextError('The video_id would be wrong,'
-                'or video is deleted or private.')
+            raise exceptions.ResponseContextError(
+                'The video_id would be wrong, or video is deleted or private.')
         contents = jsn['response'].get('continuationContents')
         return contents
 
@@ -46,33 +42,37 @@ class Parser:
 
         if contents is None:
             '''Broadcasting end or cannot fetch chat stream'''
-            raise NoContentsException('Chat data stream is empty.')
+            raise exceptions.NoContents('Chat data stream is empty.')
 
         cont = contents['liveChatContinuation']['continuations'][0]
         if cont is None:
-            raise NoContinuationsException('No Continuation')
-        metadata = (cont.get('invalidationContinuationData')  or
-                    cont.get('timedContinuationData')         or
-                    cont.get('reloadContinuationData')        or
-                    cont.get('liveChatReplayContinuationData')
+            raise exceptions.NoContinuation('No Continuation')
+        metadata = (cont.get('invalidationContinuationData')
+                    or cont.get('timedContinuationData')
+                    or cont.get('reloadContinuationData')
+                    or cont.get('liveChatReplayContinuationData')
                     )
         if metadata is None:
             if cont.get("playerSeekContinuationData"):
-                raise ChatParseException('Finished chat data')
+                raise exceptions.ChatDataFinished('Finished chat data')
             unknown = list(cont.keys())[0]
             if unknown:
-                raise ChatParseException(f"Received unknown continuation type:{unknown}")
+                raise exceptions.ReceivedUnknownContinuation(
+                    f"Received unknown continuation type:{unknown}")
             else:
-                raise ChatParseException('Cannot extract continuation data')
+                raise exceptions.FailedExtractContinuation('Cannot extract continuation data')
         return self._create_data(metadata, contents)
 
     def reload_continuation(self, contents):
         """
-        When `seektime = 0` or seektime is abbreviated , 
+        When `seektime == 0` or seektime is abbreviated ,
         check if fetched chat json has no chat data.
         If so, try to fetch playerSeekContinuationData.
         This function must be run only first fetching.
         """
+        if contents is None:
+            '''Broadcasting end or cannot fetch chat stream'''
+            raise exceptions.NoContents('Chat data stream is empty.')
         cont = contents['liveChatContinuation']['continuations'][0]
         if cont.get("liveChatReplayContinuationData"):
             # chat data exist.
@@ -81,7 +81,7 @@ class Parser:
         init_cont = cont.get("playerSeekContinuationData")
         if init_cont:
             return init_cont.get("continuation")
-        raise ChatParseException('Finished chat data')
+        raise exceptions.ChatDataFinished('Finished chat data')
 
     def _create_data(self, metadata, contents):
         actions = contents['liveChatContinuation'].get('actions')

pytchat/parser/replay.py

@@ -1,76 +0,0 @@
-import json
-from .. import config
-from .. exceptions import ( 
-    ResponseContextError, 
-    NoContentsException, 
-    NoContinuationsException )
-
-
-logger = config.logger(__name__)
-
-class Parser:
-    def parse(self, jsn):
-        """
-        このparse関数はReplayChat._listen() 関数から定期的に呼び出される。
-        引数jsnはYoutubeから取得したアーカイブ済みチャットデータの生JSONであり、
-        このparse関数によって与えられたJSONを以下に分割して返す。
-         + timeout (次のチャットデータ取得までのインターバル)
-         + chat data（チャットデータ本体）
-         + continuation （次のチャットデータ取得に必要となるパラメータ）.
-        
-        ライブ配信のチャットとアーカイブ済み動画のチャットは構造が若干異なっているが、
-        ライブチャットと同じデータ形式に変換することにより、
-        同じprocessorでライブとリプレイどちらでも利用できるようにしている。
-
-        Parameter
-        ----------
-        + jsn : dict
-            + Youtubeから取得したチャットデータのJSONオブジェクト。
-              （pythonの辞書形式に変換済みの状態で渡される）
-
-        Returns
-        -------
-        + metadata : dict
-            + チャットデータに付随するメタデータ。timeout、 動画ID、continuationパラメータで構成される。           
-        + chatdata : list[dict]
-            + チャットデータ本体のリスト。
-        """
-        if jsn is None: 
-            return {'timeoutMs':0,'continuation':None},[]
-        if jsn['response']['responseContext'].get('errors'):
-            raise ResponseContextError('動画に接続できません。'
-        '動画IDが間違っているか、動画が削除／非公開の可能性があります。')
-        contents=jsn['response'].get('continuationContents')
-        #配信が終了した場合、もしくはチャットデータが取得できない場合
-        if contents is None:
-            raise NoContentsException('チャットデータを取得できませんでした。')
-
-        cont = contents['liveChatContinuation']['continuations'][0]
-        if cont is None:
-            raise NoContinuationsException('Continuationがありません。')
-        metadata = cont.get('liveChatReplayContinuationData')
-        if metadata is None:
-            unknown = list(cont.keys())[0]
-            if unknown != "playerSeekContinuationData":
-                logger.debug(f"Received unknown continuation type:{unknown}")
-                metadata = cont.get(unknown)
-        actions = contents['liveChatContinuation'].get('actions')
-        if actions is None:
-            #後続のチャットデータなし
-            return {"continuation":None,"timeout":0,"chatdata":[]}
-        interval = self.get_interval(actions)
-        metadata.setdefault("timeoutMs",interval)
-        """アーカイブ済みチャットはライブチャットと構造が異なっているため、以下の行により
-        ライブチャットと同じ形式にそろえる"""
-        chatdata = [action["replayChatItemAction"]["actions"][0] for action in actions]
-        return metadata, chatdata
-
-    def get_interval(self, actions: list):
-        if actions is None:
-            return 0
-        start = int(actions[0]["replayChatItemAction"]["videoOffsetTimeMsec"])
-        last = int(actions[-1]["replayChatItemAction"]["videoOffsetTimeMsec"])
-        return (last - start)
-
-
-

pytchat/processors/chat_processor.py

@@ -3,6 +3,7 @@ class ChatProcessor:
     Abstract class that processes chat data.
     Receive chat data (actions) from Listener.
     '''
+
     def process(self, chat_components: list):
         '''
         Interface that represents processing of chat data.
@@ -21,7 +22,9 @@ class ChatProcessor:
         '''
         pass
 
-
-
-
-
+    def finalize(self, *args, **kwargs):
+        '''
+        Interface for finalizing the process.
+        Called when chat fetching finished.
+        '''
+        pass

pytchat/processors/combinator.py

@@ -1,5 +1,6 @@
 from .chat_processor import ChatProcessor
 
+
 class Combinator(ChatProcessor):
     '''
     Combinator combines multiple chat processors.
@@ -35,5 +36,3 @@ class Combinator(ChatProcessor):
         '''
         return tuple(processor.process(chat_components)
                      for processor in self.processors)
-            
-

pytchat/processors/compatible/processor.py

@@ -1,13 +1,13 @@
-import datetime
-import time
 from .renderer.textmessage import LiveChatTextMessageRenderer
 from .renderer.paidmessage import LiveChatPaidMessageRenderer
 from .renderer.paidsticker import LiveChatPaidStickerRenderer
 from .renderer.legacypaid import LiveChatLegacyPaidMessageRenderer
+from .renderer.membership import LiveChatMembershipItemRenderer
 from .. chat_processor import ChatProcessor
 from ... import config
 logger = config.logger(__name__)
 
+
 class CompatibleProcessor(ChatProcessor):
 
     def process(self, chat_components: list):
@@ -24,11 +24,15 @@ class CompatibleProcessor(ChatProcessor):
                 timeout += chat_component.get('timeout', 0)
                 chatdata = chat_component.get('chatdata')
 
-                if chatdata is None: break
+                if chatdata is None:
+                    break
                 for action in chatdata:
-                    if action is None: continue
-                    if action.get('addChatItemAction') is None: continue
-                    if action['addChatItemAction'].get('item') is None: continue
+                    if action is None:
+                        continue
+                    if action.get('addChatItemAction') is None:
+                        continue
+                    if action['addChatItemAction'].get('item') is None:
+                        continue
 
                     chat = self.parse(action)
                     if chat:
@@ -47,11 +51,12 @@ class CompatibleProcessor(ChatProcessor):
         action = sitem.get("addChatItemAction")
         if action:
             item = action.get("item")
-        if item is None: return None
+        if item is None:
+            return None
         rd = {}
         try:
             renderer = self.get_renderer(item)
-            if renderer == None:
+            if renderer is None:
                 return None
 
             rd["kind"] = "youtube#liveChatMessage"
@@ -75,7 +80,8 @@ class CompatibleProcessor(ChatProcessor):
             renderer = LiveChatPaidStickerRenderer(item)
         elif item.get("liveChatLegacyPaidMessageRenderer"):
             renderer = LiveChatLegacyPaidMessageRenderer(item)
+        elif item.get("liveChatMembershipItemRenderer"):
+            renderer = LiveChatMembershipItemRenderer(item)
         else:
             renderer = None
         return renderer
-

pytchat/processors/compatible/renderer/base.py

@@ -1,11 +1,12 @@
-import datetime, pytz
+import datetime
+import pytz
+
 
 class BaseRenderer:
     def __init__(self, item, chattype):
         self.renderer = list(item.values())[0]
         self.chattype = chattype
 
-
     def get_snippet(self):
 
         message = self.get_message(self.renderer)
@@ -22,7 +23,6 @@ class BaseRenderer:
             }
         }
 
-
     def get_authordetails(self):
         authorExternalChannelId = self.renderer.get("authorExternalChannelId")
         # parse subscriber type
@@ -40,7 +40,6 @@ class BaseRenderer:
             "isChatModerator": isChatModerator
         }
 
-
     def get_message(self, renderer):
         message = ''
         if renderer.get("message"):
@@ -80,4 +79,3 @@ class BaseRenderer:
         dt = datetime.datetime.fromtimestamp(int(timestamp) / 1000000)
         return dt.astimezone(pytz.utc).isoformat(
             timespec='milliseconds').replace('+00:00', 'Z')
-        

pytchat/processors/compatible/renderer/legacypaid.py

@@ -1,4 +1,6 @@
 from .base import BaseRenderer
+
+
 class LiveChatLegacyPaidMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "newSponsorEvent")
@@ -34,10 +36,7 @@ class LiveChatLegacyPaidMessageRenderer(BaseRenderer):
             "isChatModerator": isChatModerator
         }
 
-
     def get_message(self, renderer):
         message = (renderer["eventText"]["runs"][0]["text"]
                    ) + ' / ' + (renderer["detailText"]["simpleText"])
         return message
-
-

pytchat/processors/compatible/renderer/membership.py

@@ -0,0 +1,40 @@
+from .base import BaseRenderer
+
+
+class LiveChatMembershipItemRenderer(BaseRenderer):
+    def __init__(self, item):
+        super().__init__(item, "newSponsorEvent")
+
+    def get_snippet(self):
+        message = self.get_message(self.renderer)
+        return {
+            "type": self.chattype,
+            "liveChatId": "",
+            "authorChannelId": self.renderer.get("authorExternalChannelId"),
+            "publishedAt": self.get_publishedat(self.renderer.get("timestampUsec", 0)),
+            "hasDisplayContent": True,
+            "displayMessage": message,
+
+        }
+
+    def get_authordetails(self):
+        authorExternalChannelId = self.renderer.get("authorExternalChannelId")
+        # parse subscriber type
+        isVerified, isChatOwner, _, isChatModerator = (
+            self.get_badges(self.renderer)
+        )
+        return {
+            "channelId": authorExternalChannelId,
+            "channelUrl": "http://www.youtube.com/channel/" + authorExternalChannelId,
+            "displayName": self.renderer["authorName"]["simpleText"],
+            "profileImageUrl": self.renderer["authorPhoto"]["thumbnails"][1]["url"],
+            "isVerified": isVerified,
+            "isChatOwner": isChatOwner,
+            "isChatSponsor": True,
+            "isChatModerator": isChatModerator
+        }
+
+    def get_message(self, renderer):
+        message = ''.join([mes.get("text", "")
+                           for mes in renderer["headerSubtext"]["runs"]])
+        return message, [message]

pytchat/processors/compatible/renderer/paidmessage.py

@@ -3,6 +3,7 @@ from . import currency
 from .base import BaseRenderer
 superchat_regex = re.compile(r"^(\D*)(\d{1,3}(,\d{3})*(\.\d*)*\b)$")
 
+
 class LiveChatPaidMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "superChatEvent")

pytchat/processors/compatible/renderer/paidsticker.py

@@ -3,6 +3,7 @@ from . import currency
 from .base import BaseRenderer
 superchat_regex = re.compile(r"^(\D*)(\d{1,3}(,\d{3})*(\.\d*)*\b)$")
 
+
 class LiveChatPaidStickerRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "superStickerEvent")
@@ -44,5 +45,3 @@ class LiveChatPaidStickerRenderer(BaseRenderer):
             symbol = ""
             amountMicros = 0
         return amountDisplayString, symbol, amountMicros
- 
-

pytchat/processors/compatible/renderer/textmessage.py

@@ -1,4 +1,6 @@
 from .base import BaseRenderer
+
+
 class LiveChatTextMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "textMessageEvent")

pytchat/processors/default/processor.py

@@ -4,10 +4,13 @@ from .renderer.textmessage import LiveChatTextMessageRenderer
 from .renderer.paidmessage import LiveChatPaidMessageRenderer
 from .renderer.paidsticker import LiveChatPaidStickerRenderer
 from .renderer.legacypaid import LiveChatLegacyPaidMessageRenderer
+from .renderer.membership import LiveChatMembershipItemRenderer
 from .. chat_processor import ChatProcessor
 from ... import config
+
 logger = config.logger(__name__)
 
+
 class Chatdata:
     def __init__(self, chatlist: list, timeout: float):
         self.items = chatlist
@@ -25,6 +28,7 @@ class Chatdata:
             return
         await asyncio.sleep(self.interval / len(self.items))
 
+
 class DefaultProcessor(ChatProcessor):
     def process(self, chat_components: list):
 
@@ -35,27 +39,30 @@ class DefaultProcessor(ChatProcessor):
             for component in chat_components:
                 timeout += component.get('timeout', 0)
                 chatdata = component.get('chatdata')
-                if chatdata is None: continue
+                if chatdata is None:
+                    continue
                 for action in chatdata:
-                    if action is None: continue
-                    if action.get('addChatItemAction') is None: continue
-                    if action['addChatItemAction'].get('item') is None: continue
+                    if action is None:
+                        continue
+                    if action.get('addChatItemAction') is None:
+                        continue
+                    if action['addChatItemAction'].get('item') is None:
+                        continue
 
                     chat = self._parse(action)
                     if chat:
                         chatlist.append(chat)
         return Chatdata(chatlist, float(timeout))
 
-
     def _parse(self, sitem):
-
         action = sitem.get("addChatItemAction")
         if action:
             item = action.get("item")
-        if item is None: return None
+        if item is None:
+            return None
         try:
             renderer = self._get_renderer(item)
-            if renderer == None:
+            if renderer is None:
                 return None
 
             renderer.get_snippet()
@@ -74,6 +81,8 @@ class DefaultProcessor(ChatProcessor):
             renderer = LiveChatPaidStickerRenderer(item)
         elif item.get("liveChatLegacyPaidMessageRenderer"):
             renderer = LiveChatLegacyPaidMessageRenderer(item)
+        elif item.get("liveChatMembershipItemRenderer"):
+            renderer = LiveChatMembershipItemRenderer(item)
         else:
             renderer = None
         return renderer

pytchat/processors/default/renderer/base.py

@@ -1,6 +1,10 @@
 from datetime import datetime
+
+
 class Author:
     pass
+
+
 class BaseRenderer:
     def __init__(self, item, chattype):
         self.renderer = list(item.values())[0]
@@ -38,8 +42,6 @@ class BaseRenderer:
         self.author.name = self.renderer["authorName"]["simpleText"]
         self.author.imageUrl = self.renderer["authorPhoto"]["thumbnails"][1]["url"]
 
-
-
     def get_message(self, renderer):
         message = ''
         message_ex = []
@@ -50,14 +52,16 @@ class BaseRenderer:
                     if r:
                         if r.get('emoji'):
                             message += r['emoji'].get('shortcuts', [''])[0]
-                            message_ex.append(r['emoji']['image']['thumbnails'][1].get('url'))
+                            message_ex.append({
+                                'id': r['emoji'].get('emojiId').split('/')[-1],
+                                'txt': r['emoji'].get('shortcuts', [''])[0],
+                                'url': r['emoji']['image']['thumbnails'][0].get('url')
+                            })
                         else:
                             message += r.get('text', '')
                             message_ex.append(r.get('text', ''))
         return message, message_ex
 
-
-
     def get_badges(self, renderer):
         self.author.type = ''
         isVerified = False
@@ -82,12 +86,9 @@ class BaseRenderer:
                     self.get_badgeurl(badge)
         return isVerified, isChatOwner, isChatSponsor, isChatModerator
 
-
     def get_badgeurl(self, badge):
         self.author.badgeUrl = badge["liveChatAuthorBadgeRenderer"]["customThumbnail"]["thumbnails"][0]["url"]
 
-
-
     def get_datetime(self, timestamp):
         dt = datetime.fromtimestamp(timestamp / 1000000)
         return dt.strftime('%Y-%m-%d %H:%M:%S')

pytchat/processors/default/renderer/legacypaid.py

@@ -1,18 +1,15 @@
 from .base import BaseRenderer
+
+
 class LiveChatLegacyPaidMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "newSponsor")
 
-        
-
     def get_authordetails(self):
         super().get_authordetails()
         self.author.isChatSponsor = True
 
-
     def get_message(self, renderer):
         message = (renderer["eventText"]["runs"][0]["text"]
                    ) + ' / ' + (renderer["detailText"]["simpleText"])
         return message
-
-

pytchat/processors/default/renderer/membership.py

@@ -0,0 +1,15 @@
+from .base import BaseRenderer
+
+
+class LiveChatMembershipItemRenderer(BaseRenderer):
+    def __init__(self, item):
+        super().__init__(item, "newSponsor")
+
+    def get_authordetails(self):
+        super().get_authordetails()
+        self.author.isChatSponsor = True
+
+    def get_message(self, renderer):
+        message = ''.join([mes.get("text", "")
+                           for mes in renderer["headerSubtext"]["runs"]])
+        return message, [message]

pytchat/processors/default/renderer/paidmessage.py

@@ -3,11 +3,11 @@ from . import currency
 from .base import BaseRenderer
 superchat_regex = re.compile(r"^(\D*)(\d{1,3}(,\d{3})*(\.\d*)*\b)$")
 
+
 class LiveChatPaidMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "superChat")
 
-
     def get_snippet(self):
         super().get_snippet()
         amountDisplayString, symbol, amount = (
@@ -15,11 +15,10 @@ class LiveChatPaidMessageRenderer(BaseRenderer):
         )
         self.amountValue = amount
         self.amountString = amountDisplayString
-        self.currency= currency.symbols[symbol]["fxtext"] if currency.symbols.get(symbol) else symbol
+        self.currency = currency.symbols[symbol]["fxtext"] if currency.symbols.get(
+            symbol) else symbol
         self.bgColor = self.renderer.get("bodyBackgroundColor", 0)
 
-
-
     def get_amountdata(self, renderer):
         amountDisplayString = renderer["purchaseAmountText"]["simpleText"]
         m = superchat_regex.search(amountDisplayString)

pytchat/processors/default/renderer/paidsticker.py

@@ -3,11 +3,11 @@ from . import currency
 from .base import BaseRenderer
 superchat_regex = re.compile(r"^(\D*)(\d{1,3}(,\d{3})*(\.\d*)*\b)$")
 
+
 class LiveChatPaidStickerRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "superSticker")
 
-
     def get_snippet(self):
         super().get_snippet()
         amountDisplayString, symbol, amount = (
@@ -15,11 +15,11 @@ class LiveChatPaidStickerRenderer(BaseRenderer):
         )
         self.amountValue = amount
         self.amountString = amountDisplayString
-        self.currency = currency.symbols[symbol]["fxtext"] if currency.symbols.get(symbol) else symbol
+        self.currency = currency.symbols[symbol]["fxtext"] if currency.symbols.get(
+            symbol) else symbol
         self.bgColor = self.renderer.get("moneyChipBackgroundColor", 0)
-        self.sticker = "https:"+self.renderer["sticker"]["thumbnails"][0]["url"]
-            
-
+        self.sticker = "https:" + \
+            self.renderer["sticker"]["thumbnails"][0]["url"]
 
     def get_amountdata(self, renderer):
         amountDisplayString = renderer["purchaseAmountText"]["simpleText"]
@@ -31,9 +31,3 @@ class LiveChatPaidStickerRenderer(BaseRenderer):
             symbol = ""
             amount = 0.0
         return amountDisplayString, symbol, amount
-
-
- 
-
- 
-

pytchat/processors/default/renderer/textmessage.py

@@ -1,4 +1,6 @@
 from .base import BaseRenderer
+
+
 class LiveChatTextMessageRenderer(BaseRenderer):
     def __init__(self, item):
         super().__init__(item, "textMessage")

pytchat/processors/dummy_processor.py

@@ -1,8 +1,10 @@
 from .chat_processor import ChatProcessor
 
+
 class DummyProcessor(ChatProcessor):
     '''
     Dummy processor just returns received chat_components directly.
     '''
+
     def process(self, chat_components: list):
         return chat_components

pytchat/processors/html_archiver.py

@@ -1,31 +1,55 @@
-import csv
 import os
 import re
+import requests
+from base64 import standard_b64encode
 from .chat_processor import ChatProcessor
 from .default.processor import DefaultProcessor
 
+
 PATTERN = re.compile(r"(.*)\(([0-9]+)\)$")
-fmt_headers = ['datetime','elapsed','authorName','message','superchat'
-     ,'type','authorChannel']
+
+fmt_headers = ['datetime', 'elapsed', 'authorName',
+               'message', 'superchat', 'type', 'authorChannel']
 
 HEADER_HTML = '''
-<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
 <meta http-equiv="Content-Type" content="text/html;charset=UTF-8">
+<html>
+<head>
 '''
 
+TABLE_CSS = '''
+table.css {
+ border-collapse: collapse;
+}
+ 
+table.css thead{
+ border-collapse: collapse;
+ border: 1px solid #000
+}
+ 
+table.css tr td{
+ padding: 0.3em;
+ border: 1px solid #000
+}
+
+table.css th{
+ padding: 0.3em;
+ border: 1px solid #000
+}
+'''
+
+
 class HTMLArchiver(ChatProcessor):
     '''
-    HtmlArchiver saves chat data as HTML table format.
+    HTMLArchiver saves chat data as HTML table format.
     '''
-
     def __init__(self, save_path):
         super().__init__()
         self.save_path = self._checkpath(save_path)
-        with open(self.save_path, mode='a', encoding = 'utf-8') as f:
-            f.write(HEADER_HTML)
-            f.write('<table border="1" style="border-collapse: collapse">')
-            f.writelines(self._parse_html_header(fmt_headers))
         self.processor = DefaultProcessor()
+        self.emoji_table = {}  # table for custom emojis. key: emoji_id, value: base64 encoded image binary.
+        self.header = [HEADER_HTML]
+        self.body = ['<body>\n', '<table class="css">\n', self._parse_table_header(fmt_headers)]
 
     def _checkpath(self, filepath):
         splitter = os.path.splitext(os.path.basename(filepath))
@@ -56,43 +80,58 @@ class HTMLArchiver(ChatProcessor):
         """
         if chat_components is None or len(chat_components) == 0:
             return
-
-        with open(self.save_path, mode='a', encoding = 'utf-8') as f:
-            chats = self.processor.process(chat_components).items
-            for c in chats:
-                f.writelines(
-                    self._parse_html_line([
+        self.body.extend(
+            (self._parse_html_line((
                 c.datetime,
                 c.elapsedTime,
                 c.author.name,
-                        c.message,
+                self._parse_message(c.messageEx),
                 c.amountString,
                 c.author.type,
-                        c.author.channelId]
+                c.author.channelId)
+            ) for c in self.processor.process(chat_components).items)
         )
-                )
-            '''
-            #Palliative treatment#
-            Comment out below line to prevent the table
-            display from collapsing.
-            '''
-            #f.write('</table>')
 
     def _parse_html_line(self, raw_line):
-        html = ''
-        html+=' <tr>'
-        for cell in raw_line:
-            html+='<td>'+cell+'</td>'
-        html+='</tr>\n'
-        return html
+        return ''.join(('<tr>',
+                        ''.join(''.join(('<td>', cell, '</td>')) for cell in raw_line),
+                        '</tr>\n'))
 
-    def _parse_html_header(self,raw_line):
-        html = ''
-        html+='<thead>\n'
-        html+=' <tr>'
-        for cell in raw_line:
-            html+='<th>'+cell+'</th>'
-        html+='</tr>\n'
-        html+='</thead>\n'
-        return html
+    def _parse_table_header(self, raw_line):
+        return ''.join(('<thead><tr>',
+                        ''.join(''.join(('<th>', cell, '</th>')) for cell in raw_line),
+                        '</tr></thead>\n'))
         
+    def _parse_message(self, message_items: list) -> str:
+        return ''.join(''.join(('<span class="', self._set_emoji_table(item), '" title="', item['txt'], '"></span>'))
+                       if type(item) is dict else item
+                       for item in message_items)
+
+    def _encode_img(self, url):
+        resp = requests.get(url)
+        return standard_b64encode(resp.content).decode()
+
+    def _set_emoji_table(self, item: dict):
+        emoji_id = item['id']
+        if emoji_id not in self.emoji_table:
+            self.emoji_table.setdefault(emoji_id, self._encode_img(item['url']))
+        return emoji_id
+
+    def _stylecode(self, name, code, width, height):
+        return ''.join((".", name, " { display: inline-block; background-image: url(data:image/png;base64,",
+                        code, "); background-repeat: no-repeat; width: ",
+                        str(width), "; height: ", str(height), ";}"))
+    
+    def _create_styles(self):
+        return '\n'.join(('<style type="text/css">',
+                          TABLE_CSS,
+                          '\n'.join(self._stylecode(key, self.emoji_table[key], 24, 24)
+                                for key in self.emoji_table.keys()),
+                          '</style>\n'))
+    
+    def finalize(self):
+        self.header.extend([self._create_styles(), '</head>\n'])
+        self.body.extend(['</table>\n</body>'])
+        with open(self.save_path, mode='a', encoding='utf-8') as f:
+            f.writelines(self.header)
+            f.writelines(self.body)

pytchat/processors/jsonfile_archiver.py

@@ -1,4 +1,3 @@
-import datetime
 import json
 import os
 import re
@@ -6,6 +5,7 @@ from .chat_processor import ChatProcessor
 
 PATTERN = re.compile(r"(.*)\(([0-9]+)\)$")
 
+
 class JsonfileArchiver(ChatProcessor):
     """
     JsonfileArchiver saves chat data as text of JSON lines.
@@ -17,6 +17,7 @@ class JsonfileArchiver(ChatProcessor):
         it is automatically saved under a different name
         with suffix '(number)'
     """
+
     def __init__(self, save_path):
         super().__init__()
         self.save_path = self._checkpath(save_path)
@@ -32,14 +33,18 @@ class JsonfileArchiver(ChatProcessor):
             total_lines : int :
                 count of total lines written to the file.
         """
-        if chat_components is None: return
+        if chat_components is None:
+            return
         with open(self.save_path, mode='a', encoding='utf-8') as f:
             for component in chat_components:
-                if component is None: continue
+                if component is None:
+                    continue
                 chatdata = component.get('chatdata')
-                if chatdata is None: continue
+                if chatdata is None:
+                    continue
                 for action in chatdata:
-                    if action is None: continue
+                    if action is None:
+                        continue
                     json_line = json.dumps(action, ensure_ascii=False)
                     f.writelines(json_line + '\n')
                     self.line_counter += 1
@@ -62,5 +67,3 @@ class JsonfileArchiver(ChatProcessor):
                 body = f'{body}({str(counter)})'
             newpath = os.path.join(os.path.dirname(filepath), body + extention)
         return newpath
-
-

pytchat/processors/simple_display_processor.py

@@ -1,10 +1,6 @@
-import json
-import os
-import traceback
-import datetime
-import time
 from .chat_processor import ChatProcessor
-##version 2
+
+
 class SimpleDisplayProcessor(ChatProcessor):
 
     def process(self, chat_components: list):
@@ -17,28 +13,34 @@ class SimpleDisplayProcessor(ChatProcessor):
             timeout += component.get('timeout', 0)
             chatdata = component.get('chatdata')
 
-            if chatdata is None:break
+            if chatdata is None:
+                break
             for action in chatdata:
-                if action is None:continue
-                if action.get('addChatItemAction') is None:continue
-                if action['addChatItemAction'].get('item') is None:continue
+                if action is None:
+                    continue
+                if action.get('addChatItemAction') is None:
+                    continue
+                if action['addChatItemAction'].get('item') is None:
+                    continue
 
-                root = action['addChatItemAction']['item'].get('liveChatTextMessageRenderer')
+                root = action['addChatItemAction']['item'].get(
+                    'liveChatTextMessageRenderer')
 
                 if root:
                     author_name = root['authorName']['simpleText']
                     message = self._parse_message(root.get('message'))
                     purchase_amount_text = ''
                 else:
-                    root = ( action['addChatItemAction']['item'].get('liveChatPaidMessageRenderer') or 
-                            action['addChatItemAction']['item'].get('liveChatPaidStickerRenderer') )
+                    root = (action['addChatItemAction']['item'].get('liveChatPaidMessageRenderer')
+                            or action['addChatItemAction']['item'].get('liveChatPaidStickerRenderer'))
                     if root:
                         author_name = root['authorName']['simpleText']
                         message = self._parse_message(root.get('message'))
                         purchase_amount_text = root['purchaseAmountText']['simpleText']
                     else:
                         continue
-                chatlist.append(f'[{author_name}]:  {message}  {purchase_amount_text}')
+                chatlist.append(
+                    f'[{author_name}]:  {message}  {purchase_amount_text}')
         return {"timeout": timeout, "chatlist": chatlist}
 
     def _parse_message(self, message):
@@ -57,5 +59,3 @@ class SimpleDisplayProcessor(ChatProcessor):
             return tmp
         else:
             return ''
-
-

pytchat/processors/speed/calculator.py

@@ -5,6 +5,8 @@ Calculate speed of chat.
 """
 import time
 from .. chat_processor import ChatProcessor
+
+
 class RingQueue:
     """
     リング型キュー
@@ -77,6 +79,7 @@ class RingQueue:
     def item_count(self):
         return len(self.items)
 
+
 class SpeedCalculator(ChatProcessor, RingQueue):
     """
     チャットの勢いを計算する。
@@ -106,7 +109,6 @@ class SpeedCalculator(ChatProcessor, RingQueue):
             self.speed = self._calc_speed()
         return self.speed
 
-
     def _calc_speed(self):
         """
         RingQueue内のチャット勢い算出用データリストを元に、
@@ -120,8 +122,7 @@ class SpeedCalculator(ChatProcessor, RingQueue):
             # キュー内の総チャット数
             total = sum(item['chat_count'] for item in self.items)
             # キュー内の最初と最後のチャットの時間差
-            duration = (self.items[self.last_pos]['endtime'] 
-                        - self.items[self.first_pos]['starttime'])
+            duration = (self.items[self.last_pos]['endtime'] - self.items[self.first_pos]['starttime'])
             if duration != 0:
                 return int(total * 60 / duration)
             return 0
@@ -200,4 +201,3 @@ class SpeedCalculator(ChatProcessor, RingQueue):
             'starttime': int(starttime / 1000000),
             'endtime': int(endtime / 1000000)
         })
-

pytchat/processors/superchat/calculator.py

@@ -15,10 +15,12 @@ items_sticker = [
     'liveChatPaidStickerRenderer'
 ]
 
+
 class SuperchatCalculator(ChatProcessor):
     """
     Calculate the amount of SuperChat by currency.
     """
+
     def __init__(self):
         self.results = {}
 
@@ -34,11 +36,13 @@ class SuperchatCalculator(ChatProcessor):
             return self.results
         for component in chat_components:
             chatdata = component.get('chatdata')
-            if chatdata is None: continue
+            if chatdata is None:
+                continue
             for action in chatdata:
                 renderer = self._get_item(action, items_paid) or \
                     self._get_item(action, items_sticker)
-                if renderer is None: continue
+                if renderer is None:
+                    continue
                 symbol, amount = self._parse(renderer)
                 self.results.setdefault(symbol, 0)
                 self.results[symbol] += amount
@@ -69,6 +73,3 @@ class SuperchatCalculator(ChatProcessor):
                 continue
             return None
         return dict_body
-
-
-

pytchat/processors/tsv_archiver.py

@@ -5,8 +5,9 @@ from .chat_processor import ChatProcessor
 from .default.processor import DefaultProcessor
 
 PATTERN = re.compile(r"(.*)\(([0-9]+)\)$")
-fmt_headers = ['datetime','elapsed','authorName','message','superchatAmount'
-     ,'authorType','authorChannel']
+fmt_headers = ['datetime', 'elapsed', 'authorName', 'message',
+               'superchatAmount', 'authorType', 'authorChannel']
+
 
 class TSVArchiver(ChatProcessor):
     '''
@@ -64,7 +65,3 @@ class TSVArchiver(ChatProcessor):
                     c.author.type,
                     c.author.channelId
                 ])
-
-
-    
-    

pytchat/tool/extract/asyncdl.py

@@ -17,6 +17,7 @@ REPLAY_URL = "https://www.youtube.com/live_chat_replay/" \
              "get_live_chat_replay?continuation="
 MAX_RETRY_COUNT = 3
 
+
 def _split(start, end, count, min_interval_sec=120):
     """
     Split section from `start` to `end` into `count` pieces,
@@ -28,7 +29,6 @@ def _split(start, end, count, min_interval_sec = 120):
     --------
         List of the offset of each block's first chat data.
     """
-    
     if not (isinstance(start, int) or isinstance(start, float)) or \
        not (isinstance(end, int) or isinstance(end, float)):
         raise ValueError("start/end must be int or float")
@@ -40,7 +40,8 @@ def _split(start, end, count, min_interval_sec = 120):
         raise ValueError("count must be equal to or greater than 1.")
     if (end - start) / count < min_interval_sec:
         count = int((end - start) / min_interval_sec)
-        if count == 0 : count = 1
+        if count == 0:
+            count = 1
     interval = (end - start) / count
 
     if count == 1:
@@ -48,8 +49,10 @@ def _split(start, end, count, min_interval_sec = 120):
     return sorted(list(set([int(start + interval * j)
                             for j in range(count)])))
 
+
 def ready_blocks(video_id, duration, div, callback):
-    if div <= 0: raise ValueError
+    if div <= 0:
+        raise ValueError
 
     async def _get_blocks(video_id, duration, div, callback):
         async with aiohttp.ClientSession() as session:
@@ -92,6 +95,7 @@ def ready_blocks(video_id, duration, div, callback):
         _get_blocks(video_id, duration, div, callback))
     return blocks
 
+
 def fetch_patch(callback, blocks, video_id):
 
     async def _allocate_workers():
@@ -137,6 +141,7 @@ def fetch_patch(callback, blocks, video_id):
     except CancelledError:
         pass
 
+
 async def _shutdown():
     print("\nshutdown...")
     tasks = [t for t in asyncio.all_tasks()
@@ -148,7 +153,7 @@ async def _shutdown():
         except asyncio.CancelledError:
             pass
 
+
 def cancel():
     loop = asyncio.get_event_loop()
     loop.create_task(_shutdown())
-    

pytchat/tool/extract/block.py

@@ -1,4 +1,3 @@
-from . import parser
 class Block:
     """Block object represents something like a box
     to join chunk of chatdata.

pytchat/tool/extract/duplcheck.py

@@ -1,5 +1,6 @@
 from . import parser
 
+
 def check_duplicate(chatdata):
     max_range = len(chatdata) - 1
     tbl_offset = [None] * max_range
@@ -15,10 +16,8 @@ def check_duplicate(chatdata):
     def is_duplicate(i, j):
         return (
             tbl_offset[i] == tbl_offset[j]
-                and
-            tbl_id[i] == tbl_id[j]
-                and
-            tbl_type[i] == tbl_type[j]
+            and tbl_id[i] == tbl_id[j]
+            and tbl_type[i] == tbl_type[j]
         )
     print("creating table...")
     create_table(chatdata, max_range)
@@ -53,8 +52,7 @@ def check_duplicate_offset(chatdata):
     def is_duplicate(i, j):
         return (
             tbl_offset[i] == tbl_offset[j]
-                 and
-            tbl_id[i] == tbl_id[j]
+            and tbl_id[i] == tbl_id[j]
         )
 
     print("creating table...")
@@ -69,8 +67,10 @@ def check_duplicate_offset(chatdata):
         for i in range(max_range - 1)
         if is_duplicate(i, i + 1)]
 
+
 def remove_duplicate_head(blocks):
-    if len(blocks) == 1 : return blocks
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks
 
     def is_duplicate_head(index):
 
@@ -85,19 +85,19 @@ def remove_duplicate_head(blocks):
         type_1 = parser.get_type(blocks[index + 1].chat_data[0])
         return (
             blocks[index].first == blocks[index + 1].first
-                and
-            id_0 == id_1
-                and
-            type_0 == type_1
+            and id_0 == id_1
+            and type_0 == type_1
         )
     ret = [blocks[i] for i in range(len(blocks) - 1)
-        if (len(blocks[i].chat_data)>0 and 
-        not is_duplicate_head(i) )]
+           if (len(blocks[i].chat_data) > 0
+               and not is_duplicate_head(i))]
     ret.append(blocks[-1])
     return ret
 
+
 def remove_duplicate_tail(blocks):
-    if len(blocks) == 1 : return blocks    
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks
 
     def is_duplicate_tail(index):
         if len(blocks[index].chat_data) == 0:
@@ -110,23 +110,23 @@ def remove_duplicate_tail(blocks):
         type_1 = parser.get_type(blocks[index].chat_data[-1])
         return (
             blocks[index - 1].last == blocks[index].last
-                and
-            id_0 == id_1
-                and
-            type_0 == type_1
+            and id_0 == id_1
+            and type_0 == type_1
         )
 
     ret = [blocks[i] for i in range(0, len(blocks))
            if i == 0 or not is_duplicate_tail(i)]
     return ret
 
+
 def remove_overlap(blocks):
     """
     Fix overlapped blocks after ready_blocks().
     Align the last offset of each block to the first offset
     of next block (equals `end` offset of each block).
     """
-    if len(blocks) == 1 : return blocks
+    if len(blocks) == 0 or len(blocks) == 1:
+        return blocks
 
     for block in blocks:
         if block.is_last:
@@ -146,8 +146,8 @@ def remove_overlap(blocks):
     return blocks
 
 
-
 def _dump(blocks):
-    print(f"----------        first         last         end---")
+    print("----------        first         last         end---")
     for i, block in enumerate(blocks):
-        print(f"block[{i:3}]   {block.first:>10}   {block.last:>10}  {block.end:>10}")
+        print(
+            f"block[{i:3}]   {block.first:>10}   {block.last:>10}  {block.end:>10}")

pytchat/tool/extract/extractor.py

@@ -1,6 +1,5 @@
 from . import asyncdl
 from . import duplcheck
-from . import parser
 from .. videoinfo import VideoInfo
 from ... import config
 from ... exceptions import InvalidVideoIdException
@@ -8,6 +7,7 @@ from ... exceptions import InvalidVideoIdException
 logger = config.logger(__name__)
 headers = config.headers
 
+
 class Extractor:
     def __init__(self, video_id, div=1, callback=None, processor=None):
         if not isinstance(div, int) or div < 1:
@@ -40,6 +40,7 @@ class Extractor:
         return self
 
     def _set_block_end(self):
+        if len(self.blocks) > 0:
             for i in range(len(self.blocks) - 1):
                 self.blocks[i].end = self.blocks[i + 1].first
             self.blocks[-1].end = self.duration * 1000
@@ -82,10 +83,13 @@ class Extractor:
         data = self._execute_extract_operations()
         if self.processor is None:
             return data
-        return self.processor.process(
-            [{'video_id':None,'timeout':1,'chatdata' : (action
-            ["replayChatItemAction"]["actions"][0] for action in data)}]
+        ret = self.processor.process(
+            [{'video_id': None,
+              'timeout': 1,
+              'chatdata': (action["replayChatItemAction"]["actions"][0] for action in data)}]
         )
+        self.processor.finalize()
+        return ret
 
     def cancel(self):
         asyncdl.cancel()

pytchat/tool/extract/parser.py

@@ -1,12 +1,9 @@
-import json
 from ... import config
-from ... exceptions import ( 
-    ResponseContextError, 
-    NoContentsException, 
-    NoContinuationsException )
+from ... import exceptions
 
 logger = config.logger(__name__)
 
+
 def parse(jsn):
     """
     Parse replay chat data.
@@ -23,15 +20,15 @@ def parse(jsn):
     if jsn is None:
         raise ValueError("parameter JSON is None")
     if jsn['response']['responseContext'].get('errors'):
-        raise ResponseContextError(
+        raise exceptions.ResponseContextError(
             'video_id is invalid or private/deleted.')
     contents = jsn['response'].get('continuationContents')
     if contents is None:
-        raise NoContentsException('No chat data.')
+        raise exceptions.NoContents('No chat data.')
 
     cont = contents['liveChatContinuation']['continuations'][0]
     if cont is None:
-        raise NoContinuationsException('No Continuation')
+        raise exceptions.NoContinuation('No Continuation')
     metadata = cont.get('liveChatReplayContinuationData')
     if metadata:
         continuation = metadata.get("continuation")
@@ -43,12 +40,12 @@ def parse(jsn):
 def get_offset(item):
     return int(item['replayChatItemAction']["videoOffsetTimeMsec"])
 
+
 def get_id(item):
     return list((list(item['replayChatItemAction']["actions"][0].values()
                       )[0])['item'].values())[0].get('id')
 
+
 def get_type(item):
     return list((list(item['replayChatItemAction']["actions"][0].values()
                       )[0])['item'].keys())[0]
-
-

pytchat/tool/extract/patch.py

@@ -2,6 +2,7 @@ from . import parser
 from . block import Block
 from typing import NamedTuple
 
+
 class Patch(NamedTuple):
     """
     Patch represents chunk of chat data
@@ -12,6 +13,7 @@ class Patch(NamedTuple):
     first: int = None
     last: int = None
 
+
 def fill(block: Block, patch: Patch):
     block_end = block.end
     if patch.last < block_end or block.is_last:
@@ -51,4 +53,3 @@ def set_patch(block:Block, patch:Patch):
     block.chat_data.extend(patch.chats)
     block.last = patch.last
     block.remaining = block.end - block.last
-

pytchat/tool/extract/worker.py

@@ -1,8 +1,8 @@
-from . import parser
 from . block import Block
-from . patch import Patch, fill, split
+from . patch import fill, split
 from ... paramgen import arcparam
 
+
 class ExtractWorker:
     """
     ExtractWorker associates a download session with a block.
@@ -54,6 +54,7 @@ class ExtractWorker:
                 self.block.done = True
                 self.block = _search_new_block(self)
 
+
 def _search_new_block(worker) -> Block:
     index, undone_block = _get_undone_block(worker.blocks)
     if undone_block is None:
@@ -74,6 +75,7 @@ def _search_new_block(worker) -> Block:
     worker.blocks.insert(index + 1, new_block)
     return new_block
 
+
 def _get_undone_block(blocks) -> (int, Block):
     min_interval_ms = 120000
     max_remaining = 0

pytchat/tool/mining/parser.py

@@ -1,12 +1,12 @@
-import json
+import re
 from ... import config
 from ... exceptions import (
     ResponseContextError,
-    NoContentsException, 
-    NoContinuationsException )
+    NoContents, NoContinuation)
 
 logger = config.logger(__name__)
 
+
 def parse(jsn):
     """
     Parse replay chat data.
@@ -27,11 +27,11 @@ def parse(jsn):
             'video_id is invalid or private/deleted.')
     contents = jsn["response"].get('continuationContents')
     if contents is None:
-        raise NoContentsException('No chat data.')
+        raise NoContents('No chat data.')
 
     cont = contents['liveChatContinuation']['continuations'][0]
     if cont is None:
-        raise NoContinuationsException('No Continuation')
+        raise NoContinuation('No Continuation')
     metadata = cont.get('liveChatReplayContinuationData')
     if metadata:
         continuation = metadata.get("continuation")
@@ -50,15 +50,21 @@ def parse(jsn):
 def get_offset(item):
     return int(item['replayChatItemAction']["videoOffsetTimeMsec"])
 
+
 def get_id(item):
     return list((list(item['replayChatItemAction']["actions"][0].values()
                       )[0])['item'].values())[0].get('id')
 
+
 def get_type(item):
     return list((list(item['replayChatItemAction']["actions"][0].values()
                       )[0])['item'].keys())[0]
-import re                
-_REGEX_YTINIT = re.compile("window\\[\"ytInitialData\"\\]\\s*=\\s*({.+?});\\s+")
+
+
+_REGEX_YTINIT = re.compile(
+    "window\\[\"ytInitialData\"\\]\\s*=\\s*({.+?});\\s+")
+
+
 def extract(text):
 
     match = re.findall(_REGEX_YTINIT, str(text))

pytchat/tool/videoinfo.py

@@ -2,7 +2,6 @@ import json
 import re
 import requests
 from .. import config
-from .. import util
 from ..exceptions import InvalidVideoIdException
 
 headers = config.headers
@@ -63,6 +62,7 @@ item_moving_thumbnail = [
     "url"
 ]
 
+
 class VideoInfo:
     '''
     VideoInfo object retrieves YouTube video information.
@@ -76,6 +76,7 @@ class VideoInfo:
     InvalidVideoIdException :
         Occurs when video_id does not exist on YouTube.
     '''
+
     def __init__(self, video_id):
         self.video_id = video_id
         text = self._get_page_text(video_id)

pytchat/util/__init__.py

@@ -1,6 +1,9 @@
-import requests,json,datetime
+import requests
+import json
+import datetime
 from .. import config
 
+
 def extract(url):
     _session = requests.Session()
     html = _session.get(url, headers=config.headers)
@@ -10,6 +13,6 @@ def extract(url):
 
 
 def save(data, filename, extention):
-    with open(filename+"_"+(datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S')
-                            )+extention,mode ='w',encoding='utf-8') as f:
+    with open(filename + "_" + (datetime.datetime.now().strftime('%Y-%m-%d %H-%M-%S')) + extention,
+              mode='w', encoding='utf-8') as f:
         f.writelines(data)

requirements.txt

@@ -1,4 +1,5 @@
 aiohttp
+protobuf
 pytz
 requests
 urllib3

tests/test_arcparam.py

@@ -1,19 +1,22 @@
-import pytest
-from pytchat.parser.live import Parser
+import json
+import requests
 import pytchat.config as config
-import requests, json
 from pytchat.paramgen import arcparam
+from pytchat.parser.live import Parser
+
 
 def test_arcparam_0(mocker):
     param = arcparam.getparam("01234567890", -1)
-    assert param == "op2w0wRyGjxDZzhhRFFvTE1ERXlNelExTmpjNE9UQWFFLXFvM2JrQkRRb0xNREV5TXpRMU5qYzRPVEFnQVElM0QlM0QoADAAOABAAEgEUhwIABAAGAAgACoOc3RhdGljY2hlY2tzdW1AAFgDYAFoAHIECAEQAHgA" 
+    assert param == "op2w0wQmGhxDZzhLRFFvTE1ERXlNelExTmpjNE9UQWdBUT09SARgAXICCAE%3D" 
+
 
 def test_arcparam_1(mocker):
     param = arcparam.getparam("01234567890", seektime=100000)
-    assert param == "op2w0wR3GjxDZzhhRFFvTE1ERXlNelExTmpjNE9UQWFFLXFvM2JrQkRRb0xNREV5TXpRMU5qYzRPVEFnQVElM0QlM0QogNDbw_QCMAA4AEAASANSHAgAEAAYACAAKg5zdGF0aWNjaGVja3N1bUAAWANgAWgAcgQIARAAeAA%3D"
+    assert param == "op2w0wQtGhxDZzhLRFFvTE1ERXlNelExTmpjNE9UQWdBUT09KIDQ28P0AkgDYAFyAggB"
+
 
 def test_arcparam_2(mocker):
-    param = arcparam.getparam("SsjCnHOk-Sk")
+    param = arcparam.getparam("SsjCnHOk-Sk", seektime=100)
     url = f"https://www.youtube.com/live_chat_replay/get_live_chat_replay?continuation={param}&pbj=1"
     resp = requests.Session().get(url, headers=config.headers)
     jsn = json.loads(resp.text)
@@ -25,4 +28,4 @@ def test_arcparam_2(mocker):
 
 def test_arcparam_3(mocker):
     param = arcparam.getparam("01234567890")
-    assert param == "op2w0wRyGjxDZzhhRFFvTE1ERXlNelExTmpjNE9UQWFFLXFvM2JrQkRRb0xNREV5TXpRMU5qYzRPVEFnQVElM0QlM0QoATAAOABAAEgDUhwIABAAGAAgACoOc3RhdGljY2hlY2tzdW1AAFgDYAFoAHIECAEQAHgA"
+    assert param == "op2w0wQmGhxDZzhLRFFvTE1ERXlNelExTmpjNE9UQWdBUT09SARgAXICCAE%3D"

tests/test_arcparam_mining.py

@@ -1,9 +1,10 @@
-import pytest
 from pytchat.tool.mining import parser
 import pytchat.config as config
-import requests, json
+import requests
+import json
 from pytchat.paramgen import arcparam_mining as arcparam
 
+
 def test_arcparam_e(mocker):
     try:
         arcparam.getparam("01234567890", -1)
@@ -12,8 +13,6 @@ def test_arcparam_e(mocker):
         assert True
 
 
-
-
 def test_arcparam_0(mocker):
     param = arcparam.getparam("01234567890", 0)
 
@@ -25,6 +24,7 @@ def test_arcparam_1(mocker):
     print(param)
     assert param == "op2w0wQzGiBDZzhhRFFvTE1ERXlNelExTmpjNE9UQWdBUSUzRCUzREABWgUQgMLXL2AEcgIIAXgB"
 
+
 def test_arcparam_2(mocker):
     param = arcparam.getparam("PZz9NB0-Z64", 1)
     url = f"https://www.youtube.com/live_chat_replay?continuation={param}&playerOffsetMs=1000&pbj=1"
@@ -35,6 +35,7 @@ def test_arcparam_2(mocker):
     print(test_id)
     assert test_id == "ChwKGkNKSGE0YnFJeWVBQ0ZWcUF3Z0VkdGIwRm9R"
 
+
 def test_arcparam_3(mocker):
     param = arcparam.getparam("01234567890")
     assert param == "op2w0wQsGiBDZzhhRFFvTE1ERXlNelExTmpjNE9UQWdBUSUzRCUzREABYARyAggBeAE%3D"

tests/test_compatible_processor.py

@@ -1,19 +1,10 @@
 import json
-import pytest
-import asyncio,aiohttp
 from pytchat.parser.live import Parser
 from pytchat.processors.compatible.processor import CompatibleProcessor
-from pytchat.exceptions import (
-    NoLivechatRendererException,NoYtinitialdataException,
-    ResponseContextError, NoContentsException)
-
-from pytchat.processors.compatible.renderer.textmessage import LiveChatTextMessageRenderer
-from pytchat.processors.compatible.renderer.paidmessage import LiveChatPaidMessageRenderer
-from pytchat.processors.compatible.renderer.paidsticker import LiveChatPaidStickerRenderer
-from pytchat.processors.compatible.renderer.legacypaid import LiveChatLegacyPaidMessageRenderer
 
 parser = Parser(is_replay=False)
 
+
 def test_textmessage(mocker):
     '''api互換processorのテスト：通常テキストメッセージ'''
     processor = CompatibleProcessor()
@@ -40,10 +31,12 @@ def test_textmessage(mocker):
         "kind", "etag", "id", "snippet", "authorDetails"
     }
     assert ret["items"][0]["snippet"].keys() == {
-        'type', 'liveChatId', 'authorChannelId', 'publishedAt', 'hasDisplayContent', 'displayMessage', 'textMessageDetails'
+        'type', 'liveChatId', 'authorChannelId', 'publishedAt', 'hasDisplayContent', 'displayMessage',
+        'textMessageDetails'
     }
     assert ret["items"][0]["authorDetails"].keys() == {
-        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor', 'isChatModerator'
+        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor',
+        'isChatModerator'
     }
     assert ret["items"][0]["snippet"]["textMessageDetails"].keys() == {
         'messageText'
@@ -51,6 +44,7 @@ def test_textmessage(mocker):
     assert "LCC." in ret["items"][0]["id"]
     assert ret["items"][0]["snippet"]["type"] == "textMessageEvent"
 
+
 def test_newsponcer(mocker):
     '''api互換processorのテスト：メンバ新規登録'''
     processor = CompatibleProcessor()
@@ -81,7 +75,45 @@ def test_newsponcer(mocker):
 
     }
     assert ret["items"][0]["authorDetails"].keys() == {
-        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor', 'isChatModerator'
+        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor',
+        'isChatModerator'
+    }
+    assert "LCC." in ret["items"][0]["id"]
+    assert ret["items"][0]["snippet"]["type"] == "newSponsorEvent"
+
+
+def test_newsponcer_rev(mocker):
+    '''api互換processorのテスト：メンバ新規登録'''
+    processor = CompatibleProcessor()
+
+    _json = _open_file("tests/testdata/compatible/newSponsor_rev.json")
+
+    _, chatdata = parser.parse(parser.get_contents(json.loads(_json)))
+    data = {
+        "video_id": "",
+        "timeout": 7,
+        "chatdata": chatdata
+    }
+    ret = processor.process([data])
+
+    assert ret["kind"] == "youtube#liveChatMessageListResponse"
+    assert ret["pollingIntervalMillis"] == data["timeout"] * 1000
+    assert ret.keys() == {
+        "kind", "etag", "pageInfo", "nextPageToken", "pollingIntervalMillis", "items"
+    }
+    assert ret["pageInfo"].keys() == {
+        "totalResults", "resultsPerPage"
+    }
+    assert ret["items"][0].keys() == {
+        "kind", "etag", "id", "snippet", "authorDetails"
+    }
+    assert ret["items"][0]["snippet"].keys() == {
+        'type', 'liveChatId', 'authorChannelId', 'publishedAt', 'hasDisplayContent', 'displayMessage'
+
+    }
+    assert ret["items"][0]["authorDetails"].keys() == {
+        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor',
+        'isChatModerator'
     }
     assert "LCC." in ret["items"][0]["id"]
     assert ret["items"][0]["snippet"]["type"] == "newSponsorEvent"
@@ -113,10 +145,12 @@ def test_superchat(mocker):
         "kind", "etag", "id", "snippet", "authorDetails"
     }
     assert ret["items"][0]["snippet"].keys() == {
-        'type', 'liveChatId', 'authorChannelId', 'publishedAt', 'hasDisplayContent', 'displayMessage', 'superChatDetails'
+        'type', 'liveChatId', 'authorChannelId', 'publishedAt', 'hasDisplayContent', 'displayMessage',
+        'superChatDetails'
     }
     assert ret["items"][0]["authorDetails"].keys() == {
-        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor', 'isChatModerator'
+        'channelId', 'channelUrl', 'displayName', 'profileImageUrl', 'isVerified', 'isChatOwner', 'isChatSponsor',
+        'isChatModerator'
     }
     assert ret["items"][0]["snippet"]["superChatDetails"].keys() == {
         'amountMicros', 'currency', 'amountDisplayString', 'tier', 'backgroundColor'
@@ -124,6 +158,7 @@ def test_superchat(mocker):
     assert "LCC." in ret["items"][0]["id"]
     assert ret["items"][0]["snippet"]["type"] == "superChatEvent"
 
+
 def test_unregistered_currency(mocker):
     processor = CompatibleProcessor()
 

tests/test_livechat.py

@@ -1,30 +1,21 @@
-import pytest
-from pytchat.parser.live import Parser
-
 import json
-import asyncio,aiohttp
-
 from aioresponses import aioresponses
 from pytchat.core_async.livechat import LiveChatAsync
-from pytchat.exceptions import (
-    NoLivechatRendererException,NoYtinitialdataException,
-    ResponseContextError,NoContentsException)
+from pytchat.exceptions import ResponseContextError
 
 
-from pytchat.core_multithread.livechat import LiveChat
-import unittest
-from unittest import TestCase
-
 def _open_file(path):
     with open(path, mode='r', encoding='utf-8') as f:
         return f.read()
 
+
 @aioresponses()
 def test_Async(*mock):
     vid = ''
     _text = _open_file('tests/testdata/paramgen_firstread.json')
     _text = json.loads(_text)
-    mock[0].get(f"https://www.youtube.com/live_chat?v={vid}&is_popout=1", status=200, body=_text)
+    mock[0].get(
+        f"https://www.youtube.com/live_chat?v={vid}&is_popout=1", status=200, body=_text)
     try:
         chat = LiveChatAsync(video_id='')
         assert chat.is_alive()
@@ -33,6 +24,7 @@ def test_Async(*mock):
     except ResponseContextError:
         assert not chat.is_alive()
 
+
 def test_MultiThread(mocker):
     _text = _open_file('tests/testdata/paramgen_firstread.json')
     _text = json.loads(_text)
@@ -48,6 +40,3 @@ def test_MultiThread(mocker):
     except ResponseContextError:
         chat.terminate()
         assert not chat.is_alive()
-
-
-

tests/test_livechat_2.py

@@ -1,43 +1,43 @@
-import asyncio, aiohttp
-import json
-import pytest
+import asyncio
 import re
-import requests
-import sys
-import time
 from aioresponses import aioresponses
 from concurrent.futures import CancelledError
-from unittest import TestCase
 from pytchat.core_multithread.livechat import LiveChat
 from pytchat.core_async.livechat import LiveChatAsync
-from pytchat.exceptions import (
-      NoLivechatRendererException,NoYtinitialdataException,
-      ResponseContextError,NoContentsException)
-from pytchat.parser.live import Parser
 from pytchat.processors.dummy_processor import DummyProcessor
 
+
 def _open_file(path):
     with open(path, mode='r', encoding='utf-8') as f:
         return f.read()
 
+
 @aioresponses()
 def test_async_live_stream(*mock):
 
     async def test_loop(*mock):
-        pattern = re.compile(r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
+        pattern = re.compile(
+            r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
         _text = _open_file('tests/testdata/test_stream.json')
         mock[0].get(pattern, status=200, body=_text)
         chat = LiveChatAsync(video_id='', processor=DummyProcessor())
         chats = await chat.get()
         rawdata = chats[0]["chatdata"]
         # assert fetching livachat data
-        assert list(rawdata[0]["addChatItemAction"]["item"].keys())[0] == "liveChatTextMessageRenderer"
-        assert list(rawdata[1]["addChatItemAction"]["item"].keys())[0] == "liveChatTextMessageRenderer"
-        assert list(rawdata[2]["addChatItemAction"]["item"].keys())[0] == "liveChatPlaceholderItemRenderer"
-        assert list(rawdata[3]["addLiveChatTickerItemAction"]["item"].keys())[0] == "liveChatTickerPaidMessageItemRenderer"
-        assert list(rawdata[4]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidMessageRenderer"
-        assert list(rawdata[5]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidStickerRenderer"
-        assert list(rawdata[6]["addLiveChatTickerItemAction"]["item"].keys())[0] == "liveChatTickerSponsorItemRenderer"                                
+        assert list(rawdata[0]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatTextMessageRenderer"
+        assert list(rawdata[1]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatTextMessageRenderer"
+        assert list(rawdata[2]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatPlaceholderItemRenderer"
+        assert list(rawdata[3]["addLiveChatTickerItemAction"]["item"].keys())[
+            0] == "liveChatTickerPaidMessageItemRenderer"
+        assert list(rawdata[4]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatPaidMessageRenderer"
+        assert list(rawdata[5]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatPaidStickerRenderer"
+        assert list(rawdata[6]["addLiveChatTickerItemAction"]["item"].keys())[
+            0] == "liveChatTickerSponsorItemRenderer"
 
     loop = asyncio.get_event_loop()
     try:
@@ -45,12 +45,15 @@ def test_async_live_stream(*mock):
     except CancelledError:
         assert True
 
+
 @aioresponses()
 def test_async_replay_stream(*mock):
 
     async def test_loop(*mock):
-        pattern_live = re.compile(r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
-        pattern_replay =  re.compile(r'^https://www.youtube.com/live_chat_replay/get_live_chat_replay\?continuation=.*$')
+        pattern_live = re.compile(
+            r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
+        pattern_replay = re.compile(
+            r'^https://www.youtube.com/live_chat_replay/get_live_chat_replay\?continuation=.*$')
         # empty livechat -> switch to fetch replaychat
         _text_live = _open_file('tests/testdata/finished_live.json')
         _text_replay = _open_file('tests/testdata/chatreplay.json')
@@ -61,8 +64,10 @@ def test_async_replay_stream(*mock):
         chats = await chat.get()
         rawdata = chats[0]["chatdata"]
         # assert fetching replaychat data
-        assert list(rawdata[0]["addChatItemAction"]["item"].keys())[0] == "liveChatTextMessageRenderer"
-        assert list(rawdata[14]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidMessageRenderer"
+        assert list(rawdata[0]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatTextMessageRenderer"
+        assert list(rawdata[14]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatPaidMessageRenderer"
 
     loop = asyncio.get_event_loop()
     try:
@@ -70,12 +75,15 @@ def test_async_replay_stream(*mock):
     except CancelledError:
         assert True
 
+
 @aioresponses()
 def test_async_force_replay(*mock):
 
     async def test_loop(*mock):
-        pattern_live = re.compile(r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
-        pattern_replay =  re.compile(r'^https://www.youtube.com/live_chat_replay/get_live_chat_replay\?continuation=.*$')
+        pattern_live = re.compile(
+            r'^https://www.youtube.com/live_chat/get_live_chat\?continuation=.*$')
+        pattern_replay = re.compile(
+            r'^https://www.youtube.com/live_chat_replay/get_live_chat_replay\?continuation=.*$')
         # valid live data, but force_replay = True
         _text_live = _open_file('tests/testdata/test_stream.json')
         # valid replay data
@@ -84,13 +92,16 @@ def test_async_force_replay(*mock):
         mock[0].get(pattern_live, status=200, body=_text_live)
         mock[0].get(pattern_replay, status=200, body=_text_replay)
         # force replay
-        chat = LiveChatAsync(video_id='', processor = DummyProcessor(), force_replay = True)
+        chat = LiveChatAsync(
+            video_id='', processor=DummyProcessor(), force_replay=True)
         chats = await chat.get()
         rawdata = chats[0]["chatdata"]
         # assert fetching replaychat data
-        assert list(rawdata[14]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidMessageRenderer"
+        assert list(rawdata[14]["addChatItemAction"]["item"].keys())[
+            0] == "liveChatPaidMessageRenderer"
         # assert not mix livechat data
-        assert list(rawdata[2]["addChatItemAction"]["item"].keys())[0] != "liveChatPlaceholderItemRenderer"
+        assert list(rawdata[2]["addChatItemAction"]["item"].keys())[
+            0] != "liveChatPlaceholderItemRenderer"
 
     loop = asyncio.get_event_loop()
     try:
@@ -98,28 +109,32 @@ def test_async_force_replay(*mock):
     except CancelledError:
         assert True
 
+
 def test_multithread_live_stream(mocker):
 
     _text = _open_file('tests/testdata/test_stream.json')
     responseMock = mocker.Mock()
     responseMock.status_code = 200
     responseMock.text = _text
-    mocker.patch('requests.Session.get').return_value.__enter__.return_value = responseMock
+    mocker.patch(
+        'requests.Session.get').return_value.__enter__.return_value = responseMock
 
     chat = LiveChat(video_id='test_id', processor=DummyProcessor())
     chats = chat.get()
     rawdata = chats[0]["chatdata"]
     # assert fetching livachat data
-    assert list(rawdata[0]["addChatItemAction"]["item"].keys())[0] == "liveChatTextMessageRenderer"
-    assert list(rawdata[1]["addChatItemAction"]["item"].keys())[0] == "liveChatTextMessageRenderer"
-    assert list(rawdata[2]["addChatItemAction"]["item"].keys())[0] == "liveChatPlaceholderItemRenderer"
-    assert list(rawdata[3]["addLiveChatTickerItemAction"]["item"].keys())[0] == "liveChatTickerPaidMessageItemRenderer"
-    assert list(rawdata[4]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidMessageRenderer"
-    assert list(rawdata[5]["addChatItemAction"]["item"].keys())[0] == "liveChatPaidStickerRenderer"
-    assert list(rawdata[6]["addLiveChatTickerItemAction"]["item"].keys())[0] == "liveChatTickerSponsorItemRenderer"
+    assert list(rawdata[0]["addChatItemAction"]["item"].keys())[
+        0] == "liveChatTextMessageRenderer"
+    assert list(rawdata[1]["addChatItemAction"]["item"].keys())[
+        0] == "liveChatTextMessageRenderer"
+    assert list(rawdata[2]["addChatItemAction"]["item"].keys())[
+        0] == "liveChatPlaceholderItemRenderer"
+    assert list(rawdata[3]["addLiveChatTickerItemAction"]["item"].keys())[
+        0] == "liveChatTickerPaidMessageItemRenderer"
+    assert list(rawdata[4]["addChatItemAction"]["item"].keys())[
+        0] == "liveChatPaidMessageRenderer"
+    assert list(rawdata[5]["addChatItemAction"]["item"].keys())[
+        0] == "liveChatPaidStickerRenderer"
+    assert list(rawdata[6]["addLiveChatTickerItemAction"]["item"].keys())[
+        0] == "liveChatTickerSponsorItemRenderer"
     chat.terminate()
-
-    
-
-
-

tests/test_liveparam.py

@@ -5,5 +5,5 @@ def test_liveparam_0(mocker):
     _ts1= 1546268400
     param = liveparam._build("01234567890",
     *([_ts1*1000000 for i in range(5)]), topchat_only=False)
-    test_param="0ofMyAPiARp8Q2c4S0RRb0xNREV5TXpRMU5qYzRPVEFhUTZxNXdiMEJQUW83YUhSMGNITTZMeTkzZDNjdWVXOTFkSFZpWlM1amIyMHZiR2wyWlY5amFHRjBQM1k5TURFeU16UTFOamM0T1RBbWFYTmZjRzl3YjNWMFBURWdBZyUzRCUzRCiAuNbVqsrfAjAAOABAAkorCAEQABgAIAAqDnN0YXRpY2NoZWNrc3VtOgBAAEoCCAFQgLjW1arK3wJYA1CAuNbVqsrfAliAuNbVqsrfAmgBggEECAEQAIgBAKABgLjW1arK3wI%3D"
+    test_param="0ofMyANcGhxDZzhLRFFvTE1ERXlNelExTmpjNE9UQWdBUT09KIC41tWqyt8CQAFKC1CAuNbVqsrfAlgDUIC41tWqyt8CWIC41tWqyt8CaAGCAQIIAZoBAKABgLjW1arK3wI%3D"
     assert test_param == param

tests/test_parser.py

@@ -1,18 +1,17 @@
-import pytest
 from pytchat.parser.live import Parser
 import json
-import asyncio,aiohttp
 from aioresponses import aioresponses
-from pytchat.exceptions import (
-    NoLivechatRendererException,NoYtinitialdataException,
-    ResponseContextError, NoContentsException)
+from pytchat.exceptions import NoContents
 
 
 def _open_file(path):
     with open(path, mode='r', encoding='utf-8') as f:
         return f.read()
+
+
 parser = Parser(is_replay=False)
 
+
 @aioresponses()
 def test_finishedlive(*mock):
     '''配信が終了した動画を正しく処理できるか'''
@@ -23,9 +22,10 @@ def test_finishedlive(*mock):
     try:
         parser.parse(parser.get_contents(_text))
         assert False
-    except NoContentsException:
+    except NoContents:
         assert True
 
+
 @aioresponses()
 def test_parsejson(*mock):
     '''jsonを正常にパースできるか'''
@@ -37,8 +37,9 @@ def test_parsejson(*mock):
         parser.parse(parser.get_contents(_text))
         jsn = _text
         timeout = jsn["response"]["continuationContents"]["liveChatContinuation"]["continuations"][0]["timedContinuationData"]["timeoutMs"]
-        continuation = jsn["response"]["continuationContents"]["liveChatContinuation"]["continuations"][0]["timedContinuationData"]["continuation"]
-        assert 5035 == timeout
-        assert "0ofMyAPiARp8Q2c4S0RRb0xhelJMZDBsWFQwdERkalFhUTZxNXdiMEJQUW83YUhSMGNITTZMeTkzZDNjdWVXOTFkSFZpWlM1amIyMHZiR2wyWlY5amFHRjBQM1k5YXpSTGQwbFhUMHREZGpRbWFYTmZjRzl3YjNWMFBURWdBZyUzRCUzRCiPz5-Os-PkAjAAOABAAUorCAAQABgAIAAqDnN0YXRpY2NoZWNrc3VtOgBAAEoCCAFQgJqXjrPj5AJYA1CRwciOs-PkAli3pNq1k-PkAmgBggEECAEQAIgBAKABjbfnjrPj5AI%3D" == continuation
-    except:
+        continuation = jsn["response"]["continuationContents"]["liveChatContinuation"][
+            "continuations"][0]["timedContinuationData"]["continuation"]
+        assert timeout == 5035
+        assert continuation == "0ofMyAPiARp8Q2c4S0RRb0xhelJMZDBsWFQwdERkalFhUTZxNXdiMEJQUW83YUhSMGNITTZMeTkzZDNjdWVXOTFkSFZpWlM1amIyMHZiR2wyWlY5amFHRjBQM1k5YXpSTGQwbFhUMHREZGpRbWFYTmZjRzl3YjNWMFBURWdBZyUzRCUzRCiPz5-Os-PkAjAAOABAAUorCAAQABgAIAAqDnN0YXRpY2NoZWNrc3VtOgBAAEoCCAFQgJqXjrPj5AJYA1CRwciOs-PkAli3pNq1k-PkAmgBggEECAEQAIgBAKABjbfnjrPj5AI%3D"
+    except Exception:
         assert False

tests/test_speed_calculator.py

@@ -1,16 +1,10 @@
 import json
-import pytest
-import asyncio,aiohttp
 from pytchat.parser.live import Parser
-from pytchat.processors.compatible.processor import CompatibleProcessor
-from pytchat.exceptions import (
-    NoLivechatRendererException,NoYtinitialdataException,
-    ResponseContextError, NoContentsException)
-
 from pytchat.processors.speed.calculator import SpeedCalculator
 
 parser = Parser(is_replay=False)
 
+
 def test_speed_1(mocker):
     '''test speed calculation with normal json.
     test json has 15 chatdata, duration is 30 seconds,
@@ -30,6 +24,7 @@ def test_speed_1(mocker):
     ret = processor.process([data])
     assert 30 == ret
 
+
 def test_speed_2(mocker):
     '''test speed calculation with no valid chat data.
     '''
@@ -44,7 +39,8 @@ def test_speed_2(mocker):
         "chatdata": chatdata
     }
     ret = processor.process([data])
-    assert 0 == ret
+    assert ret == 0
+
 
 def test_speed_3(mocker):
     '''test speed calculation with empty data.
@@ -60,7 +56,7 @@ def test_speed_3(mocker):
         "chatdata": chatdata
     }
     ret = processor.process([data])
-    assert 0 == ret
+    assert ret == 0
 
 
 def _open_file(path):