alphard/ComfyUI-Lora-Manager
1
0
Fork 0
mirror of https://github.com/willmiao/ComfyUI-Lora-Manager.git synced 2026-03-25 07:05:43 -03:00
Code Issues Packages Projects Releases Wiki Activity

refactor(example-images): inject websocket manager

Browse Source
This commit is contained in:
pixelpaws
2025-09-23 14:40:43 +08:00
parent 49b7126278
commit 43fcce6361
5 changed files with 191 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
py/routes/example_images_routes.py
View File

@@ -32,21 +32,24 @@ class ExampleImagesRoutes:
def __init__(
self,
*,
ws_manager,
download_manager: DownloadManager | None = None,
processor=ExampleImagesProcessor,
file_manager=ExampleImagesFileManager,
) -> None:
self._download_manager = download_manager or get_default_download_manager()
if ws_manager is None:
raise ValueError("ws_manager is required")
self._download_manager = download_manager or get_default_download_manager(ws_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:
def setup_routes(cls, app: web.Application, *, ws_manager) -> None:
"""Register routes on the given aiohttp application using default wiring."""
controller = cls()
controller = cls(ws_manager=ws_manager)
controller.register(app)
def register(self, app: web.Application) -> None: