Merge branch 'main' into dev
This commit is contained in:
镀铬酸钾
240
plugins/weather.py
Normal file
240
plugins/weather.py
Normal file
@@ -0,0 +1,240 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
import re
|
||||
from datetime import datetime
|
||||
from typing import Any, Dict, List
|
||||
|
||||
import requests
|
||||
|
||||
from core.managers.command_manager import matcher
|
||||
from core.managers.image_manager import image_manager
|
||||
from core.utils.logger import logger
|
||||
from models import MessageEvent, MessageSegment
|
||||
|
||||
# 插件元数据
|
||||
__plugin_meta__ = {
|
||||
"name": "weather",
|
||||
"description": "查询天气信息，支持中国天气网数据。",
|
||||
"usage": "/天气 [城市代码] - 查询指定城市的天气信息\n例如：/天气 101190207 (南京)",
|
||||
}
|
||||
|
||||
# 城市代码映射（可以扩展）
|
||||
CITY_CODES = {
|
||||
"北京": "101010100",
|
||||
"上海": "101020100",
|
||||
"广州": "101280101",
|
||||
"深圳": "101280601",
|
||||
"南京": "101190101",
|
||||
"苏州": "101190401",
|
||||
"杭州": "101210101",
|
||||
"武汉": "101200101",
|
||||
"成都": "101270101",
|
||||
"重庆": "101040100",
|
||||
"西安": "101110101",
|
||||
"天津": "101030100",
|
||||
"沈阳": "101070101",
|
||||
"大连": "101070201",
|
||||
"青岛": "101120201",
|
||||
"济南": "101120101",
|
||||
"郑州": "101180101",
|
||||
"长沙": "101250101",
|
||||
"南昌": "101240101",
|
||||
"合肥": "101220101",
|
||||
"福州": "101230101",
|
||||
"厦门": "101230201",
|
||||
"南宁": "101300101",
|
||||
"海口": "101310101",
|
||||
"昆明": "101290101",
|
||||
"贵阳": "101260101",
|
||||
"拉萨": "101140101",
|
||||
"兰州": "101160101",
|
||||
"西宁": "101150101",
|
||||
"银川": "101170101",
|
||||
"乌鲁木齐": "101130101",
|
||||
"哈尔滨": "101050101",
|
||||
"长春": "101060101",
|
||||
"呼和浩特": "101080101",
|
||||
"太原": "101100101",
|
||||
"石家庄": "101090101",
|
||||
}
|
||||
|
||||
HEADERS = {
|
||||
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
|
||||
}
|
||||
|
||||
|
||||
def get_weather_data(city_code: str) -> Dict[str, Any]:
|
||||
"""
|
||||
获取天气数据
|
||||
|
||||
Args:
|
||||
city_code (str): 城市代码
|
||||
|
||||
Returns:
|
||||
Dict[str, Any]: 包含城市信息和天气数据的字典
|
||||
"""
|
||||
try:
|
||||
url = f"https://www.weather.com.cn/weather/{city_code}.shtml"
|
||||
response = requests.get(url, headers=HEADERS, timeout=10)
|
||||
response.encoding = "utf-8"
|
||||
html_content = response.text
|
||||
|
||||
# 提取城市信息
|
||||
city_info = (
|
||||
html_content.split('<a href="http://www.weather.com.cn/forecast/" ')[-1]
|
||||
.split("</div>")[0]
|
||||
.strip()
|
||||
)
|
||||
|
||||
city_parts = []
|
||||
city_parts.append(city_info.split("<a>")[-1].split("</a>")[0])
|
||||
|
||||
if city_info.count("_blank") == 1:
|
||||
city_parts.append(
|
||||
city_info.split("<span>></span>")[-1]
|
||||
.replace("</span>", "")
|
||||
.replace("<span>", "")
|
||||
.strip()
|
||||
)
|
||||
else:
|
||||
additional_parts = (
|
||||
city_info.split('target="_blank">')[-1]
|
||||
.replace("<span>></span> <span>", "")
|
||||
.replace("</span>", "")
|
||||
.split("</a>")
|
||||
)
|
||||
city_parts.extend(additional_parts)
|
||||
|
||||
city_name = " ".join([part for part in city_parts if part.strip()])
|
||||
|
||||
# 提取天气信息
|
||||
weather_data = []
|
||||
for i in range(7):
|
||||
try:
|
||||
weather_info = (
|
||||
html_content.split('<ul class="t clearfix">')[-1]
|
||||
.split('on">')[1]
|
||||
.split("</li>")[i]
|
||||
)
|
||||
|
||||
day = weather_info.split("<h1>")[-1].split("</h1>")[0].strip()
|
||||
weather = (
|
||||
weather_info.split('<p title="')[-1]
|
||||
.split('" class="wea">')[0]
|
||||
.strip()
|
||||
)
|
||||
|
||||
tem = (
|
||||
weather_info.split("<span>")[-1]
|
||||
.split("</i>")[0]
|
||||
.replace("</span>/<i>", " / ")
|
||||
.strip()
|
||||
)
|
||||
if len(tem) > 10:
|
||||
tem = weather_info.split("<i>")[1].split("</i>")[0].strip()
|
||||
|
||||
wind = weather_info.split('<span title="')
|
||||
wind_direction = []
|
||||
if len(wind) > 2:
|
||||
for j in range(2):
|
||||
wind_direction.append(
|
||||
wind[j + 1]
|
||||
.split('"></span>')[0]
|
||||
.replace('" class="', " / ")
|
||||
)
|
||||
else:
|
||||
wind_direction.append(
|
||||
wind[1].split('"></span>')[0].replace('" class="', " / ")
|
||||
)
|
||||
wind_power = weather_info.split("<i>")[-1].split("</i>")[0].strip()
|
||||
|
||||
wind_direction_str = (
|
||||
" / ".join(wind_direction) if wind_direction else "未知"
|
||||
)
|
||||
|
||||
weather_data.append(
|
||||
{
|
||||
"day": day,
|
||||
"weather": weather,
|
||||
"temperature": tem,
|
||||
"wind_power": wind_power,
|
||||
"wind_direction": wind_direction_str,
|
||||
}
|
||||
)
|
||||
|
||||
except (IndexError, ValueError) as e:
|
||||
logger.warning(f"解析第{i + 1}天天气数据失败: {e}")
|
||||
continue
|
||||
|
||||
return {
|
||||
"city_name": city_name,
|
||||
"weather_data": weather_data,
|
||||
"query_time": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||
"timestamp": datetime.now().strftime("%Y年%m月%d日 %H:%M"),
|
||||
}
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"获取天气数据失败: {e}")
|
||||
return None
|
||||
|
||||
|
||||
@matcher.command("天气")
|
||||
async def handle_weather(bot, event: MessageEvent, args: List[str]):
|
||||
"""
|
||||
处理天气查询指令
|
||||
|
||||
Args:
|
||||
bot: Bot实例
|
||||
event: 消息事件
|
||||
args: 指令参数
|
||||
"""
|
||||
if not args:
|
||||
# 显示支持的城市列表
|
||||
city_list = "\n".join(
|
||||
[f"{name}: {code}" for name, code in list(CITY_CODES.items())[:10]]
|
||||
)
|
||||
reply_msg = f"请指定城市名称或城市代码，例如：\n/天气 北京\n/天气 101010100\n\n支持的城市：\n{city_list}\n..."
|
||||
await event.reply(reply_msg)
|
||||
return
|
||||
|
||||
city_input = args[0].strip()
|
||||
|
||||
# 尝试匹配城市名称或直接使用城市代码
|
||||
city_code = None
|
||||
if city_input in CITY_CODES:
|
||||
city_code = CITY_CODES[city_input]
|
||||
elif re.match(r"^\d{9}$", city_input):
|
||||
city_code = city_input
|
||||
else:
|
||||
# 尝试模糊匹配城市名称
|
||||
for name, code in CITY_CODES.items():
|
||||
if city_input in name:
|
||||
city_code = code
|
||||
break
|
||||
|
||||
if not city_code:
|
||||
await event.reply(f"未找到城市 '{city_input}'，请检查城市名称或使用城市代码。")
|
||||
return
|
||||
|
||||
# 获取天气数据
|
||||
await event.reply("正在查询天气信息，请稍候...")
|
||||
weather_info = get_weather_data(city_code)
|
||||
|
||||
if not weather_info or not weather_info.get("weather_data"):
|
||||
await event.reply("获取天气信息失败，请稍后重试。")
|
||||
return
|
||||
|
||||
try:
|
||||
# 渲染HTML模板为图片
|
||||
base64_image = await image_manager.render_template_to_base64(
|
||||
"weather.html", weather_info, output_name="weather.png"
|
||||
)
|
||||
|
||||
if base64_image:
|
||||
# 发送图片消息
|
||||
await event.reply(MessageSegment.image(base64_image))
|
||||
else:
|
||||
await event.reply("生成天气图片失败，请稍后重试。")
|
||||
|
||||
except Exception as e:
|
||||
logger.error(f"渲染天气图片失败: {e}")
|
||||
await event.reply("生成天气图片时发生错误，请稍后重试。")
|
||||
Reference in New Issue
Block a user