feat: 添加每周总结功能
- 新增 get_weekly_stats_for_discord() 获取过去7天统计数据 - 新增 generate_weekly_summary_for_discord() 生成周报 - 新增 generate_weekly_summary_image() 生成周报图片 - 新增 weekly_summary.html 模板 - 添加每周日21:00定时推送 - 添加 /weekly_summary 命令手动触发
This commit is contained in:
OpenClaw Bot
parent
7f81574192
commit
8708c931c8
@ -140,6 +140,22 @@ async def deactivate_friend(ctx, steam_id):
|
||||
else:
|
||||
await ctx.respond(content=f'找不到 {steam_id}')
|
||||
|
||||
@bot.command(description='获取本周统计总结', name='weekly_summary')
|
||||
async def get_weekly_summary(ctx):
|
||||
"""手动触发每周总结"""
|
||||
await ctx.defer()
|
||||
logger.info("Manual weekly summary requested")
|
||||
try:
|
||||
data = await dota.generate_weekly_summary_for_discord()
|
||||
if data:
|
||||
await ctx.respond(content=data['content'], embeds=[discord.Embed.from_dict(embed) for embed in data['embeds']])
|
||||
else:
|
||||
await ctx.respond(content='本周暂无数据')
|
||||
except Exception as e:
|
||||
logger.error(f"Error generating weekly summary: {e}")
|
||||
await ctx.respond(content=f'生成周报失败: {str(e)}')
|
||||
|
||||
|
||||
@bot.command(description='启用朋友', name='activate_friend')
|
||||
async def activate_friend(ctx, steam_id):
|
||||
logger.info(f'activate_friend {steam_id}')
|
||||
@ -178,6 +194,35 @@ async def before_check_rank_changes():
|
||||
seconds_until_target = (target_time - now).total_seconds()
|
||||
await asyncio.sleep(seconds_until_target)
|
||||
|
||||
|
||||
@tasks.loop(hours=168) # 每7天运行一次
|
||||
async def weekly_summary(channel):
|
||||
"""每周总结任务"""
|
||||
logger.info("Generating weekly summary")
|
||||
try:
|
||||
data = await dota.generate_weekly_summary_for_discord()
|
||||
if data:
|
||||
logger.info(f"Sending weekly summary")
|
||||
await channel.send(content=data['content'], embeds=[discord.Embed.from_dict(embed) for embed in data['embeds']])
|
||||
except Exception as e:
|
||||
logger.error(f"Error generating weekly summary: {e}")
|
||||
sentry_sdk.capture_exception(e)
|
||||
|
||||
|
||||
@weekly_summary.before_loop
|
||||
async def before_weekly_summary():
|
||||
# 等待到周日晚上9点再开始第一次运行
|
||||
now = datetime.datetime.now()
|
||||
# 计算下一个周日的21:00
|
||||
days_until_sunday = (6 - now.weekday()) % 7
|
||||
if days_until_sunday == 0 and now.hour >= 21:
|
||||
days_until_sunday = 7
|
||||
target_time = (now + datetime.timedelta(days=days_until_sunday)).replace(hour=21, minute=0, second=0, microsecond=0)
|
||||
|
||||
seconds_until_target = (target_time - now).total_seconds()
|
||||
logger.info(f"Weekly summary will start in {seconds_until_target} seconds")
|
||||
await asyncio.sleep(seconds_until_target)
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
logger.info(f"We have logged in as {bot.user}")
|
||||
@ -189,4 +234,7 @@ async def on_ready():
|
||||
# 启动天梯检查任务
|
||||
check_rank_changes.start(channel)
|
||||
|
||||
# 启动每周总结任务
|
||||
weekly_summary.start(channel)
|
||||
|
||||
bot.run('MTE1MjE2NTc3NDMwNDIyMzI2Mg.GEi-17.VvuIkRy_cFD9XF6wtTagY95LKEbTxKaxy-FxGw') # 这里替换成你自己的 token
|
||||
|
||||
126
dota.py
126
dota.py
@ -604,3 +604,129 @@ def check_streaks():
|
||||
streak_updates = []
|
||||
|
||||
return notifications
|
||||
|
||||
|
||||
def get_weekly_stats_for_discord():
|
||||
"""
|
||||
获取过去7天的统计数据，返回Discord格式的消息
|
||||
"""
|
||||
import json
|
||||
|
||||
# 获取7天前的日期
|
||||
week_ago = datetime.datetime.now() - datetime.timedelta(days=7)
|
||||
|
||||
weekly_data = []
|
||||
|
||||
for friend in Friend.filter(active=True):
|
||||
# 获取该玩家最近20场比赛（从中筛选过去7天的）
|
||||
recent_matches = friend.get_recent_matches(limit=20)
|
||||
|
||||
# 筛选过去7天的比赛
|
||||
matches_this_week = []
|
||||
for match in recent_matches:
|
||||
match_time = datetime.datetime.fromtimestamp(match.start_time)
|
||||
if match_time >= week_ago:
|
||||
matches_this_week.append(match)
|
||||
|
||||
if not matches_this_week:
|
||||
continue
|
||||
|
||||
# 统计数据
|
||||
wins = 0
|
||||
losses = 0
|
||||
total_kills = 0
|
||||
total_deaths = 0
|
||||
total_assists = 0
|
||||
hero_stats = {} # {hero_id: {'wins': 0, 'losses': 0, 'games': 0}}
|
||||
|
||||
for match in matches_this_week:
|
||||
player_won = match.radiant_win == (match.player_slot < 128)
|
||||
if player_won:
|
||||
wins += 1
|
||||
else:
|
||||
losses += 1
|
||||
|
||||
total_kills += match.kills
|
||||
total_deaths += match.deaths
|
||||
total_assists += match.assists
|
||||
|
||||
# 英雄统计
|
||||
hero_id = match.hero_id
|
||||
if hero_id not in hero_stats:
|
||||
hero_stats[hero_id] = {'wins': 0, 'losses': 0, 'games': 0, 'hero_id': hero_id}
|
||||
hero_stats[hero_id]['games'] += 1
|
||||
if player_won:
|
||||
hero_stats[hero_id]['wins'] += 1
|
||||
else:
|
||||
hero_stats[hero_id]['losses'] += 1
|
||||
|
||||
total_games = wins + losses
|
||||
win_rate = (wins / total_games * 100) if total_games > 0 else 0
|
||||
kda = (total_kills + total_assists) / total_deaths if total_deaths > 0 else (total_kills + total_assists)
|
||||
|
||||
# 找出最常用英雄和最佳英雄
|
||||
most_played = max(hero_stats.values(), key=lambda x: x['games']) if hero_stats else None
|
||||
best_hero = max(
|
||||
[h for h in hero_stats.values() if h['games'] >= 2],
|
||||
key=lambda x: x['wins'] / x['games'] if x['games'] > 0 else 0,
|
||||
default=None
|
||||
)
|
||||
|
||||
weekly_data.append({
|
||||
'name': friend.name,
|
||||
'steam_id': friend.steam_id,
|
||||
'total_games': total_games,
|
||||
'wins': wins,
|
||||
'losses': losses,
|
||||
'win_rate': win_rate,
|
||||
'kda': kda,
|
||||
'total_kills': total_kills,
|
||||
'total_deaths': total_deaths,
|
||||
'total_assists': total_assists,
|
||||
'most_played_hero': most_played,
|
||||
'best_hero': best_hero,
|
||||
'hero_stats': hero_stats
|
||||
})
|
||||
|
||||
return weekly_data
|
||||
|
||||
|
||||
async def generate_weekly_summary_for_discord():
|
||||
"""
|
||||
生成每周总结报告，返回Discord格式的消息
|
||||
"""
|
||||
weekly_data = get_weekly_stats_for_discord()
|
||||
|
||||
if not weekly_data:
|
||||
return None
|
||||
|
||||
# 生成图片
|
||||
image_url = await image_generator.generate_weekly_summary_image(weekly_data)
|
||||
|
||||
content = "## 📊 本周 Dota 战报\n\n"
|
||||
|
||||
# 整体统计
|
||||
total_games = sum(d['total_games'] for d in weekly_data)
|
||||
total_wins = sum(d['wins'] for d in weekly_data)
|
||||
total_losses = sum(d['losses'] for d in weekly_data)
|
||||
|
||||
content += f"本周共打了 **{total_games}** 场比赛，**{total_wins}** 胜 **{total_losses}** 败\n\n"
|
||||
|
||||
# 个人统计
|
||||
for data in weekly_data:
|
||||
win_rate_str = f"{data['win_rate']:.1f}%"
|
||||
kda_str = f"{data['kda']:.2f}"
|
||||
content += f"**{data['name']}**: {data['wins']}胜{data['losses']}败 ({win_rate_str}) KDA {kda_str}\n"
|
||||
|
||||
result = {
|
||||
'content': content,
|
||||
'embeds': []
|
||||
}
|
||||
|
||||
if image_url:
|
||||
result['embeds'].append({
|
||||
'image': {'url': image_url},
|
||||
'color': 3447003
|
||||
})
|
||||
|
||||
return result
|
||||
|
||||
@ -236,3 +236,121 @@ class ImageGenerator:
|
||||
|
||||
return image_url
|
||||
|
||||
async def generate_weekly_summary_image(self, weekly_data):
|
||||
"""
|
||||
生成每周总结报告图片
|
||||
|
||||
Args:
|
||||
weekly_data: dota.py 中 get_weekly_stats_for_discord() 返回的数据列表
|
||||
|
||||
Returns:
|
||||
str: 上传后的图片URL
|
||||
"""
|
||||
import datetime
|
||||
|
||||
# 计算日期范围
|
||||
today = datetime.datetime.now()
|
||||
week_ago = today - datetime.timedelta(days=7)
|
||||
date_range = f"{week_ago.strftime('%m/%d')} - {today.strftime('%m/%d')}"
|
||||
|
||||
# 整体统计
|
||||
total_games = sum(d['total_games'] for d in weekly_data)
|
||||
total_wins = sum(d['wins'] for d in weekly_data)
|
||||
total_losses = sum(d['losses'] for d in weekly_data)
|
||||
overall_win_rate = round((total_wins / total_games * 100), 1) if total_games > 0 else 0
|
||||
|
||||
# 处理每个玩家的数据
|
||||
players = []
|
||||
for data in weekly_data:
|
||||
player_data = {
|
||||
'name': data['name'],
|
||||
'wins': data['wins'],
|
||||
'losses': data['losses'],
|
||||
'win_rate': round(data['win_rate'], 1),
|
||||
'kda': round(data['kda'], 2),
|
||||
'avg_kills': round(data['total_kills'] / data['total_games'], 1) if data['total_games'] > 0 else 0,
|
||||
'avg_deaths': round(data['total_deaths'] / data['total_games'], 1) if data['total_games'] > 0 else 0,
|
||||
'avg_assists': round(data['total_assists'] / data['total_games'], 1) if data['total_games'] > 0 else 0,
|
||||
}
|
||||
|
||||
# 处理最常用英雄
|
||||
if data['most_played_hero']:
|
||||
hero_id = str(data['most_played_hero']['hero_id'])
|
||||
hero_info = self.heroes_data.get(hero_id, {})
|
||||
hero_name = hero_info.get('localized_name', 'Unknown')
|
||||
hero_img_name = hero_info.get('name', '').replace('npc_dota_hero_', '')
|
||||
player_data['most_played_hero'] = {
|
||||
'name': utils.get_hero_chinese_name(hero_name),
|
||||
'img': f"https://cdn.dota2.com/apps/dota2/images/heroes/{hero_img_name}_full.png",
|
||||
'wins': data['most_played_hero']['wins'],
|
||||
'losses': data['most_played_hero']['losses'],
|
||||
'games': data['most_played_hero']['games']
|
||||
}
|
||||
else:
|
||||
player_data['most_played_hero'] = None
|
||||
|
||||
# 处理最佳英雄
|
||||
if data['best_hero']:
|
||||
hero_id = str(data['best_hero']['hero_id'])
|
||||
hero_info = self.heroes_data.get(hero_id, {})
|
||||
hero_name = hero_info.get('localized_name', 'Unknown')
|
||||
hero_img_name = hero_info.get('name', '').replace('npc_dota_hero_', '')
|
||||
win_rate = round(data['best_hero']['wins'] / data['best_hero']['games'] * 100, 1) if data['best_hero']['games'] > 0 else 0
|
||||
player_data['best_hero'] = {
|
||||
'name': utils.get_hero_chinese_name(hero_name),
|
||||
'img': f"https://cdn.dota2.com/apps/dota2/images/heroes/{hero_img_name}_full.png",
|
||||
'wins': data['best_hero']['wins'],
|
||||
'losses': data['best_hero']['losses'],
|
||||
'win_rate': win_rate
|
||||
}
|
||||
else:
|
||||
player_data['best_hero'] = None
|
||||
|
||||
players.append(player_data)
|
||||
|
||||
# 渲染模板
|
||||
template = self.env.get_template('weekly_summary.html')
|
||||
html_content = template.render(
|
||||
date_range=date_range,
|
||||
total_games=total_games,
|
||||
total_wins=total_wins,
|
||||
total_losses=total_losses,
|
||||
overall_win_rate=overall_win_rate,
|
||||
players=players
|
||||
)
|
||||
|
||||
# 使用Playwright生成图片
|
||||
async with async_playwright() as playwright:
|
||||
browser = await playwright.chromium.launch()
|
||||
page = await browser.new_page()
|
||||
await page.set_content(html_content)
|
||||
await page.set_viewport_size({"width": 800, "height": 800})
|
||||
|
||||
# 等待内容完全加载
|
||||
await page.wait_for_timeout(1000)
|
||||
|
||||
# 调整截图高度以适应内容
|
||||
body_height = await page.evaluate('document.body.scrollHeight')
|
||||
body_width = await page.evaluate('document.body.offsetWidth')
|
||||
await page.set_viewport_size({"width": body_width, "height": body_height})
|
||||
|
||||
# 设置更高的设备像素比以获得更清晰的图像
|
||||
await page.evaluate('''() => {
|
||||
window.devicePixelRatio = 2;
|
||||
}''')
|
||||
|
||||
# 截图
|
||||
image_path = f"weekly_summary_{today.strftime('%Y%m%d')}.png"
|
||||
await page.screenshot(path=image_path, full_page=True)
|
||||
await browser.close()
|
||||
|
||||
# 上传图片
|
||||
image_url = utils.upload_image(image_path, image_path)
|
||||
|
||||
# 删除本地文件
|
||||
try:
|
||||
os.remove(image_path)
|
||||
except Exception as e:
|
||||
logger.warning(f"删除本地图片文件失败: {str(e)}")
|
||||
|
||||
return image_url
|
||||
244
templates/weekly_summary.html
Normal file
244
templates/weekly_summary.html
Normal file
@ -0,0 +1,244 @@
|
||||
<!DOCTYPE html>
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="UTF-8" />
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0" />
|
||||
<style>
|
||||
html,
|
||||
body {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
background: #1a1a1a;
|
||||
}
|
||||
body {
|
||||
font-family: "Segoe UI", Arial, sans-serif;
|
||||
color: #ffffff;
|
||||
padding: 20px;
|
||||
box-sizing: border-box;
|
||||
width: 600px;
|
||||
}
|
||||
.header {
|
||||
text-align: center;
|
||||
margin-bottom: 20px;
|
||||
padding-bottom: 15px;
|
||||
border-bottom: 2px solid #4caf50;
|
||||
}
|
||||
.header h1 {
|
||||
margin: 0;
|
||||
font-size: 1.8em;
|
||||
color: #4caf50;
|
||||
}
|
||||
.header .date-range {
|
||||
font-size: 0.9em;
|
||||
color: #aaaaaa;
|
||||
margin-top: 5px;
|
||||
}
|
||||
.summary {
|
||||
background: #2a2a2a;
|
||||
border-radius: 8px;
|
||||
padding: 15px;
|
||||
margin-bottom: 20px;
|
||||
text-align: center;
|
||||
}
|
||||
.summary-stats {
|
||||
display: flex;
|
||||
justify-content: space-around;
|
||||
margin-top: 10px;
|
||||
}
|
||||
.stat-box {
|
||||
text-align: center;
|
||||
}
|
||||
.stat-value {
|
||||
font-size: 2em;
|
||||
font-weight: bold;
|
||||
color: #4caf50;
|
||||
}
|
||||
.stat-label {
|
||||
font-size: 0.8em;
|
||||
color: #aaaaaa;
|
||||
}
|
||||
.player-section {
|
||||
background: #2a2a2a;
|
||||
border-radius: 8px;
|
||||
padding: 15px;
|
||||
margin-bottom: 15px;
|
||||
border-left: 4px solid #4caf50;
|
||||
}
|
||||
.player-header {
|
||||
display: flex;
|
||||
justify-content: space-between;
|
||||
align-items: center;
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
.player-name {
|
||||
font-size: 1.3em;
|
||||
font-weight: bold;
|
||||
}
|
||||
.player-record {
|
||||
font-size: 1em;
|
||||
color: #aaaaaa;
|
||||
}
|
||||
.player-stats {
|
||||
display: flex;
|
||||
gap: 15px;
|
||||
margin-bottom: 10px;
|
||||
}
|
||||
.stat-item {
|
||||
font-size: 0.9em;
|
||||
}
|
||||
.stat-item .label {
|
||||
color: #aaaaaa;
|
||||
}
|
||||
.stat-item .value {
|
||||
font-weight: bold;
|
||||
}
|
||||
.win-rate {
|
||||
color: #4caf50;
|
||||
}
|
||||
.kda {
|
||||
color: #ff9800;
|
||||
}
|
||||
.hero-section {
|
||||
display: flex;
|
||||
gap: 15px;
|
||||
margin-top: 10px;
|
||||
padding-top: 10px;
|
||||
border-top: 1px solid #3a3a3a;
|
||||
}
|
||||
.hero-box {
|
||||
flex: 1;
|
||||
background: #3a3a3a;
|
||||
border-radius: 5px;
|
||||
padding: 10px;
|
||||
}
|
||||
.hero-box-title {
|
||||
font-size: 0.75em;
|
||||
color: #aaaaaa;
|
||||
margin-bottom: 5px;
|
||||
}
|
||||
.hero-info {
|
||||
display: flex;
|
||||
align-items: center;
|
||||
gap: 8px;
|
||||
}
|
||||
.hero-img {
|
||||
width: 40px;
|
||||
height: 22px;
|
||||
border-radius: 3px;
|
||||
object-fit: cover;
|
||||
}
|
||||
.hero-name {
|
||||
font-size: 0.9em;
|
||||
font-weight: bold;
|
||||
}
|
||||
.hero-record {
|
||||
font-size: 0.75em;
|
||||
color: #aaaaaa;
|
||||
}
|
||||
.no-data {
|
||||
color: #666666;
|
||||
font-size: 0.85em;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="header">
|
||||
<h1>📊 本周 Dota 战报</h1>
|
||||
<div class="date-range">{{ date_range }}</div>
|
||||
</div>
|
||||
|
||||
<div class="summary">
|
||||
<div class="summary-stats">
|
||||
<div class="stat-box">
|
||||
<div class="stat-value">{{ total_games }}</div>
|
||||
<div class="stat-label">总局数</div>
|
||||
</div>
|
||||
<div class="stat-box">
|
||||
<div class="stat-value" style="color: #4caf50">{{ total_wins }}</div>
|
||||
<div class="stat-label">胜场</div>
|
||||
</div>
|
||||
<div class="stat-box">
|
||||
<div class="stat-value" style="color: #f44336">{{ total_losses }}</div>
|
||||
<div class="stat-label">败场</div>
|
||||
</div>
|
||||
<div class="stat-box">
|
||||
<div class="stat-value" style="color: #2196f3">{{ overall_win_rate }}%</div>
|
||||
<div class="stat-label">总胜率</div>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
{% for player in players %}
|
||||
<div class="player-section">
|
||||
<div class="player-header">
|
||||
<span class="player-name">{{ player.name }}</span>
|
||||
<span class="player-record"
|
||||
>{{ player.wins }}胜 {{ player.losses }}败</span
|
||||
>
|
||||
</div>
|
||||
<div class="player-stats">
|
||||
<div class="stat-item">
|
||||
<span class="label">胜率: </span>
|
||||
<span class="value win-rate">{{ player.win_rate }}%</span>
|
||||
</div>
|
||||
<div class="stat-item">
|
||||
<span class="label">KDA: </span>
|
||||
<span class="value kda">{{ player.kda }}</span>
|
||||
</div>
|
||||
<div class="stat-item">
|
||||
<span class="label">场均: </span>
|
||||
<span class="value"
|
||||
>{{ player.avg_kills }}/{{ player.avg_deaths }}/{{ player.avg_assists
|
||||
}}</span
|
||||
>
|
||||
</div>
|
||||
</div>
|
||||
<div class="hero-section">
|
||||
<div class="hero-box">
|
||||
<div class="hero-box-title">最常用英雄</div>
|
||||
{% if player.most_played_hero %}
|
||||
<div class="hero-info">
|
||||
<img
|
||||
class="hero-img"
|
||||
src="{{ player.most_played_hero.img }}"
|
||||
alt="{{ player.most_played_hero.name }}"
|
||||
/>
|
||||
<div>
|
||||
<div class="hero-name">{{ player.most_played_hero.name }}</div>
|
||||
<div class="hero-record">
|
||||
{{ player.most_played_hero.wins }}胜 {{
|
||||
player.most_played_hero.losses }}败 ({{ player.most_played_hero.games
|
||||
}}场)
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% else %}
|
||||
<div class="no-data">暂无数据</div>
|
||||
{% endif %}
|
||||
</div>
|
||||
<div class="hero-box">
|
||||
<div class="hero-box-title">最佳英雄 (≥2场)</div>
|
||||
{% if player.best_hero %}
|
||||
<div class="hero-info">
|
||||
<img
|
||||
class="hero-img"
|
||||
src="{{ player.best_hero.img }}"
|
||||
alt="{{ player.best_hero.name }}"
|
||||
/>
|
||||
<div>
|
||||
<div class="hero-name">{{ player.best_hero.name }}</div>
|
||||
<div class="hero-record">
|
||||
{{ player.best_hero.wins }}胜 {{ player.best_hero.losses }}败
|
||||
({{ player.best_hero.win_rate }}%胜率)
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% else %}
|
||||
<div class="no-data">暂无数据</div>
|
||||
{% endif %}
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
{% endfor %}
|
||||
</body>
|
||||
</html>
|
||||
Loading…
x
Reference in New Issue
Block a user