use related_field_model_attr in Singleton model mixin

clinicedc · Aug 11, 2023 · 32eaa3b · 32eaa3b
1 parent 6c04306
commit 32eaa3b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/edc_crf/modelform_mixins/crf_singleton_model_form_mixin.py b/edc_crf/modelform_mixins/crf_singleton_model_form_mixin.py
@@ -23,10 +23,11 @@ def raise_if_singleton_exists(self: MyForm) -> None:
         timepoint.
         """
         if not self.instance.id:
+            opts = {
+                f"{self.related_visit_model_attr}__subject_identifier": self.subject_identifier
+            }
             try:
-                obj = self._meta.model.objects.get(
-                    subject_visit__subject_identifier=self.subject_identifier
-                )
+                obj = self._meta.model.objects.get(**opts)
             except ObjectDoesNotExist:
                 pass
             else: