From 2e0e6bb8d2a05cbbeb6aae908b038ed34c14b746 Mon Sep 17 00:00:00 2001
From: CkuT <cku@gmx.com>
Date: Sat, 6 May 2017 14:32:28 +0200
Subject: [PATCH] Add thumbnail export

---
 src/documents/management/commands/document_exporter.py | 7 +++++++
 src/documents/management/commands/document_importer.py | 6 ++++++
 2 files changed, 13 insertions(+)

diff --git a/src/documents/management/commands/document_exporter.py b/src/documents/management/commands/document_exporter.py
index 1c6ac6e44..8203f930e 100644
--- a/src/documents/management/commands/document_exporter.py
+++ b/src/documents/management/commands/document_exporter.py
@@ -62,7 +62,9 @@ class Command(Renderable, BaseCommand):
             document = document_map[document_dict["pk"]]
 
             target = os.path.join(self.target, document.file_name)
+            thumbnail_target = target + "-tumbnail.png"
             document_dict["__exported_file_name__"] = target
+            document_dict["__exported_thumbnail_name__"] = thumbnail_target
 
             print("Exporting: {}".format(target))
 
@@ -71,6 +73,11 @@ class Command(Renderable, BaseCommand):
                 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))
+
         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
index afa0164ab..be6fbc8bd 100644
--- a/src/documents/management/commands/document_importer.py
+++ b/src/documents/management/commands/document_importer.py
@@ -91,9 +91,15 @@ class Command(Renderable, BaseCommand):
                 continue
 
             doc_file = record["__exported_file_name__"]
+            thumb_file = record["__exported_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(
+                        thumb_file, document.thumbnail_path))
+                    encrypted.write(GnuPG.encrypted(unencrypted))