python脚本

.gitignore

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

test.py

@@ -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} 执行超时，进程已被终止")