v0.3.1 新增接口 set_config，用于设置相关配置，暂时仅提供前端 字幕打印机API地址设置功能

Author: Ikaros-521

README.md

@@ -112,6 +112,27 @@ python：3.10.10
 |--------      |--------|----------|--------------        |
 
 
+### 响应
+
+| 参数名  | 类型    | 描述         |
+|--------|-------- |--------------|
+| code   | int     | 状态码，200为成功，小于0为错误代码，大于0为部分成功代码 |
+| msg    | string  | 响应消息，描述请求的处理结果 |
+
+## 设置相关配置
+
+### 概述
+
+- **请求地址:** `/set_config`
+- **请求类型:** POST
+- **描述:** 用于设置相关配置
+
+### 请求参数
+
+| 参数名        | 类型   | 是否必需  | 描述                 |
+|--------      |--------|----------|--------------        |
+| captions_printer_api_url   | string | 否       | 前端 字幕打印机API请求地址，例如：http://127.0.0.1:5500/send_message     |
+
 ### 响应
 
 | 参数名  | 类型    | 描述         |
@@ -121,6 +142,8 @@ python：3.10.10
 | msg    | string  | 响应消息，描述请求的处理结果 |
 
 # 更新日志
+- v0.3.1
+    - 新增接口 set_config，用于设置相关配置，暂时仅提供前端 字幕打印机API地址设置功能
 
 - v0.3.0
     - show接口 新增参数 captions_printer 控制字幕打印机显示文本内容，不传则不发送 

api_server.py

@@ -16,7 +16,7 @@
 from utils.common import Common
 from utils.logger import Configure_logger
 from utils.video_generate import run_get_video
-from utils.models import ShowMessage, GetNonDefaultVideoCountResult, GetVideoQueueResult, CommonResult
+from utils.models import ShowMessage, GetNonDefaultVideoCountResult, GetVideoQueueResult, CommonResult, SetConfigMessage
 
 # 获取 httpx 库的日志记录器
 httpx_logger = logging.getLogger("httpx")
@@ -257,6 +257,24 @@ async def get_video_queue():
         logging.error(traceback.format_exc())
         return CommonResult(code=-1, message=f"操作失败: {str(e)}")
 
+# 设置配置
+@app.post("/set_config")
+async def set_config(msg: SetConfigMessage):
+    try:
+        await send_to_all_websockets(
+            json.dumps(
+                {
+                    "type": "set_config",
+                    "data": msg.dict()
+                }
+            )
+        )
+        await asyncio.sleep(0.5)
+        return CommonResult(code=200, message="操作成功")
+    except Exception as e:
+        logging.error(traceback.format_exc())
+        return CommonResult(code=-1, message=f"操作失败: {str(e)}")
+
 def start_browser(stop_event):
     options = webdriver.ChromeOptions()
     # 设置为开发者模式，避免被浏览器识别为自动化程序

static/index.html

@@ -27,6 +27,15 @@
     let currentVideo = video1; // 当前播放的视频元素
     let nextVideo = video2; // 下一个将要播放的视频元素
     let isWaitingForNextVideo = false; // 是否正在等待下一个视频加载
+    let captions_printer_api_url = 'http://127.0.0.1:5500/send_message'; // 字幕打印机api地址
+
+    // 设置配置
+    function set_config(data) {
+        if ("captions_printer_api_url" in data.data) {
+            captions_printer_api_url = data.data.captions_printer_api_url;
+            console.log("字幕打印机api地址已更新为：" + captions_printer_api_url);
+        }
+    }
 
     /**
      * 准备好下一个视频
@@ -113,7 +122,8 @@
 
             // 发送到字幕打印机
             if ("captions_printer" in data) {
-                getData('http://127.0.0.1:5500/send_message', {"content": data.captions_printer, "start_delay": data.captions_printer_start_delay})
+                // 根据实际字幕打印机的接口进行修改
+                getData(captions_printer_api_url, {"content": data.captions_printer, "start_delay": data.captions_printer_start_delay})
                     .then(data => {
                         console.log('成功:', data);
                     })
@@ -170,6 +180,9 @@
             case 'get_video_queue':
                 getVideoQueue();
                 break;
+            case 'set_config':
+                set_config(data);
+                break;
         }
     };
 

utils/models.py

@@ -20,6 +20,8 @@ class GetVideoQueueResult(BaseModel):
     data: list
     message: str
 
+class SetConfigMessage(BaseModel):
+    captions_printer_api_url: Optional[str] = Field(None)
 
 """
 通用