57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
import gnupg
|
|
import os
|
|
import time
|
|
|
|
from django.conf import settings
|
|
from django.core.management.base import BaseCommand, CommandError
|
|
|
|
from documents.models import Document
|
|
from paperless.db import GnuPG
|
|
|
|
|
|
class Command(BaseCommand):
|
|
|
|
help = """
|
|
Decrypt and rename all files in our collection into a given target
|
|
directory. Note that we don't export any of the parsed data since
|
|
that can always be re-collected via the consumer.
|
|
""".replace(" ", "")
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument("target")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.verbosity = 0
|
|
self.target = None
|
|
self.gpg = gnupg.GPG(gnupghome=settings.GNUPG_HOME)
|
|
BaseCommand.__init__(self, *args, **kwargs)
|
|
|
|
def handle(self, *args, **options):
|
|
|
|
self.verbosity = options["verbosity"]
|
|
self.target = options["target"]
|
|
|
|
if not os.path.exists(self.target):
|
|
raise CommandError("That path doesn't exist")
|
|
|
|
if not os.access(self.target, os.W_OK):
|
|
raise CommandError("That path doesn't appear to be writable")
|
|
|
|
if not settings.PASSPHRASE:
|
|
settings.PASSPHRASE = input("Please enter the passphrase: ")
|
|
|
|
for document in Document.objects.all():
|
|
|
|
target = os.path.join(self.target, document.parseable_file_name)
|
|
|
|
self._render("Exporting: {}".format(target), 1)
|
|
|
|
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))
|
|
|
|
def _render(self, text, verbosity):
|
|
if self.verbosity >= verbosity:
|
|
print(text)
|