mirror of
https://github.com/paperless-ngx/paperless-ngx.git
synced 2025-09-16 21:55:37 -05:00
Chore: refactor consumer plugin checks to a pre-flight plugin (#9994)
This commit is contained in:
@@ -21,6 +21,7 @@ from django.test import TransactionTestCase
|
||||
from django.test import override_settings
|
||||
|
||||
from documents.consumer import ConsumerPlugin
|
||||
from documents.consumer import ConsumerPreflightPlugin
|
||||
from documents.data_models import ConsumableDocument
|
||||
from documents.data_models import DocumentMetadataOverrides
|
||||
from documents.data_models import DocumentSource
|
||||
@@ -344,12 +345,21 @@ class GetConsumerMixin:
|
||||
) -> Generator[ConsumerPlugin, None, None]:
|
||||
# Store this for verification
|
||||
self.status = DummyProgressManager(filepath.name, None)
|
||||
doc = ConsumableDocument(
|
||||
source,
|
||||
original_file=filepath,
|
||||
mailrule_id=mailrule_id or None,
|
||||
)
|
||||
preflight_plugin = ConsumerPreflightPlugin(
|
||||
doc,
|
||||
overrides or DocumentMetadataOverrides(),
|
||||
self.status, # type: ignore
|
||||
self.dirs.scratch_dir,
|
||||
"task-id",
|
||||
)
|
||||
preflight_plugin.setup()
|
||||
reader = ConsumerPlugin(
|
||||
ConsumableDocument(
|
||||
source,
|
||||
original_file=filepath,
|
||||
mailrule_id=mailrule_id or None,
|
||||
),
|
||||
doc,
|
||||
overrides or DocumentMetadataOverrides(),
|
||||
self.status, # type: ignore
|
||||
self.dirs.scratch_dir,
|
||||
@@ -357,6 +367,7 @@ class GetConsumerMixin:
|
||||
)
|
||||
reader.setup()
|
||||
try:
|
||||
preflight_plugin.run()
|
||||
yield reader
|
||||
finally:
|
||||
reader.cleanup()
|
||||
|
Reference in New Issue
Block a user