feat: 添加掉落物的url查询API

app/api/info.py

@@ -186,6 +186,21 @@ async def get_web_config() -> InfoOut:
     return InfoOut(data={"WebConfig": data})
 
 
+@router.post(
+    "/get/materials", summary="获取材料URL", response_model=InfoOut, status_code=200
+)
+async def get_materials(
+    materials: GetMaterialsIn = Body(..., description="材料ID")
+) -> InfoOut:
+    try:
+        data = await Config.get_materials(materials.dropId)
+    except Exception as e:
+        return InfoOut(
+            code=500, status="error", message=f"{type(e).__name__}: {str(e)}", data={}
+        )
+    return InfoOut(data={"FileUrl": data})
+
+
 @router.post(
     "/get/overview", summary="信息总览", response_model=InfoOut, status_code=200
 )

app/core/config.py

@@ -30,9 +30,10 @@ import requests
 import truststore
 from pathlib import Path
 from fastapi import WebSocket
+from urllib.parse import quote
 from collections import defaultdict
 from datetime import datetime, timedelta, date, timezone
-from typing import Literal, Optional, Tuple
+from typing import Literal, Optional
 
 from app.models.ConfigBase import *
 from app.utils.constants import *
@@ -1741,6 +1742,12 @@ class AppConfig(GlobalConfig):
 
         return stage_data
 
+    async def get_materials(self, dropId):
+        """获取材料URL"""
+        if dropId not in MATERIALS_MAP:
+            raise ValueError(f"未知的材料ID: {dropId}")
+        return f"file:///{quote((Path.cwd() / f"res/images/materials/{dropId}.png").as_posix())}"
+
     async def get_script_combox(self):
         """获取脚本下拉框信息"""
 

app/models/schema.py

@@ -67,6 +67,10 @@ class GetStageIn(BaseModel):
     )
 
 
+class GetMaterialsIn(BaseModel):
+    dropId: str = Field(..., description="材料ID")
+
+
 class GlobalConfig_Function(BaseModel):
     HistoryRetentionTime: Optional[Literal[7, 15, 30, 60, 90, 180, 365, 0]] = Field(
         None, description="历史记录保留时间, 0表示永久保存"

app/utils/constants.py

@@ -206,5 +206,9 @@ MATERIALS_MAP = {
     "3261": "医疗芯片",
     "3271": "辅助芯片",
     "3281": "特种芯片",
+    "PR-A": "医疗/重装芯片",
+    "PR-B": "术师/狙击芯片",
+    "PR-C": "先锋/辅助芯片",
+    "PR-D": "近卫/特种芯片",
 }
 """掉落物索引表"""