fix: 飞书通知

api_routes.py

@@ -11,6 +11,7 @@ import shutil
 import time
 import uuid
 import subprocess
+import requests
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime
 from typing import Any, Dict, List, Optional, Tuple
@@ -36,16 +37,37 @@ except Exception:
 from fastapi.responses import JSONResponse, FileResponse, HTMLResponse
 
 import wx_access_token
-from config import logger, OUTPUT_DIR, IMAGES_DIR, DEEPFACE_AVAILABLE, \
-    DLIB_AVAILABLE, GFPGAN_AVAILABLE, DDCOLOR_AVAILABLE, REALESRGAN_AVAILABLE, \
-    UPSCALE_SIZE, CLIP_AVAILABLE, REALESRGAN_MODEL, REMBG_AVAILABLE, \
-    ANIME_STYLE_AVAILABLE, SAVE_QUALITY, \
-    AUTO_INIT_ANALYZER, AUTO_INIT_GFPGAN, AUTO_INIT_DDCOLOR, \
-    AUTO_INIT_REALESRGAN, MODELS_PATH, \
-    AUTO_INIT_REMBG, AUTO_INIT_ANIME_STYLE, RVM_AVAILABLE, AUTO_INIT_RVM, \
-    FACE_SCORE_MAX_IMAGES, FEMALE_AGE_ADJUSTMENT, \
-    FEMALE_AGE_ADJUSTMENT_THRESHOLD, CELEBRITY_SOURCE_DIR, \
-    CELEBRITY_FIND_THRESHOLD
+from config import (
+    logger,
+    OUTPUT_DIR,
+    IMAGES_DIR,
+    DEEPFACE_AVAILABLE,
+    DLIB_AVAILABLE,
+    GFPGAN_AVAILABLE,
+    DDCOLOR_AVAILABLE,
+    REALESRGAN_AVAILABLE,
+    UPSCALE_SIZE,
+    CLIP_AVAILABLE,
+    REALESRGAN_MODEL,
+    REMBG_AVAILABLE,
+    ANIME_STYLE_AVAILABLE,
+    SAVE_QUALITY,
+    AUTO_INIT_ANALYZER,
+    AUTO_INIT_GFPGAN,
+    AUTO_INIT_DDCOLOR,
+    AUTO_INIT_REALESRGAN,
+    MODELS_PATH,
+    AUTO_INIT_REMBG,
+    AUTO_INIT_ANIME_STYLE,
+    RVM_AVAILABLE,
+    AUTO_INIT_RVM,
+    FACE_SCORE_MAX_IMAGES,
+    FEMALE_AGE_ADJUSTMENT,
+    FEMALE_AGE_ADJUSTMENT_THRESHOLD,
+    CELEBRITY_SOURCE_DIR,
+    CELEBRITY_FIND_THRESHOLD,
+    FEISHU_WEBHOOK_URL,
+)
 from database import (
     record_image_creation,
     fetch_paged_image_records,
@@ -1589,8 +1611,8 @@ async def app_analyze_face(
                 "device_model": request.headers.get("x-device-model"),
                 "os_version": request.headers.get("x-os-version"),
                 "app_version": request.headers.get("x-app-version"),
-                "timezone": request.headers.get("x-timezone"),
                 "region": request.headers.get("x-region"),
+                "timezone": request.headers.get("x-timezone"),
                 "language": request.headers.get("Accept-Language"),
             }
             # 创建异步任务，不阻塞当前请求
@@ -1608,6 +1630,82 @@ async def app_analyze_face(
     )
 
 
+@api_router.post("/send_feishu_notification", tags=["系统通知"])
+@log_api_params
+async def send_feishu_notification(
+    request: Request,
+    title: str = Form(..., description="通知标题"),
+    payload: str = Form(None, description="自定义业务数据的JSON字符串"),
+):
+    """
+    发送飞书消息通知，集成设备信息与自定义内容。
+    """
+    # 提取设备信息
+    device_info = {
+        "device_id": request.headers.get("x-device-id"),
+        "device_type": request.headers.get("x-device-type"),
+        "device_model": request.headers.get("x-device-model"),
+        "os_version": request.headers.get("x-os-version"),
+        "app_version": request.headers.get("x-app-version"),
+        "region": request.headers.get("x-region"),
+    }
+
+    # 构造飞书文案
+    time_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    text_lines = [f"📢 【{title}】"]
+    text_lines.append(f"⏱️ 时间：{time_str}")
+
+    # 映射表
+    header_mapping = {
+        "device_id": "🆔 设备ID",
+        "device_type": "📱 设备类型",
+        "device_model": "🛠️ 设备型号",
+        "os_version": "💿 系统版本",
+        "app_version": "📦 App版本",
+        "region": "🌍 地区"
+    }
+
+    for key, label in header_mapping.items():
+        val = device_info.get(key)
+        if val:
+            text_lines.append(f"{label}：{val}")
+
+    # 处理自定义 payload
+    if payload:
+        try:
+            extra_data = json.loads(payload)
+            if isinstance(extra_data, dict):
+                for k, v in extra_data.items():
+                    text_lines.append(f"🔹 {k}：{v}")
+            else:
+                text_lines.append(f"🔹 内容：{payload}")
+        except json.JSONDecodeError:
+            text_lines.append(f"🔹 内容：{payload}")
+
+    full_text = "\n".join(text_lines)
+
+    # 发送通知（在线程池中执行以防阻塞）
+    def _send():
+        try:
+            if not FEISHU_WEBHOOK_URL:
+                logger.warning("FEISHU_WEBHOOK_URL is not configured, skip sending.")
+                return
+            resp = requests.post(
+                FEISHU_WEBHOOK_URL,
+                json={"msg_type": "text", "content": {"text": full_text}},
+                timeout=10
+            )
+            logger.info(f"Feishu notification sent: {resp.status_code}, {resp.text}")
+        except Exception as exc:
+            logger.warning(f"Failed to send Feishu notification: {exc}")
+
+    # 使用 run_in_executor 避免阻塞
+    loop = asyncio.get_event_loop()
+    loop.run_in_executor(executor, _send)
+
+    return {"success": True, "message": "Notification scheduled"}
+
+
 @api_router.post("/image_search", response_model=ImageFileList, tags=["图像搜索"])
 @log_api_params
 async def search_by_image(

config.py

@@ -200,6 +200,11 @@ CELEBRITY_FIND_THRESHOLD = float(
 )
 
 # ---- start ----
+# 飞书 Webhook 配置
+FEISHU_WEBHOOK_URL = os.environ.get(
+    "FEISHU_WEBHOOK_URL",
+    "https://open.feishu.cn/open-apis/bot/v2/hook/07cfc4b6-4dfd-4b5c-80a4-a66ccba83547"
+).strip()
 # 微信小程序配置（默认值仅用于本地开发）
 WECHAT_APPID = os.environ.get("WECHAT_APPID", "******").strip()
 WECHAT_SECRET = os.environ.get("WCT_SECRET", "******").strip()