批量重命名文件 Python

最近工作需要，要批量重命名服务器上的文件，将之前文件名中有空格的全部都替换为下划线。
开始准备使用shell脚本，发现python实现更简单。所以就写了这个脚本。

 

import os
import glob
import re
path="C:/test" # 批量化的初始位置
fileHandle = open ( 'C:/test/test.txt', 'w' ) #log记录rename的信息

def getthefile(path):
    for oldfn in glob.glob( path + os.sep + '*' ):
        if os.path.isdir( oldfn):# 查询是含有子目录
            getthefile( oldfn)
        else:
            if re.search(' ', oldfn):
                fn=renamefile(oldfn)
                if os.path.exists(fn):
                    index=1
                    while True:
                        filelist=fn.split('.')
                        fn="%s%s%s.%s"%(filelist[0],"_",index,filelist[1])
                        if os.path.exists(fn):
                            index=index+1
                            continue
                        break
                fileHandle.write ("%s ===> %s\n"%(oldfn,fn))
                fileHandle.flush()
                os.rename(oldfn,fn);
            else:
                fileHandle.write ("%s Nospace!!\n"%(oldfn))
                fileHandle.flush()
                continue
def renamefile(filename):
    return filename.replace(' ','_')
if __name__ == '__main__':
    getthefile(path)