From 3693a321304397ec934669c29a11dcb3676a8d6a Mon Sep 17 00:00:00 2001
From: xtaodada <xtao@xtaolink.cn>
Date: Wed, 9 Aug 2023 16:55:17 +0800
Subject: [PATCH] :sparkles: Support starrail activity - TreasureDungeon

---
 simnet/models/starrail/chronicle/activity.py | 41 ++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/simnet/models/starrail/chronicle/activity.py b/simnet/models/starrail/chronicle/activity.py
index e0947eb..37b9153 100644
--- a/simnet/models/starrail/chronicle/activity.py
+++ b/simnet/models/starrail/chronicle/activity.py
@@ -77,6 +77,39 @@ class StarRailFantasticStory(StarRailActivityBase):
     records: List[StarRailFantasticStoryRecord]
 
 
+class StarRailTreasureDungeonRecord(APIModel):
+    """Treasure Dungeon Record"""
+
+    stage_id: int
+    name: str
+    difficulty_id: int
+    avatars: List[ActivityCharacter]
+    icon: str
+    atk_buff: int
+    def_buff: int
+    used_stamina: int
+    ancient_weapon: int
+    ancient_armor: int
+    ancient_bomb: int
+    enemy_killed: int
+    finish_time: Optional[PartialTime]
+    special_buff: int
+
+    @property
+    def time_str(self) -> str:
+        """Get the time as a string."""
+        if self.finish_time is None:
+            return "N/A"
+
+        return self.finish_time.datetime.strftime("%Y.%m.%d %H:%M")
+
+
+class StarRailTreasureDungeon(StarRailActivityBase):
+    """Treasure Dungeon"""
+
+    records: List[StarRailTreasureDungeonRecord]
+
+
 class StarRailActivity(APIModel):
     """Starrail chronicle activity."""
 
@@ -105,3 +138,11 @@ class StarRailActivity(APIModel):
             return StarRailFantasticStory(**data["fantastic_story"])
 
         raise ValueError("No fantastic story activity found.")
+
+    @property
+    def treasure_dungeon(self) -> StarRailTreasureDungeon:
+        """Get the treasure dungeon activity."""
+        if data := self.find_activity("treasure_dungeon"):
+            return StarRailTreasureDungeon(**data["treasure_dungeon"])
+
+        raise ValueError("No treasure dungeon activity found.")