# -*- 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 Optimised 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("Download 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)