def addContact(self, contact: Contact) -> None:
with open("contacts.csv", "r") as inFile:
reader = csv.reader(inFile)
with open("contacts_tmp.csv", "w") as outFile:
writer = csv.writer(outFile)
for line in reader:
writer.writerow(line)
writer.writerow([contact.name(), contact.phoneNumber()])
os.remove("contacts.csv")
os.rename("contacts_tmp.csv", "contacts.csv")
def updateContact(self, contact: Contact) -> None:
with open("contacts.csv", "r") as myContactsFile:
contactsReader = csv.DictReader(myContactsFile)
with open("contacts_tmp.csv", "w") as outFile:
writer = csv.writer(outFile)
writer.writerow(contactsReader.fieldnames)
for row in contactsReader:
if row["name"] == contact.name():
writer.writerow([contact.name(), contact.phoneNumber()])
continue
writer.writerow([row["name"], row["phone number"]])
os.remove("contacts.csv")
os.rename("contacts_tmp.csv", "contacts.csv")
