use partial function

import hashlib

import os.path

from functools import partial

from modules import shared

import modules.cache

def sha256(filename, title, use_addnet_hash=False):

    if shared.opts.experimental_sqlite_cache:

        if use_addnet_hash:

            def calculate_addnet_hash_sqlite3():

                with open(filename, "rb") as file:

                    return addnet_hash_safetensors(file)

            return modules.cache.cached_data_for_file("hashes-addnet", title, filename, calculate_addnet_hash_sqlite3)

            subsection = "hashes-addnet"

            calculate_hash = partial(calculate_addnet_hash, filename)

        else:

            def calculate_sha256_sqlite3():

                hash_sha256 = hashlib.sha256()

                blksize = 1024 * 1024

                with open(filename, "rb") as f:

                    for chunk in iter(lambda: f.read(blksize), b""):

                        hash_sha256.update(chunk)

                return hash_sha256.hexdigest()

            return modules.cache.cached_data_for_file("hashes", title, filename, calculate_sha256_sqlite3)

            subsection = "hashes"

            calculate_hash = partial(calculate_sha256, filename)

        return modules.cache.cached_data_for_file(subsection, title, filename, calculate_hash)

    hashes = cache("hashes-addnet") if use_addnet_hash else cache("hashes")

    return hash_sha256.hexdigest()

def calculate_addnet_hash(filename):

    with open(filename, "rb") as f:

        return addnet_hash_safetensors(f)