Implement postprocessing

2020-02-09 21:58:09 +09:00
parent ee77807dbd
commit abb7565e3a
1 changed files with 7 additions and 1 deletions
--- a/pytchat/tool/downloader.py
+++ b/pytchat/tool/downloader.py
@@ -77,4 +77,10 @@ def download(video_id, div = 1, callback = None, processor = None):
    if duration == 0:
        print("video is live.")
        return []
-    return Downloader(video_id, duration, div, callback).download()
+    data = Downloader(video_id, duration, div, callback).download()
+    if processor is None:
+        return data
+    return processor.process(
+        [{'video_id':None,'timeout':1,'chatdata' : [action
+        ["replayChatItemAction"]["actions"][0] for action in data]}]
+    )