Cutlery/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-11-03 19:17:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

progress bar for the document importer

Browse Source
This commit is contained in:
jonaswinkler 2020-12-22 15:50:27 +01:00
parent 8dc15352ef
commit ebcb2cf694
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,8 +1,10 @@
import json
import logging
import os
import shutil
from contextlib import contextmanager
import tqdm
from django.conf import settings
from django.core.management import call_command
from django.core.management.base import BaseCommand, CommandError
@ -43,6 +45,8 @@ class Command(Renderable, BaseCommand):
def handle(self, *args, **options):
logging.getLogger().handlers[0].level = logging.ERROR
self.source = options["source"]
if not os.path.exists(self.source):
@ -111,10 +115,13 @@ class Command(Renderable, BaseCommand):
os.makedirs(settings.THUMBNAIL_DIR, exist_ok=True)
os.makedirs(settings.ARCHIVE_DIR, exist_ok=True)
for record in self.manifest:
print("Copy files into paperless...")
if not record["model"] == "documents.document":
continue
manifest_documents = list(filter(
lambda r: r["model"] == "documents.document",
self.manifest))
for record in tqdm.tqdm(manifest_documents):
document = Document.objects.get(pk=record["pk"])
@ -138,7 +145,6 @@ class Command(Renderable, BaseCommand):
create_source_path_directory(document.source_path)
print(f"Moving {document_path} to {document.source_path}")
shutil.copy(document_path, document.source_path)
shutil.copy(thumbnail_path, document.thumbnail_path)
if archive_path: