mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-09-21 03:01:27 -03:00
Will Miao
c8b9db5bf4
Checkpoint entries that cannot be restored by download (deleted, unresolvable hash, or name-only remnants with no CivitAI identifiers) now get the same remediation chain LoRAs already had: - scanner: parameterized reconnect-suggestion ranking, update/restore/ set-hash-invalid for the checkpoint entry, and clear hashInvalid on rematch write-back (was only done for LoRAs) - persistence/handlers/routes: reconnect/restore/reconnect-suggestions/ mark-hash-invalid endpoints under /api/lm/recipe/checkpoint/* - modal: checkpoint reconnect UI (deleted/hash-invalid badges, inline form with suggestions, undo for reconnected entries); download failures mark the hash invalid only on explicit unresolvable signals (not found/deleted/404/410), matching the LoRA rule - css: checkpoint undo button shares the LoRA undo styles - i18n: the 14 new keys translated in all 9 locales
148 lines
6.2 KiB
Python
148 lines
6.2 KiB
Python
"""Route registrar for recipe endpoints."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Callable, Mapping
|
|
|
|
from aiohttp import web
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class RouteDefinition:
|
|
"""Declarative definition for a recipe HTTP route."""
|
|
|
|
method: str
|
|
path: str
|
|
handler_name: str
|
|
|
|
|
|
ROUTE_DEFINITIONS: tuple[RouteDefinition, ...] = (
|
|
RouteDefinition("GET", "/loras/recipes", "render_page"),
|
|
RouteDefinition("GET", "/api/lm/recipes", "list_recipes"),
|
|
RouteDefinition("GET", "/api/lm/recipe/{recipe_id}", "get_recipe"),
|
|
RouteDefinition("GET", "/api/lm/recipes/import-remote", "import_remote_recipe"),
|
|
RouteDefinition("POST", "/api/lm/recipes/analyze-image", "analyze_uploaded_image"),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipes/analyze-local-image", "analyze_local_image"
|
|
),
|
|
RouteDefinition("POST", "/api/lm/recipes/save", "save_recipe"),
|
|
RouteDefinition("DELETE", "/api/lm/recipe/{recipe_id}", "delete_recipe"),
|
|
RouteDefinition("GET", "/api/lm/recipes/top-tags", "get_top_tags"),
|
|
RouteDefinition("GET", "/api/lm/recipes/search-tags", "search_tags"),
|
|
RouteDefinition("GET", "/api/lm/recipes/base-models", "get_base_models"),
|
|
RouteDefinition("GET", "/api/lm/recipes/roots", "get_roots"),
|
|
RouteDefinition("GET", "/api/lm/recipes/folders", "get_folders"),
|
|
RouteDefinition("GET", "/api/lm/recipes/folder-tree", "get_folder_tree"),
|
|
RouteDefinition(
|
|
"GET", "/api/lm/recipes/unified-folder-tree", "get_unified_folder_tree"
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipe/{recipe_id}/share", "share_recipe"),
|
|
RouteDefinition(
|
|
"GET", "/api/lm/recipe/{recipe_id}/share/download", "download_shared_recipe"
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipe/{recipe_id}/syntax", "get_recipe_syntax"),
|
|
RouteDefinition("PUT", "/api/lm/recipe/{recipe_id}/update", "update_recipe"),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/{recipe_id}/opened", "record_recipe_open"
|
|
),
|
|
RouteDefinition("POST", "/api/lm/recipe/move", "move_recipe"),
|
|
RouteDefinition("POST", "/api/lm/recipes/move-bulk", "move_recipes_bulk"),
|
|
RouteDefinition("POST", "/api/lm/recipe/lora/reconnect", "reconnect_lora"),
|
|
RouteDefinition("POST", "/api/lm/recipe/lora/restore", "restore_lora"),
|
|
RouteDefinition(
|
|
"GET",
|
|
"/api/lm/recipe/{recipe_id}/lora/{lora_index}/reconnect-suggestions",
|
|
"get_reconnect_suggestions",
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/lora/mark-hash-invalid", "mark_lora_hash_invalid"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/checkpoint/reconnect", "reconnect_checkpoint"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/checkpoint/restore", "restore_checkpoint"
|
|
),
|
|
RouteDefinition(
|
|
"GET",
|
|
"/api/lm/recipe/{recipe_id}/checkpoint/reconnect-suggestions",
|
|
"get_checkpoint_reconnect_suggestions",
|
|
),
|
|
RouteDefinition(
|
|
"POST",
|
|
"/api/lm/recipe/checkpoint/mark-hash-invalid",
|
|
"mark_checkpoint_hash_invalid",
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipes/find-duplicates", "find_duplicates"),
|
|
RouteDefinition("POST", "/api/lm/recipes/bulk-delete", "bulk_delete"),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipes/save-from-widget", "save_recipe_from_widget"
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipes/for-lora", "get_recipes_for_lora"),
|
|
RouteDefinition(
|
|
"GET", "/api/lm/recipes/for-checkpoint", "get_recipes_for_checkpoint"
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipes/scan", "scan_recipes"),
|
|
RouteDefinition("POST", "/api/lm/recipes/repair", "repair_recipes"),
|
|
RouteDefinition("POST", "/api/lm/recipes/cancel-repair", "cancel_repair"),
|
|
RouteDefinition("POST", "/api/lm/recipe/{recipe_id}/repair", "repair_recipe"),
|
|
RouteDefinition("POST", "/api/lm/recipes/repair-bulk", "repair_recipes_bulk"),
|
|
RouteDefinition("GET", "/api/lm/recipes/repair-progress", "get_repair_progress"),
|
|
RouteDefinition("POST", "/api/lm/recipes/rematch", "rematch_recipes"),
|
|
RouteDefinition("POST", "/api/lm/recipes/rematch-bulk", "rematch_recipes_bulk"),
|
|
RouteDefinition("POST", "/api/lm/recipe/{recipe_id}/rematch", "rematch_recipe"),
|
|
RouteDefinition("POST", "/api/lm/recipes/cancel-rematch", "cancel_rematch"),
|
|
RouteDefinition("GET", "/api/lm/recipes/rematch-progress", "get_rematch_progress"),
|
|
RouteDefinition("POST", "/api/lm/recipes/batch-import/start", "start_batch_import"),
|
|
RouteDefinition(
|
|
"GET", "/api/lm/recipes/batch-import/progress", "get_batch_import_progress"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipes/batch-import/cancel", "cancel_batch_import"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipes/batch-import/directory", "start_directory_import"
|
|
),
|
|
RouteDefinition("POST", "/api/lm/recipes/browse-directory", "browse_directory"),
|
|
RouteDefinition(
|
|
"GET", "/api/lm/recipes/check-image-exists", "check_image_exists"
|
|
),
|
|
RouteDefinition("GET", "/api/lm/recipes/import-from-url", "import_from_url"),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipes/create-from-example", "create_from_example"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/{recipe_id}/reimport", "reimport_recipe"
|
|
),
|
|
RouteDefinition(
|
|
"POST", "/api/lm/recipe/{recipe_id}/send-workflow", "send_recipe_workflow"
|
|
),
|
|
)
|
|
|
|
|
|
class RecipeRouteRegistrar:
|
|
"""Bind declarative recipe definitions to an aiohttp router."""
|
|
|
|
_METHOD_MAP = {
|
|
"GET": "add_get",
|
|
"POST": "add_post",
|
|
"PUT": "add_put",
|
|
"DELETE": "add_delete",
|
|
}
|
|
|
|
def __init__(self, app: web.Application) -> None:
|
|
self._app = app
|
|
|
|
def register_routes(
|
|
self, handler_lookup: Mapping[str, Callable[[web.Request], object]]
|
|
) -> None:
|
|
for definition in ROUTE_DEFINITIONS:
|
|
handler = handler_lookup[definition.handler_name]
|
|
self._bind_route(definition.method, definition.path, handler)
|
|
|
|
def _bind_route(self, method: str, path: str, handler: Callable[..., Any]) -> None:
|
|
add_method_name = self._METHOD_MAP[method.upper()]
|
|
add_method = getattr(self._app.router, add_method_name)
|
|
add_method(path, handler)
|