nonebot-plugin-l4d2-server 0.3.4a2__py3-none-any.whl → 0.3.5__py3-none-any.whl

nonebot_plugin_l4d2_server/l4d2_anne/anne_telecom.py

@@ -1,76 +1,76 @@
-from ..l4d2_image.steam import url_to_byte
-from bs4 import BeautifulSoup
-from typing import List
+# from ..l4d2_image.steam import url_to_byte
+# from bs4 import BeautifulSoup
+# from typing import List
 
-# 计划获取电信服ip
-class ANNE_API:
+# 暂时废弃
+# class ANNE_API:
     
-    async def __init__(
-        self,
-        STEAMID:str,
-        tag:str
-    ):
-        self.STEAMID = STEAMID
-        if tag == '中':
-            msg1 = await self.anne_msg()
-        elif tag == '长':
-            msg1 = await self.anne_msg()
-            msg1.update(await self.anne_map())
+#     async def __init__(
+#         self,
+#         STEAMID:str,
+#         tag:str
+#     ):
+#         self.STEAMID = STEAMID
+#         if tag == '中':
+#             msg1 = await self.anne_msg()
+#         elif tag == '长':
+#             msg1 = await self.anne_msg()
+#             msg1.update(await self.anne_map())
         
     
-    async def anne_msg(self):
-        """个人资料表"""
-        data_bytes = await url_to_byte('https://sb.trygek.com/l4d_stats/ranking/player.php?steamid={self.STEAMID}')
-        data_bs = BeautifulSoup(data_bytes, 'html.parser')
-        data_fom = data_bs.find_all('table')
-        n = 0
-        data_dict = {}
-        while n < 2:
-            data_list:List[dict] = []
-            detail2 = data_fom[n]
-            tr = detail2.find_all('tr')
-            for i in tr:
-                title = i.find('td', {'class': 'w-50'})
-                value = title.find_next_sibling('td')
-                new_dict = {title.text:value.text}
-                data_dict.update(new_dict)
-            data_list.append(data_dict)
-            n += 1
-        # 获取头像
-        element:str = data_fom.find_all(attrs={"style": "cursor:pointer"})[0].get("onclick")
-        player_url = element.split("'")[1]
-        data_list[0].update({"个人资料":player_url})
-        # 获取一言
-        message = data_fom.select("html body div.content.text-center.text-md-left div.container.text-left div.col-md-12.h-100 div.card-body.worldmap.d-flex.flex-column.justify-content-center.text-center span")
-        msg_list = []
-        for i in message:
-            msg_list.append(i.text)
-        data_list[0].update({"一言":msg_list})
-        return data_list
+#     async def anne_msg(self):
+#         """个人资料表"""
+#         data_bytes = await url_to_byte('https://sb.trygek.com/l4d_stats/ranking/player.php?steamid={self.STEAMID}')
+#         data_bs = BeautifulSoup(data_bytes, 'html.parser')
+#         data_fom = data_bs.find_all('table')
+#         n = 0
+#         data_dict = {}
+#         while n < 2:
+#             data_list:List[dict] = []
+#             detail2 = data_fom[n]
+#             tr = detail2.find_all('tr')
+#             for i in tr:
+#                 title = i.find('td', {'class': 'w-50'})
+#                 value = title.find_next_sibling('td')
+#                 new_dict = {title.text:value.text}
+#                 data_dict.update(new_dict)
+#             data_list.append(data_dict)
+#             n += 1
+#         # 获取头像
+#         element:str = data_fom.find_all(attrs={"style": "cursor:pointer"})[0].get("onclick")
+#         player_url = element.split("'")[1]
+#         data_list[0].update({"个人资料":player_url})
+#         # 获取一言
+#         message = data_fom.select("html body div.content.text-center.text-md-left div.container.text-left div.col-md-12.h-100 div.card-body.worldmap.d-flex.flex-column.justify-content-center.text-center span")
+#         msg_list = []
+#         for i in message:
+#             msg_list.append(i.text)
+#         data_list[0].update({"一言":msg_list})
+#         return data_list
         
-    async def anne_map(self):
-        """个人地图表"""
-        data_dict = {}
-        data_bytes = await url_to_byte('https://sb.trygek.com/l4d_stats/ranking/timedmaps.php?steamid={self.STEAMID}')
-        data_bs = BeautifulSoup(data_bytes, 'html.parser')
-        tbody = data_bs.select('tbody')
-        for tr in tbody:
-            tds = tr.select('td')
-            n = 0
-            for td in tds:
-                n += 1
-                title:str = td['data-title'][:-1]
-                data_text = td.text
-                if title == '特感数量':
-                    special_amount = data_text
-                elif title == '刷新间隔':
-                    refresh_interval = data_text
-                else:
-                    if title in data_dict:
-                        data_dict[title].append(data_text)
-                    else:
-                        data_dict[title] = [data_text]
-            if special_amount and refresh_interval:
-                data_dict['刷特时间'] = special_amount + refresh_interval
+#     async def anne_map(self):
+#         """个人地图表"""
+#         data_dict = {}
+#         data_bytes = await url_to_byte('https://sb.trygek.com/l4d_stats/ranking/timedmaps.php?steamid={self.STEAMID}')
+#         data_bs = BeautifulSoup(data_bytes, 'html.parser')
+#         tbody = data_bs.select('tbody')
+#         for tr in tbody:
+#             tds = tr.select('td')
+#             n = 0
+#             for td in tds:
+#                 n += 1
+#                 title:str = td['data-title'][:-1]
+#                 data_text = td.text
+#                 if title == '特感数量':
+#                     special_amount = data_text
+#                 elif title == '刷新间隔':
+#                     refresh_interval = data_text
+#                 else:
+#                     if title in data_dict:
+#                         data_dict[title].append(data_text)
+#                     else:
+#                         data_dict[title] = [data_text]
+#             if special_amount and refresh_interval:
+#                 data_dict['刷特时间'] = special_amount + refresh_interval
                 
-        return data_dict
+#         return data_dict

nonebot_plugin_l4d2_server/l4d2_anne/server.py

@@ -60,4 +60,5 @@ def server_key():
     except AttributeError:
         a.add('希腊那我从来没有想过这个事情')
     return  a
-            
+            
+            

nonebot_plugin_l4d2_server/l4d2_anne/startand.py

@@ -0,0 +1,18 @@
+
+NUMBER_MAP = [4,5,4,5,5,3,3,5,2,5,5,5,4,2]
+SAVE_MAP = [
+    "c1m4_atrium",
+    "c2m5_concert",
+    "c3m4_plantation",
+    "c4m5_milltown_escape",
+    "c5m5_bridge",
+    "c6m3_port",
+    "c7m3_port",
+    "c8m5_rooftop",
+    "c9m2_lots",
+    "c10m5_houseboat",
+    "c11m5_runway",
+    "c12m5_cornfield",
+    "c13m4_cutthroatcreek",
+    "c14m2_lighthouse"
+    ]

nonebot_plugin_l4d2_server/l4d2_file/__init__.py

@@ -3,6 +3,8 @@ from zipfile import ZipFile
 from time import sleep
 import sys
 import os
+import io
+from typing import List
 
 from ..utils import get_file,get_vpk
 from ..config import systems
@@ -39,7 +41,7 @@ def open_packet(name:str,down_file:Path):
     if systems == 'win':
         if name.endswith('.zip'):
             mes = 'zip文件已下载,正在解压'
-            with ZipFile(down_file, 'r') as z:
+            with support_gbk(ZipFile(down_file, 'r')) as z:
                 z.extractall(down_path)
             os.remove(down_file)
         elif name.endswith('.7z'):
@@ -99,4 +101,28 @@ def support_gbk(zip_file):
                 del name_to_info[name]
                 name_to_info[real_name] = info
                 zip_file = rar_file
-    return zip_file
+    return zip_file
+
+
+
+
+async def all_zip_to_one(data_list:List[bytes]):
+    """多压缩包文件合并"""
+    file_list = []
+    for data in data_list:
+        # 将每个bytes对象解压缩成文件对象
+        # 将文件对象存储在一个列表中
+        file_list.append(io.BytesIO(data))
+
+    # 创建一个新的BytesIO对象
+    data_file = io.BytesIO()
+
+    # 使用zipfile将列表中的文件对象添加到zipfile中
+    with ZipFile(data_file, mode='w') as zf:
+        for i, file in enumerate(file_list):
+            # 将文件名设置为"file{i}.zip"，i为文件在列表中的索引
+            filename = f"file{i}.zip"
+            zf.writestr(filename, file.getvalue())
+
+    # 获取zipfile的bytes对象
+    return data_file.getvalue()

nonebot_plugin_l4d2_server/l4d2_image/__init__.py

@@ -4,11 +4,12 @@ from nonebot.log import logger
 from nonebot_plugin_htmlrender import html_to_pic
 from typing import List,Optional
 # from .htmlimg import dict_to_dict_img
-from ..l4d2_anne.anne_telecom import ANNE_API
+# from ..l4d2_anne.anne_telecom import ANNE_API
 from ..config import TEXT_PATH
 from .download import get_head_by_user_id_and_save
 from .send_image_tool import convert_img
 import jinja2
+from ..config import l4_style
 template_path = TEXT_PATH/"template"
 
 env = jinja2.Environment(
@@ -50,6 +51,7 @@ async def dict_to_html(usr_id,DETAIL_MAP:dict,soup:BeautifulSoup):
     DETAIL_right['playtimes'] = DETAIL_MAP['游玩地图数量:']
     DETAIL_right['url'] = DETAIL_MAP['个人资料']
     DETAIL_right['one_msg'] = DETAIL_MAP['一言']
+    DETAIL_right['last_one'] = DETAIL_MAP['救援关']
     # html_text = soup.prettify()
     # for key, value in DETAIL_right.items():
     #     html_text = html_text.replace(key,value)
@@ -80,22 +82,17 @@ async def server_ip_pic(msg_list:List[dict]):
             server_info['Players'] = players_list
             print(server_info['Players'])
 
-    # 返回更新后的 msg_list
-    # template_path = TEXT_PATH/"template"
-    # template = env.get_template('ip.html')
     pic = await get_help_img(msg_list)
-    # html = await template.render_async(data = msg_list)
-    # pic = await html_to_pic(
-    #         html,
-    #         wait=0,
-    #         viewport={"width": 1080, "height": 400},
-    #         template_path=f"file://{template_path.absolute()}",)
+
     return pic
 
 
 async def get_help_img(plugins: List[dict]) -> Optional[bytes]:
     try:
-        template = env.get_template("help.html")
+        if l4_style == 'black':
+            template = env.get_template("help_dack.html")
+        else:
+            template = env.get_template("help.html")
         content = await template.render_async(plugins=plugins)
         return await html_to_pic(
             content,

nonebot_plugin_l4d2_server/l4d2_queries/__init__.py

@@ -7,6 +7,7 @@ async def queries(ip:str,port:int):
     message = 'ip:' + msg_dict['ip'] + '\n'
     message += '名称：' + msg_dict['name'] + '\n'
     message += f"地图：{msg_dict['map_']}\n"
+    message += f"延迟：{msg_dict['ping']}\n"
     message += f"玩家：{msg_dict['players']} / {msg_dict['max_players']}\n"
     return message
 
@@ -23,10 +24,12 @@ async def queries_dict(ip:str,port:int) -> dict:
     msg_dict['players'] =  msg.player_count
     msg_dict['max_players'] =  msg.max_players
     msg_dict['ip'] = str(ip) + ':' +str(port)
+    msg_dict['ping'] = f"{msg.ping*1000:.0f}ms"
     if msg_dict['players'] < msg_dict['max_players']:
         msg_dict['enabled'] = True
     else:
         msg_dict['enabled'] = False
+    print(msg_dict)
     return msg_dict
     
 async def player_queries_anne_dict(ip:str,port:int): 
@@ -73,6 +76,7 @@ async def msg_ip_to_list(message_list:list):
         max_duration_len = max([len(str(i['Duration'])) for i in message_list])
         max_score_len = max([len(str(i['Score'])) for i in message_list])
         for i in message_list:
+            print(i)
             n += 1 
             name = i['name']
             Score = i['Score']

nonebot_plugin_l4d2_server/l4d2_queries/maps.py

@@ -0,0 +1,27 @@
+import httpx
+
+
+async def seach_map(msg:str,qq:str,key:str):
+    url = "http://106.13.207.45:4015/l4d2"
+    json = {
+        "mode":"zh",
+        "map_name":msg,
+        "qq":qq,
+        "key":key
+    }
+    print(json)
+    file = httpx.post(url,json=json)
+    if file.status_code == 200:
+        return file.json()
+    elif file.status_code == 204:
+        return "没有结果"
+    elif file.status_code == 406:
+        return "参数错误"
+    elif file.status_code == 401:
+        return file.json()
+
+async def map_dict_to_str(data:dict):
+    msg = ""
+    for key,value in data[0].items():
+        msg += f"{key}:{value}\n"
+    return msg

nonebot_plugin_l4d2_server/l4d2_queries/qqgroup.py

@@ -54,6 +54,7 @@ async def qq_ip_queries(msg:List[tuple]):
 async def qq_ip_queries_pic(msg:list):
     """输入一个ip的三元元组组成的列表，返回一个输出消息的图片"""
     msg_list = []
+    pic = None
     if msg != []:
         for i in msg:
             try:
@@ -71,6 +72,8 @@ async def qq_ip_queries_pic(msg:list):
                 except errors:
                     continue
         pic = await server_ip_pic(msg_list)
+    if pic == None:
+        return None
     return pic
     
 async def get_tan_jian(msg:List[tuple],mode:int):
@@ -247,10 +250,12 @@ async def write_json(data_str:str):
                         return '删除成功喵'
                 return '序号不正确，请输入【求生更新 删除 腐竹 序号】'
         return '腐竹名不存在，请输入【求生更新 删除 腐竹 序号】'    
-    
-
-ips = ALL_HOST['云']
-ip_anne_list = []
-for one_ip in ips:
-    host,port = split_maohao(one_ip['ip'])
-    ip_anne_list.append((one_ip['id'],host,port))
+ip_anne_list=[] 
+try:
+    ips = ALL_HOST['云']
+    ip_anne_list = []
+    for one_ip in ips:
+        host,port = split_maohao(one_ip['ip'])
+        ip_anne_list.append((one_ip['id'],host,port))
+except KeyError:
+    pass

nonebot_plugin_l4d2_server/l4d2_server/web.py

@@ -0,0 +1,221 @@
+import datetime
+from typing import Optional, Union
+
+from fastapi import FastAPI
+from fastapi import Header, HTTPException, Depends
+from fastapi.responses import JSONResponse, HTMLResponse, RedirectResponse
+from jose import jwt
+from nonebot import get_bot, get_app
+from pydantic import BaseModel
+from typing import List, Dict
+from pathlib import Path
+
+from pydantic import BaseModel, Field
+
+from nonebot import get_driver, logger
+from ruamel import yaml
+
+CONFIG_PATH = Path() / 'data' / 'L4D2' / 'l4d2.yml'
+CONFIG_PATH.parent.mkdir(parents=True, exist_ok=True)
+
+driver = get_driver()
+
+from .webUI import login_page, admin_app
+
+requestAdaptor = '''
+requestAdaptor(api) {
+    api.headers["token"] = localStorage.getItem("token");
+    return api;
+},
+'''
+responseAdaptor = '''
+responseAdaptor(api, payload, query, request, response) {
+    if (response.data.detail == '登录验证失败或已失效，请重新登录') {
+        window.location.href = '/l4d2/login'
+        window.localStorage.clear()
+        window.sessionStorage.clear()
+        window.alert('登录验证失败或已失效，请重新登录')
+    }
+    return payload
+},
+'''
+
+
+def authentication():
+    def inner(token: Optional[str] = Header(...)):
+        try:
+            payload = jwt.decode(token, config_manager.config.web_secret_key, algorithms='HS256')
+            if not (username := payload.get('username')) or username != config_manager.config.web_username:
+                raise HTTPException(status_code=400, detail='登录验证失败或已失效，请重新登录')
+        except (jwt.JWTError, jwt.ExpiredSignatureError, AttributeError):
+            raise HTTPException(status_code=400, detail='登录验证失败或已失效，请重新登录')
+
+    return Depends(inner)
+
+
+COMMAND_START = driver.config.command_start.copy()
+if '' in COMMAND_START:
+    COMMAND_START.remove('')
+
+
+class ChatGroupConfig(BaseModel):
+    enable: bool = Field(True, alias='群聊学习开关')
+    ban_words: List[str] = Field([], alias='屏蔽词')
+    ban_users: List[int] = Field([], alias='屏蔽用户')
+    answer_threshold: int = Field(4, alias='回复阈值')
+    answer_threshold_weights: List[int] = Field([10, 30, 60], alias='回复阈值权重')
+    repeat_threshold: int = Field(3, alias='复读阈值')
+    break_probability: float = Field(0.25, alias='打断复读概率')
+    speak_enable: bool = Field(True, alias='主动发言开关')
+    speak_threshold: int = Field(5, alias='主动发言阈值')
+    speak_min_interval: int = Field(300, alias='主动发言最小间隔')
+    speak_continuously_probability: float = Field(0.5, alias='连续主动发言概率')
+    speak_continuously_max_len: int = Field(3, alias='最大连续主动发言句数')
+    speak_poke_probability: float = Field(0.5, alias='主动发言附带戳一戳概率')
+
+    def update(self, **kwargs):
+        for key, value in kwargs.items():
+            if key in self.__fields__:
+                self.__setattr__(key, value)
+
+
+class ChatConfig(BaseModel):
+    total_enable: bool = Field(True, alias='群聊学习总开关')
+    enable_web: bool = Field(True, alias='启用后台管理')
+    web_username: str = Field('chat', alias='后台管理用户名')
+    web_password: str = Field('admin', alias='后台管理密码')
+    web_secret_key: str = Field('49c294d32f69b732ef6447c18379451ce1738922a75cd1d4812ef150318a2ed0',
+                                alias='后台管理token密钥')
+    ban_words: List[str] = Field([], alias='全局屏蔽词')
+    ban_users: List[int] = Field([], alias='全局屏蔽用户')
+    KEYWORDS_SIZE: int = Field(3, alias='单句关键词分词数量')
+    cross_group_threshold: int = Field(3, alias='跨群回复阈值')
+    learn_max_count: int = Field(6, alias='最高学习次数')
+    dictionary: List[str] = Field([], alias='自定义词典')
+    group_config: Dict[int, ChatGroupConfig] = Field({}, alias='分群配置')
+
+    def update(self, **kwargs):
+        for key, value in kwargs.items():
+            if key in self.__fields__:
+                self.__setattr__(key, value)
+                
+class ChatConfigManager:
+
+    def __init__(self):
+        self.file_path = CONFIG_PATH
+        if self.file_path.exists():
+            self.config = ChatConfig.parse_obj(
+                yaml.load(self.file_path.read_text(encoding='utf-8'), Loader=yaml.Loader))
+        else:
+            self.config = ChatConfig()
+        self.save()
+
+    def get_group_config(self, group_id: int) -> ChatGroupConfig:
+        if group_id not in self.config.group_config:
+            self.config.group_config[group_id] = ChatGroupConfig()
+            self.save()
+        return self.config.group_config[group_id]
+
+    @property
+    def config_list(self) -> List[str]:
+        return list(self.config.dict(by_alias=True).keys())
+
+    def save(self):
+        with self.file_path.open('w', encoding='utf-8') as f:
+            yaml.dump(
+                self.config.dict(by_alias=True),
+                f,
+                indent=2,
+                Dumper=yaml.RoundTripDumper,
+                allow_unicode=True)
+            
+config_manager = ChatConfigManager()
+
+class UserModel(BaseModel):
+    username: str
+    password: str
+
+
+@driver.on_startup
+async def init_web():
+    if not config_manager.config.enable_web:
+        return
+    app: FastAPI = get_app()
+
+    @app.post('/l4d2/api/login', response_class=JSONResponse)
+    async def login(user: UserModel):
+        if user.username != config_manager.config.web_username or user.password != config_manager.config.web_password:
+            return {
+                'status': -100,
+                'msg':    '登录失败，请确认用户ID和密码无误'
+            }
+        token = jwt.encode({'username': user.username,
+                            'exp':      datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(
+                                minutes=30)}, config_manager.config.web_secret_key, algorithm='HS256')
+        return {
+            'status': 0,
+            'msg':    '登录成功',
+            'data':   {
+                'token': token
+            }
+        }
+
+    @app.get('/l4d2/api/get_group_list', response_class=JSONResponse, dependencies=[authentication()])
+    async def get_group_list_api():
+        try:
+            group_list = await get_bot().get_group_list()
+            group_list = [{'label': f'{group["group_name"]}({group["group_id"]})', 'value': group['group_id']} for group
+                          in group_list]
+            return {
+                'status': 0,
+                'msg':    'ok',
+                'data':   {
+                    'group_list': group_list
+                }
+            }
+        except ValueError:
+            return {
+                'status': -100,
+                'msg':    '获取群和好友列表失败，请确认已连接GOCQ'
+            }
+
+    @app.get('/l4d2/api/chat_global_config', response_class=JSONResponse, dependencies=[authentication()])
+    async def get_chat_global_config():
+        try:
+            bot = get_bot()
+            groups = await bot.get_group_list()
+            member_list = []
+            for group in groups:
+                members = await bot.get_group_member_list(group_id=group['group_id'])
+                member_list.extend(
+                    [{'label': f'{member["nickname"] or member["card"]}({member["user_id"]})',
+                      'value': member['user_id']}
+                     for
+                     member in members])
+            config = config_manager.config.dict(exclude={'group_config'})
+            config['member_list'] = member_list
+            return config
+        except ValueError:
+            return {
+                'status': -100,
+                'msg':    '获取群和好友列表失败，请确认已连接GOCQ'
+            }
+
+
+
+    @app.get('/l4d2', response_class=RedirectResponse)
+    async def redirect_page():
+        return RedirectResponse('/l4d2/login')
+
+    @app.get('/l4d2/login', response_class=HTMLResponse)
+    async def login_page_app():
+        return login_page.render(site_title='登录 | Learning-Chat 后台管理',
+                                 theme='ang')
+
+    @app.get('/l4d2/admin', response_class=HTMLResponse)
+    async def admin_page_app():
+        return admin_app.render(site_title='Learning-Chat 后台管理',
+                                theme='ang',
+                                requestAdaptor=requestAdaptor,
+                                responseAdaptor=responseAdaptor)
+