From eb12cdcb15ba4d4a1c51b782364909c07d30054e Mon Sep 17 00:00:00 2001
From: Karl Hudgell <karl.hudgell@bjss.com>
Date: Mon, 16 Dec 2024 15:55:51 +0000
Subject: [PATCH] build config based on processed_players

---
 lib/generate_xml.py | 34 ++++++++++++++++++++++++++--------
 1 file changed, 26 insertions(+), 8 deletions(-)

diff --git a/lib/generate_xml.py b/lib/generate_xml.py
index d4676c6..ae84c43 100644
--- a/lib/generate_xml.py
+++ b/lib/generate_xml.py
@@ -39,10 +39,30 @@ def add_uid_to_maps(maps, uid, football_manager_version):
     for record in maps.findall("record"):
         if record.attrib.get("from") == str(uid):
             return  # UID already exists
-    ET.SubElement(maps, "record", attrib={"from": str(uid), "to": f"graphics/pictures/person/{prefix}{uid}/portrait"})
+    ET.SubElement(
+        maps,
+        "record",
+        attrib={
+            "from": str(uid),
+            "to": f"graphics/pictures/person/{prefix}{uid}/portrait",
+        },
+    )
 
 
-def create_config_xml(folder_path, football_manager_version):
+def update_player_file_list(folder_path, processed_players):
+    """
+    Updates each player in the list with the full file path and `.png` extension.
+
+    Args:
+        folder_path (str): The path to the folder where the player files are stored.
+        processed_players (list): A list of player names to be updated.
+    """
+    for i in range(len(processed_players)):
+        processed_players[i] = os.path.join(folder_path, f"{processed_players[i]}.png")
+    return processed_players
+
+
+def create_config_xml(folder_path, processed_players, football_manager_version):
     """
     Create a new config.xml file based on .png files in the folder.
     """
@@ -53,10 +73,8 @@ def create_config_xml(folder_path, football_manager_version):
     maps = ET.SubElement(root, "list", id="maps")
 
     # Add records for each .png file
-    for filename in os.listdir(folder_path):
-        if filename.endswith(".png"):
-            uid = os.path.splitext(filename)[0]
-            add_uid_to_maps(maps, uid, football_manager_version)
+    for player in processed_players:
+        add_uid_to_maps(maps, player, football_manager_version)
 
     # Save the XML file
     output_path = os.path.join(folder_path, "config.xml")
@@ -65,7 +83,7 @@ def create_config_xml(folder_path, football_manager_version):
     print(f"Config XML created at: {output_path}")
 
 
-def append_to_config_xml(folder_path, uid_list, football_manager_version):
+def append_to_config_xml(folder_path, processed_players, football_manager_version):
     """
     Append new UIDs to an existing config.xml file.
     """
@@ -76,7 +94,7 @@ def append_to_config_xml(folder_path, uid_list, football_manager_version):
     maps = get_or_create_maps_list(root)
 
     # Add new UIDs
-    for uid in uid_list:
+    for uid in processed_players:
         add_uid_to_maps(maps, uid, football_manager_version)
 
     # Save the updated XML file