smokephil/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-12-19 21:16:56 +01:00
Code Issues Projects Releases Packages Wiki Activity

Use constants for manifest

Browse source
This commit is contained in:
CkuT 2017-05-08 14:54:48 +02:00
parent 2e0e6bb8d2
commit da71eab0ae
4 changed files with 15 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/documents/management/commands/document_exporter.py
View file

@ -10,6 +10,7 @@ from documents.models import Document, Correspondent, Tag
from paperless.db import GnuPG
from ...mixins import Renderable
from documents.settings import EXPORTER_FILE_NAME, EXPORTER_THUMBNAIL_NAME
class Command(Renderable, BaseCommand):
@ -63,8 +64,8 @@ class Command(Renderable, BaseCommand):
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
document_dict[EXPORTER_FILE_NAME] = target
document_dict[EXPORTER_THUMBNAIL_NAME] = thumbnail_target
print("Exporting: {}".format(target))

10
src/documents/management/commands/document_importer.py
View file

@ -10,6 +10,8 @@ from paperless.db import GnuPG
from ...mixins import Renderable
from documents.settings import EXPORTER_FILE_NAME, EXPORTER_THUMBNAIL_NAME
class Command(Renderable, BaseCommand):
@ -70,13 +72,13 @@ class Command(Renderable, BaseCommand):
if not record["model"] == "documents.document":
continue
if "__exported_file_name__" not in record:
if EXPORTER_FILE_NAME not in record:
raise CommandError(
'The manifest file contains a record which does not '
'refer to an actual document file.'
)
doc_file = record["__exported_file_name__"]
doc_file = record[EXPORTER_FILE_NAME]
if not os.path.exists(doc_file):
raise CommandError(
'The manifest file refers to "{}" which does not '
@ -90,8 +92,8 @@ class Command(Renderable, BaseCommand):
if not record["model"] == "documents.document":
continue
doc_file = record["__exported_file_name__"]
thumb_file = record["__exported_thumbnail_name__"]
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: