nonebot-plugin-l4d2-server 0.5.0__py3-none-any.whl → 0.5.2__py3-none-any.whl

nonebot_plugin_l4d2_server/l4d2_image/send_image_tool.py

@@ -1,32 +1,32 @@
-from io import BytesIO
-from pathlib import Path
-from typing import Union
-from base64 import b64encode
-
-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
-    elif isinstance(img, bytes):
-        return b64encode(img).decode()
-    else:
-        return f'[CQ:image,file=file:///{str(img)}]'
+from io import BytesIO
+from pathlib import Path
+from typing import Union
+from base64 import b64encode
+
+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
+    elif isinstance(img, bytes):
+        return b64encode(img).decode()
+    else:
+        return f'[CQ:image,file=file:///{str(img)}]'

nonebot_plugin_l4d2_server/l4d2_image/steam.py

@@ -1,83 +1,83 @@
-# from PIL import Image
-import httpx
-import aiohttp
-# from bs4 import BeautifulSoup
-from urllib.parse import unquote
-
-    
-# async def web_player(url) -> Image :
-#     """steam个人资料获取头像"""
-#     data = BeautifulSoup(await url_for_byte(url), 'html.parser')
-#     print(data)
-#     head = data.find("div", class_="playerAvatarAutoSizeInner")
-#     img_elements = head.find_all("img")
-#     if len(img_elements)== 1:
-#         for img_element in img_elements:
-#             head = await url_for_byte(img_element["src"])
-#         im = Image.open(io.BytesIO(head)).resize((225, 225)).convert("RGBA")
-#     if len(img_elements) == 2:
-#         head,headd = img_elements
-#         head = await url_for_byte(img_element["src"])
-#         head = await url_for_byte(img_element["src"])
-#     # 下面是边框，不一定有
-#     try:
-#         headd = data.select("html.responsive body.flat_page.profile_page.has_profile_background.GhostTheme.responsive_page div.responsive_page_frame.with_header div.responsive_page_content div#responsive_page_template_content.responsive_page_template_content div.no_header.profile_page.has_profile_background div.profile_header_bg div.profile_header_bg_texture div.profile_header div.profile_header_content div.playerAvatar.profile_header_size.in-game div.playerAvatarAutoSizeInner div.profile_avatar_frame img")
-#         im2 = Image.open(io.BytesIO(headd)).resize((185, 185)).convert("RGBA")
-#         r, g, b, a2 = im2.split()
-#         # im.paste(im2,(20,20),mask=a2)
-#     except IndexError:
-#         pass
-#     return im
-
-# async def url_for_byte(url):
-#     """所有代理_url终将绳之以法"""
-#     if not l4_proxies:
-#         print("代理不存在")
-#         data = await url_to_byte(url)
-#     headers = {
-#         'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
-#     }
-#     data = httpx.get(url,headers=headers,proxies=l4_proxies,timeout=60,verify=False).content
-#     return data
-async def url_to_byte(url:str,filename:str =''):
-    """获取URL数据的字节流"""
-    headers = {
-    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
-    }
-    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()
-            else:
-                return None
-
-async def url_to_byte_name(url:str,filename:str =''):
-    """获取URL数据的字节流"""
-    headers = {
-    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
-    }
-    if filename == "htp":
-        response = httpx.get(url,headers=headers,timeout=600)
-        content_disposition:str = response.headers.get("Content-Disposition")
-        if not content_disposition:
-            return None
-        elif "''" in content_disposition:
-            file_name = content_disposition.split("''")[-1]
-        else: 
-            file_name = content_disposition
-        file_name = unquote(file_name)
-        return [response.read(),file_name]
-    else:       
-        async with aiohttp.ClientSession() as session:
-            async with session.get(url, headers=headers, timeout=600) as response:
-                content_disposition:str = response.headers.get("Content-Disposition")
-                if "''" in content_disposition:
-                    file_name = content_disposition.split("''")[-1]
-                else: 
-                    file_name = content_disposition
-                file_name = unquote(file_name)
-                if response.status == 200:
-                    return [await response.read(),file_name]
-                else:
-                    return None
-            
+# from PIL import Image
+import httpx
+import aiohttp
+# from bs4 import BeautifulSoup
+from urllib.parse import unquote
+
+    
+# async def web_player(url) -> Image :
+#     """steam个人资料获取头像"""
+#     data = BeautifulSoup(await url_for_byte(url), 'html.parser')
+#     print(data)
+#     head = data.find("div", class_="playerAvatarAutoSizeInner")
+#     img_elements = head.find_all("img")
+#     if len(img_elements)== 1:
+#         for img_element in img_elements:
+#             head = await url_for_byte(img_element["src"])
+#         im = Image.open(io.BytesIO(head)).resize((225, 225)).convert("RGBA")
+#     if len(img_elements) == 2:
+#         head,headd = img_elements
+#         head = await url_for_byte(img_element["src"])
+#         head = await url_for_byte(img_element["src"])
+#     # 下面是边框，不一定有
+#     try:
+#         headd = data.select("html.responsive body.flat_page.profile_page.has_profile_background.GhostTheme.responsive_page div.responsive_page_frame.with_header div.responsive_page_content div#responsive_page_template_content.responsive_page_template_content div.no_header.profile_page.has_profile_background div.profile_header_bg div.profile_header_bg_texture div.profile_header div.profile_header_content div.playerAvatar.profile_header_size.in-game div.playerAvatarAutoSizeInner div.profile_avatar_frame img")
+#         im2 = Image.open(io.BytesIO(headd)).resize((185, 185)).convert("RGBA")
+#         r, g, b, a2 = im2.split()
+#         # im.paste(im2,(20,20),mask=a2)
+#     except IndexError:
+#         pass
+#     return im
+
+# async def url_for_byte(url):
+#     """所有代理_url终将绳之以法"""
+#     if not l4_proxies:
+#         print("代理不存在")
+#         data = await url_to_byte(url)
+#     headers = {
+#         'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
+#     }
+#     data = httpx.get(url,headers=headers,proxies=l4_proxies,timeout=60,verify=False).content
+#     return data
+async def url_to_byte(url:str,filename:str =''):
+    """获取URL数据的字节流"""
+    headers = {
+    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
+    }
+    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()
+            else:
+                return None
+
+async def url_to_byte_name(url:str,filename:str =''):
+    """获取URL数据的字节流"""
+    headers = {
+    'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:107.0) Gecko/20100101 Firefox/107.0'
+    }
+    if filename == "htp":
+        response = httpx.get(url,headers=headers,timeout=600)
+        content_disposition:str = response.headers.get("Content-Disposition")
+        if not content_disposition:
+            return None
+        elif "''" in content_disposition:
+            file_name = content_disposition.split("''")[-1]
+        else: 
+            file_name = content_disposition
+        file_name = unquote(file_name)
+        return [response.read(),file_name]
+    else:       
+        async with aiohttp.ClientSession() as session:
+            async with session.get(url, headers=headers, timeout=600) as response:
+                content_disposition:str = response.headers.get("Content-Disposition")
+                if "''" in content_disposition:
+                    file_name = content_disposition.split("''")[-1]
+                else: 
+                    file_name = content_disposition
+                file_name = unquote(file_name)
+                if response.status == 200:
+                    return [await response.read(),file_name]
+                else:
+                    return None
+            

nonebot_plugin_l4d2_server/l4d2_image/vtfs.py

@@ -1,40 +1,40 @@
-
-from PIL import Image
-from nonebot.log import logger
-from io import BytesIO
-# import sys
-# sys.modules["srctools._cy_vtf_readwrite"] = None
-from srctools.vtf import VTF, ImageFormats
-
-async def img_to_vtf(pic_byte:bytes,tag) -> BytesIO:
-    pic = BytesIO(pic_byte)
-    pic = Image.open(pic).convert('RGBA')
-    vtf_io = BytesIO()
-    vtf_ = VTF(1024, 1024, fmt = ImageFormats.DXT5,thumb_fmt = ImageFormats.DXT1,version=(7,2))
-    if tag == '覆盖':
-        logger.info(tag)
-        img2 = Image.new('RGBA',(1024, 1024), (255, 255, 255,0))
-        r, g, b, a = pic.split()
-        img2.paste(pic,mask=a)
-        pic = pic.resize((1024,1024))
-    elif tag == '填充':
-        w, h = pic.size
-        if w > h:
-            ratio = 1024/w
-            new_width = 1024
-            new_height = int(h * ratio)
-        else:
-            ratio = 1024/h
-            new_height = 1024
-            new_width = int(w * ratio)
-        pic = pic.resize((new_width, new_height), Image.ANTIALIAS)
-        background = Image.new('RGBA', (1024, 1024), (255, 255, 255, 0))
-        background.paste(pic, ((1024-new_width)//2, (1024-new_height)//2))
-        pic = background
-    else:
-        logger.info('拉伸')
-        pic = pic.resize((1024,1024))
-    largest_frame = vtf_.get() 
-    largest_frame.copy_from(pic.tobytes(), ImageFormats.RGBA8888)
-    vtf_.save(vtf_io,version=(7,2))
-    return vtf_io
+
+from PIL import Image
+from nonebot.log import logger
+from io import BytesIO
+# import sys
+# sys.modules["srctools._cy_vtf_readwrite"] = None
+from srctools.vtf import VTF, ImageFormats
+
+async def img_to_vtf(pic_byte:bytes,tag) -> BytesIO:
+    pic = BytesIO(pic_byte)
+    pic = Image.open(pic).convert('RGBA')
+    vtf_io = BytesIO()
+    vtf_ = VTF(1024, 1024, fmt = ImageFormats.DXT5,thumb_fmt = ImageFormats.DXT1,version=(7,2))
+    if tag == '覆盖':
+        logger.info(tag)
+        img2 = Image.new('RGBA',(1024, 1024), (255, 255, 255,0))
+        r, g, b, a = pic.split()
+        img2.paste(pic,mask=a)
+        pic = pic.resize((1024,1024))
+    elif tag == '填充':
+        w, h = pic.size
+        if w > h:
+            ratio = 1024/w
+            new_width = 1024
+            new_height = int(h * ratio)
+        else:
+            ratio = 1024/h
+            new_height = 1024
+            new_width = int(w * ratio)
+        pic = pic.resize((new_width, new_height), Image.ANTIALIAS)
+        background = Image.new('RGBA', (1024, 1024), (255, 255, 255, 0))
+        background.paste(pic, ((1024-new_width)//2, (1024-new_height)//2))
+        pic = background
+    else:
+        logger.info('拉伸')
+        pic = pic.resize((1024,1024))
+    largest_frame = vtf_.get() 
+    largest_frame.copy_from(pic.tobytes(), ImageFormats.RGBA8888)
+    vtf_.save(vtf_io,version=(7,2))
+    return vtf_io

nonebot_plugin_l4d2_server/l4d2_queries/__init__.py

@@ -1,114 +1,114 @@
-import a2s
-from typing import List
-
-async def queries(ip:str,port:int):
-    port = int(port)
-    msg_dict = await queries_dict(ip,port)
-    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
-
-async def queries_dict(ip:str,port:int) -> dict:
-    port = int(port)
-    ip = str(ip)
-    msg_dict = {}
-    # message_dict = await l4d(ip,port)
-    msg:a2s.SourceInfo = await a2s.ainfo((ip,port))
-    # message_dict = await l4d2.server(ip,port,times=5)
-    msg_dict['folder'] =  msg.folder
-    msg_dict['name'] =  msg.server_name
-    msg_dict['map_'] =  msg.map_name
-    msg_dict['players'] =  msg.player_count
-    msg_dict['max_players'] =  msg.max_players
-    msg_dict['rank_players'] =  f'{msg.player_count}/{msg.max_players}'
-    msg_dict['ip'] = str(ip) + ':' +str(port)
-    msg_dict['ping'] = f"{msg.ping*1000:.0f}ms"
-    msg_dict['system'] =  f"{msg.platform}.svg"
-    if msg_dict['players'] < msg_dict['max_players']:
-        msg_dict['enabled'] = True
-    else:
-        msg_dict['enabled'] = False
-    return msg_dict
-    
-async def player_queries_anne_dict(ip:str,port:int): 
-    """anne算法返回玩家"""
-    port = int(port)
-    # message_dic = await l4d2.APlayer(ip,port,times=5)
-    message_list:List[a2s.Player] = await a2s.aplayers((ip,port))
-    msg_list = []
-    if message_list != []:
-        for i in message_list:
-            msg_list.append({
-                'name':i.name,
-                'Score':i.score,
-                'Duration':await convert_duration(i.duration)
-            })
-    return msg_list
-
-# async def player_queries_dict(ip:str,port:int): 
-#     """一般算法返回玩家"""
-#     port = int(port)
-#     new_dict = {}
-#     message_dic = await l4d2.APlayer(ip,port,times=5)
-#     if message_dic == {}:
-#         new_dict['header'] = 0
-#     else:
-#         pass
-#         new_list = []
-#         for i in message_dic['Players']:
-#             new_list.append(i['Name'])
-#         new_dict.update({'Players':new_list})
-#     return new_dict
-
-async def player_queries(ip:str,port:int): 
-    port = int(port)
-    message_list = await player_queries_anne_dict(ip,port)
-    return await msg_ip_to_list(message_list)
-
-async def msg_ip_to_list(message_list:list):
-    message = ''
-    n = 0
-    if message_list == []:
-        message += '服务器里，是空空的呢\n'
-    else:
-        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']
-            if Score == '0':
-                Score = '摸'
-            Duration = i['Duration']
-            soc = "[{:>{}}]".format(Score,max_score_len)
-            dur = "{:^{}}".format(Duration, max_duration_len)
-            message += f'{soc} | {dur} | {name} \n'
-    return message
-
-async def convert_duration(duration: int) -> str:
-    minutes, seconds = divmod(duration, 60)
-    hours, minutes = divmod(minutes, 60)
-    time_str = ""
-    if hours > 0:
-        time_str += f"{int(hours)}h "
-    if minutes > 0 or hours > 0:
-        time_str += f"{int(minutes)}m "
-    time_str += f"{int(seconds)}s"
-    return time_str.strip()
-
-async def server_rule_dict(ip:str,port:int): 
-    port = int(port)
-    ip = str(ip)
-    msg_dict = {}
-    # message_dict = await l4d(ip,port)
-    try:
-        msg:dict = await a2s.arules((ip,port))
-        msg_dict['tick'] = msg['l4d2_tickrate_enabler'] + "tick"
-    except:
-        msg_dict['tick'] = ''
-    return msg_dict
-
+import a2s
+from typing import List
+
+async def queries(ip:str,port:int):
+    port = int(port)
+    msg_dict = await queries_dict(ip,port)
+    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
+
+async def queries_dict(ip:str,port:int) -> dict:
+    port = int(port)
+    ip = str(ip)
+    msg_dict = {}
+    # message_dict = await l4d(ip,port)
+    msg:a2s.SourceInfo = await a2s.ainfo((ip,port))
+    # message_dict = await l4d2.server(ip,port,times=5)
+    msg_dict['folder'] =  msg.folder
+    msg_dict['name'] =  msg.server_name
+    msg_dict['map_'] =  msg.map_name
+    msg_dict['players'] =  msg.player_count
+    msg_dict['max_players'] =  msg.max_players
+    msg_dict['rank_players'] =  f'{msg.player_count}/{msg.max_players}'
+    msg_dict['ip'] = str(ip) + ':' +str(port)
+    msg_dict['ping'] = f"{msg.ping*1000:.0f}ms"
+    msg_dict['system'] =  f"{msg.platform}.svg"
+    if msg_dict['players'] < msg_dict['max_players']:
+        msg_dict['enabled'] = True
+    else:
+        msg_dict['enabled'] = False
+    return msg_dict
+    
+async def player_queries_anne_dict(ip:str,port:int): 
+    """anne算法返回玩家"""
+    port = int(port)
+    # message_dic = await l4d2.APlayer(ip,port,times=5)
+    message_list:List[a2s.Player] = await a2s.aplayers((ip,port))
+    msg_list = []
+    if message_list != []:
+        for i in message_list:
+            msg_list.append({
+                'name':i.name,
+                'Score':i.score,
+                'Duration':await convert_duration(i.duration)
+            })
+    return msg_list
+
+# async def player_queries_dict(ip:str,port:int): 
+#     """一般算法返回玩家"""
+#     port = int(port)
+#     new_dict = {}
+#     message_dic = await l4d2.APlayer(ip,port,times=5)
+#     if message_dic == {}:
+#         new_dict['header'] = 0
+#     else:
+#         pass
+#         new_list = []
+#         for i in message_dic['Players']:
+#             new_list.append(i['Name'])
+#         new_dict.update({'Players':new_list})
+#     return new_dict
+
+async def player_queries(ip:str,port:int): 
+    port = int(port)
+    message_list = await player_queries_anne_dict(ip,port)
+    return await msg_ip_to_list(message_list)
+
+async def msg_ip_to_list(message_list:list):
+    message = ''
+    n = 0
+    if message_list == []:
+        message += '服务器里，是空空的呢\n'
+    else:
+        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']
+            if Score == '0':
+                Score = '摸'
+            Duration = i['Duration']
+            soc = "[{:>{}}]".format(Score,max_score_len)
+            dur = "{:^{}}".format(Duration, max_duration_len)
+            message += f'{soc} | {dur} | {name} \n'
+    return message
+
+async def convert_duration(duration: int) -> str:
+    minutes, seconds = divmod(duration, 60)
+    hours, minutes = divmod(minutes, 60)
+    time_str = ""
+    if hours > 0:
+        time_str += f"{int(hours)}h "
+    if minutes > 0 or hours > 0:
+        time_str += f"{int(minutes)}m "
+    time_str += f"{int(seconds)}s"
+    return time_str.strip()
+
+async def server_rule_dict(ip:str,port:int): 
+    port = int(port)
+    ip = str(ip)
+    msg_dict = {}
+    # message_dict = await l4d(ip,port)
+    try:
+        msg:dict = await a2s.arules((ip,port))
+        msg_dict['tick'] = msg['l4d2_tickrate_enabler'] + "tick"
+    except:
+        msg_dict['tick'] = ''
+    return msg_dict
+

nonebot_plugin_l4d2_server/l4d2_queries/api.py

@@ -1,43 +1,43 @@
-import httpx
-from typing import List,Union
-from nonebot.log import logger
-
-async def seach_map(msg:Union[list,str],qq:str,key:str,mode:str = 'zh'):
-    url = "http://106.13.207.45:4015/l4d2"
-    json = {
-        "mode":mode,
-        "map_name":msg,
-        "qq":qq,
-        "key":key
-    }
-    file = httpx.post(url=url,json=json)
-    if mode == 'zh':
-        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()
-    elif mode == 'ip':
-        rep:dict = file.json()
-        try:
-            logger.error(rep['error_'])
-        except:
-            pass
-        print(file.json())
-        return file.json()
-    elif mode == 'first':
-        ip_tag:list = file.json()
-        return ip_tag
-        
-
-async def map_dict_to_str(data:List[dict]):
-    msg = ""
-    for key,value in data[0].items():
-        if key == "url":
-            continue
-        msg += f"{key}:{value}\n"
-    return msg
-
+import httpx
+from typing import List,Union
+from nonebot.log import logger
+
+async def seach_map(msg:Union[list,str],qq:str,key:str,mode:str = 'zh'):
+    url = "http://106.13.207.45:4015/l4d2"
+    json = {
+        "mode":mode,
+        "map_name":msg,
+        "qq":qq,
+        "key":key
+    }
+    file = httpx.post(url=url,json=json)
+    if mode == 'zh':
+        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()
+    elif mode == 'ip':
+        rep:dict = file.json()
+        try:
+            logger.error(rep['error_'])
+        except:
+            pass
+        print(file.json())
+        return file.json()
+    elif mode == 'first':
+        ip_tag:list = file.json()
+        return ip_tag
+        
+
+async def map_dict_to_str(data:List[dict]):
+    msg = ""
+    for key,value in data[0].items():
+        if key == "url":
+            continue
+        msg += f"{key}:{value}\n"
+    return msg
+