Refactor to get the document time once

IbnNafis007 · May 8, 2017 · 3f13927 · 3f13927
1 parent da71eab
commit 3f13927
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/documents/management/commands/document_exporter.py b/src/documents/management/commands/document_exporter.py
@@ -69,15 +69,14 @@ def dump(self):
 
             print("Exporting: {}".format(target))
 
+            t = int(time.mktime(document.created.timetuple()))
             with open(target, "wb") as f:
                 f.write(GnuPG.decrypted(document.source_file))
-                t = int(time.mktime(document.created.timetuple()))
                 os.utime(target, times=(t, t))
 
             with open(thumbnail_target, "wb") as f:
                 f.write(GnuPG.decrypted(document.thumbnail_file))
-                t = int(time.mktime(document.created.timetuple()))
-                os.utime(target, times=(t, t))
+                os.utime(thumbnail_target, times=(t, t))
 
         manifest += json.loads(
             serializers.serialize("json", Correspondent.objects.all()))

diff --git a/src/documents/management/commands/document_importer.py b/src/documents/management/commands/document_importer.py
@@ -95,11 +95,13 @@ def _import_files_from_manifest(self):
             doc_file = record[EXPORTER_FILE_NAME]
             thumb_file = record[EXPORTER_THUMBNAIL_NAME]
             document = Document.objects.get(pk=record["pk"])
+
             with open(doc_file, "rb") as unencrypted:
                 with open(document.source_path, "wb") as encrypted:
                     print("Encrypting {} and saving it to {}".format(
                         doc_file, document.source_path))
                     encrypted.write(GnuPG.encrypted(unencrypted))
+
             with open(thumb_file, "rb") as unencrypted:
                 with open(document.thumbnail_path, "wb") as encrypted:
                     print("Encrypting {} and saving it to {}".format(