生成王者荣耀装备信息Markdown和Excel

Posted 2022-04-11 小小明-代码实体

王者荣耀的装备信息列表网址：https://pvp.qq.com/web201605/item.shtml

经开发者工具检查，装备数据可通过https://pvp.qq.com/web201605/js/item.json接口获取。

获取装备数据

采集并整理数据：

import requests
import pandas as pd

headers = 
    'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (Khtml, like Gecko) '
                  'Chrome/88.0.4324.104 Safari/537.36 '

target = 'https://pvp.qq.com/web201605/js/item.json'
item_list = requests.get(target, headers=headers).json()
item_df = pd.DataFrame(item_list)
item_df.sort_values(["item_type", "price", "item_id"], inplace=True)
item_df.fillna("", inplace=True)
item_df.des1 = item_df.des1.str.replace("</?p>", "", regex=True)
item_df.des2 = item_df.des2.str.replace("</?p>", "", regex=True)
item_df

查看装备图片网址，例如铁剑的网址为：

https://game.gtimg.cn/images/yxzj/img201606/itemimg/1111.jpg

可以发现规律，item_id是图片的名称。

多线程下载图片

import os
from concurrent.futures import ThreadPoolExecutor


def download_img(item_id):
    if os.path.exists(f"imgs/item_id.jpg"):
        return
    imgurl = f"http://game.gtimg.cn/images/yxzj/img201606/itemimg/item_id.jpg"
    res = requests.get(imgurl)
    with open(f"imgs/item_id.jpg", "wb") as f:
        f.write(res.content)


os.makedirs("imgs", exist_ok=True)
with ThreadPoolExecutor(max_workers=8) as executor:
    nums = executor.map(download_img, item_df.item_id)

几秒钟之内已经下载完全部图片：

生成Markdown文档

下面我们生成一下对应的Markdown文档，首先整理一下数据：

item_type_dict = 1: '攻击', 2: '法术', 3: '防御', 4: '移动', 5: '打野', 7: '游走'
item_ids = item_df.item_id.values
item_df.item_id = item_df.item_id.apply(
    lambda item_id: f"![item_id](imgs/item_id.jpg)")
item_df.item_type = item_df.item_type.map(item_type_dict)
item_df.columns = ["图片", "装备名称", "类型", "售价", "总价", "基础描述", "扩展描述"]
item_df

按装备类型分组，生成Markdown文档：

with open("王者装备说明.md", "w") as f:
    for item_type, item_split in item_df.groupby("类型", sort=False):
        f.write(f"# item_type\\n")
        item_split.drop(columns="类型", inplace=True)
        f.write(item_split.to_markdown(index=False))
        f.write("\\n\\n")

生成的Markdown如下：

不过Markdown的表格无法任意调整，图片需要点击后才会放大，下面我们考虑生成Excel表格：

生成Excel

首先整理数据：

item_df.图片 = ""
item_df.基础描述 = item_df.基础描述.str.replace("<br>", "\\n")
item_df.扩展描述 = item_df.扩展描述.str.replace("<br>", "\\n")
item_df

然后写出来Excel表：

from openpyxl.drawing.image import Image
from openpyxl.styles import Alignment

with pd.ExcelWriter("王者装备说明.xlsx", engine='openpyxl') as writer:
    item_df.to_excel(writer, sheet_name='装备说明', index=False)
    worksheet = writer.sheets['装备说明']
    worksheet.column_dimensions["A"].width = 11
    for item_id, (cell,) in zip(item_ids, worksheet.iter_rows(2, None, 1, 1)):
        worksheet.row_dimensions[cell.row].height = 67
        worksheet.add_image(Image(f"imgs/item_id.jpg"), f'Acell.row')
    worksheet.column_dimensions["F"].width = 15
    worksheet.column_dimensions["G"].width = 35
    writer.save()

生成结果如下：

经过一番人工调整后：

大家还可以根据自己想要的效果生成HTML和Word等等。