228 lines
9.0 KiB
Python
228 lines
9.0 KiB
Python
from playwright.async_api import async_playwright
|
|
from jinja2 import Environment, FileSystemLoader
|
|
import utils
|
|
import json
|
|
import os
|
|
import datetime
|
|
from loguru import logger
|
|
|
|
|
|
class ImageGenerator:
|
|
def __init__(self):
|
|
self.env = Environment(loader=FileSystemLoader('templates'))
|
|
# 加载英雄数据
|
|
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))
|
|
if hero_info:
|
|
return f"https://cdn.cloudflare.steamstatic.com{hero_info['img']}"
|
|
return None
|
|
|
|
async def generate_match_report(self, match_data):
|
|
"""
|
|
根据match_data生成比赛报告图片
|
|
match_data应该是dota.py中serialize_match方法返回的格式
|
|
"""
|
|
# 处理数据，标记最高经济和最高伤害的玩家
|
|
radiant_players = []
|
|
dire_players = []
|
|
|
|
# 分离天辉和夜魇玩家
|
|
for player in match_data['players']:
|
|
player_data = {
|
|
'name': player['personaname'] if player['personaname'] else '-',
|
|
'is_friend': bool(player.get('nickname', '')),
|
|
'hero': player['hero'],
|
|
'level': player['level'],
|
|
'kills': player['kills'],
|
|
'deaths': player['deaths'],
|
|
'assists': player['assists'],
|
|
'total_gold': player['total_gold'],
|
|
'hero_damage': player['hero_damage'],
|
|
'hero_img': self.get_hero_image_url(player['hero_id']),
|
|
'highest_gold': False,
|
|
'highest_damage': False,
|
|
'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'])
|
|
player['hero_damage'] = utils.shorten_digits(player['hero_damage'])
|
|
for player in dire_players:
|
|
player['total_gold'] = utils.shorten_digits(player['total_gold'])
|
|
player['hero_damage'] = utils.shorten_digits(player['hero_damage'])
|
|
|
|
end_time = match_data.get('end_time', '')
|
|
# '2025-03-05T01:04:30.000+08:00' get 01:04:30
|
|
end_time = end_time.split('T')[1].split('.')[0]
|
|
|
|
# 准备模板数据
|
|
template_data = {
|
|
'radiant_players': radiant_players,
|
|
'dire_players': dire_players,
|
|
'radiant_score': match_data.get('radiant_score', 0),
|
|
'dire_score': match_data.get('dire_score', 0),
|
|
'duration': match_data.get('duration', 0),
|
|
'radiant_win': match_data.get('radiant_win', False),
|
|
'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})
|
|
|
|
# 截图
|
|
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)
|
|
|
|
# 删除本地文件
|
|
try:
|
|
os.remove(image_path)
|
|
except Exception as e:
|
|
logger.warning(f"删除本地图片文件失败: {str(e)}")
|
|
|
|
return image_url
|
|
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'],
|
|
'deaths': match['deaths'],
|
|
'assists': match['assists'],
|
|
'hero_img': hero_img,
|
|
'hero_name': hero_name,
|
|
'duration_formatted': duration_formatted,
|
|
'end_time_formatted': end_time,
|
|
'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})
|
|
|
|
# 截图
|
|
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)
|
|
|
|
# 删除本地文件
|
|
try:
|
|
os.remove(image_path)
|
|
except Exception as e:
|
|
logger.warning(f"删除本地图片文件失败: {str(e)}")
|
|
|
|
return image_url
|
|
|