paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-25 10:49:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

Includes the progress bar

Browse Source
This commit is contained in:
Trenton Holmes 2022-05-31 11:20:10 -07:00 committed by Michael Shamoon
parent dfd16c5187
commit b7250477b5
1 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/documents/management/commands/document_redo_ocr.py
View File

@ -3,6 +3,7 @@ import shutil
from pathlib import Path
from typing import Type
import tqdm
from django.core.exceptions import ObjectDoesNotExist
from django.core.management.base import BaseCommand
from documents.models import Document
@ -21,6 +22,14 @@ class Command(BaseCommand):
)
def add_arguments(self, parser):
parser.add_argument(
"--no-progress-bar",
default=False,
action="store_true",
help="If set, the progress bar will not be shown",
)
parser.add_argument(
"documents",
nargs="+",
@ -33,9 +42,9 @@ class Command(BaseCommand):
all_docs = Document.objects.all()
for doc_pk in args.documents:
for doc_pk in tqdm.tqdm(args.documents, disable=options["no_progress_bar"]):
try:
self.stdout.write(f"Parsing document {doc_pk}")
self.stdout.write(self.style.INFO(f"Parsing document {doc_pk}"))
doc: Document = all_docs.get(pk=doc_pk)
except ObjectDoesNotExist:
self.stdout.write(self.style.ERROR(f"Document {doc_pk} does not exist"))