CM-833: added beneficiary as a matching model to generic fk reporter …

…type (#17)
openimis · May 16, 2024 · 24b06d2 · 24b06d2
1 parent 54e3632
commit 24b06d2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/grievance_social_protection/models.py b/grievance_social_protection/models.py
@@ -8,11 +8,12 @@
 from core import models as core_models
 from core.models import HistoryBusinessModel, User, HistoryModel
 from individual.models import Individual
+from social_protection.models import Beneficiary
 
 
 def check_if_user_or_individual(generic_field):
-    if not isinstance(generic_field, (User, Individual)):
-        raise ValueError('Reporter must be either a User or an Individual.')
+    if not isinstance(generic_field, (User, Individual, Beneficiary)):
+        raise ValueError('Reporter must be either a User or a Beneficiary or an Individual.')
 
 
 class Ticket(HistoryBusinessModel):