From e71d139a13ce29cd9167ca03530129bf696d57f0 Mon Sep 17 00:00:00 2001
From: tmeissner <programming@goodcleanfun.de>
Date: Tue, 26 Aug 2014 17:17:37 +0200
Subject: [PATCH] added handling of error in _decode_notefile() method by
 chacking its return value in method load_notes()

---
 python_3_oop/chapter02/menu.py | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/python_3_oop/chapter02/menu.py b/python_3_oop/chapter02/menu.py
index b9afce8..36aed2c 100644
--- a/python_3_oop/chapter02/menu.py
+++ b/python_3_oop/chapter02/menu.py
@@ -28,7 +28,8 @@ class Menu:
             "4": self.modify_note,
             "5": self.load_notes,
             "6": self.save_notes,
-            "7": self.quit
+            "7": self.__init__,
+            "8": self.quit
         }
 
     def display_menu(self):
@@ -40,7 +41,8 @@ Notebook Menu
 4. Modify Note
 5. Load Notes
 6. Save Notes
-7. Quit """)
+7. Reset Notes
+8. Quit """)
 
     def run(self):
         '''Display the menu and respond to choices.'''
@@ -94,8 +96,10 @@ Notebook Menu
         else:
             cipher = f.read()
             f.close()
-            self.notebook = self._decode_notefile(cipher)
-            self.notebook._set_id()
+            notebook = self._decode_notefile(cipher)
+            if notebook:
+                self.notebook = notebook
+                self.notebook._set_id()
 
     def save_notes(self):
         '''Encrypt notebook object and store it into notebook safe file'''