JOJ/JOJ3-config-generator
27
1
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects Releases Wiki Activity

style: some rename & more robust code
All checks were successful
build / build (push) Successful in 2m31s
Details
build / build (pull_request) Successful in 2m29s
Details

Browse Source
This commit is contained in:
李衍志523370910113 2025-03-05 08:49:34 +08:00
parent 6cd0cd5509
commit dfe836109c
1 changed files with 10 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
joj3_config_generator/processers/task.py
View File

@ -33,7 +33,9 @@ def get_conf_stage(
fn, parser_model = processed_dict[parser] fn, parser_model = processed_dict[parser]
fn(parser_model, conf_stage.parsers[idx]) fn(parser_model, conf_stage.parsers[idx])
elif parser == "diff": elif parser == "diff":
fix_diff(task_stage, task_conf, conf_stage.parsers[idx], conf_stage) fix_diff(
task_stage, task_conf, conf_stage.parsers[idx], conf_stage.executor
)
else: else:
continue continue
return conf_stage return conf_stage
@ -88,7 +90,7 @@ def get_executor_with(
def fix_keyword( def fix_keyword(
keyword_config: task.ParserKeyword, keyword_parser_: result.Parser keyword_config: task.ParserKeyword, keyword_parser: result.Parser
) -> None: ) -> None:
keyword_weight: List[result.KeywordConfig] = [] keyword_weight: List[result.KeywordConfig] = []
unique_weight = list(set(keyword_config.weight)) unique_weight = list(set(keyword_config.weight))
@ -102,7 +104,7 @@ def fix_keyword(
else: else:
continue continue
keyword_parser_.with_.update( keyword_parser.with_.update(
result.KeywordMatchConfig( result.KeywordMatchConfig(
matches=keyword_weight, matches=keyword_weight,
).model_dump(by_alias=True) ).model_dump(by_alias=True)
@ -134,8 +136,6 @@ def fix_dummy(
dummy_parser_config: task.ParserDummy, dummy_parser: result.Parser dummy_parser_config: task.ParserDummy, dummy_parser: result.Parser
) -> None: ) -> None:
# we don't use dummy parser in real application # we don't use dummy parser in real application
if dummy_parser_config is None:
return
dummy_parser.with_.update( dummy_parser.with_.update(
result.DummyConfig( result.DummyConfig(
score=dummy_parser_config.score, score=dummy_parser_config.score,
@ -156,9 +156,8 @@ def fix_diff(
task_stage: task.Stage, task_stage: task.Stage,
task_conf: task.Config, task_conf: task.Config,
diff_parser_config: result.Parser, diff_parser_config: result.Parser,
conf_stage: result.StageDetail, diff_executor: result.Executor,
) -> None: ) -> None:
diff_parser = diff_parser_config
skip = task_stage.skip skip = task_stage.skip
cases = task_stage.cases cases = task_stage.cases
finalized_cases = [case for case in cases if case not in skip] finalized_cases = [case for case in cases if case not in skip]
@ -215,10 +214,9 @@ def fix_diff(
) )
) )
if diff_parser: diff_parser_config.with_.update(
diff_parser.with_.update(
result.DiffConfig(name="diff", cases=parser_cases).model_dump(by_alias=True) result.DiffConfig(name="diff", cases=parser_cases).model_dump(by_alias=True)
) )
conf_stage.executor.with_.cases = stage_cases diff_executor.with_.cases = stage_cases
return return