diff --git a/python/src/contacts.py b/python/src/contacts.py index 4dce478..b76e951 100644 --- a/python/src/contacts.py +++ b/python/src/contacts.py @@ -1,8 +1,38 @@ def show_contacts(addressbook): - pass + print(None) + for contact in addressbook: + print(f"{contact['name']} ({contact['email']}): {contact['phone']}") + print(None) + def add_contact(addressbook): - pass + name = input("Enter name: ") + phone = input("Enter phone: ") + email = input("Email: ") + + new_contact = { + "name": name.strip(), + "phone": phone.strip(), + "email": email.strip(), + } + + addressbook.append(new_contact) + + print(f"\n{new_contact['name']} was added.\n") + def delete_contact(addressbook): - pass + pattern = input("Enter a part of their name: ").strip() + idx = None + for i, contact in enumerate(addressbook): + if contact["name"].find(pattern) != -1: + idx = i + break + + if idx is None: + print("\nContact Not found!\n") + return + deleted_name = addressbook[idx]["name"] + del addressbook[idx] + + print(f"\n{deleted_name} was deleted.\n") diff --git a/python/src/main.py b/python/src/main.py index 7bd07c8..308b092 100644 --- a/python/src/main.py +++ b/python/src/main.py @@ -1,11 +1,38 @@ import contacts -addressbook = [] +addressbook = [ + {"name": "Bruce Wayne", "phone": "555-123-4567", "email": "bruce@wayne.com"}, + {"name": "Clark Kent", "phone": "555-222-3333", "email": "clark@dailyplanet.com"}, + {"name": "Diana Prince", "phone": "555-444-5555", "email": "diana@amazon.com"}, +] + def menu(): - pass + print("[1] Display all contacts") + print("[2] Add a new contact") + print("[3] Delete a contact") + print("[4] Exit") + def main(): - pass + run = True + while run: + menu() + selection = int(input("Enter a selection: ")) + if selection == 1: + contacts.show_contacts(addressbook) + + elif selection == 2: + contacts.add_contact(addressbook) + + elif selection == 3: + contacts.delete_contact(addressbook) + + elif selection == 4: + run = False + else: + print("\nThat selection is not valid, please try again!\n") + print("\nGoodbye!\n") + main()