From 1f549dcd52e4f93ebfae8927d1076fc3a42422ec Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Sat, 1 Mar 2025 22:15:58 -0800
Subject: [PATCH] Clean this up

---
 src/documents/consumer.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/documents/consumer.py b/src/documents/consumer.py
index c05072a0b..04ba588d4 100644
--- a/src/documents/consumer.py
+++ b/src/documents/consumer.py
@@ -807,15 +807,16 @@ class ConsumerPlugin(
             set_permissions_for_object(permissions=permissions, object=document)
 
         if self.metadata.custom_fields:
-            for field_id in self.metadata.custom_fields:
-                field = CustomField.objects.get(pk=field_id)
+            for field in CustomField.objects.filter(
+                id__in=self.metadata.custom_fields.keys(),
+            ).distinct():
                 value_field_name = CustomFieldInstance.get_value_field_name(
                     data_type=field.data_type,
                 )
                 args = {
                     "field": field,
                     "document": document,
-                    value_field_name: self.metadata.custom_fields[field_id],
+                    value_field_name: self.metadata.custom_fields.get(field.id, None),
                 }
                 CustomFieldInstance.objects.create(**args)  # adds to document