Python之创建文件

需求：创建文件，存在的话则重新创建，不存在则创建并写内容到里面

import os

def get_fname():
  "获取文件名"
  while 1:
    fname = input('请输入文件名：')
    #判断文件是否存在，如果文件不存在则赋值fname并中断循环，最终得到文件名
    if not os.path.exists(fname):
      break
    print('文件已存在，请重试！')
  return fname

def get_content():
  "获取文件内容"
  content = []
  print('请输入文件内容，在单独的一行输入end以结束！')
  while 1:
    line = input('(end to quit)>')
    if line == 'end':
      break
    content.append(line)
  return content

def write_content(fname, content):
  "将文件内容写入文件"
  with open(fname, 'w') as fobj:
    fobj.writelines(content)
    
if __name__ == '__main__':
  fname = get_fname
  content = get_content
  content = ['%s\n' % line for line in content]
  write_content(fname, content)