From 03ba285a16f60ea39295709424edf657da7b4969 Mon Sep 17 00:00:00 2001 From: taizan-hokuto <55448286+taizan-hokuto@users.noreply.github.com> Date: Sun, 6 Sep 2020 18:53:35 +0900 Subject: [PATCH] Fix callback handling --- pytchat/cli/__init__.py | 2 -- pytchat/processors/html_archiver.py | 5 +++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/pytchat/cli/__init__.py b/pytchat/cli/__init__.py index 211da9f..c11d1b5 100644 --- a/pytchat/cli/__init__.py +++ b/pytchat/cli/__init__.py @@ -88,8 +88,6 @@ def main(): except InvalidVideoIdException: print("Invalid Video ID or URL:", video_id) - except TypeError as e: - print(e.with_traceback()) except NoContents as e: print(e) except FileNotFoundError: diff --git a/pytchat/processors/html_archiver.py b/pytchat/processors/html_archiver.py index 3a1b913..39cff09 100644 --- a/pytchat/processors/html_archiver.py +++ b/pytchat/processors/html_archiver.py @@ -43,7 +43,7 @@ class HTMLArchiver(ChatProcessor): ''' HTMLArchiver saves chat data as HTML table format. ''' - def __init__(self, save_path, callback): + def __init__(self, save_path, callback=None): super().__init__() self.save_path = self._checkpath(save_path) self.processor = DefaultProcessor() @@ -93,7 +93,8 @@ class HTMLArchiver(ChatProcessor): c.author.channelId) ) ) - self.callback(None, 1) + if self.callback: + self.callback(None, 1) def _parse_html_line(self, raw_line): return ''.join(('',