76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
import json
|
|
import os
|
|
from pathlib import Path
|
|
from typing import Any, List
|
|
|
|
import rtoml
|
|
|
|
from joj3_config_generator.models import joj1, repo, result, task
|
|
from joj3_config_generator.processers.joj1 import get_joj1_run_stage
|
|
from joj3_config_generator.processers.repo import ( # get_teapotcheck_config,
|
|
get_healthcheck_config,
|
|
get_teapot_stage,
|
|
)
|
|
from joj3_config_generator.processers.task import (
|
|
fix_diff,
|
|
fix_dummy,
|
|
fix_file,
|
|
fix_keyword,
|
|
fix_result_detail,
|
|
get_conf_stage,
|
|
get_executor_with_config,
|
|
)
|
|
|
|
|
|
def convert(repo_conf: repo.Config, task_conf: task.Config) -> result.Config:
|
|
# Create the base ResultConf object
|
|
result_conf = result.Config(
|
|
name=task_conf.task.name,
|
|
# exact folder difference specified by type
|
|
log_path=f"/home/tt/.cache/joj3/{task_conf.task.type_}.log",
|
|
expire_unix_timestamp=(
|
|
int(task_conf.release.deadline.timestamp())
|
|
if task_conf.release.deadline
|
|
else -1
|
|
),
|
|
actor_csv_path="/home/tt/.config/joj/students.csv", # students.csv position
|
|
max_total_score=repo_conf.max_total_score,
|
|
stage=result.Stage(
|
|
stages=[],
|
|
sandbox_token=repo_conf.sandbox_token,
|
|
poststages=[get_teapot_stage(repo_conf)],
|
|
),
|
|
)
|
|
|
|
# Construct healthcheck stage
|
|
if (
|
|
not repo_conf.force_skip_heatlh_check_on_test
|
|
or os.environ.get("PYTEST_CURRENT_TEST") is None
|
|
):
|
|
healthcheck_stage = get_healthcheck_config(repo_conf)
|
|
result_conf.stage.stages.append(healthcheck_stage)
|
|
stages: List[str] = []
|
|
# Convert each stage in the task configuration
|
|
for task_stage in task_conf.stages:
|
|
executor_with_config, stages = get_executor_with_config(task_stage, stages)
|
|
conf_stage = get_conf_stage(task_stage, executor_with_config)
|
|
conf_stage = fix_result_detail(task_stage, conf_stage)
|
|
conf_stage = fix_dummy(task_stage, conf_stage)
|
|
conf_stage = fix_keyword(task_stage, conf_stage)
|
|
conf_stage = fix_file(task_stage, conf_stage)
|
|
conf_stage = fix_diff(task_stage, conf_stage, task_conf)
|
|
result_conf.stage.stages.append(conf_stage)
|
|
|
|
return result_conf
|
|
|
|
|
|
def convert_joj1(joj1_conf: joj1.Config) -> task.Config:
|
|
stages = [get_joj1_run_stage(joj1_conf)]
|
|
return task.Config(
|
|
task=task.Task(
|
|
name=(""),
|
|
),
|
|
release=task.Release(deadline=None, begin_time=None),
|
|
stages=[],
|
|
)
|