def SaveMessage1(path, message): while True: try: print( "------ Would you like to save your message in a new file or in this file? ------\n" ) fileSituation = int( input("[1] New file\n[2] This file\n[0] Don't save \n-> ")) if fileSituation == 1: fileName = File.CreateFile() if fileName == "": continue File.WriteFile(fileName, message) break elif fileSituation == 2: File.WriteFile(path, message) break elif fileSituation == 0: break else: print("\n------ Invalid option ------\n") except ValueError: print("\n------ Invalid option ------\n")
def SaveMessage2(message): while True: try: print( "------ Would you like to create a new file with your message? ------\n" ) fileSituation = int(input('[1] Yes\n[2] No\n\n-> ')) if fileSituation == 1: fileName = File.CreateFile() if fileName == "": continue File.WriteFile(fileName, message) break elif fileSituation == 2: break else: print("\n------ Invalid option ------\n") except ValueError: print("\n------ Invalid option ------\n")