1580158
/
Stable-Diffusion-Webui-Civitai-Helper
mirror of https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add a clear error message about API key

pull/266/head
man90 2024-01-07 10:56:41 +01:00
parent 3671729087
commit 972ac08904
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
scripts/ch_lib/downloader.py
View File

@ -22,19 +22,21 @@ def dl(url, folder, filename, filepath):
if not folder: if not folder:
util.printD("folder is none") util.printD("folder is none")
return return
if not os.path.isdir(folder): if not os.path.isdir(folder):
util.printD("folder does not exist: "+folder) util.printD("folder does not exist: "+folder)
return return
if filename: if filename:
file_path = os.path.join(folder, filename) file_path = os.path.join(folder, filename)
# first request for header # first request for header
rh = requests.get(url, stream=True, verify=False, headers=util.def_headers, proxies=util.proxies) rh = requests.get(url, stream=True, verify=False, headers=util.def_headers, proxies=util.proxies)
# get file size # get file size
total_size = 0 total_size = int(rh.headers.get('Content-Length', -1))
total_size = int(rh.headers['Content-Length']) if (total_size < 0):
util.printD('This model requires an API key to download. More info: https://github.com/butaixianran/Stable-Diffusion-Webui-Civitai-Helper#civitai-api-key')
return
util.printD(f"File size: {total_size}") util.printD(f"File size: {total_size}")
# if file_path is empty, need to get file name from download url's header # if file_path is empty, need to get file name from download url's header
@ -50,11 +52,11 @@ def dl(url, folder, filename, filepath):
if not filename: if not filename:
util.printD("Fail to get file name from Content-Disposition: " + cd) util.printD("Fail to get file name from Content-Disposition: " + cd)
return return
if not filename: if not filename:
util.printD("Can not get file name from download url's header") util.printD("Can not get file name from download url's header")
return return
# with folder and filename, now we have the full file path # with folder and filename, now we have the full file path
file_path = os.path.join(folder, filename) file_path = os.path.join(folder, filename)
@ -121,4 +123,3 @@ def dl(url, folder, filename, filepath):
os.rename(dl_file_path, file_path) os.rename(dl_file_path, file_path)
util.printD(f"File Downloaded to: {file_path}") util.printD(f"File Downloaded to: {file_path}")
return file_path return file_path