pws_service.py

# pws_service.py
import requests
import json
from config import PWS_API_URL

def fetch_latest_pws_info() -> str:
    """
    從 MCP PWS 伺服器擷取最新的 PWS (Public Weather Service) 發布情形。
    此端點為 SSE (Server-Sent Events)，因此我們需要讀取事件串流。
    """
    try:
        # 使用 stream=True 來處理伺服器發送的事件串流
        r = requests.get(PWS_API_URL, timeout=15, stream=True)
        r.raise_for_status() # 如果 HTTP 狀態碼不是 200，則引發錯誤

        # 迭代處理從伺服器接收到的每一行資料
        for line in r.iter_lines():
            if line:
                decoded_line = line.decode('utf-8')
                # SSE 事件的資料通常以 "data: " 開頭
                if decoded_line.startswith('data:'):
                    # 移除 "data: " 前綴並去除多餘空格，取得 JSON 字串
                    json_data_str = decoded_line[5:].strip()
                    
                    try:
                        # 解析 JSON 字串
                        data = json.loads(json_data_str)
                        # 從解析後的資料中提取報告內容 (假設 key 是 'pws_report')
                        pws_report = data.get("pws_report")

                        if not pws_report:
                            return "❌ PWS 查詢失敗：無法從 API 回應中找到 'pws_report' 欄位。"

                        # 格式化最終要回傳的訊息
                        lines = ["🛰️ 最新 PWS 發布情形:", "----------------------------------", pws_report]
                        return "\n".join(lines)
                    
                    except json.JSONDecodeError:
                        return "❌ PWS 查詢失敗：伺服器回應的格式不正確 (非有效JSON)。"
        
        return "❌ PWS 查詢失敗：無法從伺服器串流中解析到有效的 PWS 資料。"

    except requests.exceptions.RequestException as e:
        return f"❌ PWS 查詢失敗：無法連接到伺服器。\n錯誤訊息：{e}"
    except Exception as e:
        return f"❌ PWS 查詢失敗：發生未預期的錯誤。\n錯誤訊息：{e}"