python-生成文本文件(43)

最新推荐文章于 2024-09-21 16:27:37 发布

原创最新推荐文章于 2024-09-21 16:27:37 发布 · 364 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一个使用Python进行文件创建及写入的实例，通过获取不存在的文件名避免覆盖已有文件，然后从用户处收集数据并将其写入指定的文本文件中。

import os

def get_fname():
        while True:
                fname = input('filename: ')
                if  not os.path.exists(fname):  #os.path.exists(),查看文件是否存在
                        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)    #向content列表中加入内容
        return content

def wfile(fname,content):
        with open(fname,'w') as fobj:
                fobj.writelines(content)    #把content列表中的数据写入文件中

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




# 测试
# [root@room9pc01 KINGSTON]# python3 43生成文本文件.py
# filename: /tmp/abc.txt
# 输入数据，输入end结束
# >abc
# >dvd
# >haha
# >end
# [root@room9pc01 KINGSTON]# cat /tmp/abc.txt
# abc
# dvd
# haha