Merge branch 'use-stableid-json'

Author: Reynald Pader

examples/zerg_rush.py

@@ -34,7 +34,7 @@ async def on_step(self, iteration):
         for queen in self.units(QUEEN).idle:
             abilities = await self.get_available_abilities(queen)
             if AbilityId.INJECTLARVA in abilities:
-                await self.do(queen(INJECTLARVA, hatchery))
+                await self.do(queen(EFFECT_INJECTLARVA, hatchery))
 
         if self.vespene >= 100:
             sp = self.units(SPAWNINGPOOL).ready

generate_id_constants.py

@@ -16,7 +16,7 @@ def clike_enum_parse(code):
 
     # parse enum blocks
     enums = {}
-    for enum in re.findall(r"enum(?: class)? ([a-zA-Z_][a-zA-Z0-9_]*) \{\s?(.+?)\s?}", code):
+    for enum in re.findall(r"enum(?: class)? ([a-zA-Z_][a-zA-Z0-9_]*) {\s?(.+?)\s?}", code):
         name, body = enum
         body = {key: int(value) for key, value in re.findall(r"([a-zA-Z_][a-zA-Z0-9_]*) = (\d+),?\s?", body)}
         enums[name] = body

generate_id_constants_from_stableid.py

@@ -0,0 +1,132 @@
+import json
+from pathlib import Path
+import platform
+
+HEADER = f"# DO NOT EDIT!\n# This file was automatically generated by \"{__file__}\"\n"
+
+PF = platform.system()
+
+HOME_DIR = str(Path.home())
+DATA_JSON = {
+    "Darwin": HOME_DIR + "/Library/Application Support/Blizzard/StarCraft II/stableid.json",
+    "Windows": HOME_DIR + "/Documents/StarCraft II/stableid.json"
+}
+
+ENUM_TRANSLATE = {
+    "Units": "UnitTypeId",
+    "Abilities": "AbilityId",
+    "Upgrades": "UpgradeId",
+    "Buffs": "BuffId",
+    "Effects": "EffectId"
+}
+
+FILE_TRANSLATE = {
+    "Units": "unit_typeid",
+    "Abilities": "ability_id",
+    "Upgrades": "upgrade_id",
+    "Buffs": "buff_id",
+    "Effects": "effect_id"
+}
+
+
+def clike_enum_parse(data):
+    # for d in data:  # Units, Abilities, Upgrades, Buffs, Effects
+
+    units = parse_simple('Units', data)
+    upgrades = parse_simple('Upgrades', data)
+    effects = parse_simple('Effects', data)
+    buffs = parse_simple('Buffs', data)
+
+    abilities = {}
+    for v in data['Abilities']:
+        key = v['buttonname']
+
+        if not key:
+            continue
+
+        key = key.upper().replace(" ", "_")
+
+        if 'name' in v:
+            key = "{}_{}".format(v['name'].upper().replace(" ", "_"), key)
+
+        if 'friendlyname' in v:
+            key = v['friendlyname'].upper().replace(" ", "_")
+
+        if key[0].isdigit():
+            key = "_" + key
+
+        if key in abilities and v['index'] == 0:
+            print(key)
+            raise ValueError
+        abilities[key] = v['id']
+
+    abilities['SMART'] = 1
+
+    enums = {}
+    enums["Units"] = units
+    enums["Abilities"] = abilities
+    enums["Upgrades"] = upgrades
+    enums["Buffs"] = buffs
+    enums["Effects"] = effects
+
+    return enums
+
+
+def parse_simple(d, data):
+    units = {}
+    for v in data[d]:
+        key = v['name']
+
+        if not key:
+            continue
+
+        if key[0].isdigit():
+            key = "_" + key
+
+        key = key.upper().replace(" ", "_")
+        units[key] = v['id']
+    return units
+
+
+def generate_python_code(enums):
+    assert {"Units", "Abilities", "Upgrades", "Buffs", "Effects"} <= enums.keys()
+
+    sc2dir = Path("sc2/")
+    idsdir = (sc2dir / "ids")
+    idsdir.mkdir(exist_ok=True)
+
+    with (idsdir / "__init__.py").open("w") as f:
+        f.write("\n".join([
+            HEADER,
+            f"__all__ = {[n.lower() for n in FILE_TRANSLATE.values()] !r}\n"
+        ]))
+
+    for name, body in enums.items():
+        class_name = ENUM_TRANSLATE[name]
+
+        code = [
+            HEADER,
+            "import enum",
+            "",
+            f"class {class_name}(enum.Enum):"
+        ]
+
+        for key, value in sorted(body.items(), key=lambda p: p[1]):
+            code.append(f"    {key} = {value}")
+
+        code += [
+            "",
+            f"for item in {class_name}:",
+            f"    assert not item.name in globals()",
+            f"    globals()[item.name] = item",
+            ""
+        ]
+
+        with (idsdir / FILE_TRANSLATE[name]).with_suffix(".py").open("w") as f:
+            f.write("\n".join(code))
+
+
+if __name__ == '__main__':
+    with open(DATA_JSON[PF], encoding='utf-8') as data_file:
+        data = json.loads(data_file.read())
+        generate_python_code(clike_enum_parse(data))

sc2/constants.py

@@ -1,2 +1,5 @@
-from .ids.unit_typeid import *
 from .ids.ability_id import *
+from .ids.buff_id import *
+from .ids.effect_id import *
+from .ids.unit_typeid import *
+from .ids.upgrade_id import *

sc2/ids/__init__.py

@@ -1,4 +1,4 @@
 # DO NOT EDIT!
-# This file was automatically generated by "generate_id_constants.py"
+# This file was automatically generated by "generate_id_constants_from_stableid.py"
 
-__all__ = ['unit_typeid', 'ability_id', 'upgrade_id', 'buff_id']
+__all__ = ['unit_typeid', 'ability_id', 'upgrade_id', 'buff_id', 'effect_id']