@echo off

setlocal enabledelayedexpansion

echo ====================================
echo        AMPY批量烧录脚本
echo ====================================
echo.


REM 检查是否提供了COM端口参数
if "%~1"=="" (
    echo 错误：未指定COM端口！
    echo 使用方法：%~nx0 COM<端口号>
    echo 例如：%~nx0 COM10
    goto :eof
)

REM 设置COM端口变量
set COM_PORT=%~1

REM 定义需要烧录的Python文件列表
set FILE_LIST=ws2812_Dev.py sdCard_Dev.py qmi8658a.py main.py gc9107_0inch85_Dev.py flash_img.bmp

REM 检查文件是否存在
for %%f in (%FILE_LIST%) do (
    if not exist "%%f" (
        echo 错误：未找到文件 "%%f"！
        goto :eof
    )
)

REM 依次烧录每个文件
for %%f in (%FILE_LIST%) do (
    echo 正在烧录文件: %%f 到端口 %COM_PORT% ...
    ampy --port %COM_PORT% put "%%f"
    
    REM 检查ampy命令执行结果
    if !errorlevel! neq 0 (
        echo 烧录文件 "%%f" 时出错！
        goto :eof
    )
    echo 文件 "%%f" 烧录成功。
    echo.
)

echo 所有文件已成功烧录到开发板！
pause