更新天气插件v1.0.0
This commit is contained in:
baby20162016
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("生成天气图片时发生错误，请稍后重试。")
|
||||||
312
templates/weather.html
Normal file
312
templates/weather.html
Normal file
@@ -0,0 +1,312 @@
|
|||||||
|
<!DOCTYPE html>
|
||||||
|
<html lang="zh-CN">
|
||||||
|
<head>
|
||||||
|
<meta charset="UTF-8">
|
||||||
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||||
|
<title>天气查询结果</title>
|
||||||
|
<style>
|
||||||
|
@import url('https://fonts.googleapis.com/css2?family=JetBrains+Mono:wght@400;500&family=Noto+Sans+SC:wght@400;500;700&display=swap');
|
||||||
|
|
||||||
|
:root {
|
||||||
|
--bg-color: #0f172a; /* 深蓝黑背景 */
|
||||||
|
--window-bg: rgba(30, 41, 59, 0.85); /* 窗口背景 */
|
||||||
|
--border-color: rgba(255, 255, 255, 0.08);
|
||||||
|
--accent: #6366f1; /* 核心强调色 - 靛蓝 */
|
||||||
|
--accent-glow: rgba(99, 102, 241, 0.4);
|
||||||
|
--text-title: #f8fafc;
|
||||||
|
--text-desc: #94a3b8;
|
||||||
|
--text-code: #a5f3fc; /* 代码高亮色 - 浅青 */
|
||||||
|
--card-bg: rgba(0, 0, 0, 0.2);
|
||||||
|
--weather-bg: #0b1120; /* 天气卡片深色背景 */
|
||||||
|
--sunny-color: #fbbf24; /* 晴天橙色 */
|
||||||
|
--cloudy-color: #6b7280; /* 多云灰色 */
|
||||||
|
--rainy-color: #3b82f6; /* 雨天蓝色 */
|
||||||
|
}
|
||||||
|
|
||||||
|
* {
|
||||||
|
margin: 0;
|
||||||
|
padding: 0;
|
||||||
|
box-sizing: border-box;
|
||||||
|
}
|
||||||
|
|
||||||
|
body {
|
||||||
|
font-family: 'Noto Sans SC', system-ui, sans-serif;
|
||||||
|
background-color: var(--bg-color);
|
||||||
|
color: var(--text-title);
|
||||||
|
display: flex;
|
||||||
|
justify-content: center;
|
||||||
|
padding: 0;
|
||||||
|
min-height: 100vh;
|
||||||
|
-webkit-font-smoothing: antialiased;
|
||||||
|
-moz-osx-font-smoothing: grayscale;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 窗口容器 */
|
||||||
|
.window {
|
||||||
|
width: 100%;
|
||||||
|
min-height: 100vh;
|
||||||
|
background: var(--window-bg);
|
||||||
|
backdrop-filter: blur(20px);
|
||||||
|
-webkit-backdrop-filter: blur(20px);
|
||||||
|
border-radius: 0;
|
||||||
|
border: none;
|
||||||
|
box-shadow: none;
|
||||||
|
overflow: auto;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 顶部标题栏 */
|
||||||
|
.header {
|
||||||
|
padding: 32px 40px;
|
||||||
|
border-bottom: 1px solid var(--border-color);
|
||||||
|
background: rgba(255, 255, 255, 0.02);
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.dots {
|
||||||
|
display: flex;
|
||||||
|
gap: 8px;
|
||||||
|
}
|
||||||
|
.dot {
|
||||||
|
width: 12px;
|
||||||
|
height: 12px;
|
||||||
|
border-radius: 50%;
|
||||||
|
}
|
||||||
|
.red { background: #ef4444; }
|
||||||
|
.yellow { background: #f59e0b; }
|
||||||
|
.green { background: #10b981; }
|
||||||
|
|
||||||
|
.title {
|
||||||
|
font-size: 18px;
|
||||||
|
font-weight: 700;
|
||||||
|
letter-spacing: 1.5px;
|
||||||
|
color: var(--text-desc);
|
||||||
|
text-transform: uppercase;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 内容区域 */
|
||||||
|
.content {
|
||||||
|
padding: 40px;
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 32px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.page-title {
|
||||||
|
margin-bottom: 10px;
|
||||||
|
}
|
||||||
|
.page-title h1 {
|
||||||
|
font-size: 48px;
|
||||||
|
background: linear-gradient(to right, #fff, #94a3b8);
|
||||||
|
-webkit-background-clip: text;
|
||||||
|
-webkit-text-fill-color: transparent;
|
||||||
|
}
|
||||||
|
.page-title p {
|
||||||
|
color: var(--text-desc);
|
||||||
|
font-size: 20px;
|
||||||
|
margin-top: 12px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 城市信息 */
|
||||||
|
.city-info {
|
||||||
|
background: var(--card-bg);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
border-radius: 16px;
|
||||||
|
padding: 32px;
|
||||||
|
text-align: center;
|
||||||
|
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.3);
|
||||||
|
}
|
||||||
|
|
||||||
|
.city-name {
|
||||||
|
font-size: 36px;
|
||||||
|
font-weight: 700;
|
||||||
|
color: var(--text-title);
|
||||||
|
margin-bottom: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.query-time {
|
||||||
|
font-size: 16px;
|
||||||
|
color: var(--text-desc);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 天气网格 */
|
||||||
|
.weather-grid {
|
||||||
|
display: grid;
|
||||||
|
grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
|
||||||
|
gap: 24px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 天气卡片 */
|
||||||
|
.weather-card {
|
||||||
|
background: var(--weather-bg);
|
||||||
|
border-radius: 16px;
|
||||||
|
padding: 24px;
|
||||||
|
box-shadow: 0 10px 30px rgba(0, 0, 0, 0.3);
|
||||||
|
border: 1px solid var(--border-color);
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
gap: 16px;
|
||||||
|
transition: transform 0.2s;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-card:hover {
|
||||||
|
transform: translateY(-2px);
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-header {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
align-items: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-day {
|
||||||
|
font-size: 24px;
|
||||||
|
font-weight: 700;
|
||||||
|
color: var(--text-title);
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-condition {
|
||||||
|
font-size: 18px;
|
||||||
|
color: var(--text-desc);
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-main {
|
||||||
|
text-align: center;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-temp {
|
||||||
|
font-size: 48px;
|
||||||
|
font-weight: 900;
|
||||||
|
color: var(--sunny-color);
|
||||||
|
margin: 16px 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-details {
|
||||||
|
display: flex;
|
||||||
|
justify-content: space-between;
|
||||||
|
font-size: 14px;
|
||||||
|
color: var(--text-desc);
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-detail-item {
|
||||||
|
display: flex;
|
||||||
|
flex-direction: column;
|
||||||
|
align-items: center;
|
||||||
|
gap: 4px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-label {
|
||||||
|
font-weight: 600;
|
||||||
|
text-transform: uppercase;
|
||||||
|
letter-spacing: 0.5px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.detail-value {
|
||||||
|
color: var(--text-title);
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 页脚 */
|
||||||
|
.footer {
|
||||||
|
margin-top: auto;
|
||||||
|
padding: 32px;
|
||||||
|
border-top: 1px solid var(--border-color);
|
||||||
|
text-align: center;
|
||||||
|
color: var(--text-desc);
|
||||||
|
font-size: 14px;
|
||||||
|
background: rgba(0, 0, 0, 0.1);
|
||||||
|
}
|
||||||
|
|
||||||
|
.footer .info-row {
|
||||||
|
margin-bottom: 8px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.footer .version-info {
|
||||||
|
font-size: 12px;
|
||||||
|
opacity: 0.8;
|
||||||
|
margin-top: 16px;
|
||||||
|
}
|
||||||
|
|
||||||
|
/* 响应式设计 */
|
||||||
|
@media (max-width: 768px) {
|
||||||
|
.content {
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.page-title h1 {
|
||||||
|
font-size: 36px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-grid {
|
||||||
|
grid-template-columns: 1fr;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-card {
|
||||||
|
padding: 20px;
|
||||||
|
}
|
||||||
|
|
||||||
|
.weather-temp {
|
||||||
|
font-size: 36px;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
</style>
|
||||||
|
</head>
|
||||||
|
<body>
|
||||||
|
<div class="window">
|
||||||
|
<div class="header">
|
||||||
|
<div class="dots">
|
||||||
|
<div class="dot red"></div>
|
||||||
|
<div class="dot yellow"></div>
|
||||||
|
<div class="dot green"></div>
|
||||||
|
</div>
|
||||||
|
<div class="title">天气查询</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="content">
|
||||||
|
<div class="page-title">
|
||||||
|
<h1>天气查询结果</h1>
|
||||||
|
<p>{{ timestamp }}</p>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="city-info">
|
||||||
|
<div class="city-name">{{ city_name }}</div>
|
||||||
|
<div class="query-time">查询时间: {{ query_time }}</div>
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="weather-grid">
|
||||||
|
{% for day_weather in weather_data %}
|
||||||
|
<div class="weather-card">
|
||||||
|
<div class="weather-header">
|
||||||
|
<div class="weather-day">{{ day_weather.day }}</div>
|
||||||
|
<div class="weather-condition">{{ day_weather.weather }}</div>
|
||||||
|
</div>
|
||||||
|
<div class="weather-main">
|
||||||
|
<div class="weather-temp">{{ day_weather.temperature }}</div>
|
||||||
|
</div>
|
||||||
|
<div class="weather-details">
|
||||||
|
<div class="weather-detail-item">
|
||||||
|
<div class="detail-label">风力</div>
|
||||||
|
<div class="detail-value">{{ day_weather.wind_power }}</div>
|
||||||
|
</div>
|
||||||
|
<div class="weather-detail-item">
|
||||||
|
<div class="detail-label">风向</div>
|
||||||
|
<div class="detail-value">{{ day_weather.wind_direction }}</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
{% endfor %}
|
||||||
|
</div>
|
||||||
|
|
||||||
|
<div class="footer">
|
||||||
|
<div class="info-row">数据来源: 中国天气网</div>
|
||||||
|
<div class="info-row">查询城市: {{ city_name }}</div>
|
||||||
|
<div class="version-info">
|
||||||
|
CalglauBot | Powered by NeoBot Framework
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</div>
|
||||||
|
</body>
|
||||||
|
</html></content>
|
||||||
Reference in New Issue
Block a user