gahusb
119ac88e1e
- StockAgent.on_screener_schedule: snapshot/refresh → screener/run(mode=auto) → telegram_payload(MarkdownV2) 발송. skipped_holiday는 무발신, 실패 시 운영자 HTML 알림. - service_proxy: refresh_screener_snapshot, run_stock_screener 추가 (각각 180s timeout, STOCK_LAB_URL 기존 env 재사용). - telegram.messaging.send_raw: parse_mode 파라미터 추가 (기본 HTML 유지, MarkdownV2 페이로드 직접 전달용). - scheduler: cron day_of_week=mon-fri hour=16 minute=30 id=stock_screener (Asia/Seoul TZ). - on_command 'run_screener' 수동 트리거 추가. - tests: 성공/휴일/스냅샷실패/run실패/이상status 5케이스.
84 lines
2.6 KiB
Python
84 lines
2.6 KiB
Python
"""고수준 메시지 전송 API."""
|
|
import uuid
|
|
from typing import Optional
|
|
|
|
from ..config import TELEGRAM_CHAT_ID
|
|
from ..db import save_telegram_callback
|
|
from .client import _enabled, api_call
|
|
from .formatter import MessageKind, format_agent_message
|
|
|
|
|
|
async def send_raw(
|
|
text: str,
|
|
reply_markup: Optional[dict] = None,
|
|
chat_id: Optional[str] = None,
|
|
parse_mode: str = "HTML",
|
|
) -> dict:
|
|
"""가장 저수준. 원문 텍스트 그대로 전송. chat_id 생략 시 기본 TELEGRAM_CHAT_ID로.
|
|
|
|
parse_mode: 기본 'HTML'. MarkdownV2 페이로드(예: 스크리너) 전송 시 명시 지정.
|
|
"""
|
|
if not _enabled():
|
|
return {"ok": False, "message_id": None}
|
|
payload = {
|
|
"chat_id": chat_id or TELEGRAM_CHAT_ID,
|
|
"text": text,
|
|
"parse_mode": parse_mode,
|
|
}
|
|
if reply_markup:
|
|
payload["reply_markup"] = reply_markup
|
|
result = await api_call("sendMessage", payload)
|
|
ok = result.get("ok", False)
|
|
return {
|
|
"ok": ok,
|
|
"message_id": result.get("result", {}).get("message_id") if ok else None,
|
|
"description": result.get("description") if not ok else None,
|
|
"error_code": result.get("error_code") if not ok else None,
|
|
}
|
|
|
|
|
|
async def send_agent_message(
|
|
agent_id: str,
|
|
kind: MessageKind,
|
|
title: str,
|
|
body: str,
|
|
task_id: Optional[str] = None,
|
|
actions: Optional[list] = None,
|
|
metadata: Optional[dict] = None,
|
|
body_is_html: bool = False,
|
|
) -> dict:
|
|
"""통합 에이전트 메시지 API. 모든 에이전트가 이걸 씀.
|
|
|
|
body_is_html=True: 호출자가 이미 HTML-safe 포맷(링크 <a> 등) 구성한 경우.
|
|
"""
|
|
text = format_agent_message(agent_id, kind, title, body, metadata, body_is_html=body_is_html)
|
|
reply_markup = None
|
|
if actions:
|
|
buttons = []
|
|
for action in actions:
|
|
cb_id = f"{action['action']}_{uuid.uuid4().hex[:8]}"
|
|
save_telegram_callback(cb_id, task_id or "", agent_id)
|
|
buttons.append({"text": action["label"], "callback_data": cb_id})
|
|
reply_markup = {"inline_keyboard": [buttons]}
|
|
return await send_raw(text, reply_markup)
|
|
|
|
|
|
async def send_approval_request(
|
|
agent_id: str,
|
|
task_id: str,
|
|
title: str,
|
|
detail: str,
|
|
) -> dict:
|
|
"""승인/거절 단축 헬퍼."""
|
|
return await send_agent_message(
|
|
agent_id=agent_id,
|
|
kind="approval",
|
|
title=title,
|
|
body=detail,
|
|
task_id=task_id,
|
|
actions=[
|
|
{"label": "✅ 승인", "action": "approve"},
|
|
{"label": "❌ 거절", "action": "reject"},
|
|
],
|
|
)
|