paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-17 10:13:56 -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 json
import logging
import os import os
import shutil import shutil
from contextlib import contextmanager from contextlib import contextmanager
import tqdm
from django.conf import settings from django.conf import settings
from django.core.management import call_command from django.core.management import call_command
from django.core.management.base import BaseCommand, CommandError from django.core.management.base import BaseCommand, CommandError
@ -43,6 +45,8 @@ class Command(Renderable, BaseCommand):
def handle(self, *args, **options): def handle(self, *args, **options):
logging.getLogger().handlers[0].level = logging.ERROR
self.source = options["source"] self.source = options["source"]
if not os.path.exists(self.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.THUMBNAIL_DIR, exist_ok=True)
os.makedirs(settings.ARCHIVE_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": manifest_documents = list(filter(
continue lambda r: r["model"] == "documents.document",
self.manifest))
for record in tqdm.tqdm(manifest_documents):
document = Document.objects.get(pk=record["pk"]) document = Document.objects.get(pk=record["pk"])
@ -138,7 +145,6 @@ class Command(Renderable, BaseCommand):
create_source_path_directory(document.source_path) 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(document_path, document.source_path)
shutil.copy(thumbnail_path, document.thumbnail_path) shutil.copy(thumbnail_path, document.thumbnail_path)
if archive_path: if archive_path: