Files
Obtainium-Emulation-Pack/scripts/generate-table.py
2026-01-26 23:22:45 -06:00

99 lines
3.4 KiB
Python

"""Generate markdown tables for Obtainium Emulation Pack README."""
import json
import sys
import urllib.parse
from collections import defaultdict
from typing import Any
from utils import get_application_url, get_display_name, should_include_app
def make_obtainium_link(app: dict[str, Any]) -> str:
"""Generate an Obtainium deep-link URL for an app."""
payload = {
"id": app["id"],
"url": app["url"],
"author": app["author"],
"name": app["name"],
"otherAssetUrls": app.get("otherAssetUrls"),
"apkUrls": app.get("apkUrls"),
"preferredApkIndex": app.get("preferredApkIndex"),
"additionalSettings": app.get("additionalSettings"),
"categories": app.get("categories"),
"overrideSource": app.get("overrideSource"),
"allowIdChange": app.get("allowIdChange"),
}
encoded = urllib.parse.quote(json.dumps(payload, separators=(",", ":")), safe="")
return f"http://apps.obtainium.imranr.dev/redirect.html?r=obtainium://app/{encoded}"
def generate_category_tables(apps: list[dict[str, Any]]) -> str:
"""Generate markdown tables grouped by category."""
# Categorize apps
categorized: defaultdict[str, list[dict[str, Any]]] = defaultdict(list)
for app in apps:
categories = app.get("categories", [])
for category in categories:
categorized[category].append(app)
markdown_sections = ["## Applications\n"]
for category in sorted(categorized.keys()):
markdown_sections.append(f"### {category}\n")
markdown_sections.append(
"| Application Name | Add to Obtainium | Included in export json? | Included in DS json? |"
)
markdown_sections.append(
"|------------------|------------------|---------------------------|----------------------|"
)
apps_in_category = sorted(
categorized[category], key=lambda app: get_display_name(app).lower()
)
for app in apps_in_category:
meta = app.get("meta", {})
if meta.get("excludeFromTable", False):
continue
display_name = (
f'<a href="{get_application_url(app)}">{get_display_name(app)}</a>'
)
obtainium_link = make_obtainium_link(app)
badge_md = f'<a href="{obtainium_link}">Add to Obtainium!</a>'
include_standard = "" if should_include_app(app, "standard") else ""
include_dual_screen = (
"" if should_include_app(app, "dual-screen") else ""
)
markdown_sections.append(
f"| {display_name} | {badge_md} | {include_standard} | {include_dual_screen} |"
)
markdown_sections.append("") # blank line between sections
return "\n".join(markdown_sections)
def main(input_file: str, output_file: str) -> None:
"""Generate category-based markdown table from applications.json."""
with open(input_file, "r", encoding="utf-8") as f:
data = json.load(f)
apps = data.get("apps", [])
markdown = generate_category_tables(apps)
with open(output_file, "w", encoding="utf-8") as f:
f.write(markdown)
print(f"Category-based markdown table written to {output_file}")
if __name__ == "__main__":
if len(sys.argv) != 3:
print("Usage: python generate-table.py input.json output.md")
sys.exit(1)
main(sys.argv[1], sys.argv[2])