zipwalk.py

import zipfile
from io import BytesIO


def zipwalk(current_file, curr_path=''):
    """Recursively walks into a zipped file tree while yielding non-zipped
    files content.
    """
    data = BytesIO(current_file.read())

    try:
        sub_zip = zipfile.ZipFile(data)

    # base case
    except zipfile.BadZipfile:
        yield curr_path, data.getvalue()

    # recurse case
    else:
        for inner_file in sub_zip.filelist:
            new_path = curr_path + '/' + inner_file.filename.lstrip('/')
            for i in zipwalk(sub_zip.open(inner_file), new_path):
                yield i