From 9daac6954ba5bc4ebafaf7416d62f901c911e71f Mon Sep 17 00:00:00 2001
From: PAN Lei <panleicim@gmail.com>
Date: Mon, 17 Jun 2024 20:26:27 +0200
Subject: [PATCH] add ua to contact list

---
 src/person_name/contact_manager.py | 8 ++++----
 src/pojo/contact_pojo.py           | 7 ++++++-
 src/utils/excel_reader.py          | 5 +++++
 3 files changed, 15 insertions(+), 5 deletions(-)

diff --git a/src/person_name/contact_manager.py b/src/person_name/contact_manager.py
index 45dc527..0089854 100755
--- a/src/person_name/contact_manager.py
+++ b/src/person_name/contact_manager.py
@@ -11,7 +11,7 @@ from src.utils.excel_reader import read_contacts, fr_phone_number_prefix, get_ra
 
 
 def upload_contacts_list():
-    _contacts_to_book = read_contacts("/Users/lpan/Desktop/contact_list_2024-05-16.xlsx")
+    _contacts_to_book = read_contacts("/Users/lpan/Desktop/contact_list_2024-06-17.xlsx")
     return _contacts_to_book
 
 
@@ -61,7 +61,7 @@ def write_new_contacts_to_excel(valid_contacts: list, file_name=str(datetime.dat
     col = 0
     # Create a workbook and add a worksheet.
     workbook = xlsxwriter.Workbook('contact_list_{}.xlsx'.format(file_name))
-    header_data = ['name', 'phone', 'passport', 'email', 'store', 'ip_country','ua']
+    header_data = ['name', 'phone', 'passport', 'email', 'store', 'ip_country', 'ua']
     worksheet = workbook.add_worksheet()
     header_format = workbook.add_format({'bold': True})
 
@@ -143,10 +143,10 @@ if __name__ == '__main__':
     # contacts_to_book = upload_contacts_list()
     # MONGO_STORE_MANAGER.upload_contact_list(contacts_to_book)
     # print("start at {}".format(datetime.datetime.now()))
-    generate_valid_contact_list_for_day(segment_number=2)
+    # generate_valid_contact_list_for_day(segment_number=2)
     # generate_contact_from_mail_list("/Users/panlei/Downloads/邮箱及密码.xlsx")
     # print("end at {}".format(datetime.datetime.now()))
-    # generate_all_contact_list()
+    generate_all_contact_list()
     # merge_contact_list_files(
     #     ["/Users/lpan/Desktop/contact_list_2024-05-07.xlsx",
     #      "/Users/lpan/Desktop/contact_list_2024-05-06.xlsx",
diff --git a/src/pojo/contact_pojo.py b/src/pojo/contact_pojo.py
index 7994156..b66a114 100644
--- a/src/pojo/contact_pojo.py
+++ b/src/pojo/contact_pojo.py
@@ -44,7 +44,8 @@ class ContactPojo:
             u'first_name': self.first_name,
             u'mail': self.mail,
             u'store': self.store,
-            u'ip_country': self.ip_country
+            u'ip_country': self.ip_country,
+            u'ua': self.ua
         }
         return dest
 
@@ -67,7 +68,11 @@ class ContactPojo:
         ip_country = "FR"
         if source.get('ip_country'):
             ip_country = source['ip_country']
+        ua = ""
+        if source.get('ua'):
+            ua = source['ua']
         result = ContactPojo(phone_number=phone, passport_number=passport, mail=email,
                              last_name=last_name, first_name=first_name, store=store)
         result.ip_country = ip_country
+        result.ua = ua
         return result
diff --git a/src/utils/excel_reader.py b/src/utils/excel_reader.py
index 7c4e14b..9c784e8 100755
--- a/src/utils/excel_reader.py
+++ b/src/utils/excel_reader.py
@@ -48,6 +48,10 @@ def read_contacts(file_name) -> list:
             store = "random"
             if contact_dict.get('store') is not None:
                 store = contact_dict['store']
+            ua = ""
+
+            if contact_dict.get('ua') is not None:
+                ua = contact_dict['ua']
 
             contact = ContactPojo(phone_number=contact_dict['phone'],
                                   last_name=last_name,
@@ -55,6 +59,7 @@ def read_contacts(file_name) -> list:
                                   passport_number=contact_dict['passport'],
                                   mail=contact_dict['email'], store=store)
             contact.ip_country = ip_country
+            contact.ua = ua
             contact_list.append(contact)
     return contact_list