From 6e116a3f1f3ee8372bd44a5ce9ef0d5f5e4fbe1f Mon Sep 17 00:00:00 2001
From: ppkantorski <ppkantorski@berkeley.edu>
Date: Sun, 7 Jul 2024 02:21:38 -0700
Subject: [PATCH] Create format_repo.py

---
 scripts/Fl4sh9174/format_repo.py | 58 ++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 scripts/Fl4sh9174/format_repo.py

diff --git a/scripts/Fl4sh9174/format_repo.py b/scripts/Fl4sh9174/format_repo.py
new file mode 100644
index 0000000..3eb6407
--- /dev/null
+++ b/scripts/Fl4sh9174/format_repo.py
@@ -0,0 +1,58 @@
+import os
+import zipfile
+import shutil
+import re
+import sys
+
+def unzip_files(folder_path):
+    print("Unzipping files...")
+    for item in os.listdir(folder_path):
+        if item.endswith('.zip'):
+            file_path = os.path.join(folder_path, item)
+            game_name = re.sub(r'\[.*?\]', '', item).replace('.zip', '').strip()
+            extract_to = os.path.join(folder_path, game_name)
+            with zipfile.ZipFile(file_path, 'r') as zip_ref:
+                zip_ref.extractall(extract_to)
+                print(f"Unzipped: {file_path} to {extract_to}")
+
+def create_formatted_structure(folder_path):
+    formatted_path = os.path.join(folder_path, 'formatted')
+    if not os.path.exists(formatted_path):
+        os.makedirs(formatted_path)
+    print(f"Creating formatted structure at: {formatted_path}")
+
+    for game_dir in os.listdir(folder_path):
+        game_dir_path = os.path.join(folder_path, game_dir)
+        if os.path.isdir(game_dir_path) and game_dir != 'formatted':
+            for root, dirs, files in os.walk(game_dir_path):
+                for file in files:
+                    if file.endswith('.pchtxt'):
+                        relative_path = os.path.relpath(root, folder_path)
+                        mod_name_match = re.search(r'\[(.*?)\]', relative_path)
+                        if mod_name_match:
+                            mod_name_with_version = mod_name_match.group(1)
+                            mod_name = re.sub(r' v[0-9.]+$', '', mod_name_with_version).strip()
+                            game_name = game_dir
+                            version = file.replace('.pchtxt', '').strip()
+                            
+                            new_dir = os.path.join(formatted_path, f"{game_name} - {mod_name}")
+                            if not os.path.exists(new_dir):
+                                os.makedirs(new_dir)
+                                print(f"Created directory: {new_dir}")
+                            
+                            shutil.move(os.path.join(root, file), os.path.join(new_dir, f"{version}.pchtxt"))
+                            print(f"Moved {file} to {os.path.join(new_dir, f'{version}.pchtxt')}")
+            shutil.rmtree(game_dir_path)  # Remove the extracted game directory after processing
+
+def main(folder_path):
+    unzip_files(folder_path)
+    create_formatted_structure(folder_path)
+    print("Files have been organized successfully.")
+
+if __name__ == "__main__":
+    if len(sys.argv) != 2:
+        print("Usage: python format_repo.py /path/to/folder/of/zips/")
+        sys.exit(1)
+    
+    folder_path = sys.argv[1]
+    main(folder_path)