"""comfyui_load_skill — lee una receta de *skill* ComfyUI de la libreria de disco. Carga `recipe.json` (version actual) o un snapshot concreto `versions/vN.json` de una skill guardada por `comfyui_save_skill`. Hermana inversa de save: el round-trip save(recipe) -> load(slug) devuelve un dict identico al guardado. `library_dir` por defecto `~/ComfyUI/skills_library`. Un slug inexistente devuelve ``{ok: False}`` sin lanzar excepcion. Impura: lee archivos de disco. """ import json import os DEFAULT_LIBRARY = "~/ComfyUI/skills_library" def _lib_dir(library_dir): return os.path.expanduser(library_dir or DEFAULT_LIBRARY) def _version_filename(version): """Normaliza la version a un nombre de archivo `vN.json`. Acepta int (1), str de digitos ("1") o ya prefijada ("v1"). Devuelve None si no se puede interpretar. """ if isinstance(version, int): return f"v{version}.json" s = str(version).strip() if s.startswith("v"): s = s[1:] if s.isdigit(): return f"v{s}.json" return None def comfyui_load_skill(slug: str, *, version=None, library_dir: str = None) -> dict: """Lee la receta de una skill (version actual o un snapshot concreto). Args: slug: slug de la skill (nombre de su carpeta en la libreria). version: si None, lee `recipe.json` (version actual). Si se pasa (int, "1" o "v1"), lee el snapshot `versions/vN.json`. keyword-only. library_dir: raiz de la libreria. Default `~/ComfyUI/skills_library`. keyword-only. Returns: dict ``{ok, recipe, slug, path, version, error}``. En exito ``ok=True`` y `recipe` es el dict guardado. Si el slug, la version o el archivo no existen, ``ok=False``, ``recipe=None`` y ``error`` describe la causa; nunca lanza. """ if not slug or not isinstance(slug, str): return {"ok": False, "recipe": None, "slug": slug, "path": "", "version": version, "error": "slug requerido (string no vacio)"} lib = _lib_dir(library_dir) skill_dir = os.path.join(lib, slug) if not os.path.isdir(skill_dir): return {"ok": False, "recipe": None, "slug": slug, "path": skill_dir, "version": version, "error": f"skill no encontrada: {slug!r}"} if version is None: target = os.path.join(skill_dir, "recipe.json") else: fname = _version_filename(version) if fname is None: return {"ok": False, "recipe": None, "slug": slug, "path": skill_dir, "version": version, "error": f"version invalida: {version!r}"} target = os.path.join(skill_dir, "versions", fname) if not os.path.isfile(target): return {"ok": False, "recipe": None, "slug": slug, "path": target, "version": version, "error": f"archivo de receta no encontrado: {target}"} try: with open(target, encoding="utf-8") as fh: recipe = json.load(fh) except (OSError, json.JSONDecodeError) as exc: return {"ok": False, "recipe": None, "slug": slug, "path": target, "version": version, "error": f"no se pudo leer la receta: {exc}"} return {"ok": True, "recipe": recipe, "slug": slug, "path": target, "version": version, "error": ""} if __name__ == "__main__": res = comfyui_load_skill("demo_skill", library_dir="/tmp/skills_demo") print(json.dumps(res, indent=2, ensure_ascii=False))