1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
# -*- coding: utf-8 -*-
import os
import zipstream
import zipfile
with zipstream.ZipFile(mode='w', compression=zipstream.ZIP_DEFLATED) as z:
z.write('LICENSE')
z.write('LICENSE', arcname='stuff/LICENSE')
with open('tests/sample.rtf', 'rb') as fp:
z.writestr('sample.rtf', fp.read())
for root, directories, files in os.walk('zipstream'):
for filename in files:
path = os.path.join(root, filename)
z.write(path, path)
with open('test.zip', 'wb') as f:
for chunk in z:
f.write(chunk)
with zipfile.ZipFile('test.zip') as z:
z.testzip()
|
