From 1b4deb068d48b45cea4edb2ecb58ec9b4545b0b9 Mon Sep 17 00:00:00 2001
From: baiqwerdvd <158065462+baiqwerdvd@users.noreply.github.com>
Date: Sun, 26 May 2024 22:27:55 +0800
Subject: [PATCH] fix

---
 .../arknightsuid_charsnapshot/get_char_snapshot.py    | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

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(