handle load of DataContainer from json properly in terms of cls.lab_level, load cls.village

Author: doluk

coc/abc.py

@@ -22,7 +22,7 @@
 SOFTWARE.
 """
 import ujson
-
+from pathlib import Path
 from typing import AsyncIterator, Any, Dict, Type, Optional, TYPE_CHECKING
 
 from .enums import Resource
@@ -33,6 +33,8 @@
 if TYPE_CHECKING:
     from .players import Player
 
+BUILDING_FILE_PATH = Path(__file__).parent.joinpath(Path("static/buildings.json"))
+
 
 class BaseClan:
     """An ABC that implements some common operations on clans, regardless of type.
@@ -150,8 +152,8 @@ def share_link(self) -> str:
 class DataContainerMetaClass(type):
     def __repr__(cls):
         attrs = [
-            ("name", cls.name),
-            ("id", cls.id),
+            ("name", cls.name if "name" in cls.__dict__ else "Unknown"),
+            ("id", cls.id if "id" in cls.__dict__ else "Unknown"),
         ]
         return "<%s %s>" % (cls.__name__, " ".join("%s=%r" % t for t in attrs),)
 
@@ -187,8 +189,47 @@ def _load_json_meta(cls, troop_meta, id, name, lab_to_townhall):
         cls.ground_target = _get_maybe_first(troop_meta, "GroundTargets", default=True)
         cls.hitpoints = try_enum(UnitStat, troop_meta.get("Hitpoints"))
 
+        # get production building
+        production_building = troop_meta.get("ProductionBuilding", [None])[0]
+        if production_building == "Barrack":
+            cls.is_elixir_troop = True
+        elif production_building == "Dark Elixir Barrack":
+            cls.is_dark_troop = True
+        elif production_building == "SiegeWorkshop":
+            cls.is_siege_machine = True
+        elif production_building == "Spell Forge":
+            cls.is_elixir_spell = True
+        elif production_building == "Mini Spell Factory":
+            cls.is_dark_spell = True
+
+        # load buildings
+        with open(BUILDING_FILE_PATH) as fp:
+            buildings = ujson.load(fp)
+
+        # without production_building, it is a hero or pet
+        if not production_building:
+            laboratory_levels = troop_meta.get("LaboratoryLevel")
+        else:
+            # it is a troop or spell
+            prod_unit = buildings.get(production_building)
+            min_prod_unit_level = troop_meta.get("BarrackLevel", [None, ])[0]
+            # there are some special troops, which have no BarrackLevel attribute
+            if not min_prod_unit_level:
+                laboratory_levels = troop_meta.get("LaboratoryLevel")
+            else:
+                # get the min th level were we can unlock by the required level of the production building
+                min_th_level = [th for i, th in enumerate(prod_unit["TownHallLevel"], start=1)
+                                if i == min_prod_unit_level]
+                # map the min th level to a lab level
+                [first_lab_level] = [lab_level for lab_level, th_level in lab_to_townhall.items()
+                                     if th_level in min_th_level]
+                # the first_lab_level is the lowest possible (there are some inconsistencies with siege machines)
+                # To handle them properly, replacing all lab_level lower than first_lab_level with first_lab_level
+                laboratory_levels = [x if x > first_lab_level else first_lab_level
+                                     for x in troop_meta.get("LaboratoryLevel")]
+
+        cls.lab_level = try_enum(UnitStat, laboratory_levels)
         cls.housing_space = _get_maybe_first(troop_meta, "HousingSpace", default=0)
-        cls.lab_level = try_enum(UnitStat, troop_meta.get("LaboratoryLevel"))
         cls.speed = try_enum(UnitStat, troop_meta.get("Speed"))
         cls.level = cls.dps and UnitStat(range(1, len(cls.dps) + 1))
 
@@ -197,6 +238,7 @@ def _load_json_meta(cls, troop_meta, id, name, lab_to_townhall):
         cls.upgrade_resource = Resource(value=troop_meta["UpgradeResource"][0])
         cls.upgrade_time = try_enum(UnitStat, [TimeDelta(hours=hours) for hours in troop_meta.get("UpgradeTimeH", [])])
         cls._is_home_village = False if troop_meta.get("VillageType") else True
+        cls.village = "home" if cls._is_home_village else "builderBase"
 
         # spells and troops
         cls.training_cost = try_enum(UnitStat, troop_meta.get("TrainingCost"))
@@ -208,18 +250,6 @@ def _load_json_meta(cls, troop_meta, id, name, lab_to_townhall):
         cls.required_th_level = try_enum(UnitStat, troop_meta.get("RequiredTownHallLevel"))
         cls.regeneration_time = try_enum(UnitStat, [TimeDelta(minutes=value) for value in troop_meta.get("RegenerationTimeMinutes", [])])
 
-        production_building = troop_meta.get("ProductionBuilding", [None])[0]
-        if production_building == "Barrack":
-            cls.is_elixir_troop = True
-        elif production_building == "Dark Elixir Barrack":
-            cls.is_dark_troop = True
-        elif production_building == "SiegeWorkshop":
-            cls.is_siege_machine = True
-        elif production_building == "Spell Forge":
-            cls.is_elixir_spell = True
-        elif production_building == "Mini Spell Factory":
-            cls.is_dark_spell = True
-
         cls.is_loaded = True
         return cls