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")
