44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from joj3_config_generator.models import joj1, task
|
|
from joj3_config_generator.models.common import Memory, Time
|
|
from joj3_config_generator.models.const import DEFAULT_CPU_LIMIT, DEFAULT_MEMORY_LIMIT
|
|
|
|
|
|
def get_joj1_run_stage(joj1_config: joj1.Config) -> task.Stage:
|
|
cases_conf = []
|
|
for i, case in enumerate(joj1_config.cases):
|
|
cases_conf.append(
|
|
task.Stage(
|
|
score=case.score,
|
|
command=case.execute_args if case.execute_args else "",
|
|
limit=task.Limit(
|
|
cpu=Time(case.time) if case.time else DEFAULT_CPU_LIMIT,
|
|
mem=(Memory(case.memory) if case.memory else DEFAULT_MEMORY_LIMIT),
|
|
),
|
|
)
|
|
)
|
|
for i, case in enumerate(joj1_config.cases):
|
|
cases_conf[i].in_ = case.input
|
|
cases_conf[i].out_ = case.output
|
|
run_config = task.Stage(
|
|
name="This is the converted joj1 run stage",
|
|
parsers=["diff", "result-status"],
|
|
score=100,
|
|
limit=task.Limit(
|
|
cpu=(
|
|
Time(joj1_config.cases[0].time)
|
|
if joj1_config.cases[0].time is not None
|
|
else DEFAULT_CPU_LIMIT
|
|
),
|
|
mem=(
|
|
Memory(joj1_config.cases[0].memory)
|
|
if joj1_config.cases[0].memory is not None
|
|
else DEFAULT_MEMORY_LIMIT
|
|
),
|
|
),
|
|
cases={f"case{i}": cases_conf[i] for i, _ in enumerate(cases_conf)},
|
|
) # TODO: no strong pattern match here, use dict instead
|
|
return run_config
|
|
|
|
|
|
# TODO: get formatted joj1 config, match the criterion in the doc
|