diff --git a/ArknightsUID/arknightsuid_charsnapshot/get_char_snapshot.py b/ArknightsUID/arknightsuid_charsnapshot/get_char_snapshot.py index b37ecfd..a266aaf 100644 --- a/ArknightsUID/arknightsuid_charsnapshot/get_char_snapshot.py +++ b/ArknightsUID/arknightsuid_charsnapshot/get_char_snapshot.py @@ -153,9 +153,14 @@ async def get_char_snapshot(uid: str): avatar_id = "avatar_activity_AW" else: avatar_id = status.avatar.id_ - avatar_img = Image.open(TEXT_PATH / "ui_player_avatar_list_h2" / f"{avatar_id}.png").resize( - (235, 235) - ) + try: + avatar_img = Image.open( + TEXT_PATH / "ui_player_avatar_list_h2" / f"{avatar_id}.png" + ).resize((235, 235)) + except FileNotFoundError: + avatar_img = Image.open(TEXT_PATH / "ui_char_avatar" / f"{avatar_id}.png").resize( + (235, 235) + ) avatar_fg_draw = ImageDraw.Draw(avatar_fg) avatar_fg_draw.text(