From 50f474ae92b26447fda99a72ee9174a944254bba Mon Sep 17 00:00:00 2001 From: "florian on nixos (Florian Brandes)" Date: Thu, 24 Mar 2022 21:30:34 +0100 Subject: [PATCH] run black Signed-off-by: florian on nixos (Florian Brandes) --- src/documents/tasks.py | 3 +++ src/documents/tests/test_tasks.py | 28 +++++++++++++++++++++------- 2 files changed, 24 insertions(+), 7 deletions(-) diff --git a/src/documents/tasks.py b/src/documents/tasks.py index 2f0fe30c4..4bb3804d2 100644 --- a/src/documents/tasks.py +++ b/src/documents/tasks.py @@ -21,6 +21,7 @@ from pdf2image import convert_from_path from pikepdf import Pdf from pyzbar import pyzbar from whoosh.writing import AsyncWriter + # barcode decoder logger = logging.getLogger("paperless.tasks") @@ -156,6 +157,7 @@ def separate_pages(filepath: str, pages_to_split_on: list) -> list: logger.debug(f"Temp files are {str(document_paths)}") return document_paths + def save_to_dir(filepath, target_dir=settings.CONSUMPTION_DIR): """ Copies filepath to target_dir. @@ -165,6 +167,7 @@ def save_to_dir(filepath, target_dir=settings.CONSUMPTION_DIR): else: logger.warning(f"{str(filepath)} or {str(target_dir)} don't exist.") + def consume_file( path, override_filename=None, diff --git a/src/documents/tests/test_tasks.py b/src/documents/tests/test_tasks.py index 2cf68c216..ab692f6b3 100644 --- a/src/documents/tests/test_tasks.py +++ b/src/documents/tests/test_tasks.py @@ -93,7 +93,9 @@ class TestTasks(DirectoriesMixin, TestCase): def test_barcode_reader(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t.pbm", + os.path.dirname(__file__), + "samples", + "patch-code-t.pbm", ) img = Image.open(test_file) self.assertEqual(tasks.barcode_reader(img), ["b'PATCHT'"]) @@ -105,7 +107,9 @@ class TestTasks(DirectoriesMixin, TestCase): def test_scan_file_for_separating_barcodes(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t.pdf", ) pages = tasks.scan_file_for_separating_barcodes(test_file) self.assertEqual(pages, [0]) @@ -117,21 +121,27 @@ class TestTasks(DirectoriesMixin, TestCase): def test_scan_file_for_separating_barcodes3(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t-middle.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t-middle.pdf", ) pages = tasks.scan_file_for_separating_barcodes(test_file) self.assertEqual(pages, [1]) def test_separate_pages(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t-middle.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t-middle.pdf", ) pages = tasks.separate_pages(test_file, [1]) self.assertEqual(len(pages), 2) def test_save_to_dir(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t.pdf", ) tempdir = tempfile.mkdtemp(prefix="paperless-", dir=settings.SCRATCH_DIR) tasks.save_to_dir(test_file, tempdir) @@ -140,7 +150,9 @@ class TestTasks(DirectoriesMixin, TestCase): def test_save_to_dir2(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t.pdf", ) nonexistingdir = "/nowhere" if os.path.isdir(nonexistingdir): @@ -157,7 +169,9 @@ class TestTasks(DirectoriesMixin, TestCase): def test_barcode_splitter(self): test_file = os.path.join( - os.path.dirname(__file__), "samples", "patch-code-t-middle.pdf", + os.path.dirname(__file__), + "samples", + "patch-code-t-middle.pdf", ) tempdir = tempfile.mkdtemp(prefix="paperless-", dir=settings.SCRATCH_DIR) separators = tasks.scan_file_for_separating_barcodes(test_file)