nonebot-plugin-l4d2-server 0.6.6__py3-none-any.whl → 1.0.0a1__py3-none-any.whl

nonebot_plugin_l4d2_server/l4d2_file/utils.py

@@ -1,104 +0,0 @@
-import io
-import zipfile
-from pathlib import Path
-from typing import Callable, Dict, List
-from zipfile import ZipFile
-
-import rarfile
-from nonebot.log import logger
-from pyunpack import Archive
-
-from ..l4d2_utils.config import systems
-from ..l4d2_utils.utils import get_file, get_vpk
-
-
-async def updown_l4d2_vpk(map_paths: Path, name: str, url: str):
-    """从url下载压缩包并解压到位置"""
-    original_vpk_files = get_vpk(map_paths)
-    down_file = Path(map_paths, name)
-    if await get_file(url, down_file) is None:
-        return None
-    msg = open_packet(name, down_file)
-    logger.info(msg)
-
-    extracted_vpk_files = get_vpk(map_paths)
-    # 获取新增vpk文件的list
-    return list(set(extracted_vpk_files) - set(original_vpk_files))
-
-
-SUPPORTED_EXTENSIONS = (".zip", ".7z", ".rar")
-
-
-def unzip_zipfile(down_file: Path, down_path: Path):
-    """解压zip文件"""
-    with support_gbk(zipfile.ZipFile(down_file, "r")) as z:
-        z.extractall(down_path)
-    down_file.unlink()
-
-
-def unpack_7zfile(down_file: Path, down_path: Path):
-    """解压7z文件"""
-    Archive(str(down_file)).extractall(str(down_path))
-    down_file.unlink()
-
-
-def unpack_rarfile(down_file: Path, down_path: Path):
-    """解压rar文件"""
-    with rarfile.RarFile(down_file, "r") as z:
-        z.extractall(down_path)
-    down_file.unlink()
-
-
-def open_packet(name: str, down_file: Path) -> str:
-    """解压压缩包"""
-    down_path = down_file.parent
-    logger.info("文件名为：" + name)
-    logger.info(f"系统为{systems}")
-
-    if name.endswith(".vpk"):
-        return "vpk文件已下载"
-
-    for ext in SUPPORTED_EXTENSIONS:
-        if name.endswith(ext):
-            mes = f"{ext[1:]}文件已下载,正在解压"
-            unpack_funcs: Dict[str, Callable] = {
-                ".zip": unzip_zipfile,
-                ".7z": unpack_7zfile,
-                ".rar": unpack_rarfile,
-            }
-            unpack_func = unpack_funcs.get(ext)
-            if not unpack_func:
-                raise ValueError(f"不支持的拓展名: {ext}")
-            unpack_func(down_file, down_path)
-            return mes
-
-    raise ValueError(f"不支持的文件: {name}")
-
-
-def support_gbk(zip_file: ZipFile):
-    """
-    压缩包中文恢复
-    """
-    if type(zip_file) == ZipFile:
-        name_to_info = zip_file.NameToInfo
-        # copy map first
-        for name, info in name_to_info.copy().items():
-            real_name = name.encode("cp437").decode("gbk")
-            if real_name != name:
-                info.filename = real_name
-                del name_to_info[name]
-                name_to_info[real_name] = info
-    return zip_file
-
-
-async def all_zip_to_one(data_list: List[bytes]):
-    """多压缩包文件合并"""
-    file_list = [io.BytesIO(data).getbuffer() for data in data_list]
-    data_file = io.BytesIO()
-
-    with ZipFile(data_file, mode="w") as zf:
-        for i, file in enumerate(file_list):
-            filename = f"file{i}.zip"
-            zf.writestr(filename, file)
-
-    return data_file.getbuffer()

nonebot_plugin_l4d2_server/l4d2_image/htmlimg.py

@@ -1,18 +0,0 @@
-class DATAPANDS:
-    def __init__(self, data_dict: dict) -> None:
-        self.new_data = {}
-        self.data_dict = data_dict
-        for key in data_dict:
-            self.dict_pan(key)
-
-    def dict_pan(self, key):
-        """dict转化为图像所需要的dict量化"""
-        # 删除不必要的数据
-        if key == "刷特模式":
-            self.data_dict.pop(key)
-        elif key == "游戏模式" or key == "特感数量" or key == "刷新间隔":
-            for item in self.data_dict[key]:
-                if item in self.new_data:
-                    self.new_data[item] += 1
-                else:
-                    self.new_data[item] = 1

nonebot_plugin_l4d2_server/l4d2_image/images.py

@@ -1,92 +0,0 @@
-import sys
-from base64 import b64encode
-from io import BytesIO
-from pathlib import Path
-from typing import Union, overload
-
-import aiofiles
-from PIL import Image
-
-MAIN_PATH = Path() / "data" / "GenshinUID"
-sys.path.append(str(MAIN_PATH))
-CONFIG_PATH = MAIN_PATH / "config.json"
-RESOURCE_PATH = MAIN_PATH / "resource"
-WIKI_PATH = MAIN_PATH / "wiki"
-CU_BG_PATH = MAIN_PATH / "bg"
-CU_CHBG_PATH = MAIN_PATH / "chbg"
-WEAPON_PATH = RESOURCE_PATH / "weapon"
-GACHA_IMG_PATH = RESOURCE_PATH / "gacha_img"
-CHAR_PATH = RESOURCE_PATH / "chars"
-CHAR_STAND_PATH = RESOURCE_PATH / "char_stand"
-CHAR_SIDE_PATH = RESOURCE_PATH / "char_side"
-CHAR_NAMECARD_PATH = RESOURCE_PATH / "char_namecard"
-REL_PATH = RESOURCE_PATH / "reliquaries"
-ICON_PATH = RESOURCE_PATH / "icon"
-TEMP_PATH = RESOURCE_PATH / "temp"
-CARD_PATH = RESOURCE_PATH / "card"
-GUIDE_PATH = WIKI_PATH / "guide"
-TEXT2D_PATH = Path(__file__).parents[2] / "resource" / "texture2d"
-
-PLAYER_PATH = MAIN_PATH / "players"
-
-
-TEXT2D_PATH = Path(__file__).parents[2] / "resource" / "texture2d"
-FETTER_PATH = TEXT2D_PATH / "fetter"
-TALENT_PATH = TEXT2D_PATH / "talent"
-WEAPON_BG_PATH = TEXT2D_PATH / "weapon"
-WEAPON_AFFIX_PATH = TEXT2D_PATH / "weapon_affix"
-LEVEL_PATH = TEXT2D_PATH / "level"
-
-BG_PATH = Path(__file__).parent / "bg"
-TEXT_PATH = Path(__file__).parent / "texture2d"
-ring_pic = Image.open(TEXT_PATH / "ring.png")
-mask_pic = Image.open(TEXT_PATH / "mask.png")
-NM_BG_PATH = BG_PATH / "nm_bg"
-SP_BG_PATH = BG_PATH / "sp_bg"
-
-bg_path = CU_BG_PATH if list(CU_BG_PATH.iterdir()) != [] else NM_BG_PATH
-
-
-@overload
-async def convert_img(img: Image.Image, is_base64: bool = False) -> bytes: ...
-
-
-@overload
-async def convert_img(img: Image.Image, is_base64: bool = True) -> str: ...
-
-
-@overload
-async def convert_img(img: bytes, is_base64: bool = False) -> str: ...
-
-
-@overload
-async def convert_img(img: Path, is_base64: bool = False) -> str: ...
-
-
-async def convert_img(
-    img: Union[Image.Image, str, Path, bytes],
-    is_base64: bool = False,
-):
-    """
-    :说明:
-      将PIL.Image对象转换为bytes或者base64格式。
-    :参数:
-      * img (Image): 图片。
-      * is_base64 (bool): 是否转换为base64格式, 不填默认转为bytes。
-    :返回:
-      * res: bytes对象或base64编码图片。
-    """
-    if isinstance(img, Image.Image):
-        img = img.convert("RGB")
-        result_buffer = BytesIO()
-        img.save(result_buffer, format="PNG", quality=80, subsampling=0)
-        res = result_buffer.getvalue()
-        if is_base64:
-            res = "base64://" + b64encode(res).decode()
-        return res
-    if isinstance(img, bytes):
-        pass
-    else:
-        async with aiofiles.open(img, "rb") as fp:
-            img = await fp.read()
-    return f"[CQ:image,file=base64://{b64encode(img).decode()}]"

nonebot_plugin_l4d2_server/l4d2_image/one.py

@@ -1,44 +0,0 @@
-from pathlib import Path
-from typing import List
-
-from PIL import Image
-
-# 半透明素材
-half_whitel_image_path = Path(__file__).parent.parent.joinpath("data/img/white.png")
-half_whitel_image = Image.open(half_whitel_image_path)
-
-
-async def one_server_img(msg: dict):
-    """单个服务器的dict信息变成图片"""
-    img_background = await adjust_image_size(msg["Players"])
-    img_background = await adjust_server_name(img_background, msg["name"])
-
-
-async def adjust_image_size(text_list):
-    """初始化背景"""
-    initial_width = 1080
-    initial_height = 600
-    height_increment = 100
-
-    num_players = len(text_list.get("Player", []))
-    final_height = initial_height + num_players * height_increment
-
-    return Image.new("RGBA", (initial_width, final_height), "white")
-
-
-async def adjust_server_name(image: Image.Image, name: str):
-    """写标题"""
-    if image and name:
-        ...
-
-
-async def adjust_ping(image: Image.Image, ping: str):
-    """写ping"""
-    if image and ping:
-        ...
-
-
-async def adjust_player(image: Image.Image, player: List[dict]):
-    """写玩家"""
-    if image and player:
-        ...

nonebot_plugin_l4d2_server/l4d2_image/send_image_tool.py

@@ -1,32 +0,0 @@
-from base64 import b64encode
-from io import BytesIO
-from pathlib import Path
-from typing import Union
-
-from PIL import Image
-
-
-async def convert_img(
-    img: Union[Image.Image, str, Path, bytes],
-    is_base64: bool = False,
-):
-    """
-    :说明:
-      将PIL.Image对象转换为bytes或者base64格式。
-    :参数:
-      * img (Image): 图片。
-      * is_base64 (bool): 是否转换为base64格式, 不填默认转为bytes。
-    :返回:
-      * res: bytes对象或base64编码图片。
-    """
-    if isinstance(img, Image.Image):
-        img = img.convert("RGBA")
-        result_buffer = BytesIO()
-        img.save(result_buffer, format="PNG", quality=80, subsampling=0)
-        res = result_buffer.getvalue()
-        if is_base64:
-            res = b64encode(res).decode()
-        return res
-    if isinstance(img, bytes):
-        return b64encode(img).decode()
-    return f"[CQ:image,file=file:///{img!s}]"

nonebot_plugin_l4d2_server/l4d2_image/steam.py

@@ -1,63 +0,0 @@
-# from PIL import Image
-# from bs4 import BeautifulSoup
-from urllib.parse import unquote
-
-import aiohttp
-import httpx
-
-headers = {
-    "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0",  # noqa: E501
-}
-
-
-async def url_to_byte(url: str):
-    """获取URL数据的字节流"""
-
-    async with aiohttp.ClientSession() as session:
-        async with session.get(url, headers=headers, timeout=600) as response:
-            if response.status == 200:
-                return await response.read()
-            return None
-
-
-async def url_to_msg(url: str):
-    """获取URL数据的字节流"""
-
-    async with aiohttp.ClientSession() as session:
-        async with session.get(url, headers=headers, timeout=600) as response:
-            if response.status == 200:
-                return await response.text()
-            return None
-
-
-async def url_to_byte_name(url: str, filename: str = ""):
-    """获取URL数据的字节流"""
-
-    if filename == "htp":
-        response = httpx.get(url, headers=headers, timeout=600)  # noqa: ASYNC100
-        content_disposition = response.headers.get("Content-Disposition")
-        if not content_disposition:
-            return None
-        if "''" in content_disposition:
-            file_name = content_disposition.split("''")[-1]
-        else:
-            file_name = content_disposition
-        file_name = unquote(file_name)
-        if response.content and file_name:
-            return [response.content, file_name]
-        return None
-    async with aiohttp.ClientSession() as session:
-        async with session.get(url, headers=headers, timeout=600) as response:
-            content_disposition = response.headers.get("Content-Disposition")
-            if not content_disposition:
-                return None
-            if "''" in content_disposition:
-                file_name = content_disposition.split("''")[-1]
-            else:
-                file_name = content_disposition
-            if not file_name:
-                file_name = "anyone"
-            file_name = unquote(file_name)
-            if response.status == 200:
-                return [await response.read(), file_name]
-            return None

nonebot_plugin_l4d2_server/l4d2_push/__init__.py

@@ -1,225 +0,0 @@
-from pathlib import Path
-from typing import Dict, Optional, Union
-
-import a2s
-
-try:
-    import ujson as json
-except ImportError:
-    import json
-
-from nonebot import get_bot, get_driver, on_command, require
-from nonebot.adapters.onebot.v11 import GroupMessageEvent, Message, MessageSegment
-from nonebot.adapters.onebot.v11.permission import GROUP_ADMIN, GROUP_OWNER
-from nonebot.log import logger
-from nonebot.matcher import Matcher
-from nonebot.params import CommandArg
-from nonebot.permission import SUPERUSER
-from nonebot_plugin_saa import MessageFactory
-
-from ..l4d2_queries import get_ip_to_mes
-from ..l4d2_queries.utils import json_server_to_tag_dict
-from ..l4d2_utils.config import l4_config
-from ..l4d2_utils.utils import extract_last_digit, split_maohao
-
-require("nonebot_plugin_apscheduler")
-try:
-    scheduler = require("nonebot_plugin_apscheduler").scheduler
-except Exception:
-    scheduler = None
-driver = get_driver()
-sch_json = Path("data/L4D2/scheduler.json")
-if not sch_json.exists():
-    with sch_json.open("w") as f:
-        json.dump({}, f, ensure_ascii=False)
-
-add_rss = on_command(
-    "add_rss",
-    aliases={"求生定时添加"},
-    priority=30,
-    permission=SUPERUSER | GROUP_ADMIN | GROUP_OWNER,
-)
-del_rss = on_command(
-    "del_rss",
-    aliases={"求生定时删除"},
-    priority=30,
-    permission=SUPERUSER | GROUP_ADMIN | GROUP_OWNER,
-)
-
-
-@add_rss.handle()
-async def _(event: GroupMessageEvent, matcher: Matcher, args: Message = CommandArg()):
-    group_id = event.group_id
-    if not group_id:
-        return
-    msg = args.extract_plain_text()
-    command, message = await extract_last_digit(msg)
-    push_msg = await get_ip_to_mes(msg=message, command=command)
-    if not push_msg:
-        return
-    if push_msg in ["服务器无响应", None]:
-        await matcher.finish("无响应的服务器，请检查")
-    else:
-        return_msg = await add_or_update_data(group_id, msg)
-        await MessageFactory(push_msg).send()
-        # if isinstance(push_msg, bytes):
-        #     await matcher.finish(MessageSegment.image(push_msg))
-        # elif isinstance(push_msg, Union[Message, MessageSegment]):
-        #     await matcher.finish(push_msg)
-        # else:
-        #     await matcher.send(push_msg)
-        if return_msg == "add":
-            await matcher.send(f"已添加群定时任务【{msg}】{l4_config.l4_push_times}次")
-        elif return_msg in ["update", "change"]:
-            await matcher.send(f"已更新群定时任务【{msg}】{l4_config.l4_push_times}次")
-
-
-@del_rss.handle()
-async def _(event: GroupMessageEvent, matcher: Matcher):
-    group_id = event.group_id
-    if not group_id:
-        return
-    await add_or_update_data(group_id, "", ad_mode="del")
-    await matcher.finish("已删除群定时任务")
-
-
-async def add_or_update_data(group_i: int, some_str: str = "", ad_mode: str = "add"):
-    """添加或者删除定时任务
-    mode == [new,update,del,change]
-    """
-    group_id = str(group_i)
-    sch_json = Path("data/L4D2/scheduler.json")
-    if ad_mode == "add":
-        if sch_json.exists():
-            with sch_json.open(encoding="utf-8") as f:
-                scheduler_data = json.load(f)
-            try:
-                msg_dict = scheduler_data[group_id]
-                # times = msg_dict["times"]
-                old_msg = msg_dict["msg"]
-                scheduler_data[group_id] = {
-                    "times": l4_config.l4_push_times,
-                    "msg": some_str,
-                }
-                mode = "update" if old_msg == some_str else "change"
-            except Exception:
-                scheduler_data[group_id] = {
-                    "times": l4_config.l4_push_times,
-                    "msg": some_str,
-                }
-                mode = "new"
-
-        else:
-            scheduler_data = {
-                group_id: {"times": l4_config.l4_push_times, "msg": some_str},
-            }
-            mode = "new"
-
-        with sch_json.open("w", encoding="utf-8") as f:
-            json.dump(scheduler_data, f, ensure_ascii=False)
-
-    else:
-        if sch_json.exists():
-            with sch_json.open() as f:
-                scheduler_data: Dict[str, Dict[str, Union[str, int]]] = json.load(f)
-            try:
-                msg_dict = scheduler_data[group_id]
-                # times = msg_dict["times"]
-                old_msg = msg_dict["msg"]
-                scheduler_data[group_id] = {"times": 0, "msg": old_msg}
-            except Exception:
-                scheduler_data[group_id] = {"times": 0, "msg": some_str}
-        else:
-            scheduler_data = {group_id: {"times": 0, "msg": some_str}}
-        mode = "del"
-
-        with sch_json.open("w", encoding="utf-8") as f:
-            json.dump(scheduler_data, f, ensure_ascii=False)
-
-    return mode
-
-
-async def rss_ip():
-    """推送一次"""
-    sch_json = Path("data/L4D2/scheduler.json")
-
-    if sch_json.exists():
-        with sch_json.open(encoding="utf-8") as f:
-            scheduler_data: Dict[str, Dict[str, Union[int, str]]] = json.load(f)
-
-            for key, value in scheduler_data.items():
-                try:
-                    recipient_id = int(key)
-                    count = value["times"]
-                    msg = str(value["msg"])
-                    count = int(count)
-
-                    if count > 0:
-                        msg_read = await send_message(recipient_id, msg, value)
-                        if msg_read and isinstance(msg_read, str):
-                            scheduler_data[key]["ip_detail"] = msg_read
-                        count -= 1
-
-                    scheduler_data[key]["times"] = count
-                except TypeError:
-                    continue
-
-        with sch_json.open(mode="w", encoding="utf-8") as f:
-            json.dump(scheduler_data, f, ensure_ascii=False)
-
-
-async def send_message(
-    recipient_id: int,
-    msg: str,
-    value: Optional[Dict[str, Union[int, str]]] = None,
-):
-    # 执行发送消息的操作，参数可以根据需要进行传递和使用
-    command, message = await extract_last_digit(msg)
-    push_msg = await get_ip_to_mes(msg=message, command=command)
-    if isinstance(push_msg, bytes):
-        await get_bot().send_group_msg(
-            group_id=recipient_id,
-            message=MessageSegment.image(push_msg),
-        )
-        return None
-    if msg and isinstance(push_msg, str):
-        # 单服务器
-        message = await json_server_to_tag_dict(msg, command)
-        if len(message) == 0 or not value:
-            # 关键词不匹配，忽略
-            return None
-        try:
-            old_msg = value.get("ip_detail", {})
-            if not isinstance(old_msg, dict):
-                return None
-            ip = str(message["ip"])
-            host, port = split_maohao(ip)
-            msg_: a2s.SourceInfo = await a2s.ainfo((host, port))
-            value["map_"] = msg_.map_name
-            value["rank_players"] = f"{msg_.player_count}/{msg_.max_players}"
-            if (
-                old_msg["map_"] == value["map_"]
-                and old_msg["rank_players"] == value["rank_players"]
-            ):
-                logger.info(f"{msg}{command}人数和地图未发生变化")
-            else:
-                await get_bot().send_group_msg(group_id=recipient_id, message=push_msg)
-        except Exception as e:
-            logger.warning(e)
-
-        return value
-    return None
-
-
-async def server_is_change():
-    """检测服务器是否发生变化"""
-
-
-if scheduler:
-    logger.success("已成功启动求生定时推送")
-    scheduler.add_job(
-        rss_ip,
-        "interval",
-        minutes=l4_config.l4_push_interval,
-        id="rss_ip",
-    )