mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-03-24 14:42:11 -03:00
70 lines
2.6 KiB
Python
70 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Callable, Mapping
|
|
|
|
from aiohttp import web
|
|
|
|
from .example_images_route_registrar import ExampleImagesRouteRegistrar
|
|
from .handlers.example_images_handlers import (
|
|
ExampleImagesDownloadHandler,
|
|
ExampleImagesFileHandler,
|
|
ExampleImagesHandlerSet,
|
|
ExampleImagesManagementHandler,
|
|
)
|
|
from ..utils.example_images_download_manager import DownloadManager
|
|
from ..utils.example_images_file_manager import ExampleImagesFileManager
|
|
from ..utils.example_images_processor import ExampleImagesProcessor
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class ExampleImagesRoutes:
|
|
"""Route controller for example image endpoints."""
|
|
|
|
def __init__(
|
|
self,
|
|
*,
|
|
download_manager=DownloadManager,
|
|
processor=ExampleImagesProcessor,
|
|
file_manager=ExampleImagesFileManager,
|
|
) -> None:
|
|
self._download_manager = download_manager
|
|
self._processor = processor
|
|
self._file_manager = file_manager
|
|
self._handler_set: ExampleImagesHandlerSet | None = None
|
|
self._handler_mapping: Mapping[str, Callable[[web.Request], web.StreamResponse]] | None = None
|
|
|
|
@classmethod
|
|
def setup_routes(cls, app: web.Application) -> None:
|
|
"""Register routes on the given aiohttp application using default wiring."""
|
|
|
|
controller = cls()
|
|
controller.register(app)
|
|
|
|
def register(self, app: web.Application) -> None:
|
|
"""Bind the controller's handlers to the aiohttp router."""
|
|
|
|
registrar = ExampleImagesRouteRegistrar(app)
|
|
registrar.register_routes(self.to_route_mapping())
|
|
|
|
def to_route_mapping(self) -> Mapping[str, Callable[[web.Request], web.StreamResponse]]:
|
|
"""Return the registrar-compatible mapping of handler names to callables."""
|
|
|
|
if self._handler_mapping is None:
|
|
handler_set = self._build_handler_set()
|
|
self._handler_set = handler_set
|
|
self._handler_mapping = handler_set.to_route_mapping()
|
|
return self._handler_mapping
|
|
|
|
def _build_handler_set(self) -> ExampleImagesHandlerSet:
|
|
logger.debug("Building ExampleImagesHandlerSet with %s, %s, %s", self._download_manager, self._processor, self._file_manager)
|
|
download_handler = ExampleImagesDownloadHandler(self._download_manager)
|
|
management_handler = ExampleImagesManagementHandler(self._processor)
|
|
file_handler = ExampleImagesFileHandler(self._file_manager)
|
|
return ExampleImagesHandlerSet(
|
|
download=download_handler,
|
|
management=management_handler,
|
|
files=file_handler,
|
|
)
|