# import asyncio # import base64 # import io # import json # import os # from datetime import datetime # from nonebot import logger # from PIL import Image, ImageOps, ImageDraw, ImageFont # from nonebot.internal.params import ArgPlainText # from nonebot.params import CommandArg # 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, GroupMessageEvent, Message # from nonebot.plugin import PluginMetadata # from typing import Optional, Union # from nonebot.matcher import Matcher # # __plugin_meta__ = PluginMetadata( # name="DeadLock助手", # description="DeadLock助手，我要成为DeadLock高手", # usage="nekoscore：neko查分 [steam好友代码]\n" # "nekobind：neko绑定 [steam好友代码]\n" # "nekorank：neko排行[TODO]\n" # "nekomatch：dl比赛详情[TODO]\n", # type="application", # extra={ # # } # ) # # neko_score = on_command("nekoscore", aliases={"dl查分"}) # neko_bind = on_command("nekobind", aliases={"dl绑定"}) # neko_unbind = on_command("nekounbind", aliases={"取消dl绑定"}) # neko_rank = on_command("nekorank", aliases={"dl排行"}) # neko_match = on_command("nekomatch", aliases={"dl比赛详情"}) # # 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") # # # @neko_score.handle() # async def get_neko_score(ev: MessageEvent, rgs: Message = CommandArg()): # steam_id = rgs.extract_plain_text() # # 查询绑定情况 # if steam_id is None or steam_id == "" or steam_id.isdigit() is False: # bind_steam_id = await get_neko_bind(ev.group_id, ev.user_id) # if bind_steam_id is None: # await neko_score.finish("请输入正确的steam好友代码！") # return # else: # steam_id = bind_steam_id # await neko_score.send(f"正在获取{steam_id}的数据") # url = f"https://deadlock.blast.tv/users/{steam_id}" # time_present1 = get_present_time() # result = await capture_screenshot(url, time_present1) # if result != "success": # await neko_score.finish(MessageSegment.reply(ev.message_id) + result) # img_path1 = os.path.join(save_path, f"{time_present1}.png") # images = gen_ms_img(Image.open(img_path1)) # mes = (MessageSegment.reply(ev.message_id), images) # await neko_score.send(mes) # os.remove(img_path1) # # # @neko_bind.handle() # async def play_neko_bind(ev: MessageEvent, rgs: Message = CommandArg()): # # 获取群号 # group_id = ev.group_id # steam_id = rgs.extract_plain_text() # if steam_id is None or steam_id == "" or steam_id.isdigit() is False: # await neko_score.finish("请输入正确的steam好友代码！") # return # qq_id = ev.user_id # res = await to_neko_bind(group_id, steam_id, qq_id) # await neko_score.send(res) # # # @neko_unbind.handle() # async def play_neko_unbind(ev: MessageEvent, rgs: Message = CommandArg()): # # 获取群号 # group_id = ev.group_id # steam_id = rgs.extract_plain_text() # qq_id = ev.user_id # res = await to_neko_unbind(group_id, qq_id) # if res is None: # await neko_score.finish(f"{qq_id}未绑定过") # else: # await neko_score.send(f"{qq_id}已经取消与{res}的绑定") # # # async def to_neko_bind(group_id, steam_id, qq_id): # if os.path.exists(f"{data_path}/{group_id}.json"): # with open(f"{data_path}/{group_id}.json", 'r') as file: # json_data = json.load(file) # else: # # 如果文件不存在，创建一个新的空字典 # json_data = {} # # 检查指定的 key 是否存在 # if str(qq_id) in json_data: # # 如果 key 存在，将对应键值更新 # old_steam_id = json_data[str(qq_id)] # json_data[str(qq_id)] = steam_id # # 将更新后的数据写回 JSON 文件 # with open(f"{data_path}/{group_id}.json", 'w') as file: # json.dump(json_data, file, indent=4) # return f"已经将[ {str(qq_id)} ]的绑定[ {old_steam_id} ]更新到了[ {str(steam_id)} ]" # else: # # 如果 key 不存在，添加新的 key-value 对 # json_data[str(qq_id)] = steam_id # # # 将更新后的数据写回 JSON 文件 # with open(f"{data_path}/{group_id}.json", 'w') as file: # json.dump(json_data, file, indent=4) # return f"[ {str(qq_id)} ]已经绑定[ {steam_id} ]" # # # async def get_neko_bind(group_id, qq_id): # if os.path.exists(f"{data_path}/{group_id}.json"): # with open(f"{data_path}/{group_id}.json", 'r') as file: # json_data = json.load(file) # else: # # 如果文件不存在，创建一个新的空字典 # json_data = { # "000000": 000000 # } # # 检查指定的 key 是否存在 # if str(qq_id) in json_data: # return json_data[str(qq_id)] # else: # return None # # # async def to_neko_unbind(group_id, qq_id): # if os.path.exists(f"{data_path}/{group_id}.json"): # with open(f"{data_path}/{group_id}.json", 'r') as file: # json_data = json.load(file) # else: # # 如果文件不存在，创建一个新的空字典 # json_data = { # "000000": 000000 # } # # 检查指定的 key 是否存在 # if str(qq_id) in json_data: # remove_value = json_data.pop(str(qq_id)) # # 将更新后的数据写回 JSON 文件 # with open(f"{data_path}/{group_id}.json", 'w') as file: # json.dump(json_data, file, indent=4) # return remove_value # else: # return None # # # def get_present_time() -> int: # return int(datetime.timestamp(datetime.now())) # # # async def capture_screenshot(url: str, time_present: int): # async with async_playwright() as p: # browser = await p.chromium.launch() # page = await browser.new_page() # try: # # 设置视口大小 # await page.set_viewport_size({"width": 1600, "height": 900}) # await page.goto(url) # await page.wait_for_load_state('networkidle') # # except Exception as e: # return f"访问网站异常{type(e)}`{e}`" # # await asyncio.sleep(1) # i_path = os.path.join(save_path, f'{time_present}.png') # await page.screenshot( # path=i_path, # full_page=True # ) # logger.info("正在压缩图片...") # await asyncio.sleep(2) # img_convert = Image.open(i_path) # img_convert.save(i_path, quality=80) # logger.info("图片保存成功！") # await browser.close() # return "success" # # # def gen_ms_img(image: Union[bytes, Image.Image]) -> MessageSegment: # if isinstance(image, bytes): # return MessageSegment.image( # pic2b64(Image.open(io.BytesIO(image))) # ) # else: # return MessageSegment.image( # pic2b64(image) # ) # # # def pic2b64(pic: Image) -> str: # buf = io.BytesIO() # pic.save(buf, format='PNG') # base64_str = base64.b64encode(buf.getvalue()).decode() # return 'base64://' + base64_str