From 80978332cd572101d3a3cd80b0fef7ab49a1f4f2 Mon Sep 17 00:00:00 2001
From: qwerdvd <105906879+qwerdvd@users.noreply.github.com>
Date: Sat, 2 Sep 2023 20:09:52 +0800
Subject: [PATCH] =?UTF-8?q?:bug:=20=E4=BF=AE=E6=AD=A3building.training.tra?=
 =?UTF-8?q?inee=E5=8F=AF=E8=83=BD=E4=B8=BANone=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ArknightsUID/arknightsuid_ap/draw_ap_card.py | 2 +-
 ArknightsUID/utils/models/skland/models.py   | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ArknightsUID/arknightsuid_ap/draw_ap_card.py b/ArknightsUID/arknightsuid_ap/draw_ap_card.py
index b7d7b01..8ebbfc3 100644
--- a/ArknightsUID/arknightsuid_ap/draw_ap_card.py
+++ b/ArknightsUID/arknightsuid_ap/draw_ap_card.py
@@ -275,7 +275,7 @@ async def draw_ap_img(uid: str) -> Image.Image:
         img.paste(grey_bar_bg1_img, (-20, 910), grey_bar_bg1_img)
 
     # training char check
-    if player_info.building.training:
+    if player_info.building.training and player_info.building.training.trainee:
         training_char = player_info.building.training.trainee.charId
         remain_secs = player_info.building.training.remainSecs
         remain_time = 0
diff --git a/ArknightsUID/utils/models/skland/models.py b/ArknightsUID/utils/models/skland/models.py
index b931d02..5305f4d 100644
--- a/ArknightsUID/utils/models/skland/models.py
+++ b/ArknightsUID/utils/models/skland/models.py
@@ -410,13 +410,13 @@ class BuildingLabor(Struct):
 class BuildingTraining(Struct):
     slotId: str
     level: int
-    trainee: BuildingTrainingTrainee
     remainPoint: float
     speed: float
     lastUpdateTime: int
     remainSecs: int
     slotState: int
-    trainer: BuildingTrainingTrainer | None = None
+    trainee: BuildingTrainingTrainee | None 
+    trainer: BuildingTrainingTrainer | None
 
 
 class BuildingHire(Struct):