fix TypeErrors with unknown mon level

pokealarm · Apr 8, 2023 · aed43d8 · aed43d8
1 parent 7d684af
commit aed43d8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/PokeAlarm/Events/MonEvent.py b/PokeAlarm/Events/MonEvent.py
@@ -89,7 +89,7 @@ def __init__(self, data):
         self.atk_iv = check_for_none(int, data.get("individual_attack"), Unknown.TINY)
         self.def_iv = check_for_none(int, data.get("individual_defense"), Unknown.TINY)
         self.sta_iv = check_for_none(int, data.get("individual_stamina"), Unknown.TINY)
-        if Unknown.is_not(self.atk_iv, self.def_iv, self.sta_iv):
+        if Unknown.is_not(self.atk_iv, self.def_iv, self.sta_iv, self.mon_lvl):
             self.iv = 100 * (self.atk_iv + self.def_iv + self.sta_iv) / float(45)
             (
                 self.great_product,

diff --git a/PokeAlarm/Utils.py b/PokeAlarm/Utils.py
@@ -752,7 +752,7 @@ def get_xl_candy_costs():
 
 
 def calculate_candy_cost(start_level, target_level, evo_candy_cost=0):
-    if start_level >= target_level:
+    if Unknown.is_(start_level) or start_level >= target_level:
         return (0, 0)
 
     candy_costs = get_candy_costs()
@@ -771,7 +771,7 @@ def calculate_candy_cost(start_level, target_level, evo_candy_cost=0):
 
 
 def calculate_stardust_cost(start_level, target_level):
-    if start_level >= target_level:
+    if Unknown.is_(start_level) or start_level >= target_level:
         return 0
 
     stardust_costs = get_stardust_costs()