[Shubham] Add : Manage mutiple addressbook operation functions.

Author: CodeWizard786-shubham

AddressBook_Service.py

@@ -18,10 +18,112 @@ def __str__(self):
 class AddressBook:
 
  def __init__(self):
- self.contacts = {}
+ """
+ Description : 
+ Default constructor.
+ Resturns : address books dictionary
+ """
+ self.address_books = {}
+
+ def add_address_books(self,address_book_name):
+ """
+ Description : 
+ This function creates a address book if it is already not present.
+ Parameters :
+ address_boo_name : address book name input from the user.
+ Returns :
+ none
+ prints message
+ """
+ if address_book_name in self.address_books:
+ print(f"Address Book with name '{address_book_name}' already exists.") 
+ else:
+ self.address_books[address_book_name] = {}
+ print(f"Address Book with name '{address_book_name}' is successfully created")
+ 
+
+ def fetch_address_book(self,address_book_name):
+ """
+ Description : 
+ This function feches address books from the address books dictionary.
+ Parameters :
+ address_bbok_name : name of the address book to fetch.
+ Returns :
+ address_book_value : contacts from address book
+ """
+ for address_book_key, address_book_value in self.address_books.items():
+ if address_book_key == address_book_name:
+ return address_book_value
+ return None 
+
+
+ def remove_address_book(self,address_book_name):
+ address_book = self.fetch_address_book(address_book_name)
+
+ if address_book:
+ self.address_books.pop(address_book_name)
+ print("Removed Address Book successfully")
+ else:
+ print(f"Address Book with name {address_book_name} not found")
+
+
+ def update_address_book(self,address_book_name):
+ if address_book_name not in self.address_books:
+ print(f"Address Book with name '{address_book_name}' does not exists.") 
+ else:
+ while True:
+ print()
+ print("--Contacts--")
+ print("-----------------------")
+ print("1.Add contact")
+ print("2.Update Contact")
+ print("3.Remove contact")
+ print("4.Print Contact")
+ print("5.Exit")
+ print("-----------------------")
+ choice = int(input("Enter choice: ")) 
+ if choice == 1:
+ self.add_contact(address_book_name)
+
+ elif choice == 2:
+ first_name = input("Enter first name to update: ")
+ self.update_contact(address_book_name,first_name)
+
+ elif choice == 3:
+ first_name = input("Enter first name to remove: ")
+ self.remove_contact(address_book_name,first_name)
+
+ elif choice == 4:
+ self.display_contacts(address_book_name)
+
+ elif choice == 5:
+ return False
+ else:
+ print("[error] Selected option is wrong")
+
+ # display address Book
+ def display_address_books(self):
+ """
+ Description : 
+ This function prints the contents from the address_books dictionary.
+ Parameters :
+ self : address_books dictionary which is declared in default constructor.
+ Returns :
+ none
+ prints address book name , contact from address book
+ """
+ for address_book_name, address_book in self.address_books.items():
+ print()
+ print(f"| AddressBook:{address_book_name} |")
+ print("|------------------------------------------------------------------------------------------------------------------------------|")
+ contact_number = 1
+ for contact_name, contact in address_book.items():
+ print(f"| Contact {contact_number}: {contact_name} : {contact}|")
+ contact_number += 1
+ 
 
  # add contacts
- def add_contact(self):
+ def add_contact(self,address_book_name):
  """
  Description : 
  This function passes the inputs from the user to conacts constructor and then appends the object of class contact to contacts list.
@@ -41,26 +143,27 @@ def add_contact(self):
  phone_number =int(input("Enter phone number: "))
  email_id = input("Enter email id: ")
  contact = Contact(first_name,second_name,address,city,state,zip,phone_number,email_id)
- self.contacts[first_name]= contact
+ self.address_books[address_book_name][first_name] = contact
  print("Contact Added Succesfully")
 
 
- def fetch_contact(self, first_name):
+ def fetch_contact(self, address_book_name, first_name):
  """
- Description:
- This function fetches a contact from the contacts list using the first name.
- Parameters:
- first_name: First name of the contact to retrieve.
- Returns:
- contact: The contact from the contacts list.
+ Descrption : 
+ This function searches a contact from address book it it exist or not.
+ Parameters :
+ self : default constructor
+ address_book_name : name of the address_book for fetching
+ first_name : first name from the address book to fetch. 
+ 
  """
- for contact_key, contact_value in self.contacts.items():
-  if contact_key == first_name:
-  return contact_value
+ address_book = self.address_books.get(address_book_name)
+ if address_book:
+ return address_book.get(first_name)
  return None
 
  # update address Book 
- def update_contact(self,first_name):
+ def update_contact(self,address_book_name,first_name):
  """
  Description : 
  This function checks if the first name in present in contact 
@@ -71,63 +174,65 @@ def update_contact(self,first_name):
  none
  prints update message.
  """
- contact = self.fetch_contact(first_name)
+ contact = self.fetch_contact(address_book_name,first_name)
  if contact:
  print("Please enter contact details for update")
- first_name =input("Enter first name: ")
+ new_first_name =input("Enter first name: ")
  second_name = input("Enter second name: ")
  address =input("Enter address: ")
  city =input("Enter city: ")
  state =input("Enter state: ")
  zip =int(input("Enter zip: "))
  phone_number =int(input("Enter phone number: "))
  email_id = input("Enter email id: ")
- new_contact = Contact(first_name,second_name,address,city,state,zip,phone_number,email_id)
- self.contacts[first_name] = new_contact
+ new_contact = Contact(new_first_name,second_name,address,city,state,zip,phone_number,email_id)
+ self.address_books[address_book_name][new_first_name] = new_contact
+
+ if first_name != new_first_name:
+ del self.address_books[address_book_name][first_name]
+
  print("Contact updated successfully")
  else:
  input("Contact not found. Hit enter to continue")
 
 
  # remove contact from address Book 
- def remove_contact(self,first_name):
+ def remove_contact(self,address_book_name,first_name):
  """
  Description : 
  This function checks if the first name in present in contact 
  if it is present, take the inputs from user again ,find the index number and update the contact. 
  Parameters :
- self : self.contacts list.
+ self : self.addressbooks.
+ address_book_name : get name of addressbook
+ first_name : first name of the contact to remove.
  Returns :
  none
  prints update message.
  """
- contact = self.fetch_contact(first_name)
+ contact = self.fetch_contact(address_book_name,first_name)
  if contact:
- del self.contacts[first_name]
+ del self.address_books[address_book_name][first_name]
  print("Contact removed successfully")
  else:
- input("Contact not found. Hit enter to continue")
+ print("Contact not found")
 
 
  # display contacts from address Book
- def display_contact(self):
- """
- Description : 
- This function prints the object of class contact.
- Parameters :
- self : contacts dict containing multiple contacts 
- Returns:
- none
- prints key value from contact object
- """
- # iterate over contacts dict to print each contact
- print("|-------------------------------------------------------------------------------------------------|")
- for key ,values in self.contacts.items():
- print(f"{key} \n [{values}]")
+ def display_contacts(self, address_book_name):
+ address_book = self.address_books.get(address_book_name)
+ if address_book:
+ print("|-----------------------------------------------------------------|")
+ print(f"| Contacts in '{address_book_name}' |") 
+ print("|-----------------------------------------------------------------|")
+ for contact in address_book.values():
+ print(contact)
  print()
- print("|-------------------------------------------------------------------------------------------------|")
+ print("|-----------------------------------------------------------------|")
+ else:
+ print(f"contacts not found in '{address_book_name}'")
 
- input("Contacts displayed, Hit Enter to continue!")
+ input("Contacts displayed. Hit Enter to continue!")