Create and Extract zip file example

create_and_extract_zip.py

import io
import zipfile

# Create and extract zip file example

# to file
with zipfile.ZipFile("test.zip", 'w') as z:
    z.writestr('aaa/bbb/1.txt', b'aaa')
    z.writestr('aaa/ccc/2.txt', b'ccc')
    z.writestr('aaa/ccc/3.bin', b'\x00\x01\x02')

# to memory
mf = io.BytesIO()
with zipfile.ZipFile(mf, 'w') as z:
    z.writestr('aaa/bbb/1.txt', b'aaa')
    z.writestr('aaa/ccc/2.txt', b'ccc')
    z.writestr('aaa/ccc/3.bin', b'\x00\x01\x02')

assert mf.getvalue() == open("test.zip", "rb").read()

# extract zip
with zipfile.ZipFile("test.zip") as z:
    names = z.namelist()
    assert names == ['aaa/bbb/1.txt', 'aaa/ccc/2.txt', 'aaa/ccc/3.bin']
    values = []
    for name in names:
        with z.open(name) as fp:
            values.append(fp.read())
    assert values == [b'aaa', b'ccc', b'\x00\x01\x02']