Merge pull request #16 from lesthaeghet/master

Bugfix for #15 for compressing large files
allanlei · May 6, 2016 · 2ebe493 · 2ebe493
2 parents d45fd94 + da7d3ed
commit 2ebe493
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/zipstream/__init__.py b/zipstream/__init__.py
@@ -262,7 +262,10 @@ def __write(self, filename=None, iterable=None, arcname=None, compress_type=None
         else:
             zinfo.compress_type = compress_type
 
-        zinfo.file_size = 0
+        if st:
+            zinfo.file_size = st[6]
+        else:
+            zinfo.file_size = 0
         zinfo.flag_bits = 0x00
         zinfo.flag_bits |= 0x08                 # ZIP flag bits, bit 3 indicates presence of data descriptor
         zinfo.header_offset = self.fp.tell()    # Start of header bytes