135 lines
5.4 KiB
Python
135 lines
5.4 KiB
Python
from playwright.sync_api import sync_playwright
|
|
from jinja2 import Environment, FileSystemLoader
|
|
import utils
|
|
import json
|
|
import os
|
|
|
|
|
|
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
|
|
|
|
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生成图片
|
|
with sync_playwright() as playwright:
|
|
browser = playwright.chromium.launch()
|
|
page = browser.new_page()
|
|
page.set_content(html_content)
|
|
page.set_viewport_size({"width": 800, "height": 800})
|
|
|
|
# 等待内容完全加载
|
|
page.wait_for_timeout(1000)
|
|
|
|
# 调整截图高度以适应内容
|
|
body_height = page.evaluate('document.body.scrollHeight')
|
|
body_width = page.evaluate('document.body.offsetWidth')
|
|
page.set_viewport_size({"width": body_width, "height": body_height})
|
|
|
|
# 截图
|
|
image_path = "match_report_%s.png" % match_data.get('match_id')
|
|
page.screenshot(path=image_path, full_page=True)
|
|
browser.close()
|
|
image_url = utils.upload_image(image_path, image_path)
|
|
os.remove(image_path)
|
|
return image_url
|
|
return None |