nonebot-plugin-l4d2-server 0.5.1__py3-none-any.whl → 0.5.3__py3-none-any.whl

nonebot_plugin_l4d2_server/l4d2_image/image.py

@@ -0,0 +1,292 @@
+from io import BytesIO
+from pathlib import Path
+from base64 import b64encode
+from typing import Union, overload
+from typing import Union,Tuple,Optional
+import math
+import random
+
+import aiofiles
+from PIL import Image
+from httpx import get
+import sys
+from pathlib import Path
+
+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'@overload
+
+if list(CU_BG_PATH.iterdir()) != []:
+    bg_path = CU_BG_PATH
+else:
+    bg_path = NM_BG_PATH
+
+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
+    elif 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()}]'
+
+
+async def get_color_bg(
+    based_w: int, based_h: int, bg: Optional[str] = None
+) -> Image.Image:
+    image = ''
+    if bg:
+        path = SP_BG_PATH / f'{bg}.jpg'
+        if path.exists():
+            image = Image.open(path)
+    CI_img = CustomizeImage(image, based_w, based_h)
+    img = CI_img.bg_img
+    color = CI_img.bg_color
+    color_mask = Image.new('RGBA', (based_w, based_h), color)
+    enka_mask = Image.open(TEXT2D_PATH / 'mask.png').resize((based_w, based_h))
+    img.paste(color_mask, (0, 0), enka_mask)
+    return img
+
+
+
+class CustomizeImage:
+    def __init__(
+        self, image: Union[str, Image.Image], based_w: int, based_h: int
+    ) -> None:
+
+        self.bg_img = self.get_image(image, based_w, based_h)
+        self.bg_color = self.get_bg_color(self.bg_img, is_light=True)
+        self.text_color = self.get_text_color(self.bg_color)
+        self.highlight_color = self.get_highlight_color(self.bg_color)
+        self.char_color = self.get_char_color(self.bg_color)
+        self.bg_detail_color = self.get_bg_detail_color(self.bg_color)
+        self.char_high_color = self.get_char_high_color(self.bg_color)
+
+    @staticmethod
+    def get_image(
+        image: Union[str, Image.Image], based_w: int, based_h: int
+    ) -> Image.Image:
+        # 获取背景图片
+        if isinstance(image, Image.Image):
+            edit_bg = image
+        elif image:
+            edit_bg = Image.open(BytesIO(get(image).content)).convert('RGBA')
+        else:
+            path = random.choice(list(bg_path.iterdir()))
+            edit_bg = Image.open(path).convert('RGBA')
+
+        # 确定图片的长宽
+        bg_img = crop_center_img(edit_bg, based_w, based_h)
+        return bg_img
+
+    @staticmethod
+    def get_dominant_color(pil_img: Image.Image) -> Tuple[int, int, int]:
+        img = pil_img.copy()
+        img = img.convert("RGBA")
+        img = img.resize((1, 1), resample=0)
+        dominant_color = img.getpixel((0, 0))
+        return dominant_color
+
+    @staticmethod
+    def get_bg_color(
+        edit_bg: Image.Image, is_light: Optional[bool] = False
+    ) -> Tuple[int, int, int]:
+        # 获取背景主色
+        color = 8
+        q = edit_bg.quantize(colors=color, method=2)
+        bg_color = (0, 0, 0)
+        if is_light:
+            based_light = 195
+        else:
+            based_light = 120
+        temp = 9999
+        for i in range(color):
+            bg = tuple(
+                q.getpalette()[  # type:ignore
+                    i * 3 : (i * 3) + 3  # noqa:E203
+                ]
+            )
+            light_value = bg[0] * 0.3 + bg[1] * 0.6 + bg[2] * 0.1
+            if abs(light_value - based_light) < temp:
+                bg_color = bg
+                temp = abs(light_value - based_light)
+        return bg_color
+
+    @staticmethod
+    def get_text_color(bg_color: Tuple[int, int, int]) -> Tuple[int, int, int]:
+        # 通过背景主色（bg_color）确定文字主色
+        r = 125
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        text_color = (
+            math.floor(bg_color[0] + r if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + r if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + r if bg_color[2] + r <= 255 else 255),
+        )
+        return text_color
+
+    @staticmethod
+    def get_char_color(bg_color: Tuple[int, int, int]) -> Tuple[int, int, int]:
+        r = 140
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        char_color = (
+            math.floor(bg_color[0] + 5 if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + 5 if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + 5 if bg_color[2] + r <= 255 else 255),
+        )
+        return char_color
+
+    @staticmethod
+    def get_char_high_color(
+        bg_color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        r = 140
+        d = 20
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        char_color = (
+            math.floor(bg_color[0] + d if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] + d if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] + d if bg_color[2] + r <= 255 else 255),
+        )
+        return char_color
+
+    @staticmethod
+    def get_bg_detail_color(
+        bg_color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        r = 140
+        if max(*bg_color) > 255 - r:
+            r *= -1
+        bg_detail_color = (
+            math.floor(bg_color[0] - 20 if bg_color[0] + r <= 255 else 255),
+            math.floor(bg_color[1] - 20 if bg_color[1] + r <= 255 else 255),
+            math.floor(bg_color[2] - 20 if bg_color[2] + r <= 255 else 255),
+        )
+        return bg_detail_color
+
+    @staticmethod
+    def get_highlight_color(
+        color: Tuple[int, int, int]
+    ) -> Tuple[int, int, int]:
+        red_color = color[0]
+        green_color = color[1]
+        blue_color = color[2]
+
+        highlight_color = {
+            'red': red_color - 127 if red_color > 127 else 127,
+            'green': green_color - 127 if green_color > 127 else 127,
+            'blue': blue_color - 127 if blue_color > 127 else 127,
+        }
+
+        max_color = max(highlight_color.values())
+
+        name = ''
+        for _highlight_color in highlight_color:
+            if highlight_color[_highlight_color] == max_color:
+                name = str(_highlight_color)
+
+        if name == 'red':
+            return red_color, highlight_color['green'], highlight_color['blue']
+        elif name == 'green':
+            return highlight_color['red'], green_color, highlight_color['blue']
+        elif name == 'blue':
+            return highlight_color['red'], highlight_color['green'], blue_color
+        else:
+            return 0, 0, 0  # Error
+        
+        
+def crop_center_img(
+    img: Image.Image, based_w: int, based_h: int
+) -> Image.Image:
+    # 确定图片的长宽
+    based_scale = '%.3f' % (based_w / based_h)
+    w, h = img.size
+    scale_f = '%.3f' % (w / h)
+    new_w = math.ceil(based_h * float(scale_f))
+    new_h = math.ceil(based_w / float(scale_f))
+    if scale_f > based_scale:
+        resize_img = img.resize((new_w, based_h), Image.ANTIALIAS)
+        x1 = int(new_w / 2 - based_w / 2)
+        y1 = 0
+        x2 = int(new_w / 2 + based_w / 2)
+        y2 = based_h
+    else:
+        resize_img = img.resize((based_w, new_h), Image.ANTIALIAS)
+        x1 = 0
+        y1 = int(new_h / 2 - based_h / 2)
+        x2 = based_w
+        y2 = int(new_h / 2 + based_h / 2)
+    crop_img = resize_img.crop((x1, y1, x2, y2))
+    return crop_img

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