Set timestamp for the downloaded book

[m_librarian.git] / m_librarian / download.py

diff --git a/m_librarian/download.py b/m_librarian/download.py
index 75f2b114ba9ce27ed88ed8cc0524c3074b2e5647..361f7aa735549a783e80a0f4adcef6ae67464f43 100755 (executable)
--- a/m_librarian/download.py
+++ b/m_librarian/download.py
@@ -1,6 +1,7 @@
 #! /usr/bin/env python
 
 import os
+from time import mktime
 from shutil import copyfileobj
 from zipfile import ZipFile
 from .config import get_config
@@ -11,15 +12,19 @@ __all__ = ['download']
 _library_path = None
 
 
-def download(archive, filename):
-    global _library_path
-    if _library_path is None:
-        _library_path = get_config().get('library', 'path')
+def download(archive, filename, date, path=None):
+    if path is None:
+        global _library_path
+        if _library_path is None:
+            _library_path = get_config().get('library', 'path')
+        path = _library_path
 
-    zf = ZipFile(os.path.join(_library_path, archive),  'r')
+    zf = ZipFile(os.path.join(path, archive),  'r')
     infile = zf.open(filename)
     outfile = open(filename, 'wb')
     copyfileobj(infile, outfile)
     outfile.close()
     infile.close()
     zf.close()
+    dt = mktime(date.timetuple())
+    os.utime(filename, (dt, dt))