From ffcafa4480ee6f132d993bb0210ddcd86c23c41d Mon Sep 17 00:00:00 2001 From: Ritchie Lincoln Date: Sat, 7 Sep 2013 12:41:51 -0600 Subject: [PATCH] give user the option to edit cheatsheet when trying to create an existing cheatsheet --- cheat | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/cheat b/cheat index 6d1bddc..faf8861 100755 --- a/cheat +++ b/cheat @@ -64,8 +64,12 @@ def edit_cheatsheet(cheat, cheatsheets): def create_cheatsheet(cheat, cheatsheets): "Create a new cheatsheet." if cheat in cheatsheets: - print ('Cheatsheet "%s" already exists' % cheat) - exit() + print ('Cheatsheet "%s" already exists...' % cheat) + ans = raw_input('Would you like to edit cheatsheet for "%s" (Y/N)? ' % cheat) + if ans.lower() in ['y', 'yes']: + edit_cheatsheet(cheat, cheatsheets) + else: + exit() import cheatsheets as cs