47 lines
1.9 KiB
Python
47 lines
1.9 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
from pathlib import Path
|
|
|
|
|
|
MEDIA_ROOT = Path(os.environ.get("MEDIA_ROOT", "data/media")).resolve()
|
|
|
|
|
|
def published_assets(entity_type: str | None = None) -> list[dict]:
|
|
manifest = json.loads((MEDIA_ROOT / "manifest.json").read_text(encoding="utf-8"))
|
|
result = []
|
|
for item in manifest.get("assets", []):
|
|
if item.get("status") != "approved" or not item.get("sha256") or not item.get("local_path"):
|
|
continue
|
|
if entity_type and item.get("entity_type") != entity_type:
|
|
continue
|
|
source_page = str(item.get("source_page") or "")
|
|
source = "rf4db" if "rf4db.com" in source_page else "rf4map" if "rf4map.ru" in source_page else "rf4-official"
|
|
result.append({
|
|
"id": item["sha256"],
|
|
"entity_type": item.get("entity_type"),
|
|
"entity_key": item.get("entity_key"),
|
|
"label": item.get("label"),
|
|
"width": item.get("width"),
|
|
"height": item.get("height"),
|
|
"content_type": item.get("content_type"),
|
|
"image_url": f"/api/v1/media/assets/{item['sha256']}",
|
|
"source_system": source,
|
|
"source_url": source_page,
|
|
})
|
|
return sorted(result, key=lambda item: (str(item["entity_type"]), str(item["label"] or "").casefold(), item["id"]))
|
|
|
|
|
|
def published_file(digest: str) -> tuple[Path, str] | None:
|
|
if len(digest) != 64 or any(char not in "0123456789abcdef" for char in digest):
|
|
return None
|
|
manifest = json.loads((MEDIA_ROOT / "manifest.json").read_text(encoding="utf-8"))
|
|
item = next((row for row in manifest.get("assets", []) if row.get("status") == "approved" and row.get("sha256") == digest), None)
|
|
if not item:
|
|
return None
|
|
target = (MEDIA_ROOT / item["local_path"]).resolve()
|
|
if not target.is_relative_to(MEDIA_ROOT.resolve()) or not target.is_file():
|
|
return None
|
|
return target, str(item["content_type"])
|