Add cli option

2020-09-03 19:31:34 +09:00
parent 4eb18279fe
commit 3bcad12cf6
2 changed files with 8 additions and 3 deletions
--- a/pytchat/cli/arguments.py
+++ b/pytchat/cli/arguments.py
@@ -18,7 +18,7 @@ class Arguments(metaclass=Singleton):
        VERSION: str = 'version'
        OUTPUT: str = 'output_dir'
        VIDEO_IDS: str = 'video_id'
-
+        SAVE_ERROR_DATA: bool = 'save_error_data'
    def __init__(self,
                 arguments: Optional[Dict[str, Union[str, bool, int]]] = None):
        """
@@ -34,6 +34,7 @@ class Arguments(metaclass=Singleton):
        self.print_version: bool = arguments[Arguments.Name.VERSION]
        self.output: str = arguments[Arguments.Name.OUTPUT]
        self.video_ids: List[int] = []
+        self.save_error_data: bool = arguments[Arguments.Name.SAVE_ERROR_DATA]
        # Videos
        if arguments[Arguments.Name.VIDEO_IDS]:
            self.video_ids = [video_id