154 lines
5 KiB
Python
154 lines
5 KiB
Python
import os
|
|
import json
|
|
import yaml
|
|
|
|
from markdownify import markdownify
|
|
|
|
from utils.qualities import QUALITIES
|
|
from utils.strings import get_file_name
|
|
|
|
cache = {}
|
|
|
|
|
|
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
|
|
|
|
if cache.get(trash_id):
|
|
trash_scores = cache[trash_id].get("trash_scores", {})
|
|
if not trash_scores:
|
|
print(f"No trash scores found in cache for {custom_format_name}")
|
|
return 0
|
|
|
|
return trash_scores.get(trash_score_set, trash_scores.get("default", 0))
|
|
|
|
for fname in os.listdir(custom_formats_dir):
|
|
if fname.endswith(".yml"):
|
|
target_file = os.path.join(custom_formats_dir, fname)
|
|
|
|
if not target_file or not os.path.exists(target_file):
|
|
print(f"Target file {target_file} does not exist. Skipping...")
|
|
continue
|
|
|
|
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}")
|
|
continue
|
|
|
|
cache[trash_id] = data
|
|
|
|
if data["trash_id"] != trash_id:
|
|
continue
|
|
|
|
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": get_file_name(name), "score": score})
|
|
return profile_format
|
|
|
|
|
|
def get_quality_id(quality_name):
|
|
return next(
|
|
quality["id"] for quality in QUALITIES if quality["name"] == quality_name
|
|
)
|
|
|
|
|
|
def collect_qualities(items):
|
|
qualities = []
|
|
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": get_quality_id(sub_item), "name": sub_item}
|
|
)
|
|
else:
|
|
quality["id"] = get_quality_id(item.get("name", ""))
|
|
qualities.append(quality)
|
|
|
|
return qualities
|
|
|
|
|
|
def get_upgrade_until(quality_name, profile_qualities):
|
|
found_quality = next(
|
|
quality for quality in profile_qualities if quality["name"] == quality_name
|
|
)
|
|
if found_quality:
|
|
found_quality = found_quality.copy()
|
|
if not found_quality.get("description"):
|
|
found_quality.pop("description", None)
|
|
found_quality.pop("qualities", None)
|
|
return found_quality
|
|
|
|
|
|
def collect_profile(service, input_json, output_dir):
|
|
# Compose YAML structure
|
|
name = input_json.get("name", "")
|
|
trash_id = input_json.get("trash_id", "")
|
|
profile_qualities = collect_qualities(input_json.get("items", []))
|
|
yml_data = {
|
|
"name": get_file_name(name),
|
|
"description": f"""[Profile from TRaSH-Guides.](https://trash-guides.info/{service.capitalize()}/{service}-setup-quality-profiles)
|
|
|
|
{markdownify(input_json.get('trash_description', ''))}""".strip(),
|
|
"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": profile_qualities,
|
|
"upgrade_until": get_upgrade_until(input_json.get("cutoff"), profile_qualities),
|
|
"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"{get_file_name(name)}.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)
|
|
with open(file_path, "r", encoding="utf-8") as f:
|
|
data = json.load(f)
|
|
collect_profile(service, data, output_dir)
|