def run(self):
# This block was moved from server file to try and make it handled here instead
phonebook = Phonebook() # called from the phonebook.py
filename = "AddressBook.txt"
while True:
try:
file = open(filename, "r") # opens in read only
contents = file.readlines()
for line in contents:
information = line.split(
" "
) # This splits each line of the text file by spaces
phonebook.add(information[0], information[1])
break
except ReferenceError:
print("File not found.")
filename = input(
"Please enter the name of the phonebook to be loaded: ")
# End of test block
# create string of phonebook to send to client on connection
start_book = phonebook.__str__()
encrypt(start_book)
self.client.send(bytes(encrypt(start_book.encode()))
) # encrypts start_book for delivery to client
self.client.send(bytes("Welcome to the phone book application!", ))
while True:
message = decode(self.client.recv(BUFSIZE), )
decrypt(message) # decrypts incoming message
if not message:
print("Client disconnected")
self.client.close()
break
else:
request = message.split()
command = request[0]
if command == "FIND":
number = self.phonebook.get(request[1])
if not number:
reply = "Number not found."
else:
reply = "The number is " + number
else:
self.phonebook.add(request[1], request[2])
# write it to the file:
filename = "AddressBook.txt"
phonebook_file = open(filename, "a") # opens in read only
phonebook_file.write(request[1] + " " + request[2] + "\n")
phonebook_file.close()
reply = "Name and number added to phone book and file.\nPress Update to update the table."
self.client.send(bytes(reply, ))
