Saving progress on lab 11.

Author: x8bitReignbeaux

code/zach/contacts.csv

@@ -1,4 +1,4 @@
 Name,Phone,Email
-Zachary Christman,703-555-6666,[email protected]
-Sara Beth,703-666-7777,sarabeth@test.comJustin Young, 1234567890, JustinYoung@test.com
-Justin Young, 1234567890, [email protected]
+Zachary LaFever,703-555-6666,[email protected]
+Sara Beth,703-666-7777,[email protected]
+Justin Old, 1234567890, [email protected]

code/zach/lab11-contact-list.py

@@ -4,16 +4,17 @@ def version1():
     contacts = []
     index = 1
     headers = lines[0].split(",")
-    
+
     while index < len(lines):
         contact = lines[index].split(",")
         contacts.append(dict(zip(headers, contact)))
         index += 1
-    
+
     return contacts
 
-def version2():
-    
+
+def version2_3():
+    contacts = version1()
     query_type = input("What would you like to do?\nCreate\nRetrieve\nUpdate\nDelete\n> ")
 
     if query_type == 'Create':
@@ -22,31 +23,76 @@ def version2():
             phone = input("What is the contact's phone? ")
             email = input("What is the contact's email? ")
             file.write(f'\n{name}, {phone}, {email}')
-            
+
     elif query_type == 'Retrieve':
-        contacts = version1()
-        name = input("What is the contact's name? ")
-        index = 0 
+        name = input("What is the contact's name you would like to find? ")
+        index = 0
 
         while index < len(contacts):
             if name == contacts[index]['Name']:
                 return print(contacts[index])
             index += 1
         else:
             return print("User not found.")
-    
-    #elif query_type == 'Update':
-        
-    #elif query_type == 'Delete':
+
+    elif query_type == 'Update':
+        name = input("What is the contact's name you would like to update? ")
+        index = 0
+
+        while index < len(contacts):
+            if name == contacts[index]['Name']:
+                with open('contacts.csv', 'rt') as file:
+                    data = file.read()
+                    update_type = input('Would you like to update their Name, Phone, or Email? ')
+                    
+                    if update_type == 'Name':
+                        name_change = input('What should their name change to? ')
+                        data = data.replace(contacts[index]['Name'], name_change)
+                        file.close()
+                        file = open('contacts.csv', 'wt')
+                        file.write(data)
+                        file.close()
+                    elif update_type == 'Phone':
+                        phone_change = input('What should their phone number be? ')
+                        data = data.replace(contacts[index]['Phone'], phone_change)
+                        file.close()
+                        file = open('contacts.csv', 'wt')
+                        file.write(data)
+                        file.close()
+                    elif update_type == 'Email':
+                        email_change = input('What should their email be? ')
+                        data = data.replace(contacts[index]['Email'], email_change)
+                        file.close()
+                        file = open('contacts.csv', 'wt')
+                        file.write(data)
+                        file.close()
+                    else:
+                        print('Not a valid selection.')
+                    
+            index += 1
+        else:
+            return print("User not found.")
 
-    #else:
-    
-def version3():
-    
-    return 0
+    elif query_type == 'Delete':
+        name = input("What is the contact's name you would like to delete? ")
+        index = 0
+
+        while index < len(contacts):
+            if name == contacts[index]['Name']:
+                del contacts[index]
+                data = f"{','.join(contacts[0].keys())}\n" 
+                for dict in contacts:
+                    data = data + f"{','.join(dict.values())}\n"
+            index += 1
+        file = open('contacts.csv', 'wt')
+        file.write(data)
+        file.close
+                
+    else:
+        print("Invalid selection.")
 
 def main():
     #version1()
-    version2()
+    version2_3()
 
 main()