Browse Source

added method remove_note() to remove note with given id from note list; added menu option to call remove_note()

master
T. Meissner 10 years ago
parent
commit
ace441084b
1 changed files with 18 additions and 8 deletions
  1. +18
    -8
      python_3_oop/chapter02/menu.py

+ 18
- 8
python_3_oop/chapter02/menu.py View File

@ -26,10 +26,11 @@ class Menu:
"2": self.search_notes, "2": self.search_notes,
"3": self.add_note, "3": self.add_note,
"4": self.modify_note, "4": self.modify_note,
"5": self.load_notes,
"6": self.save_notes,
"7": self.__init__,
"8": self.quit
"5": self.remove_note,
"6": self.load_notes,
"7": self.save_notes,
"8": self.__init__,
"9": self.quit
} }
def display_menu(self): def display_menu(self):
@ -39,10 +40,11 @@ Notebook Menu
2. Search Notes 2. Search Notes
3. Add Note 3. Add Note
4. Modify Note 4. Modify Note
5. Load Notes
6. Save Notes
7. Reset Notes
8. Quit """)
5. Remove Note
6. Load Notes
7. Save Notes
8. Reset Notes
9. Quit """)
def run(self): def run(self):
'''Display the menu and respond to choices.''' '''Display the menu and respond to choices.'''
@ -87,6 +89,14 @@ Notebook Menu
if not self.notebook.modify_tags(id, tags): if not self.notebook.modify_tags(id, tags):
print("Note with id {0} doesn't exist.".format(id)) print("Note with id {0} doesn't exist.".format(id))
def remove_note(self):
'''Remove note with given id from note list'''
id = input("Enter a note id: ")
if self.notebook._remove_note(id):
print("Note with id {0} removed.".format(id))
else:
print("Note with id {0} doesn't exist.".format(id))
def load_notes(self): def load_notes(self):
'''Decrypt notebook safe file and load it into notebook object''' '''Decrypt notebook safe file and load it into notebook object'''
try: try:


Loading…
Cancel
Save