paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-08-10 00:18:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Format Python code with black

Browse Source
This commit is contained in:
kpj
2022-02-27 15:26:41 +01:00
parent 13885968e3
commit fc695896dd
136 changed files with 6142 additions and 3811 deletions
Download Patch File Download Diff File Expand all files Collapse all files

51
src/documents/management/commands/document_importer.py
View File

@@ -12,8 +12,11 @@ from django.db.models.signals import post_save, m2m_changed
from filelock import FileLock
from documents.models import Document
from documents.settings import EXPORTER_FILE_NAME, EXPORTER_THUMBNAIL_NAME, \
EXPORTER_ARCHIVE_NAME
from documents.settings import (
EXPORTER_FILE_NAME,
EXPORTER_THUMBNAIL_NAME,
EXPORTER_ARCHIVE_NAME,
)
from ...file_handling import create_source_path_directory
from ...signals.handlers import update_filename_and_move_files
@@ -32,7 +35,9 @@ class Command(BaseCommand):
help = """
Using a manifest.json file, load the data from there, and import the
documents it refers to.
""".replace(" ", "")
""".replace(
" ", ""
)
def add_arguments(self, parser):
parser.add_argument("source")
@@ -40,7 +45,7 @@ class Command(BaseCommand):
"--no-progress-bar",
default=False,
action="store_true",
help="If set, the progress bar will not be shown"
help="If set, the progress bar will not be shown",
)
def __init__(self, *args, **kwargs):
@@ -67,26 +72,27 @@ class Command(BaseCommand):
self.manifest = json.load(f)
self._check_manifest()
with disable_signal(post_save,
receiver=update_filename_and_move_files,
sender=Document):
with disable_signal(m2m_changed,
receiver=update_filename_and_move_files,
sender=Document.tags.through):
with disable_signal(
post_save, receiver=update_filename_and_move_files, sender=Document
):
with disable_signal(
m2m_changed,
receiver=update_filename_and_move_files,
sender=Document.tags.through,
):
# Fill up the database with whatever is in the manifest
call_command("loaddata", manifest_path)
self._import_files_from_manifest(options['no_progress_bar'])
self._import_files_from_manifest(options["no_progress_bar"])
print("Updating search index...")
call_command('document_index', 'reindex')
call_command("document_index", "reindex")
@staticmethod
def _check_manifest_exists(path):
if not os.path.exists(path):
raise CommandError(
"That directory doesn't appear to contain a manifest.json "
"file."
"That directory doesn't appear to contain a manifest.json " "file."
)
def _check_manifest(self):
@@ -98,15 +104,15 @@ class Command(BaseCommand):
if EXPORTER_FILE_NAME not in record:
raise CommandError(
'The manifest file contains a record which does not '
'refer to an actual document file.'
"The manifest file contains a record which does not "
"refer to an actual document file."
)
doc_file = record[EXPORTER_FILE_NAME]
if not os.path.exists(os.path.join(self.source, doc_file)):
raise CommandError(
'The manifest file refers to "{}" which does not '
'appear to be in the source directory.'.format(doc_file)
"appear to be in the source directory.".format(doc_file)
)
if EXPORTER_ARCHIVE_NAME in record:
@@ -125,14 +131,11 @@ class Command(BaseCommand):
print("Copy files into paperless...")
manifest_documents = list(filter(
lambda r: r["model"] == "documents.document",
self.manifest))
manifest_documents = list(
filter(lambda r: r["model"] == "documents.document", self.manifest)
)
for record in tqdm.tqdm(
manifest_documents,
disable=progress_bar_disable
):
for record in tqdm.tqdm(manifest_documents, disable=progress_bar_disable):
document = Document.objects.get(pk=record["pk"])