Generate preview image for PDF

1 files changed, 45 insertions, 0 deletions

diff --git a/ishtar_common/utils.py b/ishtar_common/utils.py
index 173338b4f..bfb5d0e32 100644
--- a/ishtar_common/utils.py
+++ b/ishtar_common/utils.py
@@ -2117,3 +2117,48 @@ class IshtarFileSystemStorage(FileSystemStorage):
             if not os.path.exists(os.readlink(path_name)):
                 os.remove(path_name)
         return os.path.exists(path_name)
+
+
+def generate_pdf_preview(item, save=True, tempdir=None, page_number=None):
+    if not settings.PDFTOPPM_BINARY:
+        return
+    preview_tmp_name = None
+    if not item.image and item.source and item.source_page_range:
+        tempdir = tempfile.mkdtemp("-ishtarppmtopdf")
+        try:
+            page_number = int(
+                item.source_page_range.split("-")[0].split(";")[0].split(",")[0])
+        except ValueError:
+            page_number = 1
+        returned = generate_pdf_preview(item.source, save=False, tempdir=tempdir,
+                                        page_number=page_number)
+        if not returned:
+            return
+        tempdir, preview_tmp_name = returned
+    elif (not page_number and item.image) \
+            or not item.associated_file \
+            or not item.associated_file.path \
+            or not item.associated_file.path.lower().endswith(".pdf"):
+        return
+    if not page_number:
+        page_number = 1
+
+    if not tempdir:
+        tempdir = tempfile.mkdtemp("-ishtarppmtopdf")
+    if not preview_tmp_name:
+        preview_tmp_name = tempdir + os.path.sep + "preview"
+        args = (settings.PDFTOPPM_BINARY, "-singlefile", "-jpeg", "-f", str(page_number),
+                item.associated_file.path, preview_tmp_name)
+
+        try:
+            popen = subprocess.Popen(args)
+            popen.wait(timeout=5)
+        except subprocess.SubprocessError:
+            return
+    if not save:
+        return tempdir, preview_tmp_name
+
+    with open(preview_tmp_name + ".jpg", "rb") as preview:
+        django_file = File(preview)
+        getattr(item, "image").save("page.jpg", django_file, save=True)
+    shutil.rmtree(tempdir)