307 lines
11 KiB
Python
307 lines
11 KiB
Python
|
# importing the module
|
||
|
|
import sys
|
||
|
|
|
||
|
|
|
||
|
|
# this function will be the first to run as soon as the main function executes
|
||
|
|
def writeFile(pb):
|
||
|
|
|
||
|
|
b = open("records.txt", 'w')
|
||
|
|
|
||
|
|
for record in pb:
|
||
|
|
for item in record:
|
||
|
|
if item != record[len(record) - 1]:
|
||
|
|
b.write(str(item) + ",")
|
||
|
|
else:
|
||
|
|
b.write(str(item) + "\n")
|
||
|
|
|
||
|
|
b.close()
|
||
|
|
|
||
|
|
def readFile():
|
||
|
|
|
||
|
|
pb = []
|
||
|
|
f = open("records.txt", 'r')
|
||
|
|
while True:
|
||
|
|
d = f.readline().strip().split(',')
|
||
|
|
if len(d) == 1:
|
||
|
|
break
|
||
|
|
|
||
|
|
pb.append(d)
|
||
|
|
|
||
|
|
f.close()
|
||
|
|
return pb
|
||
|
|
|
||
|
|
def initial_phonebook():
|
||
|
|
rows, cols = int(input("Please enter initial number of contacts: ")), 5
|
||
|
|
|
||
|
|
|
||
|
|
# We are collecting the initial number of contacts the user wants to have in the
|
||
|
|
# phonebook already. User may also enter 0 if he doesn't wish to enter any.
|
||
|
|
phone_book = readFile()
|
||
|
|
if rows >= 1:
|
||
|
|
for i in range(rows):
|
||
|
|
print("\nEnter contact %d details in the following order (ONLY):" % (i + 1))
|
||
|
|
print("NOTE: * indicates mandatory fields")
|
||
|
|
print("....................................................................")
|
||
|
|
temp = []
|
||
|
|
for j in range(cols):
|
||
|
|
|
||
|
|
# We have taken the conditions for values of j only for the personalized fields
|
||
|
|
# such as name, number, e-mail id, dob, category etc
|
||
|
|
if j == 0:
|
||
|
|
temp.append(str(input("Enter name*: ")))
|
||
|
|
|
||
|
|
# We need to check if the user has left the name empty as its mentioned that
|
||
|
|
# name & number are mandatory fields.
|
||
|
|
# So implement a condition to check as below.
|
||
|
|
if temp[j] == '' or temp[j] == ' ':
|
||
|
|
sys.exit(
|
||
|
|
"Name is a mandatory field. Process exiting due to blank field...")
|
||
|
|
# This will exit the process if a blank field is encountered.
|
||
|
|
|
||
|
|
if j == 1:
|
||
|
|
temp.append(int(input("Enter number*: ")))
|
||
|
|
# We do not need to check if user has entered the number because int automatically
|
||
|
|
# takes care of it. Int value cannot accept a blank as that counts as a string.
|
||
|
|
# So process automatically exits without us using the sys package.
|
||
|
|
|
||
|
|
if j == 2:
|
||
|
|
temp.append(str(input("Enter e-mail address: ")))
|
||
|
|
# Even if this field is left as blank, None will take the blank's place
|
||
|
|
if temp[j] == '' or temp[j] == ' ':
|
||
|
|
temp[j] = None
|
||
|
|
|
||
|
|
if j == 3:
|
||
|
|
temp.append(str(input("Enter date of birth(dd/mm/yy): ")))
|
||
|
|
# Whatever format the user enters dob in, it won't make a difference to the compiler
|
||
|
|
# Only while searching the user will have to enter query exactly the same way as
|
||
|
|
# he entered during the input so as to ensure accurate searches
|
||
|
|
if temp[j] == '' or temp[j] == ' ':
|
||
|
|
# Even if this field is left as blank, None will take the blank's place
|
||
|
|
temp[j] = None
|
||
|
|
if j == 4:
|
||
|
|
temp.append(
|
||
|
|
str(input("Enter category(Family/Friends/Work/Others): ")))
|
||
|
|
# Even if this field is left as blank, None will take the blank's place
|
||
|
|
if temp[j] == "" or temp[j] == ' ':
|
||
|
|
temp[j] = None
|
||
|
|
|
||
|
|
phone_book.append(temp)
|
||
|
|
# By this step we are appending a list temp into a list phone_book
|
||
|
|
# That means phone_book is a 2-D array and temp is a 1-D array
|
||
|
|
|
||
|
|
for record in phone_book:
|
||
|
|
for item in record:
|
||
|
|
print(f"{item:10}", end="\t")
|
||
|
|
print("")
|
||
|
|
return phone_book
|
||
|
|
|
||
|
|
|
||
|
|
def menu():
|
||
|
|
# We created this simple menu function for
|
||
|
|
# code reusability & also for an interactive console
|
||
|
|
# Menu func will only execute when called
|
||
|
|
print("********************************************************************")
|
||
|
|
print("\t\t\tSMARTPHONE DIRECTORY", flush=False)
|
||
|
|
print("********************************************************************")
|
||
|
|
print("\tYou can now perform the following operations on this phonebook\n")
|
||
|
|
print("1. Add a new contact")
|
||
|
|
print("2. Remove an existing contact")
|
||
|
|
print("3. Delete all contacts")
|
||
|
|
print("4. Search for a contact")
|
||
|
|
print("5. Display all contacts")
|
||
|
|
print("6. Save & Exit phonebook")
|
||
|
|
|
||
|
|
# Out of the provided 6 choices, user needs to enter any 1 choice among the 6
|
||
|
|
# We return the entered choice to the calling function wiz main in our case
|
||
|
|
choice = int(input("Please enter your choice: "))
|
||
|
|
|
||
|
|
return choice
|
||
|
|
|
||
|
|
|
||
|
|
def add_contact(pb):
|
||
|
|
# Adding a contact is the easiest because all you need to do is:
|
||
|
|
# append another list of details into the already existing list
|
||
|
|
dip = []
|
||
|
|
for i in range(len(pb[0])):
|
||
|
|
if i == 0:
|
||
|
|
dip.append(str(input("Enter name: ")))
|
||
|
|
if i == 1:
|
||
|
|
dip.append(int(input("Enter number: ")))
|
||
|
|
if i == 2:
|
||
|
|
dip.append(str(input("Enter e-mail address: ")))
|
||
|
|
if i == 3:
|
||
|
|
dip.append(str(input("Enter date of birth(dd/mm/yy): ")))
|
||
|
|
if i == 4:
|
||
|
|
dip.append(
|
||
|
|
str(input("Enter category(Family/Friends/Work/Others): ")))
|
||
|
|
pb.append(dip)
|
||
|
|
# And once you modify the list, you return it to the calling function wiz main, here.
|
||
|
|
return pb
|
||
|
|
|
||
|
|
|
||
|
|
def remove_existing(pb):
|
||
|
|
# This function is to remove a contact's details from existing phonebook
|
||
|
|
query = str(
|
||
|
|
input("Please enter the name of the contact you wish to remove: "))
|
||
|
|
# We'll collect name of the contact and search if it exists in our phonebook
|
||
|
|
|
||
|
|
temp = 0
|
||
|
|
# temp is a checking variable here. We assigned a value 0 to temp.
|
||
|
|
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][0]:
|
||
|
|
temp += 1
|
||
|
|
# Temp will be incremented & it won't be 0 anymore in this function's scope
|
||
|
|
|
||
|
|
print(pb.pop(i))
|
||
|
|
# The pop function removes entry at index i
|
||
|
|
|
||
|
|
print("This query has now been removed")
|
||
|
|
# printing a confirmation message after removal.
|
||
|
|
# This ensures that removal was successful.
|
||
|
|
# After removal we will return the modified phonebook to the calling function
|
||
|
|
# which is main in our program
|
||
|
|
|
||
|
|
return pb
|
||
|
|
if temp == 0:
|
||
|
|
# Now if at all any case matches temp should've incremented but if otherwise,
|
||
|
|
# temp will remain 0 and that means the query does not exist in this phonebook
|
||
|
|
print("Sorry, you have entered an invalid query.\
|
||
|
|
Please recheck and try again later.")
|
||
|
|
|
||
|
|
return pb
|
||
|
|
|
||
|
|
|
||
|
|
def delete_all(pb):
|
||
|
|
# This function will simply delete all the entries in the phonebook pb
|
||
|
|
# It will return an empty phonebook after clearing
|
||
|
|
return pb.clear()
|
||
|
|
|
||
|
|
|
||
|
|
def search_existing(pb):
|
||
|
|
# This function searches for an existing contact and displays the result
|
||
|
|
choice = int(input("Enter search criteria\n\n\
|
||
|
|
1. Name\n2. Number\n3. Email-id\n4. DOB\n5. Category(Family/Friends/Work/Others)\
|
||
|
|
\nPlease enter: "))
|
||
|
|
# We're doing so just to ensure that the user experiences a customized search result
|
||
|
|
|
||
|
|
temp = []
|
||
|
|
check = -1
|
||
|
|
|
||
|
|
if choice == 1:
|
||
|
|
# This will execute for searches based on contact name
|
||
|
|
query = str(
|
||
|
|
input("Please enter the name of the contact you wish to search: "))
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][0]:
|
||
|
|
check = i
|
||
|
|
temp.append(pb[i])
|
||
|
|
|
||
|
|
elif choice == 2:
|
||
|
|
# This will execute for searches based on contact number
|
||
|
|
query = int(
|
||
|
|
input("Please enter the number of the contact you wish to search: "))
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][1]:
|
||
|
|
check = i
|
||
|
|
temp.append(pb[i])
|
||
|
|
|
||
|
|
elif choice == 3:
|
||
|
|
# This will execute for searches based on contact's e-mail address
|
||
|
|
query = str(input("Please enter the e-mail ID\
|
||
|
|
of the contact you wish to search: "))
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][2]:
|
||
|
|
check = i
|
||
|
|
temp.append(pb[i])
|
||
|
|
|
||
|
|
elif choice == 4:
|
||
|
|
# This will execute for searches based on contact''s date of birth
|
||
|
|
query = str(input("Please enter the DOB (in dd/mm/yyyy format ONLY)\
|
||
|
|
of the contact you wish to search: "))
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][3]:
|
||
|
|
check = i
|
||
|
|
temp.append(pb[i])
|
||
|
|
|
||
|
|
elif choice == 5:
|
||
|
|
# This will execute for searches based on contact category
|
||
|
|
query = str(
|
||
|
|
input("Please enter the category of the contact you wish to search: "))
|
||
|
|
for i in range(len(pb)):
|
||
|
|
if query == pb[i][4]:
|
||
|
|
check = i
|
||
|
|
temp.append(pb[i])
|
||
|
|
# All contacts under query category will be shown using this feature
|
||
|
|
|
||
|
|
else:
|
||
|
|
# If the user enters any other choice then the search will be unsuccessful
|
||
|
|
print("Invalid search criteria")
|
||
|
|
return -1
|
||
|
|
# returning -1 indicates that the search was unsuccessful
|
||
|
|
|
||
|
|
# all the searches are stored in temp and all the results will be displayed with
|
||
|
|
# the help of display function
|
||
|
|
|
||
|
|
if check == -1:
|
||
|
|
return -1
|
||
|
|
# returning -1 indicates that the query did not exist in the directory
|
||
|
|
else:
|
||
|
|
display_all(temp)
|
||
|
|
return check
|
||
|
|
# we're just returning a index value wiz not -1 to calling function just to notify
|
||
|
|
# that the search worked successfully
|
||
|
|
|
||
|
|
|
||
|
|
# this function displays all content of phonebook pb
|
||
|
|
def display_all(pb):
|
||
|
|
if not pb:
|
||
|
|
# if display function is called after deleting all contacts then the len will be 0
|
||
|
|
# And then without this condition it will throw an error
|
||
|
|
print("List is empty: []")
|
||
|
|
else:
|
||
|
|
for record in pb:
|
||
|
|
for item in record:
|
||
|
|
print(f"{item:10}", end="\t")
|
||
|
|
print("")
|
||
|
|
|
||
|
|
|
||
|
|
def thanks(pb):
|
||
|
|
|
||
|
|
writeFile(pb)
|
||
|
|
|
||
|
|
# A simple gesture of courtesy towards the user to enhance user experience
|
||
|
|
print("********************************************************************")
|
||
|
|
print("Thank you for using our Smartphone directory system.")
|
||
|
|
print("Please visit again!")
|
||
|
|
print("********************************************************************")
|
||
|
|
sys.exit("Goodbye, have a nice day ahead!")
|
||
|
|
|
||
|
|
|
||
|
|
# Main function code
|
||
|
|
print("....................................................................")
|
||
|
|
print("Hello dear user, welcome to our smartphone directory system")
|
||
|
|
print("You may now proceed to explore this directory")
|
||
|
|
print("....................................................................")
|
||
|
|
# This is solely meant for decoration purpose only.
|
||
|
|
# You're free to modify your interface as per your will to make it look interactive
|
||
|
|
|
||
|
|
ch = 1
|
||
|
|
pb = initial_phonebook()
|
||
|
|
while ch in (1, 2, 3, 4, 5):
|
||
|
|
ch = menu()
|
||
|
|
if ch == 1:
|
||
|
|
pb = add_contact(pb)
|
||
|
|
elif ch == 2:
|
||
|
|
pb = remove_existing(pb)
|
||
|
|
elif ch == 3:
|
||
|
|
pb = delete_all(pb)
|
||
|
|
elif ch == 4:
|
||
|
|
d = search_existing(pb)
|
||
|
|
if d == -1:
|
||
|
|
print("The contact does not exist. Please try again")
|
||
|
|
elif ch == 5:
|
||
|
|
display_all(pb)
|
||
|
|
else:
|
||
|
|
thanks(pb)
|