Add generated TRaSH-Guides with scripts

scripts/utils/profiles.py

@@ -0,0 +1,115 @@
+import os
+import json
+import yaml
+
+def find_score_for_custom_format(trash_score_set, custom_format_name, trash_id, output_dir):
+  custom_formats_dir = os.path.join(output_dir, '..', 'custom_formats')
+  target_file = None
+  for fname in os.listdir(custom_formats_dir):
+      if fname.endswith('.yml') and trash_id in fname:
+          target_file = os.path.join(custom_formats_dir, fname)
+          break
+
+  if not target_file or not os.path.exists(target_file):
+      print(f"Custom format with trash_id {trash_id} not found in {custom_formats_dir}")
+      return 0
+
+  with open(target_file, 'r', encoding='utf-8') as f:
+    data = yaml.safe_load(f)
+
+  if not data or 'trash_id' not in data:
+    print(f"Invalid custom format data for {custom_format_name}")
+    return 0
+
+  if data['trash_id'] != trash_id:
+    # TODO: Better log
+    print(f"Trash ID {trash_id} not found in trash_score_set for custom format {custom_format_name}")
+    return 0
+
+  trash_scores = data.get('trash_scores', {})
+  if not trash_scores:
+    print(f"No trash scores found in {custom_format_name}")
+    return 0
+
+  return trash_scores.get(trash_score_set, trash_scores.get('default', 0))
+
+
+def collect_profile_formats(trash_score_set, format_items, output_dir):
+    profile_format = []
+    for name, trash_id in format_items.items():
+      score = find_score_for_custom_format(trash_score_set, name, trash_id, output_dir)
+      if score == 0:
+          continue
+
+      profile_format.append({
+          'name': name,
+          'score': score
+      })
+    return profile_format
+
+
+def collect_qualities(items):
+  qualities = []
+  quality_id = 1
+  quality_collection_id = -1
+  for item in items:
+      if item.get('allowed', False) is False:
+          continue
+
+      quality = {
+          'name': item.get('name', ''),
+      }
+      if item.get('items') is not None:
+          quality['id'] = quality_collection_id
+          quality_collection_id -= 1
+          quality['description'] = ''
+          quality['qualities'] = []
+          for sub_item in item['items']:
+              quality['qualities'].append({
+                'id': quality_id,
+                'name': sub_item
+              })
+              quality_id += 1
+      else:
+        quality['id'] = quality_id
+        quality_id += 1
+      qualities.append(quality)
+
+  return qualities
+
+def collect_profile(service, file_name, input_json, output_dir):
+    # Compose YAML structure
+    name = input_json.get('name', '')
+    trash_id = input_json.get('trash_id', '')
+    yml_data = {
+        'name': name,
+        'description': f"""Profile from TRaSH-Guides.
+{input_json.get('trash_description', '')}""",
+        'trash_id': trash_id,
+        'tags': [],
+        'upgradesAllowed': input_json.get('upgradeAllowed', True),
+        'minCustomFormatScore': input_json.get('minFormatScore', 0),
+        'upgradeUntilScore': input_json.get('cutoffFormatScore', 0),
+        'minScoreIncrement': input_json.get('minUpgradeFormatScore', 0),
+        'qualities': collect_qualities(input_json.get('items', [])),
+        'custom_formats': collect_profile_formats(input_json.get('trash_score_set'), input_json.get('formatItems', {}), output_dir),
+        'language': input_json.get('language', 'any').lower(),
+    }
+
+    # Output path
+    output_path = os.path.join(output_dir, f"{file_name}-{trash_id}.yml")
+    with open(output_path, 'w', encoding='utf-8') as f:
+        yaml.dump(yml_data, f, sort_keys=False, allow_unicode=True)
+    print(f"Generated: {output_path}")
+
+def collect_profiles(service, input_dir, output_dir):
+    for root, _, files in os.walk(input_dir):
+        for filename in files:
+            if not filename.endswith('.json'):
+                continue
+
+            file_path = os.path.join(root, filename)
+            file_stem = os.path.splitext(filename)[0]  # Filename without extension
+            with open(file_path, 'r', encoding='utf-8') as f:
+                data = json.load(f)
+            collect_profile(service, file_stem, data, output_dir)