用python把所有出现snprintf的c文件添加_snprintf

用VS2013编译ffmpeg时，报snprintf找不到的错误。

需要在调用snprintf的地方加上这个宏：

#define snprintf _snprintf

然而，ffmpeg 依赖的库源码太多，不可能手动修改，于是有了下面py脚本，功用：扫描目录下所有C文件，查找文件中是否含有snprintf，若有，则在文件开头添加上述宏定义。

代码如下：

import os
import sys

def get_files(root_path):  # 注意root_path前加上r
    '''
    获得目录root_path下（包括各级子目录）所有文件的路径
    '''
    file_list = []
    for i in os.listdir(root_path):
        path = root_path + r'\\' + i
        if os.path.isfile(path):
            file_list.append(path)
        elif os.path.isdir(path):
            files = get_files(path)
            for f in files:
                file_list.append(f)
    return file_list


def word_in_files(root_path, word):
    '''
    获得目录root_path下（包括各级子目录）所有包含字符串word的文件的路径
    '''
    file_list = get_files(root_path)
    for path in file_list:
        if ".c" in path:    # 筛选c文件
            with open(path, "r+", encoding="utf8", errors='ignore') as f:
                content = f.read()
                if word in content:
                    f.seek(0, 0)
                    f.write('#define snprintf _snprintf\n'+content)

def main(argv=None):
	word_in_files(r'E:\ffmpeg\code', 'snprintf')

if __name__=="__main__":
	sys.exit(main())