Python Mini Project 1 - Address Book.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# P1 - Address Book\n",
    "\n",
    "## Author: Swaroop Kallakuri\n",
    "\n",
    "## In this small mini project we will be creating a simple address book application that will store, search and delete records."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# To-DO:\n",
    "1. test the application and create a list of errors\n",
    "2. design the smoothflow to have a nicest experience\n",
    "3. Smooth exit and bye message"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "#Pickle library to persist the data\n",
    "import pickle, os\n",
    "\n",
    "#main Class address book which has methods for adding, displaying the contacts\n",
    "class AddressBook(object):\n",
    "    def __init__(self, name = None, address = None, email = None, phone = None):\n",
    "        self.name = name\n",
    "        self.address = address\n",
    "        self.email = email\n",
    "        self.phone = phone\n",
    "        self.contacts = {}\n",
    "        self.filename = 'addressbook'\n",
    "\n",
    "    def __str__(self):\n",
    "        return '[Name: {0} | Address: {1} | Email: {2} | Phone: {3}]'.format(self.name, self.address, self.email, self.phone)\n",
    "\n",
    "    def __repr__(self):\n",
    "        return '[Name: {0} | Address: {1} | Email: {2} | Phone: {3}]'.format(self.name, self.address, self.email, self.phone)\n",
    "\n",
    "    # Adding details provided by the user in our Address Book\n",
    "    def addContacts(self):\n",
    "        try:\n",
    "            if os.path.exists(self.filename) and os.path.getsize(self.filename) > 0:\n",
    "                myAddressBook = open(self.filename, 'rb')\n",
    "                data = pickle.load(myAddressBook)\n",
    "                myAddressBook.close()\n",
    "            else:\n",
    "                myAddressBook = open(self.filename, 'wb')\n",
    "                data = {}\n",
    "\n",
    "            contact = self.getDetailsFromUser()\n",
    "            data[contact['Name']] = contact\n",
    "            myAddressBook = open(self.filename, 'wb')\n",
    "            pickle.dump(data, myAddressBook)\n",
    "            myAddressBook.close()\n",
    "            print('Contact Added Successfully!')\n",
    "        except:\n",
    "            print('There was an error! Contact was not added.')\n",
    "        finally:\n",
    "            myAddressBook.close()\n",
    "\n",
    "    # Getting the details from the user to adding the Address Book\n",
    "    def getDetailsFromUser(self):\n",
    "        try:\n",
    "            self.contacts['Name'] = str(input('Enter Contact\\'s Full Name: '))\n",
    "            self.contacts['Address'] = str(input('Enter Contact\\'s Address: '))\n",
    "            self.contacts['Email'] = str(input('Enter Contact\\'s Email Address: '))\n",
    "            self.contacts['Phone'] = int(input('Enter Contact\\'s Phone Number: '))\n",
    "            return self.contacts\n",
    "        except KeyboardInterrupt as error:\n",
    "            raise error\n",
    "\n",
    "    # To display ALL the contact in our Address Book\n",
    "    def displayContacts(self):\n",
    "        if os.path.exists(self.filename) and os.path.getsize(self.filename) > 0:\n",
    "            myAddressBook = open(self.filename, 'rb')\n",
    "            data = pickle.load(myAddressBook)\n",
    "            myAddressBook.close()\n",
    "            if data:\n",
    "                for records in data.values():\n",
    "                    print(records)\n",
    "            myAddressBook.close()\n",
    "        else:\n",
    "            print('No Record in database.')\n",
    "\n",
    "    # To search for a specific contact in our Address Book\n",
    "    def searchContacts(self):\n",
    "        if os.path.exists(self.filename) and os.path.getsize(self.filename) > 0:\n",
    "            myAddressBook = open(self.filename, 'rb')\n",
    "            data = pickle.load(myAddressBook)\n",
    "            myAddressBook.close()\n",
    "            try:\n",
    "                contactToSearch = input('Enter the name of the contact to search: ')\n",
    "                counter = 0\n",
    "                for contact in data.values():\n",
    "                    if contactToSearch in contact['Name']:\n",
    "                        print(data[contact['Name']])\n",
    "                        counter += 1\n",
    "                if counter == 0:\n",
    "                    print('No record found whose name is:', contactToSearch)\n",
    "            except:\n",
    "                print('Error occured!')\n",
    "        else:\n",
    "            print('No Record in database.')\n",
    "\n",
    "    # For modifying contacts\n",
    "    def modifyContacts(self):\n",
    "        if os.path.exists(self.filename) and os.path.getsize(self.filename) > 0:\n",
    "            myAddressBook = open(self.filename, 'rb')\n",
    "            data = pickle.load(myAddressBook)\n",
    "            myAddressBook.close()\n",
    "            try:\n",
    "                contactToModify = input('Enter the name of the contact to modify (Only enter full name): ')\n",
    "                # Search for the record to update\n",
    "                for contact in data.values():\n",
    "                    if contactToModify == contact['Name']:\n",
    "                        contact = data[contactToModify]\n",
    "                        break\n",
    "                option = int(input('1. To modify name, 2. To modify address, 3. To modify email, 4. To modify phone: '))\n",
    "                if option == 1:\n",
    "                    contact['Name'] = input('Enter Name to modify: ')\n",
    "                    del data[contactToModify]\n",
    "                    data[contact['Name']] = contact\n",
    "                    print('Successful')\n",
    "                elif option == 2:\n",
    "                     contact['Address'] = input('Enter Address to modify: ')\n",
    "                     del data[contactToModify]\n",
    "                     data[contactToModify] = contact\n",
    "                     print('Successful')\n",
    "                elif option == 3:\n",
    "                    contact['Email'] = input('Enter Email to modify: ')\n",
    "                    del data[contactToModify]\n",
    "                    data[contactToModify] = contact\n",
    "                    print('Successful')\n",
    "                elif option == 4:\n",
    "                    contact['Phone'] = input('Enter Phone to modify: ')\n",
    "                    del data[contactToModify]\n",
    "                    data[contactToModify] = contact\n",
    "                    print('Successful')\n",
    "                else:\n",
    "                    print('Incorrect option selected.')\n",
    "            except:\n",
    "                print('Error occured. No such record found. Try Again!')\n",
    "            finally:\n",
    "                myAddressBook = open(self.filename, 'wb')\n",
    "                pickle.dump(data, myAddressBook)\n",
    "                myAddressBook.close()\n",
    "        else:\n",
    "            print('No Record in database.')\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Enter 1. To Add Contacts 2. For Searching a Contact 3. For Modifying a Contact 4. To Display Contacts 5. To Exit\n",
      "Enter your choice: 1\n",
      "Enter Contact's Full Name: Swaroop\n",
      "Enter Contact's Address: 38660 lexington Street\n",
      "Enter Contact's Email Address: ksjpswaroop@gmail.com\n",
      "Enter Contact's Phone Number: 2019129101\n",
      "Contact Added Successfully!\n",
      "Enter your choice: 5\n",
      "Enter your choice: 5\n",
      "Enter your choice: 5\n",
      "Enter your choice: 4\n",
      "{'Address': '38660 lexington Street', 'Email': 'ksjpswaroop@gmail.com', 'Phone': 2019129101, 'Name': 'Swaroop'}\n",
      "Enter your choice: 3\n",
      "Enter the name of the contact to modify (Only enter full name): 2\n",
      "1. To modify name, 2. To modify address, 3. To modify email, 4. To modify phone: 1\n",
      "Enter Name to modify: swaroop\n",
      "Error occured. No such record found. Try Again!\n"
     ]
    },
    {
     "ename": "KeyboardInterrupt",
     "evalue": "",
     "output_type": "error",
     "traceback": [
      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/ipykernel/kernelbase.py\u001b[0m in \u001b[0;36m_input_request\u001b[0;34m(self, prompt, ident, parent, password)\u001b[0m\n\u001b[1;32m    729\u001b[0m             \u001b[0;32mtry\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 730\u001b[0;31m                 \u001b[0mident\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mreply\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0msession\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrecv\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mstdin_socket\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0;36m0\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    731\u001b[0m             \u001b[0;32mexcept\u001b[0m \u001b[0mException\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/jupyter_client/session.py\u001b[0m in \u001b[0;36mrecv\u001b[0;34m(self, socket, mode, content, copy)\u001b[0m\n\u001b[1;32m    795\u001b[0m         \u001b[0;32mtry\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 796\u001b[0;31m             \u001b[0mmsg_list\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0msocket\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrecv_multipart\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mmode\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcopy\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mcopy\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    797\u001b[0m         \u001b[0;32mexcept\u001b[0m \u001b[0mzmq\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mZMQError\u001b[0m \u001b[0;32mas\u001b[0m \u001b[0me\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/zmq/sugar/socket.py\u001b[0m in \u001b[0;36mrecv_multipart\u001b[0;34m(self, flags, copy, track)\u001b[0m\n\u001b[1;32m    394\u001b[0m         \"\"\"\n\u001b[0;32m--> 395\u001b[0;31m         \u001b[0mparts\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0;34m[\u001b[0m\u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mrecv\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mflags\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mcopy\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mcopy\u001b[0m\u001b[0;34m,\u001b[0m \u001b[0mtrack\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0mtrack\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m]\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    396\u001b[0m         \u001b[0;31m# have first part already, only loop while more to receive\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32mzmq/backend/cython/socket.pyx\u001b[0m in \u001b[0;36mzmq.backend.cython.socket.Socket.recv (zmq/backend/cython/socket.c:7683)\u001b[0;34m()\u001b[0m\n",
      "\u001b[0;32mzmq/backend/cython/socket.pyx\u001b[0m in \u001b[0;36mzmq.backend.cython.socket.Socket.recv (zmq/backend/cython/socket.c:7460)\u001b[0;34m()\u001b[0m\n",
      "\u001b[0;32mzmq/backend/cython/socket.pyx\u001b[0m in \u001b[0;36mzmq.backend.cython.socket._recv_copy (zmq/backend/cython/socket.c:2344)\u001b[0;34m()\u001b[0m\n",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/zmq/backend/cython/checkrc.pxd\u001b[0m in \u001b[0;36mzmq.backend.cython.checkrc._check_rc (zmq/backend/cython/socket.c:9621)\u001b[0;34m()\u001b[0m\n",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m: ",
      "\nDuring handling of the above exception, another exception occurred:\n",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m                         Traceback (most recent call last)",
      "\u001b[0;32m<ipython-input-3-37ecbaa3772d>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m()\u001b[0m\n\u001b[1;32m      3\u001b[0m     \u001b[0mprint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'Enter 1. To Add Contacts 2. For Searching a Contact 3. For Modifying a Contact 4. To Display Contacts 5. To Exit'\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      4\u001b[0m     \u001b[0;32mwhile\u001b[0m \u001b[0;32mTrue\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m----> 5\u001b[0;31m         \u001b[0mchoice\u001b[0m \u001b[0;34m=\u001b[0m \u001b[0mint\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0minput\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m'Enter your choice: '\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m      6\u001b[0m         \u001b[0;32mif\u001b[0m \u001b[0mchoice\u001b[0m \u001b[0;34m==\u001b[0m \u001b[0;36m1\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m      7\u001b[0m             \u001b[0mmyBook\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0maddContacts\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/ipykernel/kernelbase.py\u001b[0m in \u001b[0;36mraw_input\u001b[0;34m(self, prompt)\u001b[0m\n\u001b[1;32m    703\u001b[0m             \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_parent_ident\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    704\u001b[0m             \u001b[0mself\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0m_parent_header\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 705\u001b[0;31m             \u001b[0mpassword\u001b[0m\u001b[0;34m=\u001b[0m\u001b[0;32mFalse\u001b[0m\u001b[0;34m,\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    706\u001b[0m         )\n\u001b[1;32m    707\u001b[0m \u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;32m/anaconda/lib/python3.5/site-packages/ipykernel/kernelbase.py\u001b[0m in \u001b[0;36m_input_request\u001b[0;34m(self, prompt, ident, parent, password)\u001b[0m\n\u001b[1;32m    733\u001b[0m             \u001b[0;32mexcept\u001b[0m \u001b[0mKeyboardInterrupt\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    734\u001b[0m                 \u001b[0;31m# re-raise KeyboardInterrupt, to truncate traceback\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m--> 735\u001b[0;31m                 \u001b[0;32mraise\u001b[0m \u001b[0mKeyboardInterrupt\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m    736\u001b[0m             \u001b[0;32melse\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m    737\u001b[0m                 \u001b[0;32mbreak\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
      "\u001b[0;31mKeyboardInterrupt\u001b[0m: "
     ]
    }
   ],
   "source": [
    "if __name__ == '__main__':\n",
    "    myBook = AddressBook()\n",
    "    print('Enter 1. To Add Contacts 2. For Searching a Contact 3. For Modifying a Contact 4. To Display Contacts 5. To Exit')\n",
    "    while True:\n",
    "        choice = int(input('Enter your choice: '))\n",
    "        if choice == 1:\n",
    "            myBook.addContacts()\n",
    "        elif choice == 2:\n",
    "            myBook.searchContacts()\n",
    "        elif choice == 3:\n",
    "            myBook.modifyContacts()\n",
    "        elif choice == 4:\n",
    "            myBook.displayContacts()\n",
    "        elif choice == 5:\n",
    "            exit()\n",
    "        else:\n",
    "            print('Invalid Option. Try Again!')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.5.4"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}