chore: Remove draw_result.py as it is no longer needed for match result visualization
This commit is contained in:
Ching L
parent
44336e5ff0
commit
b5c58f842a
110
draw_result.py
110
draw_result.py
@ -1,110 +0,0 @@
|
|||||||
from PIL import Image, ImageDraw, ImageFont
|
|
||||||
|
|
||||||
icon_width = 256
|
|
||||||
icon_height = 144
|
|
||||||
banner_height = 70
|
|
||||||
|
|
||||||
# 创建空白画布
|
|
||||||
width = icon_width * 5
|
|
||||||
height = icon_height * 2 * 5 + banner_height
|
|
||||||
background_color = (255, 255, 255)
|
|
||||||
image = Image.new("RGB", (width, height), background_color)
|
|
||||||
draw = ImageDraw.Draw(image)
|
|
||||||
|
|
||||||
# 定义字体和颜色
|
|
||||||
font_color = (0, 0, 0)
|
|
||||||
font = ImageFont.truetype("arial.ttf", 34)
|
|
||||||
|
|
||||||
# 定义玩家信息的位置
|
|
||||||
player_info_x = 0
|
|
||||||
player_info_y = banner_height
|
|
||||||
player_info_2_x = player_info_x + icon_width
|
|
||||||
player_info_2_x_end = int(player_info_2_x + icon_width * 1.5)
|
|
||||||
lv_height = int(icon_height / 2)
|
|
||||||
|
|
||||||
# 添加天辉标题
|
|
||||||
radiant_color = "#66bb6a"
|
|
||||||
draw.rectangle([0, 0, player_info_2_x_end, 70], fill=radiant_color)
|
|
||||||
draw.text((20, 20), "天辉", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 添加玩家信息
|
|
||||||
for i in range(5):
|
|
||||||
# 左侧等级信息
|
|
||||||
draw.rectangle([player_info_x, player_info_y, player_info_2_x, player_info_y + lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_x + 10, player_info_y + 10), "LV. 25", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 左侧头像信息
|
|
||||||
icon = Image.open("hero_img/muerta.png")
|
|
||||||
# icon = icon.resize((144, 256))
|
|
||||||
image.paste(icon, (player_info_x, player_info_y+lv_height))
|
|
||||||
|
|
||||||
# 左侧KDA信息
|
|
||||||
draw.rectangle([player_info_x, player_info_y+lv_height+icon_height, player_info_2_x, player_info_y+lv_height+icon_height+lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_x + 10, player_info_y+lv_height+icon_height + 20), "10/0/10", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧玩家名称
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y, player_info_2_x_end, player_info_y + lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + 20), "muerta", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧金钱信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height, player_info_2_x_end, player_info_y + lv_height+lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height + 20), "金钱：10000", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧伤害信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height*2, player_info_2_x_end, player_info_y + lv_height*3], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height*2 +20), "伤害：23456", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧段位信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height*3, player_info_2_x_end, player_info_y + lv_height*4], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height*3 +20), "中军 3", font=font, fill=font_color)
|
|
||||||
|
|
||||||
player_info_y += lv_height * 2 + icon_height
|
|
||||||
|
|
||||||
# 添加夜魇标题
|
|
||||||
dire_color = "#f44336"
|
|
||||||
draw.rectangle([player_info_2_x_end, 0, width, 70], fill=dire_color)
|
|
||||||
draw.text((player_info_2_x_end + 20,20), "夜魇", font=font, fill=(255, 255, 255))
|
|
||||||
|
|
||||||
player_info_x = int(width / 2)
|
|
||||||
player_info_y = 70
|
|
||||||
player_info_2_x = player_info_x + icon_width
|
|
||||||
player_info_2_x_end = player_info_2_x + icon_width * 1.5
|
|
||||||
lv_height = 72
|
|
||||||
|
|
||||||
|
|
||||||
# 添加玩家信息
|
|
||||||
for i in range(5):
|
|
||||||
# 左侧等级信息
|
|
||||||
draw.rectangle([player_info_x, player_info_y, player_info_2_x, player_info_y + lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_x + 10, player_info_y + 10), "LV. 25", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 左侧头像信息
|
|
||||||
icon = Image.open("hero_img/muerta.png")
|
|
||||||
# icon = icon.resize((144, 256))
|
|
||||||
image.paste(icon, (player_info_x, player_info_y+lv_height))
|
|
||||||
|
|
||||||
# 左侧KDA信息
|
|
||||||
draw.rectangle([player_info_x, player_info_y+lv_height+icon_height, player_info_2_x, player_info_y+lv_height+icon_height+lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_x + 10, player_info_y+lv_height+icon_height + 20), "10/0/10", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧玩家名称
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y, player_info_2_x_end, player_info_y + lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + 20), "muerta", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧金钱信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height, player_info_2_x_end, player_info_y + lv_height+lv_height], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height + 20), "金钱：10000", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧伤害信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height*2, player_info_2_x_end, player_info_y + lv_height*3], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height*2 +20), "伤害：23456", font=font, fill=font_color)
|
|
||||||
|
|
||||||
# 右侧段位信息
|
|
||||||
draw.rectangle([player_info_2_x, player_info_y + lv_height*3, player_info_2_x_end, player_info_y + lv_height*4], fill="#808080")
|
|
||||||
draw.text((player_info_2_x, player_info_y + lv_height*3 +20), "中军 3", font=font, fill=font_color)
|
|
||||||
|
|
||||||
player_info_y += lv_height * 2 + icon_height
|
|
||||||
|
|
||||||
|
|
||||||
# 保存图片
|
|
||||||
image.save("dota_match_result.png")
|
|
||||||
125
project-info.md
Normal file
125
project-info.md
Normal file
@ -0,0 +1,125 @@
|
|||||||
|
# DotaBot 项目文档
|
||||||
|
|
||||||
|
## 项目概述
|
||||||
|
|
||||||
|
DotaBot 是一个 Discord 机器人，用于跟踪 Dota 2 玩家的比赛数据并在 Discord 频道中分享这些信息。它可以监控指定玩家的最近比赛，生成比赛报告图片，并通知连胜/连败等特殊事件。
|
||||||
|
|
||||||
|
## 项目架构
|
||||||
|
|
||||||
|
### 核心组件
|
||||||
|
|
||||||
|
1. **Discord Bot**: 基于 py-cord 库实现的 Discord 机器人
|
||||||
|
2. **Dota 2 API 集成**: 使用 OpenDota API 获取比赛数据
|
||||||
|
3. **数据库**: 使用 Peewee ORM 管理本地数据
|
||||||
|
4. **图片生成**: 使用 Playwright 和 Jinja2 生成比赛报告图片
|
||||||
|
5. **图片存储**: 使用 Cloudflare R2 存储生成的图片
|
||||||
|
|
||||||
|
### 文件结构
|
||||||
|
|
||||||
|
```
|
||||||
|
├── discord_bot.py # Discord 机器人主程序
|
||||||
|
├── dota.py # Dota 2 数据处理和模型定义
|
||||||
|
├── image_generator.py # 图片生成模块
|
||||||
|
├── utils.py # 工具函数
|
||||||
|
├── matches.py # 比赛数据处理
|
||||||
|
├── heroes.json # 英雄数据
|
||||||
|
├── templates/ # HTML 模板目录
|
||||||
|
│ ├── match_report.html # 比赛报告模板
|
||||||
|
│ └── recent_matches.html # 最近比赛模板
|
||||||
|
├── env.ini # 配置文件
|
||||||
|
└── dotabot.log # 日志文件
|
||||||
|
```
|
||||||
|
|
||||||
|
## 依赖关系
|
||||||
|
```
|
||||||
|
project-info.md
|
||||||
|
discord_bot.py
|
||||||
|
├── dota.py
|
||||||
|
│ ├── utils.py
|
||||||
|
│ └── image_generator.py
|
||||||
|
└── utils.py
|
||||||
|
image_generator.py
|
||||||
|
├── utils.py
|
||||||
|
├── templates/match_report.html
|
||||||
|
└── templates/recent_matches.html
|
||||||
|
dota.py
|
||||||
|
├── utils.py
|
||||||
|
└── image_generator.py
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## 功能模块
|
||||||
|
|
||||||
|
### 1. Discord Bot (discord_bot.py)
|
||||||
|
|
||||||
|
Discord 机器人的主程序，负责与 Discord 交互，包括：
|
||||||
|
- 定时任务获取最新比赛数据
|
||||||
|
- 处理用户命令
|
||||||
|
- 发送比赛报告和通知
|
||||||
|
|
||||||
|
主要命令：
|
||||||
|
- `/recent_matches [name] [match_count]`: 获取指定玩家的最近比赛
|
||||||
|
- `/list_friends`: 列出所有已添加的好友
|
||||||
|
- `/add_friend [steam_id] [name]`: 添加新好友
|
||||||
|
- `/mod_friend [steam_id] [name]`: 修改好友信息
|
||||||
|
- `/activate_friend [steam_id]`: 启用好友
|
||||||
|
- `/deactivate_friend [steam_id]`: 禁用好友
|
||||||
|
|
||||||
|
### 2. Dota 数据处理 (dota.py)
|
||||||
|
|
||||||
|
处理 Dota 2 比赛数据，定义数据模型，包括：
|
||||||
|
- `Match`: 比赛数据模型
|
||||||
|
- `Friend`: 好友数据模型
|
||||||
|
- 比赛数据序列化
|
||||||
|
- 连胜/连败检测
|
||||||
|
|
||||||
|
主要功能：
|
||||||
|
- `get_friends_recent_matches()`: 获取所有好友的最近比赛
|
||||||
|
- `serialize_match_for_discord()`: 将比赛数据格式化为 Discord 消息
|
||||||
|
- `check_streaks()`: 检查连胜/连败情况
|
||||||
|
|
||||||
|
### 3. 图片生成 (image_generator.py)
|
||||||
|
|
||||||
|
使用 Playwright 和 Jinja2 生成比赛报告图片，包括：
|
||||||
|
- `ImageGenerator` 类：负责生成比赛报告和最近比赛图片
|
||||||
|
- 使用 HTML 模板渲染比赛数据
|
||||||
|
- 使用 Playwright 将 HTML 转换为图片
|
||||||
|
- 上传图片到 Cloudflare R2 存储
|
||||||
|
|
||||||
|
主要方法：
|
||||||
|
- `generate_match_report()`: 生成单场比赛报告图片
|
||||||
|
- `generate_recent_matches_image()`: 生成最近比赛汇总图片
|
||||||
|
|
||||||
|
### 4. 工具函数 (utils.py)
|
||||||
|
|
||||||
|
提供各种辅助功能，包括：
|
||||||
|
- 时间格式转换
|
||||||
|
- 英雄名称中英文转换
|
||||||
|
- 数字格式化
|
||||||
|
- 图片上传到 Cloudflare R2
|
||||||
|
- 日志记录
|
||||||
|
|
||||||
|
主要函数：
|
||||||
|
- `convert_seconds_to_hms()`: 将秒数转换为时分秒
|
||||||
|
- `get_hero_chinese_name()`: 获取英雄的中文名称
|
||||||
|
- `upload_image()`: 上传图片到 Cloudflare R2
|
||||||
|
- `get_ranking()`: 获取天梯段位名称
|
||||||
|
|
||||||
|
## 数据流
|
||||||
|
|
||||||
|
1. **定时任务流程**:
|
||||||
|
- Discord Bot 定时调用 `get_friends_recent_matches()`
|
||||||
|
- 获取所有活跃好友的最新比赛
|
||||||
|
- 对于新比赛，调用 `serialize_match_for_discord()` 格式化数据
|
||||||
|
- 调用 `ImageGenerator.generate_match_report()` 生成比赛报告图片
|
||||||
|
- 发送格式化的比赛数据和图片到 Discord 频道
|
||||||
|
|
||||||
|
2. **用户命令流程**:
|
||||||
|
- 用户发送命令 (如 `/recent_matches`)
|
||||||
|
- Discord Bot 处理命令并调用相应函数
|
||||||
|
- 对于 `/recent_matches`，调用 `Friend.serialize_recent_matches_for_discord()`
|
||||||
|
- 调用 `ImageGenerator.generate_recent_matches_image()` 生成图片
|
||||||
|
- 发送结果到 Discord
|
||||||
|
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user