学习Python第十五天：第一个程序python程序

第一个程序：ZIP文件口令破解机

编写ZIP文件口令破解机要从学习zipfile库的使用方法着手，打开pythn解释器，我们用help('zipfile')命令进一步了解这个库，并重点看一下zipfile类中的extractall()方法，这个类和这个方法对我们编程破解有口令保护的ZIP文件时很有用的，请注意extractall()方法用可选参数指定密码的方式。

import zipfile
help('zipfile')

让我们快速编写一个脚本来测试一下ZIP文件库的用法，导入库后，用带有口令保护的ZIP文件的文件名，实例化一个ZIPfile类，要解压这个ZIP文件，我们使用extractall()方法，并在可选参数pwd上填入口令。

import zipfile


zfile = zipfile.ZipFile("evil.zip")
zfile.extractall(pwd="secret")

接下来我们要执行脚本以确保其正常运行，注意，在执行前，我们当前的工作目录只有脚本和zip文件，执行脚本后，它会将evil.zip的内容解压到一个名为evil的新创建的目录中，该目录包含有口令保护的zip文件中的文件。

如果用一个错误的密码去执行这个脚本会发生什么呢，让我们在脚本中增加一些捕获和处理异常的代码，显示错误的信息。

import zipfile


zfile = zipfile.ZipFile("evil.zip")
try:
    zfile.extractall(pwd="oranges")
except Exception as e:
    print('e')

这时我们会看到报错了，因此我们可以用因口令不正确抛出的异常来测试我们的字典文件（即代码中的dictionary.txt)中是否有zip文件的口令，实例化一个zipfile类之后，我们打开字典文件，遍历并测试字典中的每个单词，如果extractall()函数的执行没有出错，则打印一条信息，输出正确的口令，如果报错，就忽略这个异常，继续测试下一个口令。

import zipfile


zfile = zipfile.ZipFile("evil.zip")
passfile = open('dictionary.txt')
for line in passfile.readline():
    password = line.strip('\n')
    try:
        zfile.extractall(pwd=password)
        print('password=',password)
        exit(0)
    except Exception as e:
        pass

现在再来清理一下我们的代码，我们要用函数模块化脚本，而非线性执行的程序。

import zipfile

def extractfile(zfile,password):
    try:
        zfile.extractall(pwd=password)
        return password
    except:
        return
    
def main():
    zfile =zipfile.ZipFile('evil.zip')
    passfile = open('dictionary.txx')
    for line in passfile.readline():
        password = line.strip('\n')
        guess = extractfile(zfile,password)
        if guess:
            print("密码是：",password)
if __name__ =='__main__':
    main()

在将程序模块化分离成函数之后，我们现在还能去提高性能，我们可以利用线程同时测试多个口令，而不是只能逐个测试库中的单词，对库中的每个单词，我们都会生成一个新的线程去测试它。

import zipfile
from threading import Thread

def extractfile(zfile,password):
    try:
        zfile.extractall(pwd=password)
        print("密码是：",password)
    except:
        pass
def main():
    zfile = zipfile.ZipFile('evil.zip')
    passfile = open('dictionary.txt')
    for line in passfile.readline():
        password = line.strip('\n')
        t = Thread(target=extractfile,args=(zfile,password))
        t.start
if __name__ =='__main__':
    main()

现在，我们的zip口令破解脚本就算完工啦。今天的学习分享就到这里了，谢谢大家的观看，如果有什么不对的地方还请大家多多指正。