From b59f3d9b063a81e677ae09137e6191b7147d30a9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=AD=A3=E8=90=BD?=
 <34079036+jiluoQAQ@users.noreply.github.com>
Date: Thu, 18 May 2023 22:44:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=B1=E6=B8=8A=E6=9F=A5=E8=AF=A2=E5=AE=8C?=
 =?UTF-8?q?=E5=96=84=20(#39)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 StarRailUID/starrailuid_abyss/draw_abyss_card.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/StarRailUID/starrailuid_abyss/draw_abyss_card.py b/StarRailUID/starrailuid_abyss/draw_abyss_card.py
index ef3e5cc..e08bf63 100644
--- a/StarRailUID/starrailuid_abyss/draw_abyss_card.py
+++ b/StarRailUID/starrailuid_abyss/draw_abyss_card.py
@@ -231,8 +231,14 @@ async def draw_abyss_img(
     )
 
     for index_floor, level in enumerate(raw_abyss_data['all_floor_detail']):
-        if index_floor >= 3:
-            break
+        if floor:
+            if abyss_list[str(floor)] == level['name']:
+                index_floor = 0
+            else:
+                continue
+        else:
+            if index_floor >= 3:
+                break
         floor_pic = Image.open(TEXT_PATH / 'floor_bg.png')
         level_star = level['star_num']
         floor_name = level['name']