now using exception handling instead of if/else when checking valid menu input; added docstrings

python_3_oop/chapter02/menu.py

@@ -47,28 +47,32 @@ Notebook Menu
             self.display_menu()
             choice = input("Enter an option: ")
             action = self.choices.get(choice)
-            if action:
+            try:
                 action()
-            else:
+            except TypeError:
                 print("{0} is not a valid choice".format(choice))
 
     def show_notes(self, notes=None):
+        '''Display all notes stored in notebook object'''
         if not notes:
             notes = self.notebook.notes
         for note in notes:
             print("{0}: {1}\n{2}".format(note.id, note.tags, note.memo))
 
     def search_notes(self):
+        '''Search for a note containing given string'''
         filter = input("Search for: ")
         notes = self.notebook.search(filter)
         self.show_notes(notes)
 
     def add_note(self):
+        '''Add a given not to notebook object'''
         memo = input("Enter a memo: ")
         self.notebook.new_note(memo)
         print("Your note has been added.")
 
     def modify_note(self):
+        '''Modify tag and memo of note with given id'''
         id = input("Enter a note id: ")
         memo = input("Enter a memo: ")
         tags = input("Enter tags: ")
@@ -80,6 +84,7 @@ Notebook Menu
                 print("Note with id {0} doesn't exist.".format(id))
 
     def load_notes(self):
+        '''Decrypt notebook safe file and load it into notebook object'''
         try:
             f = open(self.savefile, 'rb')
         except IOError:
@@ -96,6 +101,7 @@ Notebook Menu
                 self.notebook = pickle.loads(plain)
 
     def save_notes(self):
+        '''Encrypt notebook object and store it into notebook safe file'''
         plain = pickle.dumps(self.notebook, pickle.HIGHEST_PROTOCOL)
         crypt = Fernet(self._get_password())
         cipher = crypt.encrypt(plain)
@@ -108,6 +114,7 @@ Notebook Menu
             f.close()
 
     def _get_password(self):
+        '''Request passphrase and derive key from it'''
         passphrase = getpass.getpass()
         kdf = PBKDF2HMAC(
             algorithm = hashes.SHA256(),
@@ -119,6 +126,7 @@ Notebook Menu
         return base64.urlsafe_b64encode(kdf.derive(passphrase.encode('utf-8')))
 
     def quit(self):
+        '''Quit application'''
         print("Thank you for using your notebook today.")
         sys.exit(0)