add trainer id to trainer fragment headers for grunts

original-commit-id: 9ea35422f8e54811a6ad51c06bb4300c146bc832

1 files changed, 5 insertions, 1 deletions

diff --git a/crystal.py b/crystal.py
index 1e836ac..4a14b1c 100644
--- a/crystal.py
+++ b/crystal.py
@@ -3435,11 +3435,15 @@ class TrainerFragment(Command):
         # give this object a possibly better label
         label = "Trainer"
         if "uses_numeric_trainer_ids" in trainer_group_names[trainer_group].keys():
-            label += string.capwords(trainer_group_names[trainer_group]["constant"]).replace("Lt_surge", "Lt_Surge")
+            label += string.capwords(trainer_group_names[trainer_group]["constant"])
+            if len(trainer_group_names[trainer_group]["trainer_names"]) > 1:
+                label += str(trainer_id)
         else:
             label += string.capwords(trainer_group_names[trainer_group]["constant"]) + \
                      string.capwords(trainer_group_names[trainer_group]["trainer_names"][trainer_id-1])
 
+        label = label.replace("Gruntm", "GruntM").replace("Gruntf", "GruntF").replace("Lt_surge", "LtSurge")
+
         self.label = Label(name=label, address=self.address, object=self)
 
         # ---- give better labels to the objects created by TrainerFragment ----