python 随机提取zipfile内容

Posted 2021-05-10

#!/usr/bin/env python

import os
import random
import zipfile


def is_zipfile(f):
    return os.path.isfile(f) and os.path.splitext(f)[1].lower() == ".zip"


def unzip(f, target_dir):
    zfile = zipfile.ZipFile(f)
    selected_filename = random.choice(zfile.namelist())
    (directory, filename) = os.path.split(selected_filename)

    if len(filename):
        print("  %s" % filename)
        zfile.extract(filename, target_dir)
    zfile.close()


target_dir = "samples"
print("extracting to '%s'" % os.path.abspath(target_dir))
for f in os.listdir():
    if is_zipfile(f):
        unzip(f, target_dir)