Format code

pytchat/cli/__init__.py

@@ -1,33 +1,31 @@
 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 .. processors.html_archiver import HTMLArchiver
 from .. tool.extract.extractor import Extractor
 from .. tool.videoinfo import VideoInfo
 from .. import __version__
 
 '''
-Most of CLI modules refer to 
+Most of CLI modules refer to
 Petter Kraabøl's Twitch-Chat-Downloader
 https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 
 '''
+
+
 def main():
-    # Arguments 
+    # Arguments
     parser = argparse.ArgumentParser(description=f'pytchat v{__version__}')
     parser.add_argument('-v', f'--{Arguments.Name.VIDEO}', type=str,
-         help='Video IDs separated by commas without space.\n'
-         'If ID starts with a hyphen (-), enclose the ID in square brackets.')
+                        help='Video IDs separated by commas without space.\n'
+                        'If ID starts with a hyphen (-), enclose the ID in square brackets.')
     parser.add_argument('-o', f'--{Arguments.Name.OUTPUT}', type=str,
-         help='Output directory (end with "/"). default="./"', default='./')
+                        help='Output directory (end with "/"). default="./"', default='./')
     parser.add_argument(f'--{Arguments.Name.VERSION}', action='store_true',
-         help='Settings version')
+                        help='Settings version')
     Arguments(parser.parse_args().__dict__)
     if Arguments().print_version:
         print(f'pytchat v{__version__}')
@@ -37,24 +35,26 @@ def main():
     if Arguments().video_ids:
         for video_id in Arguments().video_ids:
             if '[' in video_id:
-                video_id = video_id.replace('[','').replace(']','')
+                video_id = video_id.replace('[', '').replace(']', '')
             try:
                 info = VideoInfo(video_id)
                 print(f"Extracting...\n"
                       f" video_id: {video_id}\n"
                       f" channel:  {info.get_channel_name()}\n"
                       f" title:    {info.get_title()}")
-                path = Path(Arguments().output+video_id+'.html')
+                path = Path(Arguments().output + video_id + '.html')
                 print(f"output path: {path.resolve()}")
-                Extractor(video_id, 
-                  processor = HTMLArchiver(Arguments().output+video_id+'.html'),
-                  callback = _disp_progress
-                ).extract()
+                Extractor(video_id,
+                          processor=HTMLArchiver(
+                              Arguments().output + video_id + '.html'),
+                          callback=_disp_progress
+                          ).extract()
                 print("\nExtraction end.\n")
             except (InvalidVideoIdException, NoContentsException) as e:
                 print(e)
         return
     parser.print_help()
 
-def _disp_progress(a,b):
-    print('.',end="",flush=True)
+
+def _disp_progress(a, b):
+    print('.', end="", flush=True)

pytchat/cli/arguments.py

@@ -2,12 +2,13 @@ from typing import Optional, Dict, Union, List
 from .singleton import Singleton
 
 '''
-This modules refer to 
+This modules refer to
 Petter Kraabøl's Twitch-Chat-Downloader
 https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 '''
 
+
 class Arguments(metaclass=Singleton):
     """
     Arguments singleton
@@ -18,11 +19,11 @@ class Arguments(metaclass=Singleton):
         OUTPUT: str = 'output'
         VIDEO: str = 'video'
 
-    def __init__(self, 
-        arguments: Optional[Dict[str, Union[str, bool, int]]] = None):
+    def __init__(self,
+                 arguments: Optional[Dict[str, Union[str, bool, int]]] = None):
         """
         Initialize arguments
-        :param arguments: Arguments from cli 
+        :param arguments: Arguments from cli
         (Optional to call singleton instance without parameters)
         """
 
@@ -35,5 +36,5 @@ class Arguments(metaclass=Singleton):
         self.video_ids: List[int] = []
         # Videos
         if arguments[Arguments.Name.VIDEO]:
-            self.video_ids = [video_id 
-                for video_id in arguments[Arguments.Name.VIDEO].split(',')]
+            self.video_ids = [video_id
+                              for video_id in arguments[Arguments.Name.VIDEO].split(',')]

pytchat/cli/singleton.py

@@ -1,9 +1,11 @@
 '''
-This modules refer to 
+This modules refer to
 Petter Kraabøl's Twitch-Chat-Downloader
 https://github.com/PetterKraabol/Twitch-Chat-Downloader
 (MIT License)
 '''
+
+
 class Singleton(type):
     """
     Abstract class for singletons
@@ -16,4 +18,4 @@ class Singleton(type):
         return cls._instances[cls]
 
     def get_instance(cls, *args, **kwargs):
-        cls.__call__(*args, **kwargs)
+        cls.__call__(*args, **kwargs)