import os def get_fname(): while True: fname = input('filename: ') if not os.path.exists(fname): break print('%s already exists. Try again' % fname) return fname def get_content(): content = [] print('输入数据,输入end结束') while True: line = input('>') if line == 'end': break content.append(line) return content def wfile(fname,content): with open(fname, 'w') as fobj: fobj.write(content) if __name__ == '__manin__': fname = get_fname() content = get_content() content = ['%s\n' % line for line in content] wfile(fname,content) wfile('test.txt','hello world!')
执行后会在当前目录生成一个test.txt文件: