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

refactor: Implement download management routes and update API endpoints for LoRA

Browse Source
This commit is contained in:
Will Miao
2025-07-23 16:11:02 +08:00
parent c2e00b240e
commit ea9370443d
6 changed files with 124 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/utils/routes_common.py
View File

@@ -566,9 +566,10 @@ class ModelRouteUtils:
return web.Response(text=str(e), status=500)
@staticmethod
async def handle_download_model(request: web.Request, download_manager: DownloadManager) -> web.Response:
async def handle_download_model(request: web.Request) -> web.Response:
"""Handle model download request"""
try:
download_manager = await ServiceRegistry.get_download_manager()
data = await request.json()
# Get or generate a download ID
@@ -663,17 +664,17 @@ class ModelRouteUtils:
}, status=500)
@staticmethod
async def handle_cancel_download(request: web.Request, download_manager: DownloadManager) -> web.Response:
async def handle_cancel_download(request: web.Request) -> web.Response:
"""Handle cancellation of a download task
Args:
request: The aiohttp request
download_manager: The download manager instance
Returns:
web.Response: The HTTP response
"""
try:
download_manager = await ServiceRegistry.get_download_manager()
download_id = request.match_info.get('download_id')
if not download_id:
return web.json_response({
@@ -701,17 +702,17 @@ class ModelRouteUtils:
}, status=500)
@staticmethod
async def handle_list_downloads(request: web.Request, download_manager: DownloadManager) -> web.Response:
async def handle_list_downloads(request: web.Request) -> web.Response:
"""Get list of active downloads
Args:
request: The aiohttp request
download_manager: The download manager instance
Returns:
web.Response: The HTTP response with list of downloads
"""
try:
download_manager = await ServiceRegistry.get_download_manager()
result = await download_manager.get_active_downloads()
return web.json_response(result)
except Exception as e: