Generate treasure hoards

questhelper/map/quest_map.py

@@ -3,6 +3,8 @@ import enum
 
 from typing import List
 
+from map.treasure_hoard import TreasureHoard
+
 
 class EncounterType(enum.Enum):
     INACCESSIBLE = 'inaccessible'
@@ -60,6 +62,15 @@ class MapEncounter:
         return str(self.type)
 
 
+@dataclasses.dataclass
+class BattleEncounter(MapEncounter):
+    treasure_hoard: TreasureHoard
+
+    @property
+    def description(self) -> str:
+        return f'{self.type} [{self.treasure_hoard.description}]'
+
+
 class QuestMap:
     def __init__(self):
         # List of rows of encounters, each row has 3 encounters.