From a7a2300f03e87493e2ea4ad4943b98a51d86b09e Mon Sep 17 00:00:00 2001 From: BoYanZh Date: Thu, 29 May 2025 00:59:27 -0400 Subject: [PATCH] feat: support health check score in repo.toml --- joj3_config_generator/models/repo.py | 1 + joj3_config_generator/transformers/repo.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/joj3_config_generator/models/repo.py b/joj3_config_generator/models/repo.py index 41b7d94..03af230 100644 --- a/joj3_config_generator/models/repo.py +++ b/joj3_config_generator/models/repo.py @@ -27,3 +27,4 @@ class Config(BaseModel): root: Path = Path(".") path: Path = Path("repo.toml") grading_repo_name: str = f"{socket.gethostname().split('-')[0]}-joj" + health_check_score: int = Field(1) diff --git a/joj3_config_generator/transformers/repo.py b/joj3_config_generator/transformers/repo.py index b05f0b9..193a294 100644 --- a/joj3_config_generator/transformers/repo.py +++ b/joj3_config_generator/transformers/repo.py @@ -88,7 +88,10 @@ def get_health_check_stage(repo_conf: repo.Config) -> result.StageDetail: ), ), parsers=[ - result.Parser(name="healthcheck", with_=result.ScoreConfig(score=1)), + result.Parser( + name="healthcheck", + with_=result.ScoreConfig(score=repo_conf.health_check_score), + ), result.Parser(name="debug", with_=result.ScoreConfig(score=0)), ], )