Fix a tiny bug

This commit is contained in:
quqiOnfree 2025-08-26 03:31:48 +08:00
parent 57a741ca17
commit 0effcb90d5

14
main.py
View file

@ -14,7 +14,7 @@ import base64
BACKGROUND_URL = "https://www.loliapi.com/acg/"
def generate_java_status_image(addr: str, output_image_path: str):
def generate_java_status_image(addr: str):
try:
ip, type = dns_lookup(addr)
status = java_status(ip)
@ -36,9 +36,13 @@ def generate_java_status_image(addr: str, output_image_path: str):
f"players: {data['players']['online']}/{data['players']['max']}",
]
image = create_image(background_data, base64.b64decode(status.icon.split(",")[1]), text_list, motd_list)
image.save(output_image_path)
if status.icon:
image = create_image(background_data, base64.b64decode(status.icon.split(",")[1]), text_list, motd_list)
else:
image = create_image(background_data, None, text_list, motd_list)
return image
if __name__ == "__main__":
generate_java_status_image("mc.hypixel.net", "output_image.png")
image = generate_java_status_image("mc.hypixel.net")
if image:
image.save("output_image.png")