girineko/McStatus-Img
1
0
Fork 0
mirror of https://github.com/Murasame-Dev/McStatus-Img.git synced 2025-12-16 13:17:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add: define a function to process the icon

Browse source
This commit is contained in:
GiriNeko 2025-08-26 17:16:50 +08:00
parent 82339c3463
commit 7eb0dba03a
1 changed files with 23 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

33
main.py
View file

@ -16,13 +16,18 @@ import asyncio
BACKGROUND_URL = "https://www.loliapi.com/acg/"
DEFAULT_ICON = "./minecraft-creeper-face.png"
if DEFAULT_ICON.startswith("http"):
default_icon = asyncio.run(download_image_with_httpx_auto_redirect(DEFAULT_ICON))
if not default_icon:
default_icon = None
else:
with open(DEFAULT_ICON, "rb") as f:
default_icon = f.read()
async def get_icon_image(url: str):
if url.startswith("http"):
icon_data = await download_image_with_httpx_auto_redirect(url)
if icon_data:
return icon_data
else:
return None
else:
def read_file(path):
with open(path, "rb") as f:
return f.read()
return await asyncio.to_thread(read_file, url)
async def generate_java_status_image(addr: str):
loop = asyncio.get_event_loop()
@ -37,7 +42,11 @@ async def generate_java_status_image(addr: str):
background_data = await download_image_with_httpx_auto_redirect(BACKGROUND_URL)
if not background_data:
background_data = None
image_data = await get_icon_image(DEFAULT_ICON)
if not image_data:
image_data = None
motd_list = data['motd'].split("\n")
text_list = [
f"ip: {data['ip']}",
@ -58,7 +67,7 @@ async def generate_java_status_image(addr: str):
image = await loop.run_in_executor(None,
create_image,
background_data,
default_icon,
image_data,
text_list,
motd_list)
return image
@ -76,6 +85,10 @@ async def generate_bedrock_status_image(addr: str):
background_data = await download_image_with_httpx_auto_redirect(BACKGROUND_URL)
if not background_data:
background_data = None
image_data = await get_icon_image(DEFAULT_ICON)
if not image_data:
image_data = None
motd_list = data['motd'].split("\n")
text_list = [
@ -88,7 +101,7 @@ async def generate_bedrock_status_image(addr: str):
image = await loop.run_in_executor(None,
create_image,
background_data,
default_icon,
image_data,
text_list,
motd_list)
return image