FastAPI与WSL子进程交互：文件路径传递的正确姿势

subprocess.run(['wsl','./process', '-m', ' 1 ', ' -L ', '{img1.filename}' ,' -R ','{img2.filename}'])

subprocess.run(
    [
        "wsl",
        "./process",
        "-m",
        "1",
        "-L",
        img1.filename,  # 直接使用变量
        "-R",
        img2.filename,  # 直接使用变量
    ]
)

import shutil
from fastapi import FastAPI, File, HTTPException, UploadFile
import os
import subprocess
from pydantic import ValidationError

app = FastAPI()

# 定义允许的文件扩展名
ALLOWED_EXTENSIONS = ('.jpg', '.jpeg', '.png')

def validate_file_type(filename: str):
    """验证文件类型是否在允许的扩展名列表中。"""
    ext = os.path.splitext(filename)[1]
    if ext.lower() not in ALLOWED_EXTENSIONS:
        raise HTTPException(status_code=422, detail=f"只允许 {', '.join(ALLOWED_EXTENSIONS)} 类型的文件")

@app.post("/process-images")
async def process_images(img1: UploadFile = File(...), img2: UploadFile = File(...)):
    """
    处理上传的图片，并调用WSL子进程进行进一步处理。
    """
    try:
        # 1. 文件类型验证
        validate_file_type(img1.filename)
        validate_file_type(img2.filename)

        # 2. 保存上传文件到本地
        # 建议使用更安全的路径，例如临时目录或专门的上传目录
        # 这里为了演示，直接使用文件名，实际应用中应考虑路径冲突和安全性
        file_path_img1 = img1.filename # 或者 os.path.join("uploads", img1.filename)
        file_path_img2 = img2.filename # 或者 os.path.join("uploads", img2.filename)

        with open(file_path_img1, "wb") as buffer:
            shutil.copyfileobj(img1.file, buffer)

        with open(file_path_img2, "wb") as buffer:
            shutil.copyfileobj(img2.file, buffer)

        # 3. 调用WSL子进程
        # 重点：直接传递变量 img1.filename 和 img2.filename
        # 推荐使用 check=True 来自动检查子进程的退出码
        subprocess.run(
            [
                "wsl",
                "./process",
                "-m",
                "1",
                "-L",
                file_path_img1,  # 正确传递文件名变量
                "-R",
                file_path_img2,  # 正确传递文件名变量
            ],
            check=True,  # 如果子进程返回非零退出码，则会抛出 CalledProcessError
            capture_output=True, # 捕获子进程的stdout和stderr，便于调试
            text=True # 如果 capture_output=True，则以文本模式捕获输出
        )

        # 4. 清理临时文件 (可选但推荐)
        # os.remove(file_path_img1)
        # os.remove(file_path_img2)

        return {"message": "图片处理请求已发送成功"}

    except ValidationError as e:
        # Pydantic 验证错误
        raise HTTPException(status_code=422, detail=str(e))
    except subprocess.CalledProcessError as e:
        # 子进程执行失败
        print(f"WSL process failed. Stdout: {e.stdout}, Stderr: {e.stderr}")
        raise HTTPException(status_code=500, detail=f"WSL处理程序执行失败: {e.stderr.strip()}")
    except Exception as e:
        # 其他内部服务器错误
        raise HTTPException(status_code=500, detail=f"内部服务器错误: {str(e)}")