diff --git a/joj3_config_generator/processers/task.py b/joj3_config_generator/processers/task.py index c398acb..78d73b1 100644 --- a/joj3_config_generator/processers/task.py +++ b/joj3_config_generator/processers/task.py @@ -6,7 +6,7 @@ from typing import Any, Callable, Dict, List, Tuple from joj3_config_generator.models import result, task from joj3_config_generator.models.common import Memory, Time from joj3_config_generator.models.const import JOJ3_CONFIG_ROOT -from joj3_config_generator.models.task import Parser as parser_enum +from joj3_config_generator.models.task import Parser as ParserEnum def get_conf_stage( @@ -33,7 +33,7 @@ def get_conf_stage( if parser in processed_dict: fn, parser_model = processed_dict[parser] fn(parser_model, conf_stage.parsers[idx]) - elif parser == parser_enum.DIFF: + elif parser == ParserEnum.DIFF: fix_diff( task_stage, conf_stage.parsers[idx], @@ -47,18 +47,18 @@ def get_conf_stage( def get_processed_dict( task_stage: task.Stage, -) -> Dict[parser_enum, Tuple[Callable[[Any, result.Parser], None], Any]]: +) -> Dict[ParserEnum, Tuple[Callable[[Any, result.Parser], None], Any]]: processed_dict: Dict[ - parser_enum, Tuple[Callable[[Any, result.Parser], None], Any] + ParserEnum, Tuple[Callable[[Any, result.Parser], None], Any] ] = { - parser_enum.CLANG_TIDY: (fix_keyword, task_stage.clangtidy), - parser_enum.KEYWORD: (fix_keyword, task_stage.keyword), - parser_enum.CPPCHECK: (fix_keyword, task_stage.cppcheck), - parser_enum.CPPLINT: (fix_keyword, task_stage.cpplint), - parser_enum.RESULT_DETAIL: (fix_result_detail, task_stage.result_detail), - parser_enum.DUMMY: (fix_dummy, task_stage.dummy), - parser_enum.RESULT_STATUS: (fix_dummy, task_stage.result_status), - parser_enum.FILE: (fix_file, task_stage.file), + ParserEnum.CLANG_TIDY: (fix_keyword, task_stage.clangtidy), + ParserEnum.KEYWORD: (fix_keyword, task_stage.keyword), + ParserEnum.CPPCHECK: (fix_keyword, task_stage.cppcheck), + ParserEnum.CPPLINT: (fix_keyword, task_stage.cpplint), + ParserEnum.RESULT_DETAIL: (fix_result_detail, task_stage.result_detail), + ParserEnum.DUMMY: (fix_dummy, task_stage.dummy), + ParserEnum.RESULT_STATUS: (fix_dummy, task_stage.result_status), + ParserEnum.FILE: (fix_file, task_stage.file), } return processed_dict