Update basic_contact.py

george0st · Apr 16, 2024 · 5506269 · 5506269
1 parent fa7d4f4
commit 5506269
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/generator/basic_contact.py b/generator/basic_contact.py
@@ -53,10 +53,16 @@ def generate(self, count):
                     if self.rnd_bool() else ContactEnum.Phone
 
                 # "name": "contact-email"
-                model['contact-email']=self.fake.email() if contact_detail & ContactEnum.Email else ""
+                if contact_detail & ContactEnum.Email:
+                    model['contact-email'] = self.fake.email()
+                else:
+                    model['contact-email'] = ""
 
                 # "name": "contact-phone"
-                model['contact-phone']=self.fake.phone_number() if contact_detail & ContactEnum.Phone else ""                     
+                if contact_detail & ContactEnum.Phone:
+                    model['contact-phone'] = self.fake.phone_number()
+                else:
+                    model['contact-phone'] = ""
 
                 # "name": "contact-state"
                 if count==0:
@@ -70,4 +76,3 @@ def generate(self, count):
                 model['record-date']=self.gmodel["NOW"]
 
                 self.model.append(model)
-