246 lines
8.5 KiB
Python
246 lines
8.5 KiB
Python
|
import random
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
import httpx
|
|||
|
|
import asyncio
|
|||
|
|
import io
|
|||
|
|
import json
|
|||
|
|
import os
|
|||
|
|
import re
|
|||
|
|
from nonebot import logger
|
|||
|
|
from io import BytesIO
|
|||
|
|
from datetime import datetime
|
|||
|
|
from PIL import Image, ImageOps, ImageDraw, ImageFont
|
|||
|
|
from nonebot.internal.params import ArgPlainText
|
|||
|
|
from nonebot.typing import T_State
|
|||
|
|
from playwright.async_api import async_playwright
|
|||
|
|
from nonebot import on_command
|
|||
|
|
from nonebot.adapters.onebot.v11 import MessageEvent, Bot, MessageSegment
|
|||
|
|
from nonebot.plugin import PluginMetadata
|
|||
|
|
from typing import Optional, Union
|
|||
|
|
from .utils import *
|
|||
|
|
from nonebot.matcher import Matcher
|
|||
|
|
|
|||
|
|
__plugin_meta__ = PluginMetadata(
|
|||
|
|
name="绝地潜兵小助手",
|
|||
|
|
description="绝地潜兵小助手,提供一些小功能，让超级地球更加美好！",
|
|||
|
|
usage="简报：获取星系战争简要概况\n"
|
|||
|
|
"随机战备：让机器人帮你随机一套战备",
|
|||
|
|
type="application",
|
|||
|
|
homepage="https://github.com/sansenhoshi/nonebot_plugin_helldivers_tools"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
basic_path = os.path.dirname(__file__)
|
|||
|
|
save_path = os.path.join(basic_path, "temp")
|
|||
|
|
img_path = os.path.join(basic_path, "img")
|
|||
|
|
data_path = os.path.join(basic_path, "data")
|
|||
|
|
|
|||
|
|
|
|||
|
|
war_situation = on_command("简报", aliases={"简报"})
|
|||
|
|
|
|||
|
|
se_situation = on_command("超级地球战况", aliases={"超级地球战况"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
@war_situation.handle()
|
|||
|
|
async def get_war_info(ev: MessageEvent):
|
|||
|
|
await war_situation.send("正在获取前线战况……\n本地化需要30s左右，请民主的等待")
|
|||
|
|
url1 = r"https://hd2galaxy.com/"
|
|||
|
|
url2 = r"https://helldiverscompanion.com/#"
|
|||
|
|
url3 = r"https://helldiverscompanion.com/#hellpad/planets/super_earth/regions"
|
|||
|
|
time_present1 = get_present_time()
|
|||
|
|
result = await screen_shot(url2, time_present1)
|
|||
|
|
if result != "success":
|
|||
|
|
await war_situation.finish(MessageSegment.reply(ev.message_id) + result)
|
|||
|
|
img_path1 = os.path.join(save_path, f"{time_present1}.png")
|
|||
|
|
logger.info(img_path1)
|
|||
|
|
images = gen_ms_img(Image.open(img_path1))
|
|||
|
|
mes = (MessageSegment.reply(ev.message_id), images)
|
|||
|
|
await war_situation.send(mes)
|
|||
|
|
os.remove(img_path1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
@se_situation.handle()
|
|||
|
|
async def get_se_info(ev: MessageEvent):
|
|||
|
|
await war_situation.send("正在获取……\n本地化需要30s左右，请民主的等待")
|
|||
|
|
url1 = r"https://hd2galaxy.com/"
|
|||
|
|
url2 = r"https://helldiverscompanion.com/#"
|
|||
|
|
url3 = r"https://helldiverscompanion.com/#hellpad/planets/super_earth/regions"
|
|||
|
|
time_present1 = get_present_time()
|
|||
|
|
result = await screen_shot_2(url3, time_present1)
|
|||
|
|
if result != "success":
|
|||
|
|
await war_situation.finish(MessageSegment.reply(ev.message_id) + result)
|
|||
|
|
img_path1 = os.path.join(save_path, f"{time_present1}.png")
|
|||
|
|
logger.info(img_path1)
|
|||
|
|
images = gen_ms_img(Image.open(img_path1))
|
|||
|
|
mes = (MessageSegment.reply(ev.message_id), images)
|
|||
|
|
await war_situation.send(mes)
|
|||
|
|
os.remove(img_path1)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def download_url(url: str) -> bytes:
|
|||
|
|
async with httpx.AsyncClient() as client:
|
|||
|
|
for i in range(3):
|
|||
|
|
try:
|
|||
|
|
resp = await client.get(url)
|
|||
|
|
if resp.status_code != 200:
|
|||
|
|
continue
|
|||
|
|
return resp.content
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Error downloading {url}, retry {i}/3: {str(e)}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
random_helldivers = on_command("随机战备", aliases={"随机战备"}, block=True)
|
|||
|
|
|
|||
|
|
PROMPT = """ ***超级地球武装部***
|
|||
|
|
请发送你需要的随机规则
|
|||
|
|
|
|||
|
|
1：纯随机（不推荐）
|
|||
|
|
|
|||
|
|
套路随机(按一定规则随机)
|
|||
|
|
|
|||
|
|
2：2红/1蓝/1绿
|
|||
|
|
3：2绿/1红/1蓝
|
|||
|
|
4：2蓝/1绿/1红
|
|||
|
|
|
|||
|
|
5：3红/1蓝
|
|||
|
|
6：3绿/1蓝
|
|||
|
|
|
|||
|
|
7：2红/2蓝
|
|||
|
|
8：2蓝/2绿
|
|||
|
|
9：2绿/2红
|
|||
|
|
|
|||
|
|
10：4红
|
|||
|
|
11：4绿"""
|
|||
|
|
|
|||
|
|
@random_helldivers.got("pick_type", prompt=PROMPT)
|
|||
|
|
|
|||
|
|
# 用户选择
|
|||
|
|
async def got_random_helldivers(event: MessageEvent, pick_type: str = ArgPlainText()):
|
|||
|
|
logger.info(f"用户选择的战备类型: {pick_type}")
|
|||
|
|
if not is_number(pick_type):
|
|||
|
|
await random_helldivers.reject(f"您输入的 {pick_type} 非数字，请重新输入1到11，或者输入0退出")
|
|||
|
|
elif int(pick_type) not in range(12):
|
|||
|
|
await random_helldivers.reject(f"您输入的 {pick_type} 不在范围内，请重新输入1到11，或者输入0退出")
|
|||
|
|
elif int(pick_type) == 0:
|
|||
|
|
logger.info("用户选择退出随机战备")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
mix_msg = (MessageSegment.reply(event.message_id),)
|
|||
|
|
|
|||
|
|
type_combinations = {
|
|||
|
|
2: {'red': 2, 'blue': 1, 'green': 1},
|
|||
|
|
3: {'green': 2, 'red': 1, 'blue': 1},
|
|||
|
|
4: {'blue': 2, 'green': 1, 'red': 1},
|
|||
|
|
5: {'red': 3, 'blue': 1},
|
|||
|
|
6: {'green': 3, 'blue': 1},
|
|||
|
|
7: {'red': 2, 'blue': 2},
|
|||
|
|
8: {'blue': 2, 'green': 2},
|
|||
|
|
9: {'green': 2, 'red': 2},
|
|||
|
|
10: {'red': 4},
|
|||
|
|
11: {'green': 4}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if int(pick_type) == 1:
|
|||
|
|
logger.info("用户选择纯随机战备")
|
|||
|
|
result = await get_random_equipment(4) # 4 random equipments
|
|||
|
|
else:
|
|||
|
|
combination = type_combinations.get(int(pick_type))
|
|||
|
|
logger.info(f"用户选择的战备组合: {combination}")
|
|||
|
|
result = await get_equipment_by_combination(combination)
|
|||
|
|
|
|||
|
|
final_msg = MessageSegment.text("您的随机结果是：\n")
|
|||
|
|
img_base_str = pic2b64(Image.open(result))
|
|||
|
|
image_turple = MessageSegment.image(img_base_str)
|
|||
|
|
mix_msg += (final_msg, image_turple)
|
|||
|
|
|
|||
|
|
await random_helldivers.finish(mix_msg)
|
|||
|
|
|
|||
|
|
async def get_random_equipment(count):
|
|||
|
|
data_config = os.path.join(basic_path, "data")
|
|||
|
|
with open(data_config + "/equipment.json", "r", encoding="utf-8") as file:
|
|||
|
|
data = json.load(file)
|
|||
|
|
|
|||
|
|
indices = select_random_equipment(len(data), count)
|
|||
|
|
selected_equipment = [data[i] for i in indices]
|
|||
|
|
|
|||
|
|
return create_image(selected_equipment)
|
|||
|
|
|
|||
|
|
async def get_equipment_by_combination(type_combination):
|
|||
|
|
data_config = os.path.join(basic_path, "data")
|
|||
|
|
with open(data_config + "/equipment.json", "r", encoding="utf-8") as file:
|
|||
|
|
data = json.load(file)
|
|||
|
|
|
|||
|
|
equipment_by_type = categorize_equipment_by_type(data)
|
|||
|
|
selected_equipment = select_equipment_by_type(equipment_by_type, type_combination)
|
|||
|
|
|
|||
|
|
logger.debug(f"根据组合选择的装备: {selected_equipment}")
|
|||
|
|
return create_image(selected_equipment)
|
|||
|
|
|
|||
|
|
def categorize_equipment_by_type(data):
|
|||
|
|
equipment_by_type = {}
|
|||
|
|
for item in data:
|
|||
|
|
equip_type = item['type']
|
|||
|
|
if equip_type not in equipment_by_type:
|
|||
|
|
equipment_by_type[equip_type] = []
|
|||
|
|
equipment_by_type[equip_type].append(item)
|
|||
|
|
return equipment_by_type
|
|||
|
|
|
|||
|
|
def select_equipment_by_type(equipment_by_type, type_combination):
|
|||
|
|
selected_equipment = []
|
|||
|
|
backpack_count = 0
|
|||
|
|
|
|||
|
|
for equip_type, count in type_combination.items():
|
|||
|
|
if equip_type in equipment_by_type:
|
|||
|
|
available_items = equipment_by_type[equip_type]
|
|||
|
|
random.shuffle(available_items)
|
|||
|
|
selected = 0
|
|||
|
|
for item in available_items:
|
|||
|
|
if selected < count:
|
|||
|
|
if item['backpack'] == 1 and backpack_count >= 1:
|
|||
|
|
continue
|
|||
|
|
selected_equipment.append(item)
|
|||
|
|
selected += 1
|
|||
|
|
if item['backpack'] == 1:
|
|||
|
|
backpack_count += 1
|
|||
|
|
else:
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
return selected_equipment
|
|||
|
|
|
|||
|
|
def select_random_equipment(max_equipment, count):
|
|||
|
|
return random.sample(range(max_equipment), count)
|
|||
|
|
|
|||
|
|
def create_image(selected_equipment):
|
|||
|
|
new_img = Image.new('RGBA', (800, 500), (0, 0, 0, 1000))
|
|||
|
|
logo_path = img_path + "/super earth.png"
|
|||
|
|
logo = Image.open(logo_path)
|
|||
|
|
new_img = image_paste(logo, new_img, (682, 20))
|
|||
|
|
draw = ImageDraw.Draw(new_img)
|
|||
|
|
ch_text_font = ImageFont.truetype(data_path + '/font/msyh.ttc', 36)
|
|||
|
|
pos_horizon = 20
|
|||
|
|
|
|||
|
|
for equipment in selected_equipment:
|
|||
|
|
name = equipment['name']
|
|||
|
|
path = basic_path + "/" + equipment['path'].replace("\\", "/")
|
|||
|
|
icon = Image.open(path)
|
|||
|
|
icon = icon.resize((100, 100))
|
|||
|
|
new_img = image_paste(icon, new_img, (20, pos_horizon))
|
|||
|
|
draw.text((140, pos_horizon + 5), '战备名称：', fill='white', font=ch_text_font)
|
|||
|
|
draw.text((140, pos_horizon + 50), f'{name}', fill='white', font=ch_text_font)
|
|||
|
|
pos_horizon += 120
|
|||
|
|
|
|||
|
|
b_io = BytesIO()
|
|||
|
|
new_img = ImageOps.expand(new_img, border=10, fill="white")
|
|||
|
|
new_img.save(b_io, format="PNG")
|
|||
|
|
return b_io
|
|||
|
|
|
|||
|
|
def image_paste(paste_image, under_image, pos):
|
|||
|
|
if paste_image.mode == 'RGBA':
|
|||
|
|
under_image.paste(paste_image, pos, mask=paste_image.split()[3])
|
|||
|
|
else:
|
|||
|
|
under_image.paste(paste_image, pos)
|
|||
|
|
return under_image
|
|||
|
|
|
|||
|
|
def is_number(s):
|
|||
|
|
return bool(re.match(r'^[0-9]+$', s))
|