44 lines
1.0 KiB
Python
44 lines
1.0 KiB
Python
# -*- coding: UTF-8 -*-
|
|
import hashlib
|
|
import requests
|
|
import shutil
|
|
|
|
# print for debugging
|
|
def printD(msg):
|
|
print(f"Civitai Helper: {msg}")
|
|
|
|
|
|
def gen_file_sha256(filname, is_low_memory=True):
|
|
printD("Calculate SHA256")
|
|
hash_sha256 = hashlib.sha256()
|
|
with open(filname, "rb") as f:
|
|
if is_low_memory:
|
|
printD("Using Memory Optimized SHA256")
|
|
for chunk in iter(lambda: f.read(4096), b""):
|
|
hash_sha256.update(chunk)
|
|
else:
|
|
hash_sha256.update(f.read())
|
|
|
|
|
|
hash_value = hash_sha256.hexdigest()
|
|
printD("sha256: " + hash_value)
|
|
return hash_value
|
|
|
|
|
|
# get preview image
|
|
def download_file(url, path):
|
|
printD("Downloading file from: " + url)
|
|
# get file
|
|
r = requests.get(url, stream=True)
|
|
if not r.ok:
|
|
printD("Get error code: " + str(r.status_code))
|
|
printD(r.text)
|
|
return
|
|
|
|
# write to file
|
|
with open(path, 'wb') as f:
|
|
r.raw.decode_content = True
|
|
shutil.copyfileobj(r.raw, f)
|
|
|
|
printD("File downloaded to: " + path)
|