Add date_time parameter to writestr() too

Arjan Schrijver · Arjan Schrijver · commit a2b5b8f7902c · 2020-09-14T10:48:53.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,11 +2,14 @@
 
 This file details the changes that were made after forking v1.1.4 from https://github.com/allanlei/python-zipstream.
 
-## v1.1.5 (2019-03-18)
-* Support Zip64 when compressing iterables and strings (https://github.com/allanlei/python-zipstream/pull/25)
+## Unreleased
+* New datetime parameter in write_iter (https://github.com/arjan-s/python-zipstream/pull/8)
+
+## v1.1.7 (2019-10-22)
+* Stream data in the order it was received (https://github.com/arjan-s/python-zipstream/pull/4)
 
 ## v1.1.6 (2019-06-06)
 * Add partial flushing of ZipStreams (https://github.com/arjan-s/python-zipstream/pull/1)
 
-## v1.1.7 (2019-10-22)
-* Stream data in the order it was received (https://github.com/arjan-s/python-zipstream/pull/4)
+## v1.1.5 (2019-03-18)
+* Support Zip64 when compressing iterables and strings (https://github.com/allanlei/python-zipstream/pull/25)
diff --git a/zipstream/__init__.py b/zipstream/__init__.py
@@ -225,13 +225,13 @@ def write_iter(self, arcname, iterable, compress_type=None, buffer_size=None, da
         kwargs = {'arcname': arcname, 'iterable': iterable, 'compress_type': compress_type, 'buffer_size': buffer_size, 'date_time': date_time}
         self.paths_to_write.append(kwargs)
 
-    def writestr(self, arcname, data, compress_type=None, buffer_size=None):
+    def writestr(self, arcname, data, compress_type=None, buffer_size=None, date_time=None):
         """
         Writes a str into ZipFile by wrapping data as a generator
         """
         def _iterable():
             yield data
-        return self.write_iter(arcname, _iterable(), compress_type=compress_type, buffer_size=buffer_size)
+        return self.write_iter(arcname, _iterable(), compress_type=compress_type, buffer_size=buffer_size, date_time=date_time)
 
     def __write(self, filename=None, iterable=None, arcname=None, compress_type=None, buffer_size=None, date_time=None):
         """Put the bytes from filename into the archive under the name
