wangbo/ComfyUI-Manager
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI-Manager.git synced 2025-12-17 02:12:58 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix parse aria2 host (#829)

Browse Source
This commit is contained in:
dishuostec 2024-07-01 19:51:16 +08:00 committed by GitHub
parent 17895916f8
commit 96dbc71ad4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
glob/manager_downloader.py
View File

@ -1,11 +1,14 @@
import os import os
from urllib.parse import urlparse
aria2 = os.getenv('COMFYUI_MANAGER_ARIA2_SERVER') aria2 = os.getenv('COMFYUI_MANAGER_ARIA2_SERVER')
HF_ENDPOINT = os.getenv('HF_ENDPOINT') HF_ENDPOINT = os.getenv('HF_ENDPOINT')
if aria2 is not None: if aria2 is not None:
secret = os.getenv('COMFYUI_MANAGER_ARIA2_SECRET') secret = os.getenv('COMFYUI_MANAGER_ARIA2_SECRET')
host, port = aria2.split(':') url = urlparse(aria2)
port = url.port
host = url.scheme + '://' + url.hostname
import aria2p import aria2p
aria2 = aria2p.API(aria2p.Client(host=host, port=port, secret=secret)) aria2 = aria2p.API(aria2p.Client(host=host, port=port, secret=secret))