python - 写入文件并询问用户文件名

Question

    import sys
    outfile = open( r'/Users/user1/Desktop/myDoc.txt', 'w' ) 
    outfile.write("hello world")
    outfile.close()

我试图创建一个程序来创建一个文件，并将 hello world 写入该文件，有两件事。现在我已经设置好了，因此它将保存到我的桌面并将文件命名为“myDoc.html”。我是否可以以某种方式纠正文件名，以便将其保存到任何人的桌面上，以便程序询问用户如何命名文件但保留 .txt 文件扩展名？

Answer 1

import os
outfile = input('Enter file name: ')
outfile = os.path.join(os.path.expanduser('~'),'Desktop',outfile)    
with open(outfile +'.txt', 'w') as f:
    f.write("hello world")