mirror of
https://github.com/willmiao/ComfyUI-Lora-Manager.git
synced 2026-08-09 07:20:15 -03:00
8e724538bd
Fix ~950 basedpyright errors across the backend: - Convert ineffective # type: ignore comments to # pyright: ignore[rule] - Add missing generic type arguments (Dict[str, Any], list[Any], ...) - Annotate dynamic dict literals and runtime-initialized attributes - Widen CivitAI provider tuple signatures in recipe parsers - Remove dead LoraRoutes handlers calling nonexistent LoraService methods - Suppress unavoidable ServiceRegistry import cycles (basedpyright counts function-local imports as cycle edges)
30 lines
1021 B
Python
30 lines
1021 B
Python
"""Progress callback implementations backed by the shared WebSocket manager."""
|
|
|
|
from typing import Any, Dict, Protocol
|
|
|
|
from .model_file_service import ProgressCallback
|
|
from .websocket_manager import ws_manager
|
|
|
|
|
|
class ProgressReporter(Protocol):
|
|
"""Protocol representing an async progress callback."""
|
|
|
|
async def on_progress(self, progress_data: Dict[str, Any]) -> None:
|
|
"""Handle a progress update payload."""
|
|
|
|
|
|
class WebSocketProgressCallback(ProgressCallback):
|
|
"""WebSocket implementation of progress callback."""
|
|
|
|
async def on_progress(self, progress_data: Dict[str, Any]) -> None:
|
|
"""Send progress data via WebSocket."""
|
|
await ws_manager.broadcast_auto_organize_progress(progress_data)
|
|
|
|
|
|
class WebSocketBroadcastCallback:
|
|
"""Generic WebSocket progress callback broadcasting to all clients."""
|
|
|
|
async def on_progress(self, payload: Dict[str, Any]) -> None:
|
|
"""Send the provided payload to all connected clients."""
|
|
await ws_manager.broadcast(payload)
|