adding contacts ex

Author: sbtries

1 Python/docs/class_code_examples/fileIO/contacts/PhoneBook.py

@@ -0,0 +1,84 @@
+class PhoneBook:
+    # attributes, characteristics
+    def __init__(self):
+        self.contacts = [] #list of contacts
+        self.path = '' #path to csv
+        self.headers = [] #stored headers
+
+
+    def load(self, path):
+        """
+        returns a list of dictionaries representating a contact list csv
+        reads csv
+        """
+        self.path = path #set path
+        with open(self.path) as f:
+            lines = f.read().split('\n')
+
+        # process csv into list of dictionaries
+        self.headers = lines[0].split(',')
+        for i in range(1, len(lines)):
+            row = lines[i].split(',')
+            contact = dict(zip(self.headers, row))
+            self.contacts.append(contact)
+        return len(self.contacts)
+
+    def count(self):
+        """return the length of self.contacts"""
+
+        return len(self.contacts)
+
+    def save(self):
+        """
+        1) open 'self.file' with open 'w' for write
+        2) convert the dictionaries into lines
+        3) & write each line into to the file
+        """
+        lines = []
+        for contact in self.contacts:
+            lines.append( ",".join(list(contact.values())))
+
+        with open(self.path, 'w') as phone_book_file:
+            phone_book_file.write('\n'.join(lines))
+
+
+    def print(self):
+        """
+        loop over self.contacts
+        print the information for each contact
+        """
+        for contact in self.contacts:
+            print(contact)
+
+    def add(self, name, phone_number, state):
+        """
+        create a new dictionary
+        add the new dictionary to self.contacts
+        """
+
+        new_contact = {
+            "name": f"{name}",
+            "phone_number": f"{phone_number}",
+            "state": f"{state}"
+        }
+
+        self.contacts.append(new_contact)
+
+    def remove(self, name):
+        """
+        find the contact in self-contacts with the given name
+        remove the element at that index
+        """
+        for i in self.contacts:
+            if i["name"] == name:
+                self.contacts.remove(i)
+
+    def update(self, old_name, new_name, new_phone_number, new_state):
+        """
+        find the contact in self.contacts with the given old_name
+        set that contacts' name, phone number, etc to the given values
+        """
+        for i in self.contacts:
+            if i["name"] == old_name:
+                self.add(new_name, new_phone_number, new_state)
+                self.remove(old_name)

1 Python/docs/class_code_examples/fileIO/contacts/UserInterface.py

@@ -0,0 +1,50 @@
+import PhoneBook
+
+def main():
+    phone_book = PhoneBook.PhoneBook()  # create an instance of our class
+    print('Welcome to the Phone Book')
+    while True:
+        command = input('Enter a command: ')
+        if command == 'load':
+            path = input('Enter the path to your CSV: ')
+            phone_book.load(path)
+            print(f'Loaded {phone_book.count()} contacts.')
+        elif command == 'save':
+            phone_book.save()
+            print(f'Saved {phone_book.count()} contacts.')
+        elif command == 'print':
+            phone_book.print()
+        elif command == 'add':
+            print('Enter info of contact to add:')
+            name = input('Name: ')
+            phone_number = input('Phone Number: ')
+            state = input('state: ')
+            phone_book.add(name, phone_number, state)
+        elif command == 'remove':
+            name = input('Name of contact to remove: ')
+            phone_book.remove(name)
+        elif command == 'update':
+            print('Enter info of contact to add:')
+            old_name = input('Name of contact to update: ')
+            new_name = input('New Name: ')
+            new_phone_number = input('New Phone Number: ')
+            new_state = input('New state: ')
+            phone_book.update(old_name, new_name, new_phone_number,
+                                new_state)
+        elif command == 'help':
+            print('Available commands:')
+            print('load   - load all contacts from the file')
+            print('save   - save contacts to a file')
+            print('print  - print all contacts')
+            print('add    - add a new contact')
+            print('remove - remove a contact')
+            print('update - update a contact')
+            print('q   - quit the program')
+        elif command == 'q':
+            break
+        else:
+            print('Command not recognized')
+
+
+if __name__ == "__main__":
+    main()

1 Python/docs/class_code_examples/fileIO/contacts/contact.csv

@@ -0,0 +1,3 @@
+Sarah Beth,123432112,VA
+
+Frog,123456,florida

1 Python/docs/class_code_examples/fileIO/contacts/work_contacts.csv

@@ -0,0 +1,2 @@
+name,phone number,state,title,office
+Sarah Beth,123432112,VA,teacher,home

1 Python/docs/class_code_examples/fileIO/contacts/workphonebook.py

@@ -0,0 +1,5 @@
+import PhoneBook
+
+book = PhoneBook.PhoneBook()
+book.load('work_contacts.csv')
+book.print()