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

feat: initial yaml2json (by nuvole)

Browse Source
This commit is contained in:
王韵晨520370910012 2024-09-28 10:05:28 +08:00
parent 46d8dec76d
commit 74777e9bec
No known key found for this signature in database
GPG Key ID: F19C8A1AD51F8FC9
1 changed files with 175 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

175
yaml2json.py Normal file
View File

@ -0,0 +1,175 @@
import yaml
import json
import os
def matlab_json_init():
output_json = {
"sandboxExecServer": "172.17.0.1:5051",
"outputPath": "/tmp/joj3_result.json",
"stages": []
}
healthcheck_json = {
"name": "healthcheck",
"executor": {
"name": "sandbox",
"with": {
"default": {
"args": [
"./healthcheck",
"-root=.",
"-meta=readme",
"-whitelist=stderr",
"-whitelist=stdout",
"-whitelist=.*\\.toml",
"-whitelist=.*\\.md",
"-whitelist=healthcheck",
"-whitelist=.*\\.json",
"-whitelist=.git.*"
],
"env": [
"PATH=/usr/bin:/bin"
],
"cpuLimit": 10000000000,
"memoryLimit": 104857600,
"procLimit": 50,
"copyInDir": ".",
"copyIn": {
"healthcheck": {
"src": "./../../../../../../build/healthcheck",
"copyOut": [
"stdout",
"stderr"
]
}
},
"stdin": {
"content": ""
},
"stdout": {
"name": "stdout",
"max": 4096
},
"stderr": {
"name": "stderr",
"max": 4096
}
}
}
},
"parser": {
"name": "healthcheck",
"with": {
"score": 10,
"comment": " + comment from json conf"
}
}
}
run_json = {
"name": "run",
"executor": {
"name": "sandbox",
"with": {
"default": {
"args": [
""
],
"env": [
"PATH=/usr/bin:/bin"
],
"cpuLimit": 20000000000,
"memoryLimit": 104857600,
"clockLimit": 40000000000,
"procLimit": 50,
"copyOut": [
"stdout",
"stderr"
],
"stdout": {
"name": "stdout",
"max": 4096
},
"stderr": {
"name": "stderr",
"max": 4096
},
# matlab don't need this
# "copyInCached": {
# "a": "a"
# }
},
"cases": []
}
},
"parser": {
"name": "diff",
"with": {
"cases": []
}
}
}
output_json["stages"].append(healthcheck_json)
output_json["stages"].append(run_json)
return output_json
def get_cases(output_json, yaml_data):
for case in yaml_data['cases']:
print(yaml_data['cases'])
input_entry = {
"stdin":{
"src": case["input"]
}
}
output_entry = {
"outputs": {
"score": 100,
"fileName": "stdout",
"answerPath": case["output"]
}
}
output_json["stages"][1]["executor"]["with"]["cases"].append(input_entry)
output_json["stages"][1]["parser"]["with"]["cases"].append(output_entry)
return output_json
# Function to merge YAML content into the JSON structure
def yaml_to_custom_json(yaml_file, json_file):
# Load YAML data from the input file
with open(yaml_file, 'r') as f:
yaml_data = yaml.safe_load(f)
# Define the base JSON structure as per your example
output_json = matlab_json_init()
# memory limit migration
memory_str = yaml_data['default']['memory']
memory_limit = int(memory_str[:-1]) * 1024 * 1024
output_json["stages"][0]["executor"]["with"]["default"]["memoryLimit"] = memory_limit
# time limit migration
time_str = yaml_data['default']['time']
cpu_limit = int(time_str[:-1]) * 1000000000
clock_limit = 2 * cpu_limit
output_json['stages'][1]["executor"]["with"]["default"]["cpuLimit"] = cpu_limit
output_json["stages"][1]["executor"]["with"]["default"]["clockLimit"] = clock_limit
# test cases migration
# testcases input migration
# # testcases output migration
output_json = get_cases(output_json, yaml_data)
# execution args migration
args = "octave " + assignment_name + ".m"
output_json["stages"][1]["executor"]["with"]["default"]["args"] = args.split()
# Write the output JSON to the specified file
with open(json_file, 'w') as f:
json.dump(output_json, f, indent=2)
# i/p of files
yaml_file = './ex4/config.yaml'
json_file = './output.json'
assignment_name = "ex4"
yaml_to_custom_json(yaml_file, json_file)
print(f"YAML content has been successfully converted to JSON and saved to {json_file}")