entzip0r.py

#!/usr/bin/python

import os

import fnmatch

import zipfile

import tempfile

def gen_find(filepat,top):

    for path, dirlist, filelist in os.walk(top):

        for name in fnmatch.filter(filelist,filepat):

            yield os.path.join(path,name)

# zz = file object auf ein .zip file

def gen_open_files_in_zip(zippath, zz):

    myzip = zipfile.ZipFile(zz, 'r')

    # inhalt des zipfiles durchgehen:

    for f in myzip.infolist():

        new_zippath = zippath + '|' + f.filename

        if f.filename.endswith('.zip'):

            # zwischendurch auf die platte speichern

            tmpf = tempfile.TemporaryFile()

            tmpf.write(myzip.open(f).read())

            # rekursion

            temp_gen = gen_open_files_in_zip(new_zippath, tmpf)

            for t in temp_gen:

                yield t[0], t[1]

            tmpf.close() #tempfile wird geloescht

        else:

            yield new_zippath, myzip.open(f)

# gibt einen iterator ueber tupel aus dateiname und geoeffnetem file object oder ZipExtFile zurueck

# auch dateinamen in geschachtelten zipfiles.

def gen_open_files(filenames):

    for f in filenames:

        if f.endswith('.zip'):

            z = open(f, 'r')

            temp = gen_open_files_in_zip(f, z)

            for t in temp:

                yield t

        else:

            yield f, open(f)

if __name__ == '__main__':

    allfiles = gen_find('*.*', '/tmp/ziptests')

    zip_contents = gen_open_files(allfiles)

    for z in zip_contents:

        print z[0], 'enthaelt:', z[1].read().strip()