Skip to content

Commit 80c8b08

Browse files
committed
[IMP] phone numbers possible to have more than 1
1 parent df953f2 commit 80c8b08

1 file changed

Lines changed: 14 additions & 12 deletions

File tree

spp_demo_common/models/demo_data_generator.py

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -528,18 +528,20 @@ def generate_phone_number(self, fake):
528528
return cleaned
529529

530530
def create_phone_numbers(self, fake, registrant):
531-
phone_number = self.generate_phone_number(fake)
532-
date_collected = self.get_random_date(
533-
fake,
534-
datefrom=registrant.registration_date,
535-
dateto=fields.Date.today(),
536-
)
537-
phone_vals = {
538-
"partner_id": registrant.id,
539-
"phone_no": phone_number,
540-
"date_collected": date_collected,
541-
}
542-
self.env["g2p.phone.number"].create(phone_vals)
531+
num_phone_numbers = random.randint(1, 5)
532+
for _ in range(num_phone_numbers):
533+
phone_number = self.generate_phone_number(fake)
534+
date_collected = self.get_random_date(
535+
fake,
536+
datefrom=registrant.registration_date,
537+
dateto=fields.Date.today(),
538+
)
539+
phone_vals = {
540+
"partner_id": registrant.id,
541+
"phone_no": phone_number,
542+
"date_collected": date_collected,
543+
}
544+
self.env["g2p.phone.number"].create(phone_vals)
543545
registrant.phone_number_ids_change()
544546

545547
def create_gps_coordinates(self, fake, registrant):

0 commit comments

Comments
 (0)