lgcosta/paperless-ngx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Prefetch actions and triggers, add some additional checks before using fields which may be None

Browse Source
This commit is contained in:
Trenton H 2024-03-04 07:18:01 -08:00 committed by shamoon
parent 85d2d47dc1
commit ca40a99c6d
2 changed files with 36 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/documents/consumer.py
View File

@ -7,6 +7,7 @@ from enum import Enum
from pathlib import Path
from subprocess import CompletedProcess
from subprocess import run
from typing import TYPE_CHECKING
from typing import Optional
import magic
@ -66,7 +67,12 @@ class WorkflowTriggerPlugin(
"""
msg = ""
overrides = DocumentMetadataOverrides()
for workflow in Workflow.objects.filter(enabled=True).order_by("order"):
for workflow in (
Workflow.objects.filter(enabled=True)
.prefetch_related("actions")
.prefetch_related("triggers")
.order_by("order")
):
action_overrides = DocumentMetadataOverrides()
if document_matches_workflow(
@ -75,6 +81,8 @@ class WorkflowTriggerPlugin(
WorkflowTrigger.WorkflowTriggerType.CONSUMPTION,
):
for action in workflow.actions.all():
if TYPE_CHECKING:
assert isinstance(action, WorkflowAction)
msg += f"Applying {action} from {workflow}\n"
if action.type == WorkflowAction.WorkflowActionType.ASSIGNMENT:
if action.assign_title is not None:
@ -122,10 +130,7 @@ class WorkflowTriggerPlugin(
# Removal actions overwrite the current overrides
if action.remove_all_tags:
overrides.tag_ids = []
elif (
action.remove_tags.all().count() > 0
and len(overrides.tag_ids) > 0
):
elif action.remove_tags.all().count() > 0 and overrides.tag_ids:
for tag in action.remove_tags.all():
if tag.pk in overrides.tag_ids:
overrides.tag_ids.remove(tag.pk)
@ -164,7 +169,7 @@ class WorkflowTriggerPlugin(
overrides.custom_field_ids = []
elif (
action.remove_custom_fields.all().count() > 0
and len(overrides.custom_field_ids) > 0
and overrides.custom_field_ids
):
for field in action.remove_custom_fields.all():
if field.pk in overrides.custom_field_ids:
@ -186,15 +191,19 @@ class WorkflowTriggerPlugin(
overrides.change_users = []
overrides.change_groups = []
else:
if overrides.view_users:
for user in action.remove_view_users.all():
if user.pk in overrides.view_users:
overrides.view_users.remove(user.pk)
if overrides.change_users:
for user in action.remove_change_users.all():
if user.pk in overrides.change_users:
overrides.change_users.remove(user.pk)
if overrides.view_groups:
for group in action.remove_view_groups.all():
if group.pk in overrides.view_groups:
overrides.view_groups.remove(group.pk)
if overrides.change_groups:
for group in action.remove_change_groups.all():
if group.pk in overrides.change_groups:
overrides.change_groups.remove(group.pk)

9
src/documents/signals/handlers.py
View File

@ -531,10 +531,15 @@ def run_workflow(
document: Document,
logging_group=None,
):
for workflow in Workflow.objects.filter(
for workflow in (
Workflow.objects.filter(
enabled=True,
triggers__type=trigger_type,
).order_by("order"):
)
.prefetch_related("actions")
.prefetch_related("triggers")
.order_by("order")
):
if matching.document_matches_workflow(
document,
workflow,