Python - Generate a SHA3-512 hash with optional salt

hash-sha3-512.py

# Generate a SHA3-512 hash with optional salt

import re

# requires pycryptodome
# - https://pycryptodome.readthedocs.io/en/latest/src/hash/sha3_512.html
from Crypto.Hash import SHA3_512


def generate_hash(s, salt=None):
    """Generate a SHA3-512 hash from a string with optional salt.

    Returns a string containing a 128 character hex digest.
    """
    if salt:
        s += salt
    sha3_hash = SHA3_512.new()
    sha3_hash.update(s.encode())
    return sha3_hash.hexdigest()


# example usage:
if __name__ == "__main__":
    s = "some data"
    salt = "some salt"
    digest = generate_hash(s, salt)
    print(digest)