From 0effcb90d5952bd6c2b8f5a9cbb6fb93ee6c81bc Mon Sep 17 00:00:00 2001 From: quqiOnfree Date: Tue, 26 Aug 2025 03:31:48 +0800 Subject: [PATCH] Fix a tiny bug --- main.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/main.py b/main.py index f878cb5..3237231 100644 --- a/main.py +++ b/main.py @@ -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") \ No newline at end of file + image = generate_java_status_image("mc.hypixel.net") + if image: + image.save("output_image.png") \ No newline at end of file