check specificly if pet is randomized

ArchipelagoMW · Jun 13, 2024 · 01a83b3 · 01a83b3
1 parent bf608f5
commit 01a83b3
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/worlds/stardew_valley/content/feature/friendsanity.py b/worlds/stardew_valley/content/feature/friendsanity.py
@@ -58,6 +58,10 @@ class FriendsanityFeature(ABC):
     def get_randomized_hearts(self, villager: Villager) -> Tuple[int, ...]:
         ...
 
+    @property
+    def is_pet_randomized(self):
+        return bool(self.get_pet_randomized_hearts())
+
     @abstractmethod
     def get_pet_randomized_hearts(self) -> Tuple[int, ...]:
         ...

diff --git a/worlds/stardew_valley/logic/pet_logic.py b/worlds/stardew_valley/logic/pet_logic.py
@@ -23,7 +23,7 @@ def has_pet_hearts(self, hearts: int = 1) -> StardewRule:
         if hearts == 0:
             return True_()
 
-        if self.content.features.friendsanity.is_enabled:
+        if self.content.features.friendsanity.is_pet_randomized:
             return self.received_pet_hearts(hearts)
 
         return self.can_befriend_pet(hearts)