paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-05-23 12:58:18 -05:00
Code Issues Packages Projects Releases Wiki Activity

Use match for date field conditionals

Browse Source
This commit is contained in:
shamoon 2024-11-22 13:30:02 -08:00
parent 3a3dd3608e
commit 8ba5b535c3
1 changed files with 21 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

53
src/documents/tasks.py
View File

@ -345,38 +345,27 @@ def check_scheduled_workflows():
logger.debug(
f"Checking trigger {trigger} with offset {offset_td} against field: {trigger.schedule_date_field}",
)
if (
trigger.schedule_date_field
== WorkflowTrigger.ScheduleDateField.ADDED
):
documents = Document.objects.filter(
added__lt=timezone.now() - offset_td,
)
elif (
trigger.schedule_date_field
== WorkflowTrigger.ScheduleDateField.CREATED
):
documents = Document.objects.filter(
created__lt=timezone.now() - offset_td,
)
elif (
trigger.schedule_date_field
== WorkflowTrigger.ScheduleDateField.MODIFIED
):
documents = Document.objects.filter(
modified__lt=timezone.now() - offset_td,
)
elif (
trigger.schedule_date_field
== WorkflowTrigger.ScheduleDateField.CUSTOM_FIELD
):
cf_instances = CustomFieldInstance.objects.filter(
field=trigger.schedule_date_custom_field,
value_date__lt=timezone.now() - offset_td,
)
documents = Document.objects.filter(
id__in=cf_instances.values_list("document", flat=True),
)
match trigger.schedule_date_field:
case WorkflowTrigger.ScheduleDateField.ADDED:
documents = Document.objects.filter(
added__lt=timezone.now() - offset_td,
)
case WorkflowTrigger.ScheduleDateField.CREATED:
documents = Document.objects.filter(
created__lt=timezone.now() - offset_td,
)
case WorkflowTrigger.ScheduleDateField.MODIFIED:
documents = Document.objects.filter(
modified__lt=timezone.now() - offset_td,
)
case WorkflowTrigger.ScheduleDateField.CUSTOM_FIELD:
cf_instances = CustomFieldInstance.objects.filter(
field=trigger.schedule_date_custom_field,
value_date__lt=timezone.now() - offset_td,
)
documents = Document.objects.filter(
id__in=cf_instances.values_list("document", flat=True),
)
if documents.count() > 0:
logger.debug(
f"Found {documents.count()} documents for trigger {trigger}",