diff --git a/text_based.py b/text_based.py index f68e6ba..ca6462b 100644 --- a/text_based.py +++ b/text_based.py @@ -404,6 +404,11 @@ class EditProductMenu(Menu): self.session.close() self.pause() return + elif what == None: + print 'Edit aborted' + return + else: + print 'What what?' class ShowUserMenu(Menu):