使用Python暴力破解压缩文件密码

有一个加密的压缩文件，只记得密码前缀，但知道结尾是4位数字，想用Python暴力破解下。
刚开始想用Python标准库zipfile实现，先创建个密码为password的压缩文件test来测试：

from zipfile import ZipFile

zf = ZipFile("test.zip", "r")
zf.extractall(pwd="password")

发现错误：
TypeError: pwd: expected bytes, got str
期望的pwd是一个bytes类型的，简单，把最后一行改成：
zf.extractall(pwd=b"password")
这下好了，直接Bad Password

查了下，说Python的zipfile库不支持AES加密，解决方案参考：
https://stackoverflow.com/questions/15553150/python-unzip-aes-128-encrypted-file

在这里我是通过执行7z解决的，测试环境为Ubuntu，可通过sudo apt install p7zip-full来安装该命令。
代码如下：

import subprocess
import shutil
import os

passwordPrefix = "something"

for i in ["0"*(4-len(str(x))) + str(x) for x in range(10000)]:
    cmd = "7z x -P{} test.zip".format(passwordPrefix + i)
    
    # 解压失败会产生0字节的文件，所以当文件不再是0字节时说明解压成功
    if os.stat("test/test.txt").st_size == 0:
        shutil.rmtree("test")  # 删除解压失败时产生的目录及目录下的文件
        subprocess.run(cmd, shell=True)
    else:
        print(passwordPrefix + i)
        break