Create pws_service.py

pws_service.py

@@ -0,0 +1,46 @@
+# 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}"