feat: 添加每周总结功能

- 新增 get_weekly_stats_for_discord() 获取过去7天统计数据
- 新增 generate_weekly_summary_for_discord() 生成周报
- 新增 generate_weekly_summary_image() 生成周报图片
- 新增 weekly_summary.html 模板
- 添加每周日21:00定时推送
- 添加 /weekly_summary 命令手动触发

discord_bot.py

@@ -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}")
@@ -188,5 +233,8 @@ async def on_ready():
   
   # 启动天梯检查任务
   check_rank_changes.start(channel)
+  
+  # 启动每周总结任务
+  weekly_summary.start(channel)
 
 bot.run('MTE1MjE2NTc3NDMwNDIyMzI2Mg.GEi-17.VvuIkRy_cFD9XF6wtTagY95LKEbTxKaxy-FxGw') # 这里替换成你自己的 token

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

image_generator.py

@@ -13,7 +13,7 @@ class ImageGenerator:
         # 加载英雄数据
         with open('heroes.json', 'r', encoding='utf-8') as f:
             self.heroes_data = json.load(f)
-        
+
     def get_hero_image_url(self, hero_id):
         """根据英雄id获取图片URL"""
         hero_info = self.heroes_data.get(str(hero_id))
@@ -29,7 +29,7 @@ class ImageGenerator:
         # 处理数据，标记最高经济和最高伤害的玩家
         radiant_players = []
         dire_players = []
-        
+
         # 分离天辉和夜魇玩家
         for player in match_data['players']:
             player_data = {
@@ -48,42 +48,42 @@ class ImageGenerator:
                 'rank': player.get('rank'),
                 'rank_tier': player.get('rank_tier'),
             }
-            
+
             # 如果是好友，使用昵称
             if player.get('nickname'):
                 player_data['name'] = player['nickname']
-                
+
             # 转换英雄名称为中文
             player_data['hero'] = utils.get_hero_chinese_name(player['hero'])
-                
+
             if player['is_radiant']:
                 radiant_players.append(player_data)
             else:
                 dire_players.append(player_data)
-        
+
         # 找出天辉最高经济和伤害
         if radiant_players:
             max_gold_player = max(radiant_players, key=lambda p: int(p['total_gold']))
             max_damage_player = max(radiant_players, key=lambda p: int(p['hero_damage']))
-            
+
             for player in radiant_players:
                 if player == max_gold_player:
                     player['highest_gold'] = True
                 if player == max_damage_player:
                     player['highest_damage'] = True
-        
+
         # 找出夜魇最高经济和伤害
         if dire_players:
             max_gold_player = max(dire_players, key=lambda p: int(p['total_gold']))
             max_damage_player = max(dire_players, key=lambda p: int(p['hero_damage']))
-            
+
             for player in dire_players:
                 if player == max_gold_player:
                     player['highest_gold'] = True
                 if player == max_damage_player:
                     player['highest_damage'] = True
-        
+
-        
+
         # 短数字
         for player in radiant_players:
             player['total_gold'] = utils.shorten_digits(player['total_gold'])
@@ -107,77 +107,77 @@ class ImageGenerator:
             'start_time': match_data.get('start_time', ''),
             'end_time': end_time,
         }
-        
+
         # 渲染模板
         template = self.env.get_template('match_report.html')
         html_content = template.render(**template_data)
-        
+
         # 使用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"match_report_{match_data.get('match_id')}.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
-        return None 
+        return None
 
     async def generate_recent_matches_image(self, player_name, matches):
         """
         生成玩家最近比赛的图片报告
-        
+
         Args:
             player_name: 玩家名称
             matches: 比赛数据列表
-        
+
         Returns:
             str: 上传后的图片URL
         """
         # 处理比赛数据
         processed_matches = []
-        
+
         for match in matches:
             # 获取英雄图片
             hero_id = str(match['hero_id'])
             hero_img = f"https://cdn.dota2.com/apps/dota2/images/heroes/{self.heroes_data[hero_id]['name'].replace('npc_dota_hero_', '')}_full.png"
-            
+
             # 格式化时间
             end_time = datetime.datetime.fromtimestamp(match['end_time']).strftime('%Y-%m-%d %H:%M')
-            
+
             # 格式化时长
             duration_hms = utils.convert_seconds_to_hms(match['duration'])
             duration_formatted = f"{duration_hms[0]}:{duration_hms[1]:02d}:{duration_hms[2]:02d}"
-            
+
             # 获取英雄中文名
             hero_name = utils.get_hero_chinese_name(self.heroes_data[hero_id]['name'])
-            
+
             processed_match = {
                 'win': match['win'],
                 'kills': match['kills'],
@@ -190,15 +190,134 @@ class ImageGenerator:
                 'party_size': match['party_size'],
                 'average_rank': match['average_rank']
             }
-            
+
             processed_matches.append(processed_match)
-        
+
         # 渲染模板
         template = self.env.get_template('recent_matches.html')
         html_content = template.render(
             player_name=player_name,
             matches=processed_matches
         )
+
+        # 使用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"recent_matches_{player_name.replace(' ', '_')}_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.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 
+
+    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:
@@ -221,7 +340,7 @@ class ImageGenerator:
             }''')
             
             # 截图
-            image_path = f"recent_matches_{player_name.replace(' ', '_')}_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.png"
+            image_path = f"weekly_summary_{today.strftime('%Y%m%d')}.png"
             await page.screenshot(path=image_path, full_page=True)
             await browser.close()
             
@@ -233,6 +352,5 @@ class ImageGenerator:
                 os.remove(image_path)
             except Exception as e:
                 logger.warning(f"删除本地图片文件失败: {str(e)}")
-                
+            
-            return image_url 
+            return image_url
-            

templates/weekly_summary.html

@@ -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>