Fix callback handling

This commit is contained in:
taizan-hokuto
2020-09-06 18:53:35 +09:00
parent 4e829a25d4
commit 03ba285a16
2 changed files with 3 additions and 4 deletions

View File

@@ -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:

View File

@@ -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,6 +93,7 @@ class HTMLArchiver(ChatProcessor):
c.author.channelId)
)
)
if self.callback:
self.callback(None, 1)
def _parse_html_line(self, raw_line):