[FIX] farmer generator

Author: emjay0921

spp_base_farmer_registry_demo/models/generate_demo_data.py

@@ -187,6 +187,9 @@ def get_individual_vals(self, fake):
         })
 
         return individual_vals
+    
+    def head_member_getter(self, group):
+        return True
 
     def generate_groups(self, fake):
         """Override to include farmer registry related data creation"""

spp_demo_common/models/demo_data_generator.py

@@ -146,6 +146,7 @@ def _generate_demo_data(self, fake):
         have_head_member = False
         new_group_name = False
         for _ in range(num_members):
+            head_membership = self.head_member_getter(group)
             is_head_member = random.choice([True, False]) if not have_head_member else False
 
             # Check if last member and no head member assigned yet
@@ -154,15 +155,20 @@ def _generate_demo_data(self, fake):
 
             individual = self.generate_individuals(fake)
             membership_vals = self.get_group_membership_vals(fake, group, individual)
-            if is_head_member:
-                # Check if the group doesn't have a head member before proceeding
-                if not group.group_membership_ids.filtered(lambda x: x.kind in [self.env.ref("g2p_registry_membership.group_membership_kind_head").id]):
-                    have_head_member = True
-                    new_group_name = individual.family_name
-                    group.name = new_group_name
-                    membership_vals["kind"] = [(4, self.env.ref("g2p_registry_membership.group_membership_kind_head").id)]
+            if is_head_member and not head_membership:
+                have_head_member = True
+                new_group_name = individual.family_name
+                group.name = new_group_name
+                membership_vals["kind"] = [(4, self.env.ref("g2p_registry_membership.group_membership_kind_head").id)]
+                
+            have_head_member = True if head_membership else False
 
             self.env["g2p.group.membership"].create(membership_vals)
+    
+    def head_member_getter(self, group):
+        memberships = self.env["g2p.group.membership"].search([("group", "=", group.id)])
+        head_membership = memberships.filtered(lambda x: x.kind in [self.env.ref("g2p_registry_membership.group_membership_kind_head").id])
+        return head_membership
 
     def _async_generate_demo_data(self):
         jobs = []