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

nonebot_plugin_l4d2_server/__init__.py

@@ -17,15 +17,15 @@
 
 from nonebot import get_driver, require
 
-require("nonebot_plugin_htmlrender")  # noqa: F401
+require("nonebot_plugin_htmlrender")
 require("nonebot_plugin_alconna")
+require("nonebot_plugin_tortoise_orm")
 
-from nonebot import require
-from nonebot.plugin import PluginMetadata, inherit_supported_adapters
+from nonebot.plugin import PluginMetadata, inherit_supported_adapters  # noqa: E402
 
 from . import __main__ as __main__  # noqa: E402
 from .config import ConfigModel  # noqa: E402
-from .l4_help import __version__
+from .l4_help import __version__  # noqa: E402
 
 driver = get_driver()
 

nonebot_plugin_l4d2_server/__main__.py

@@ -15,25 +15,37 @@
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 """
 
-from typing import Optional
+from typing import TYPE_CHECKING, List, Optional
 
-from nonebot import require
 from nonebot.adapters import Message
 from nonebot.log import logger
 from nonebot.matcher import Matcher
 from nonebot.params import CommandArg, CommandStart, RawCommand
 from nonebot.plugin import on_command
-
-require("nonebot_plugin_alconna")
 from nonebot_plugin_alconna import UniMessage
 
+from .config import config
 from .l4_help import get_l4d2_core_help
-from .l4_request import COMMAND, get_ip_server, get_server_detail, reload_ip
+from .l4_request import (
+    COMMAND,
+    get_all_server_detail,
+    get_ip_server,
+    get_server_detail,
+    reload_ip,
+)
+from .utils.api.request import L4API
+
+if TYPE_CHECKING:
+    from .utils.api.models import OutServer
+
+reload_ip()
 
-l4_help = on_command("l4帮助", aliases={"l4help", "l4d2帮助"})
-l4_request = on_command("anne", aliases=COMMAND)
-l4_reload = on_command("l4重载", aliases={"l4刷新"})
+l4_help = on_command("l4help", aliases={"l4帮助", "l4d2帮助"})
+l4_request = on_command("anne", aliases=COMMAND, priority=10)
+l4_reload = on_command("l4reload", aliases={"l4刷新,l4重载"})
+l4_all = on_command("l4all", aliases={"l4全服"})
 l4_connect = on_command("connect", aliases={"l4连接"})
+l4_find_player = on_command("l4find", aliases={"l4查找"})
 
 
 @l4_help.handle()
@@ -41,7 +53,6 @@ async def _(matcher: Matcher):
     """帮助"""
     logger.info("开始执行[l4d2帮助]")
     im = await get_l4d2_core_help()
-    print(type(im))
     if isinstance(im, str):
         await matcher.finish(im)
     await UniMessage.image(raw=im).send()
@@ -65,6 +76,8 @@ async def _(
     if command == "anne":
         command = "云"
     _id: Optional[str] = args.extract_plain_text()
+    if _id is not None and not _id.isdigit() and _id:
+        return
     if not _id:
         _id = None
     logger.info(f"组:{command} ;数字:{_id}")
@@ -72,9 +85,34 @@ async def _(
     if msg is not None:
         if isinstance(msg, str):
             await UniMessage.text(msg).finish()
-        await UniMessage.image(raw=msg).finish()
+        if isinstance(msg, bytes):
+            await UniMessage.image(raw=msg).finish()
     else:
-        await UniMessage.text("没有这个服呢").send()
+        await UniMessage.text("服务器无响应").finish()
+
+
+@l4_find_player.handle()
+async def _(
+    args: Message = CommandArg(),
+):
+    msg: str = args.extract_plain_text().strip()
+    tag_list: List[str] = msg.split(" ", maxsplit=1)
+    if len(tag_list) < 2:
+        return await UniMessage.text("格式错误，正确格式：/l4find 组名 玩家名").finish()
+    group, name = tag_list
+    out: List[OutServer] = await get_server_detail(group, is_img=False)  # type: ignore
+    out_msg = "未找到玩家"
+    for one in out:
+        for player in one["player"]:
+            if name in player.name:
+                out_msg = await get_ip_server(f"{one['host']}:{one['port']}")
+
+    return await UniMessage.text(out_msg).finish()
+
+
+@l4_all.handle()
+async def _():
+    await UniMessage.text(await get_all_server_detail()).finish()
 
 
 @l4_connect.handle()
@@ -84,7 +122,19 @@ async def _(args: Message = CommandArg()):
         await UniMessage.text(await get_ip_server(ip)).finish()
 
 
+# anne部分
+if config.l4_anne:
+    logger.info("加载anne功能")
+    from .l4_anne import *  # noqa: F403
+
+
 @l4_reload.handle()
-async def _():
-    reload_ip()
-    logger.success("重载ip完成")
+async def _(args: Message = CommandArg()):
+    arg = args.extract_plain_text().strip()
+    print(arg)
+    if not arg:
+        reload_ip()
+        logger.success("重载ip完成")
+        await UniMessage.text("重载ip完成").finish()
+    else:
+        print(await L4API.get_sourceban())

nonebot_plugin_l4d2_server/config.py

@@ -5,6 +5,7 @@ from pydantic import BaseModel
 
 DATAPATH = Path(__file__).parent.joinpath("data")
 DATAOUT = Path("data/L4D2")
+print(DATAOUT.absolute())
 server_all_path = DATAOUT / "l4d2"
 server_all_path.mkdir(parents=True, exist_ok=True)
 
@@ -12,7 +13,7 @@ ICONPATH = DATAPATH / "icon"
 
 
 class ConfigModel(BaseModel):
-    l4_anne: bool = False
+    l4_anne: bool = True
     """是否启用anne电信服相关功能"""
     l4_enable: bool = True
     """是否全局启用求生功能"""
@@ -25,6 +26,10 @@ class ConfigModel(BaseModel):
     l4_players: int = 4
     """查询总图的时候展示的用户数量"""
     l4_style: str = "default"
+    """图片风格"""
+    l4_font: str = str(Path(__file__).parent.joinpath("data/font/loli.ttf"))
+    """字体"""
+    l4_show_ip: bool = True
 
 
 config = get_plugin_config(ConfigModel)

nonebot_plugin_l4d2_server/l4_anne/__init__.py

@@ -0,0 +1,124 @@
+from nonebot import log as log, on_command  # noqa: N999
+from nonebot.adapters import Event, Message
+from nonebot.log import logger
+from nonebot.params import CommandArg
+from nonebot_plugin_alconna import UniMessage
+
+from ..utils.api.request import L4API
+from ..utils.database.models import SteamUser
+from ..utils.utils import get_message_at
+from .ranne import get_anne_rank_out
+
+anne_bind = on_command("Banne", aliases={"l4绑定", "anne绑定"}, priority=5, block=True)
+anne_search = on_command("Sanne", aliases={"anne搜索"})
+anne_rank = on_command("Ranne", aliases={"anne成绩"}, block=True, priority=1)
+
+anne_del = on_command(
+    "Danne",
+    aliases={"l4删除", "anne删除", "l4解绑", "anne解绑"},
+    priority=5,
+    block=True,
+)
+
+
+@anne_search.handle()
+async def _(args: Message = CommandArg()):
+    name: str = args.extract_plain_text().strip()
+    print(name)
+    user_list = await L4API.get_anne_steamid(name)
+    if user_list is None:
+        await UniMessage.text("未找到玩家").finish()
+    msg = f"---有{len(user_list)}个玩家---"
+    for index, user in enumerate(user_list, start=1):
+        if index >= 10:
+            break
+        msg += f"""
+        {index}. {user["name"]} | [{user["score"]}] | {user["play_time"]}
+        {user["steamid"]}
+        """
+    if msg:
+        await UniMessage.text("\n".join(msg.splitlines())).finish()
+    else:
+        await UniMessage.text("没有找到玩家").finish()
+
+
+@anne_bind.handle()
+async def _(ev: Event, args: Message = CommandArg()):
+    arg: str = args.extract_plain_text()
+    if not arg:
+        await UniMessage.text("虚空绑定？").finish()
+
+    user = await SteamUser.get_or_none(userid=int(ev.get_user_id()))
+    if user is None:
+        user = await SteamUser.create(userid=int(ev.get_user_id()))
+
+    if len(arg) == 17:
+        # steamid64
+        logger.info(f"SteamID64:{arg}")
+        user.SteamID64 = arg
+        msg = "绑定steamid64"
+
+    elif arg.startswith("STEAM_"):
+        # steamid
+        logger.info(f"SteamID:{arg}")
+        user.SteamID = arg
+        msg = "绑定steamid"
+
+    else:
+        # name
+        logger.info(f"Name:{arg}")
+        user.Name = arg
+        msg = "绑定名字"
+    await user.save()
+
+    return await UniMessage.text(f"{msg}成功").finish()
+
+
+@anne_del.handle()
+async def _(ev: Event):
+    if record := await SteamUser.get_or_none(userid=int(ev.get_user_id())):
+        logger.info(f"删除用户:{record}")
+        await record.delete()
+        await record.save()
+        return await UniMessage.text("删除成功").finish()
+    return await UniMessage.text("没有绑定信息呢").finish()
+
+
+@anne_rank.handle()
+async def _(ev: Event, args: Message = CommandArg()):
+    uid = await get_message_at(str(ev.json()))
+    if uid is None:
+        uid = int(int(ev.get_user_id()))
+    steamid = ""
+    arg: str = args.extract_plain_text().strip()
+    logger.info(f"arg:{arg}")
+    # 优先从数据库查询
+    if not arg:
+        msg = await SteamUser.get_or_none(userid=uid)
+        if msg is not None:
+            steamid = msg.SteamID
+            if not steamid:
+                name = msg.Name
+                if not name:
+                    await UniMessage.text("未绑定名字/steamid").finish()
+                msg_dict = await L4API.get_anne_steamid(name)
+                if not msg_dict:
+                    await UniMessage.text("绑定的昵称找不到呢").finish()
+                steamid = msg_dict[0]["steamid"]
+            logger.info(f"steamid:{steamid}")
+
+    # 再从arg中查找
+    else:
+        if arg.startswith("STEAM_"):
+            steamid = arg
+        else:
+            arg_dict = await L4API.get_anne_steamid(arg)
+            if not arg_dict:
+                await UniMessage.text("未找到该昵称玩家").finish()
+            steamid = arg_dict[0]["steamid"]
+    if not steamid:
+        await UniMessage.text("未找到玩家,请使用指令`l4搜索`查找").finish()
+    out_msg = await get_anne_rank_out(steamid)
+    if out_msg is None:
+        await UniMessage.text("未找到玩家").finish()
+    await UniMessage.text(out_msg).finish()

nonebot_plugin_l4d2_server/l4_anne/ranne.py

@@ -0,0 +1,18 @@
+# coding=utf-8
+# anne战绩查询
+from ..utils.api.request import L4API
+
+
+async def get_anne_rank_out(steamid: str):
+    msg = await L4API.get_anne_playerdetail(steamid)
+    if msg is None:
+        return None
+    return f"""电信anne查询结果：
+    昵称：{msg['info']['name']}
+    排名：{msg['detail']['rank']}
+    分数：{msg['detail']["source"]}
+    击杀：{msg['detail']["kills"]}
+    爆头率：{msg['detail']["avg_source"]}
+    时间：{msg['info']['playtime']}
+    上次：{msg['info']['lasttime']}
+    """

nonebot_plugin_l4d2_server/l4_help/Help.json

@@ -3,59 +3,96 @@
     "desc": "无需任何权限",
     "data": [
       {
-        "name": "l4全服",
-        "desc": "返回可以查询的所有服务器组",
-        "eg": "l4全服",
+        "name": "(组)",
+        "desc": "返回一个组的服务器信息",
+        "eg": "/云",
         "need_ck": false,
         "need_sk": false,
         "need_admin": false
       },
       {
-        "name": "(组指令)",
-        "desc": "返回一个组的服务器信息",
-        "eg": "云",
+        "name": "(组)(id)",
+        "desc": "查询单服信息",
+        "eg": "/云1",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "l4查找(组)(玩家)",
+        "desc": "查询用户所在服务器状态",
+        "eg": "/l4查找 云 爱丽数码",
         "need_ck": false,
         "need_sk": false,
         "need_admin": false
       },
       {
-        "name": "雀魂切换uid",
-        "desc": "切换你的雀魂UID",
-        "eg": "雀魂切换uid123456",
+        "name": "l4全服",
+        "desc": "返回可以查询的所有服务器组(未完成)",
+        "eg": "/l4全服",
         "need_ck": false,
         "need_sk": false,
         "need_admin": false
       }
     ]
   },
-  "查询服务": {
-    "desc": "查询一些基本信息, 需要在金之间拥有一定场次的对局",
+  "电信服anne服务": {
+    "desc": "电信anne服相关服务",
     "data": [
       {
-        "name": "雀魂搜索",
-        "desc": "搜索雀魂玩家",
-        "eg": "雀魂搜索Wuyi",
-        "need_ck": true,
+        "name": "Banne xx",
+        "desc": "绑定昵称/steamid",
+        "eg": "/Banne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "Ranne(玩家)",
+        "desc": "查询anne个人信息,没有参数查自己",
+        "eg": "/Ranne 爱丽数码",
+        "need_ck": false,
         "need_sk": false,
         "need_admin": false
       },
       {
-        "name": "雀魂查询",
-        "desc": "查询雀魂数据一览",
-        "eg": "雀魂查询1148850",
-        "need_ck": true,
+        "name": "Sanne(玩家)",
+        "desc": "按昵称查询anne玩家信息",
+        "eg": "/Sanne 爱丽数码",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": false
+      },
+      {
+        "name": "Danne(玩家)",
+        "desc": "删除anne玩家信息",
+        "eg": "/Danne 爱丽数码",
+        "need_ck": false,
         "need_sk": false,
         "need_admin": false
       }
     ]
   },
+  "管理员服务": {
+    "desc": "管理可以使用的指令",
+    "data": [
+      {
+        "name": "l4刷新",
+        "desc": "重载存储的ip数据",
+        "eg": "l4刷新",
+        "need_ck": false,
+        "need_sk": false,
+        "need_admin": true
+      }
+    ]
+  },
   "其他服务": {
     "desc": "一些杂项服务",
     "data": [
       {
-        "name": "雀魂帮助",
-        "desc": "呼出该帮助界面",
-        "eg": "雀魂帮助",
+        "name": "connect",
+        "desc": "直接发送求生链接返回信息",
+        "eg": "connect 127.0.0.1:27015",
         "need_ck": false,
         "need_sk": false,
         "need_admin": false

nonebot_plugin_l4d2_server/l4_help/__init__.py

@@ -10,7 +10,7 @@ from ..l4_image.convert import core_font
 from ..l4_image.model import PluginHelp
 from .draw import get_help
 
-__version__ = "1.0.0a1"
+__version__ = "1.0.0a2"
 TEXT_PATH = Path(__file__).parent / "texture2d"
 HELP_DATA = Path(__file__).parent / "Help.json"
 
@@ -39,7 +39,7 @@ async def get_l4d2_core_help() -> Union[bytes, str]:
     bg_out.paste(bg_new, None, bg_new)
 
     return await get_help(
-        "CS2UID",
+        "L4插件",
         f"版本号:{__version__}",
         help_data,
         bg_out,

nonebot_plugin_l4d2_server/l4_image/__init__.py

@@ -1,15 +1,15 @@
-from typing import List
+from typing import List  # noqa:
 
 from ..utils.api.models import OutServer
 from .html_img import server_ip_pic
 
 
-async def msg_to_image(server_dict: List[OutServer], mode="html"):
+async def msg_to_image(server_dict: List[OutServer], mode_: str = "html"):
     """信息构造图片"""
-    if mode == "html":
+    if mode_ == "html":
         """用浏览器作图"""
         return await server_ip_pic(server_dict)
-    if mode == "pil":
+    if mode_ == "pil":
         """用pil作图"""
         return None
     """返回文字"""

nonebot_plugin_l4d2_server/l4_image/anne_pil.py

@@ -0,0 +1,17 @@
+from pathlib import Path
+
+from nonebot_plugin_l4d2_server.utils.api.models import AnnePlayer2
+from PIL import Image, ImageFont
+
+from ..config import config
+
+font = ImageFont.truetype(config.l4_font)
+
+anne_path = Path(__file__).parent / "img" / "anne"
+
+
+async def anne_player_info(msg: AnnePlayer2):
+    back_img = Image.open(anne_path / "back1.jpg")
+    base_img = Image.new("RGBA", (back_img.size), (255, 255, 255, 50))
+    back_img.paste(base_img, (0, 0), base_img)
+    return msg

nonebot_plugin_l4d2_server/l4_image/convert.py

@@ -169,7 +169,7 @@ async def text2pic(text: str, max_size: int = 800, font_size: int = 24):
         core_font(font_size),
         "black",
         max_size - 80,
-        True,
+        True,  # noqa: FBT003
     )
     img = img.crop((0, 0, max_size, int(y + 80)))
     return await convert_img(img)

nonebot_plugin_l4d2_server/l4_image/download.py

@@ -33,7 +33,7 @@ async def download_url(url: str) -> bytes:
 async def download_head(user_id: str) -> bytes:
     url = f"http://q1.qlogo.cn/g?b=qq&nk={user_id}&s=640"
     data = await download_url(url)
-    if hashlib.md5(data).hexdigest() == "acef72340ac0e914090bd35799f5594e":
+    if hashlib.md5(data).hexdigest() == "acef72340ac0e914090bd35799f5594e":  # noqa: S324
         url = f"http://q1.qlogo.cn/g?b=qq&nk={user_id}&s=100"
         data = await download_url(url)
     return data
@@ -50,7 +50,7 @@ def square_to_circle(im: ImageS):
     return im
 
 
-async def get_head_by_user_id_and_save(user_id):
+async def get_head_by_user_id_and_save(user_id: str):
     """qq转头像"""
     user_id = str(user_id)
 
@@ -61,7 +61,7 @@ async def get_head_by_user_id_and_save(user_id):
         os.listdir(default_header_path),
     )
     default_head = default_head_path / random.choice(os.listdir(default_head_path))
-    ## im头像 im2头像框 im3合成
+    # im头像 im2头像框 im3合成
     if user_head_path.exists():
         logger.info("使用本地头像")
         im = Image.open(user_head_path).resize((280, 280)).convert("RGBA")
@@ -86,8 +86,8 @@ async def get_head_by_user_id_and_save(user_id):
                 return None
     im2 = Image.open(default_head).resize((450, 450)).convert("RGBA")
     im3 = Image.new("RGBA", im2.size, (255, 255, 255, 0))
-    r, g, b, a1 = im.split()
-    r, g, b, a2 = im2.split()
+    _, _, _, a1 = im.split()
+    _, _, _, a2 = im2.split()
     im = square_to_circle(im)
     im3.paste(im, (75, 75), mask=a1)
     im3.paste(im2, mask=a2)