如何在python中实现简化的彩虹表

import hashlib

def generate_rainbow_table():
    rainbow_table = {}
    for i in range(10000):
        password = str(i).zfill(4)  # 将数字转换为4位长度的字符串
        hash_value = hashlib.md5(password.encode()).hexdigest()  # 使用MD5哈希算法计算哈希值
        rainbow_table[hash_value] = password
    return rainbow_table

def crack_password(hash_value, rainbow_table):
    for _ in range(4):  # 假设密码最多为4位
        if hash_value in rainbow_table:
            return rainbow_table[hash_value]
        hash_value = hashlib.md5(hash_value.encode()).hexdigest()
    return "Password not found in the rainbow table."

# 生成彩虹表
rainbow_table = generate_rainbow_table()

# 破解密码
hash_value = "5d41402abc4b2a76b9719d911017c592"  # 假设这是一个密码的哈希值
password = crack_password(hash_value, rainbow_table)
print("Password:", password)