python开发_fileinput

python中，fileinput模块对读取文件操作提供了一些有用的方法

下面是我做的demo：

运行效果：

======================================

代码部分：

======================================

#python fileinput
'''
    fileinput:
    优点：
        可以同时读取多个文件
        可以获取到正在读取的文件的filename
        ....
    #######################################
    This module implements a helper class
    and functions to quickly write a loop
    over standard input or a list of files.
    If you just want to read or write one
    file see open().
    #正如API中所描述的一样：
    如果需要读/写文件推荐使用open()方法
'''

import fileinput
import os

def get_file_content(files):
    '''读取(多个)文件中的内容，以字符串的形式返回'''
    if files != None:
        lines = ''
        with fileinput.input(files) as fp:
            for line in fp:
                lines += line
            return lines
    else:
        print('files is None')

def get_file_name(file):
    '''只有文件被读的时候，才会取得filename，否则返回None'''
    if os.path.exists(file) and os.path.isfile(file):
        names = []
        for line in fileinput.input(file):
            name = fileinput.filename()
            if name != None:
                fileinput.nextfile()
            names.append(name)  
        return names
    else:
        print('the path [{}] is not exist!'.format(file))
        

def main():
    files = ('c:\temp.txt', 'c:\test.txt')
    file = 'c:\temp.txt'
    content = get_file_content(files)
    print(content)
    name = get_file_name(file)
    print(name)

if __name__ == '__main__':
    main()