2569718930@qq.com
286 lines
10 KiB
Python
286 lines
10 KiB
Python
import requests
|
|
import html
|
|
from loguru import logger
|
|
from datetime import datetime
|
|
|
|
|
|
class TelegramNotifier:
|
|
"""
|
|
Telegram 消息推送模块
|
|
支持信号推送、预警推送和市场异常提醒
|
|
"""
|
|
|
|
def __init__(self, config: dict):
|
|
self.config = config
|
|
self.token = config.get("bot_token")
|
|
self.chat_id = config.get("chat_id")
|
|
self.proxy = config.get("proxy")
|
|
|
|
self.session = requests.Session()
|
|
if self.proxy:
|
|
if not self.proxy.startswith("http"):
|
|
self.proxy = f"http://{self.proxy}"
|
|
self.session.proxies = {"http": self.proxy, "https": self.proxy}
|
|
|
|
logger.info("Telegram 通知器初始化完成。")
|
|
|
|
@staticmethod
|
|
def _escape_html(text: str) -> str:
|
|
"""Escape HTML special characters"""
|
|
if not isinstance(text, str):
|
|
text = str(text)
|
|
return html.escape(text, quote=False)
|
|
|
|
def _send_message(self, text: str):
|
|
"""发送 Telegram 消息的主函数 (支持多个 ID)"""
|
|
if not self.token or not self.chat_id:
|
|
logger.warning("未配置 Telegram Token 或 Chat ID，无法发送消息。")
|
|
return False
|
|
|
|
# 支持逗号分隔的多个 ID
|
|
chat_ids = str(self.chat_id).replace(" ", "").split(",")
|
|
url = f"https://api.telegram.org/bot{self.token}/sendMessage"
|
|
|
|
all_successful = True
|
|
for cid in chat_ids:
|
|
if not cid:
|
|
continue
|
|
|
|
payload = {
|
|
"chat_id": cid,
|
|
"text": text,
|
|
"parse_mode": "HTML",
|
|
"disable_web_page_preview": True,
|
|
}
|
|
|
|
try:
|
|
response = self.session.post(url, json=payload, timeout=10)
|
|
if response.status_code != 200:
|
|
error_msg = response.text
|
|
if "chat not found" in error_msg.lower():
|
|
logger.error(
|
|
f"Telegram 消息发送给 {cid} 失败 (400): Chat ID {cid} 无效或机器人尚未被加入该聊天。请在 Telegram 中发送 /id 给机器人确认正确的 Chat ID。"
|
|
)
|
|
else:
|
|
logger.error(
|
|
f"Telegram 消息发送给 {cid} 失败 ({response.status_code}): {error_msg}"
|
|
)
|
|
all_successful = False
|
|
else:
|
|
logger.info(f"Telegram 消息发送给 {cid} 成功。")
|
|
except Exception as e:
|
|
logger.error(f"Telegram 消息发送给 {cid} 异常: {e}")
|
|
all_successful = False
|
|
return all_successful
|
|
|
|
def send_signal(
|
|
self,
|
|
market_name: str,
|
|
full_title: str,
|
|
option: str,
|
|
score: float,
|
|
prediction: str,
|
|
confidence: int,
|
|
analysis_list: list,
|
|
price: float,
|
|
market_url: str,
|
|
local_time: str = None,
|
|
target_date: str = None,
|
|
):
|
|
"""发送交易信号推送"""
|
|
stars = "⭐" * int(score) + "☆" * (5 - int(score))
|
|
timestamp_utc = datetime.utcnow().strftime("%H:%M")
|
|
|
|
analysis_text = "\n".join(
|
|
[
|
|
f"✅ {self._escape_html(item)}" if "✅" not in item else item
|
|
for item in analysis_list
|
|
]
|
|
)
|
|
|
|
local_time_text = (
|
|
f"🕒 当地时间: <b>{self._escape_html(local_time)}</b>\n"
|
|
if local_time
|
|
else ""
|
|
)
|
|
target_date_text = self._escape_html(target_date) if target_date else "待定"
|
|
|
|
text = (
|
|
f"🎯 <b>交易信号 #{self._escape_html(market_name.split(' ')[0])}</b>\n\n"
|
|
f"📍 城市: <b>{self._escape_html(market_name)}</b>\n"
|
|
f"🏆 市场: <i>{self._escape_html(full_title)}</i>\n"
|
|
f"📝 选项: <b>{self._escape_html(option)}</b>\n"
|
|
f"💰 当前价格: <b>{price}¢</b>\n"
|
|
f"═══════════════════\n"
|
|
f"📊 信号评分: {stars} ({score}/5)\n"
|
|
f"🤖 模型预测: {self._escape_html(prediction)}\n"
|
|
f"📈 置信度: {confidence}%\n\n"
|
|
f"分析汇总:\n"
|
|
f"{analysis_text}\n"
|
|
f"═══════════════════\n"
|
|
f"{local_time_text}"
|
|
f"📅 结算日期: <b>{target_date_text}</b>\n"
|
|
f"🔗 <a href='{market_url}'>点击进入市场</a>\n\n"
|
|
f"⏰ 信号时间: {timestamp_utc} UTC"
|
|
)
|
|
return self._send_message(text)
|
|
|
|
def send_combined_alert(
|
|
self,
|
|
city: str,
|
|
alerts: list,
|
|
local_time: str = None,
|
|
forecast_temp: str = None,
|
|
total_volume: float = 0,
|
|
brackets_count: int = 0,
|
|
strategy_tips: list = None,
|
|
metar_data: dict = None,
|
|
):
|
|
"""发送简约版合并预警 (含 METAR 航空气象数据)"""
|
|
if not alerts:
|
|
return
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
# UTC+8 北京时间
|
|
now_bj = datetime.utcnow() + timedelta(hours=8)
|
|
timestamp_bj = now_bj.strftime("%H:%M")
|
|
|
|
# 1. METAR 航空气象数据区块
|
|
metar_text = ""
|
|
if metar_data and metar_data.get("current", {}).get("temp") is not None:
|
|
icao = metar_data.get("icao", "N/A")
|
|
temp = metar_data["current"]["temp"]
|
|
unit = "°F" if metar_data.get("unit") == "fahrenheit" else "°C"
|
|
|
|
# 解析观测时间 (格式: 2026-02-07T11:00:00.000Z)
|
|
obs_time_raw = metar_data.get("observation_time", "")
|
|
if "T" in obs_time_raw:
|
|
obs_time = obs_time_raw.split("T")[1][:5] + " UTC"
|
|
else:
|
|
obs_time = obs_time_raw or "N/A"
|
|
|
|
# 可选：风速信息
|
|
wind_kt = metar_data["current"].get("wind_speed_kt")
|
|
wind_text = f" | 风速:{wind_kt}kt" if wind_kt else ""
|
|
|
|
metar_text = (
|
|
f"✈️ <b>机场实测 ({icao}):</b>\n"
|
|
f" 🌡️ {temp:.1f}{unit}{wind_text}\n"
|
|
f" 🕐 观测: {obs_time}\n\n"
|
|
)
|
|
|
|
# 2. 信号详情构建
|
|
items_text = ""
|
|
for a in alerts:
|
|
items_text += f"{a['msg']}\n\n"
|
|
|
|
# 3. 策略建议（如果有）
|
|
tips_text = ""
|
|
if strategy_tips:
|
|
tips_text = (
|
|
"💡 <b>策略建议:</b>\n"
|
|
+ "\n".join([f"• {self._escape_html(tip)}" for tip in strategy_tips])
|
|
+ "\n\n"
|
|
)
|
|
|
|
# 4. 总体布局
|
|
text = (
|
|
f"🔔 <b>城市监控报告 #{self._escape_html(city)}</b>\n\n"
|
|
f"📍 城市: {self._escape_html(city)}\n"
|
|
f"{metar_text}"
|
|
f"📊 <b>实时异动:</b>\n"
|
|
f"{items_text}"
|
|
f"{tips_text}"
|
|
f"═══════════════════\n"
|
|
f"🕒 当地时间: {self._escape_html(local_time or 'N/A')}\n"
|
|
f"⏰ 预警时间: {timestamp_bj} (北京时间)"
|
|
)
|
|
return self._send_message(text)
|
|
|
|
def send_anomaly(
|
|
self,
|
|
city_tag: str,
|
|
market_name: str,
|
|
detected_anomaly: str,
|
|
stats: dict,
|
|
whales: list,
|
|
current_price: float,
|
|
local_time: str = None,
|
|
):
|
|
"""发送市场异常推送"""
|
|
from datetime import datetime, timedelta
|
|
|
|
# UTC+8 北京时间
|
|
timestamp_bj = (datetime.utcnow() + timedelta(hours=8)).strftime("%H:%M")
|
|
|
|
whale_text = "\n".join([f"- {self._escape_html(w)}" for w in whales])
|
|
stats_text = "\n".join(
|
|
[
|
|
f"{self._escape_html(k)}: {self._escape_html(v)}"
|
|
for k, v in stats.items()
|
|
]
|
|
)
|
|
local_time_text = (
|
|
f"🕒 当地时间: <b>{self._escape_html(local_time)}</b>\n"
|
|
if local_time
|
|
else ""
|
|
)
|
|
|
|
text = (
|
|
f"👀 <b>市场异常 #{self._escape_html(city_tag)}</b>\n\n"
|
|
f"📍 城市: {self._escape_html(city_tag)}\n"
|
|
f"🏆 市场: {self._escape_html(market_name)}\n\n"
|
|
f"🚨 <b>检测到异常:</b>\n"
|
|
f"{self._escape_html(detected_anomaly)}\n"
|
|
f"{stats_text}\n\n"
|
|
f"🐋 <b>大户动向:</b>\n"
|
|
f"{whale_text}\n\n"
|
|
f"💰 当前价格: <b>{current_price}¢</b>\n"
|
|
f"═══════════════════\n"
|
|
f"{local_time_text}"
|
|
f"⏰ 信号时间: {timestamp_bj} (北京时间)"
|
|
)
|
|
return self._send_message(text)
|
|
|
|
def send_alert(
|
|
self,
|
|
city_tag: str,
|
|
market_name: str,
|
|
price: float,
|
|
trigger: str,
|
|
prev_price: float,
|
|
change: str,
|
|
quick_analysis: list,
|
|
local_time: str = None,
|
|
):
|
|
"""发送价格预警推送"""
|
|
from datetime import datetime, timedelta
|
|
|
|
# UTC+8 北京时间
|
|
timestamp_bj = (datetime.utcnow() + timedelta(hours=8)).strftime("%H:%M")
|
|
|
|
analysis_text = "\n".join(
|
|
[f"- {self._escape_html(item)}" for item in quick_analysis]
|
|
)
|
|
local_time_text = (
|
|
f"🕒 当地时间: <b>{self._escape_html(local_time)}</b>\n"
|
|
if local_time
|
|
else ""
|
|
)
|
|
|
|
text = (
|
|
f"⚡ <b>价格预警 #{self._escape_html(city_tag)}</b>\n\n"
|
|
f"📍 城市: {self._escape_html(city_tag)}\n"
|
|
f"🏆 市场: {self._escape_html(market_name)}\n"
|
|
f"💰 报价: <b>{price}¢ ↗️</b>\n\n"
|
|
f"触发条件: {self._escape_html(trigger)}\n"
|
|
f"变动详情: {prev_price}¢ -> {price}¢ ({self._escape_html(change)})\n\n"
|
|
f"📊 <b>快速分析:</b>\n"
|
|
f"{analysis_text}\n\n"
|
|
f"═══════════════════\n"
|
|
f"{local_time_text}"
|
|
f"⏰ 预警时间: {timestamp_bj} (北京时间)"
|
|
)
|
|
return self._send_message(text)
|