You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
24 lines
573 B
24 lines
573 B
"""
|
|
Generate zip test data files.
|
|
"""
|
|
|
|
import zipfile
|
|
|
|
|
|
def make_zip_file(tree, dst):
|
|
"""
|
|
Zip the files in tree into a new zipfile at dst.
|
|
"""
|
|
with zipfile.ZipFile(dst, 'w') as zf:
|
|
for name, contents in walk(tree):
|
|
zf.writestr(name, contents)
|
|
zipfile._path.CompleteDirs.inject(zf)
|
|
return dst
|
|
|
|
|
|
def walk(tree, prefix=''):
|
|
for name, contents in tree.items():
|
|
if isinstance(contents, dict):
|
|
yield from walk(contents, prefix=f'{prefix}{name}/')
|
|
else:
|
|
yield f'{prefix}{name}', contents
|
|
|