Wei Yang

import hashlib

import tlsh


def calculate_md5(input_str):
    return hashlib.md5(input_str.encode("utf-8")).hexdigest()


def calculate_sha256(input_str):
    """Calculate SHA256 hash of a string."""
    return hashlib.sha256(input_str.encode("utf-8")).hexdigest()

def calculate_tlsh(data):
    """Calculate TLSH fuzzy hash of data if available.

    Args:
        data: The string data to calculate the hash for

    Returns:
        The TLSH hash as a string, or None if not available

    Note:
        TLSH requires at least 50 bytes of data to generate a hash
    """
    try:
        if len(data) >= 50:
            return tlsh.hash(data.encode("utf-8"))
    except Exception:
        pass
    return None