diff --git a/cheat b/cheat index 7c7d639..09a4b50 100755 --- a/cheat +++ b/cheat @@ -87,8 +87,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