diff options
-rw-r--r-- | downloadmanager.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/downloadmanager.py b/downloadmanager.py index bde77bf..ec0923f 100644 --- a/downloadmanager.py +++ b/downloadmanager.py @@ -480,12 +480,12 @@ class _SaveDocumentProgressListener(object): def _create_zip(self): bundle = zipfile.ZipFile(self._zip_path, 'w', zipfile.ZIP_DEFLATED) - - bundle.write(os.path.join(self._temp_dir, 'index.html'), 'index.html') - data_path = os.path.join(self._temp_dir, 'data') - for i in os.listdir(data_path): - bundle.write(os.path.join(data_path, i), - os.path.join('data', i)) + + for root, dirs, files in os.walk(self._temp_dir): + for f in files: + bundle.write(os.path.join(self._temp_dir, root, f), + os.path.join(root, f)) + bundle.close() def _create_journal_object(self): |