2021-09-20-python-创建文件脚本

创建文件脚本

#!/usr/bin/python3
import os

def get_fname():
    while 1:
        fname = input('文件名：')
        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_file(fname,content):
    with open(fname, 'w') as fjob:
        fjob.writelines(content)

if __name__ == '__main__':
    fname = get_fname()
    content = get_content()
    content = ['%s\n' % line for line in content]
    write_file(fname,content)