33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
import json
|
|
|
|
import pandas as pandas
|
|
from definitions import CONTACT_LIST_FILE
|
|
from pojo.contact_pojo import ContactPojo
|
|
|
|
|
|
class ExcelHelper:
|
|
|
|
def __init__(self):
|
|
self._df = pandas.Series()
|
|
|
|
def write_to_exel(self, file_name, data_list: list):
|
|
new_df = pandas.Series(data_list)
|
|
self._df = pandas.concat([self._df, new_df])
|
|
self._df.to_excel(file_name)
|
|
|
|
def read_contacts(self, file_name=CONTACT_LIST_FILE) -> list:
|
|
contact_list_in_json = pandas.read_excel(file_name).to_json(orient='records')
|
|
contact_dict_list = json.loads(contact_list_in_json)
|
|
contact_list = []
|
|
for contact_dict in contact_dict_list:
|
|
name = contact_dict['name'].split(' ')
|
|
first_name = name[0]
|
|
last_name = name[-1]
|
|
contact = ContactPojo(phone_number=contact_dict['phone'],
|
|
last_name=last_name,
|
|
first_name=first_name,
|
|
passport_number=contact_dict['passport'],
|
|
mail=contact_dict['email'])
|
|
contact_list.append(contact)
|
|
return contact_list
|