paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-02 13:45:10 -05:00
Code Issues Packages Projects Releases Wiki Activity

Minor tweaks to getting the document thumbnail path. Adds text thumbnail as webp

Browse Source
This commit is contained in:
Trenton Holmes 2022-06-10 06:56:28 -07:00
parent 58f2c6a5fc
commit 6844f8f2bf
2 changed files with 17 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/documents/models.py
View File

@ -3,6 +3,7 @@ import logging
import os import os
import re import re
from collections import OrderedDict from collections import OrderedDict
from typing import Optional
import dateutil.parser import dateutil.parser
import pathvalidate import pathvalidate
@ -228,7 +229,7 @@ class Document(models.Model):
verbose_name = _("document") verbose_name = _("document")
verbose_name_plural = _("documents") verbose_name_plural = _("documents")
def __str__(self): def __str__(self) -> str:
# Convert UTC database time to local time # Convert UTC database time to local time
created = datetime.date.isoformat(timezone.localdate(self.created)) created = datetime.date.isoformat(timezone.localdate(self.created))
@ -242,7 +243,7 @@ class Document(models.Model):
return res return res
@property @property
def source_path(self): def source_path(self) -> str:
if self.filename: if self.filename:
fname = str(self.filename) fname = str(self.filename)
else: else:
@ -257,11 +258,11 @@ class Document(models.Model):
return open(self.source_path, "rb") return open(self.source_path, "rb")
@property @property
def has_archive_version(self): def has_archive_version(self) -> bool:
return self.archive_filename is not None return self.archive_filename is not None
@property @property
def archive_path(self): def archive_path(self) -> Optional[str]:
if self.has_archive_version: if self.has_archive_version:
return os.path.join(settings.ARCHIVE_DIR, str(self.archive_filename)) return os.path.join(settings.ARCHIVE_DIR, str(self.archive_filename))
else: else:
@ -271,7 +272,7 @@ class Document(models.Model):
def archive_file(self): def archive_file(self):
return open(self.archive_path, "rb") return open(self.archive_path, "rb")
def get_public_filename(self, archive=False, counter=0, suffix=None): def get_public_filename(self, archive=False, counter=0, suffix=None) -> str:
result = str(self) result = str(self)
if counter: if counter:
@ -292,17 +293,18 @@ class Document(models.Model):
return get_default_file_extension(self.mime_type) return get_default_file_extension(self.mime_type)
@property @property
def thumbnail_path(self): def thumbnail_path(self) -> str:
file_name = f"{self.pk:07}.webp" png_file_name = f"{self.pk:07}.png"
webp_file_name = f"{self.pk:07}.webp"
if self.storage_type == self.STORAGE_TYPE_GPG: if self.storage_type == self.STORAGE_TYPE_GPG:
file_name += ".gpg" png_file_name += ".gpg"
webp_file_name += ".gpg"
thumb = os.path.join(settings.THUMBNAIL_DIR, file_name) thumb = os.path.join(settings.THUMBNAIL_DIR, webp_file_name)
if os.path.exists(thumb): if not os.path.exists(thumb):
return thumb thumb = os.path.join(settings.THUMBNAIL_DIR, png_file_name)
else: return thumb
return os.path.splitext(thumb)[0] + ".png"
@property @property
def thumbnail_file(self): def thumbnail_file(self):

4
src/paperless_text/parsers.py
View File

@ -30,8 +30,8 @@ class TextDocumentParser(DocumentParser):
) )
draw.text((5, 5), read_text(), font=font, fill="black") draw.text((5, 5), read_text(), font=font, fill="black")
out_path = os.path.join(self.tempdir, "thumb.png") out_path = os.path.join(self.tempdir, "thumb.webp")
img.save(out_path) img.save(out_path, format="WEBP")
return out_path return out_path