TBNilles/ComfyUI-Model-Manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove download debug.

Browse Source
This commit is contained in:
Christian Bastian
2024-02-13 01:11:50 -05:00
parent 1fcda7bc17
commit 241e06e17d
1 changed files with 6 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
__init__.py
View File

@@ -391,8 +391,7 @@ def download_file(url, filename, overwrite):
if not overwrite and os.path.isfile(filename): if not overwrite and os.path.isfile(filename):
raise Exception("File already exists!") raise Exception("File already exists!")
# TODO: clear any previous failed partial download file filename_temp = filename + ".download"
dl_filename = filename + ".download"
def_headers = { def_headers = {
"User-Agent": "Mozilla/5.0 (iPad; CPU OS 12_2 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148", "User-Agent": "Mozilla/5.0 (iPad; CPU OS 12_2 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148",
@@ -411,8 +410,8 @@ def download_file(url, filename, overwrite):
raise Exception("Unable to download") raise Exception("Unable to download")
downloaded_size = 0 downloaded_size = 0
if rh.status_code == 200 and os.path.exists(dl_filename): if rh.status_code == 200 and os.path.exists(filename_temp):
downloaded_size = os.path.getsize(dl_filename) downloaded_size = os.path.getsize(filename_temp)
headers = {"Range": "bytes=%d-" % downloaded_size} headers = {"Range": "bytes=%d-" % downloaded_size}
headers["User-Agent"] = def_headers["User-Agent"] headers["User-Agent"] = def_headers["User-Agent"]
@@ -439,16 +438,14 @@ def download_file(url, filename, overwrite):
# Civitai download link # Civitai download link
pass pass
print("temp file is " + dl_filename)
total_size = int(rh.headers.get("Content-Length", 0)) # TODO: pass in total size earlier total_size = int(rh.headers.get("Content-Length", 0)) # TODO: pass in total size earlier
basename, ext = os.path.splitext(filename) print("Download file: " + filename)
print("Start download " + basename)
if total_size != 0: if total_size != 0:
print("Download file size: " + str(total_size)) print("Download file size: " + str(total_size))
mode = "wb" if overwrite else "ab" mode = "wb" if overwrite else "ab"
with open(dl_filename, mode) as f: with open(filename_temp, mode) as f:
for chunk in r.iter_content(chunk_size=1024): for chunk in r.iter_content(chunk_size=1024):
if chunk is not None: if chunk is not None:
downloaded_size += len(chunk) downloaded_size += len(chunk)
@@ -472,13 +469,12 @@ def download_file(url, filename, overwrite):
print() print()
if overwrite and os.path.isfile(filename): if overwrite and os.path.isfile(filename):
os.remove(filename) os.remove(filename)
os.rename(dl_filename, filename) os.rename(filename_temp, filename)
@server.PromptServer.instance.routes.post("/model-manager/download") @server.PromptServer.instance.routes.post("/model-manager/download")
async def download_model(request): async def download_model(request):
body = await request.json() body = await request.json()
json.dump(body, sys.stdout, indent=4)
overwrite = body.get("overwrite", False) overwrite = body.get("overwrite", False)