Added script to generate [build]_abilities.csv file from balance data exports.

Author: frugs

generate_abilities.py

@@ -0,0 +1,51 @@
+import argparse
+import csv
+import glob
+import os
+import xml.etree.ElementTree
+
+
+def generate_abilities(balance_data_path):
+    abilities = []
+
+    for xml_file_path in glob.glob(os.path.join(balance_data_path, "*.xml")):
+        tree = xml.etree.ElementTree.parse(xml_file_path)
+        root = tree.getroot()
+
+        ability_elements = root.findall(".//abilities/ability")
+        abilities.extend(
+            [
+                (element.get("index"), element.get("id"))
+                for element in ability_elements
+                if element.get("index") and element.get("id")
+            ])
+
+        unit_name = root.get("id")
+
+
+
+    return abilities
+
+
+def main():
+    parser = argparse.ArgumentParser(description='Generate a [BUILD]_abilities.csv file from exported balance data.')
+    parser.add_argument('build_version', metavar='BUILD_VERSION', type=int,
+                        help='the build version of the balance data export')
+    parser.add_argument('balance_data_path', metavar='BALANCE_DATA_PATH', type=str,
+                        help='the path to the balance data export')
+
+    args = parser.parse_args()
+
+    abilities = generate_abilities(args.balance_data_path)
+
+    if not abilities:
+        raise ValueError("No balance data found at provided balance data path.")
+
+    with open('{}_abilities.txt'.format(args.build_version), 'w', newline='') as csvfile:
+        csv_writer = csv.writer(csvfile, delimiter=',')
+        for ability_index, ability_name in abilities:
+            csv_writer.writerow([ability_index, ability_name])
+
+
+if __name__ == "__main__":
+    main()