python脚本

This commit is contained in:
wangfiox 2024-08-17 20:52:28 +08:00
parent ab4819e15e
commit 32259fa217
2 changed files with 53 additions and 0 deletions

1
.gitignore vendored
View File

@ -10,3 +10,4 @@ build
!**/expect/output !**/expect/output
**/*.toml **/*.toml
zci.toml zci.toml
err

52
test.py Normal file
View File

@ -0,0 +1,52 @@
import os
import subprocess
import shutil
from pathlib import Path
# 配置
gcc = "gcc"
timeout = None # 设置超时时间,单位:秒
# 设置路径
base_dir = "./h-p/" # 基本路径
asm_dir = Path(base_dir) / "asm"
bin_dir = Path(base_dir) / "bin"
input_dir = Path(base_dir) / "input"
err_dir = Path(base_dir) / "err"
# 创建 bin 和 err 目录
bin_dir.mkdir(parents=True, exist_ok=True)
err_dir.mkdir(parents=True, exist_ok=True)
# 遍历 asm 文件夹,编译 .s 文件
for asm_file in asm_dir.glob("*.s"):
executable_name = asm_file.stem
executable_path = bin_dir / executable_name
# 编译 .s 文件为可执行文件
compile_command = [gcc, asm_file, "lib/sylib.c", "-o", executable_path]
try:
subprocess.run(compile_command, check=True)
except subprocess.CalledProcessError as e:
print(f"编译 {asm_file} 失败: {e}")
continue
# 查找对应的 .in 输入文件
input_file = input_dir / f"{executable_name}.in"
input_arg = input_file if input_file.exists() else None
# 错误输出重定向文件
stderr_file = err_dir / f"{executable_name}.txt"
# 运行可执行文件,处理输入和错误输出
try:
with stderr_file.open("w") as stderr:
if input_arg:
with input_arg.open("r") as stdin:
subprocess.run(
[executable_path], stdin=stdin, stderr=stderr, timeout=timeout
)
else:
subprocess.run([executable_path], stderr=stderr, timeout=timeout)
except subprocess.TimeoutExpired:
print(f"{executable_name} 执行超时,进程已被终止")