Home > Article > Backend Development > python批量修改文件名的实现代码
#coding:utf-8 #批量修改文件名 import os import re import datetime re_st = r'(\d+)\+\s?\((\d+)\)' #用于匹配旧的文件名,需含分组 re_match_old_file_name = re.compile(re_st) #要修改的目录 WORKING_PATH = r'F:\Gallery' #---------------------------------------------------------------------- def rename_fomat(name): """ 文件重命名格式组织模块(一般修改这里就可以了) NOTE:返回类型必须是unicode """ if name: re_rn = re_match_old_file_name.findall(name) if re_rn and re_rn != []: re_rn = re_rn[0] num = int(re_rn) new_nm = u'NO.%04d' % ( num) return new_nm #---------------------------------------------------------------------- def logs(error): """ 错误记录 """ log = '' LOG_FILE = open(r'./log.txt', 'a') live_info =""" ========== Time : %s title : %s Path : %s ========== """ % ( datetime.datetime.now(), str(error['title']), str(error['index']), ) log += live_info errors = error['error_paths'] for item in errors: item = '%s\n' % item log += item log = log.encode('utf-8') try: LOG_FILE.write(log) except IOError: print u'写入日志失败' finally: LOG_FILE.close() #---------------------------------------------------------------------- def rename(old, new): """ 文件重命名模块 return: 0:rename success 1:the new path is exists -1:rename failed """ if not os.path.exists(new): try: os.renames(old, new) return 0 except IOError: print 'path error:', new return -1 else: return 1 #---------------------------------------------------------------------- def get_dirs(path): """ 获取目录列表 """ if os.path.exists(path): return os.listdir(path) else: return -1 #---------------------------------------------------------------------- def get_input_result(word, choice): """ 获取正确的输入结果 """ correct_result = set(choice) word = '===%s?\n[in]:' % (word) while True: in_choice = raw_input(word) if in_choice in correct_result: return in_choice #---------------------------------------------------------------------- def batch_rename(index, dirs = []): """ 批量修改文件 """ index = unicode(index) errors = [] if dirs == []: dirs = get_dirs(path = index) if dirs and dirs != []: for item in dirs: item = unicode(item) new_name = rename_fomat(item) if new_name : old_pt = u'%s\\%s'% (index, item) new_pt = u'%s\\%s'% (index, new_name) res_rn = rename(old_pt, new_pt) if res_rn != 0: errors.append(item) else: errors.append(item) if errors and errors != []: print 'Rename Failed:' logs({ 'index': index, 'title': 'Rename Failed' , 'error_paths': errors, }) for i, item in enumerate(errors): print item, '|', if i % 5 == 4: print '' print '' else: return -1 #---------------------------------------------------------------------- def batch_rename_test(index): """ 测试 返回过滤结果 """ index = unicode(index) errors = [] correct = [] dirs = get_dirs(path = index) if dirs and dirs != []: for x, item in enumerate(dirs): item = unicode(item) new_name = rename_fomat(item) if new_name : correct.append(item) old_pt = u'%s\\%s'% (index, item) new_pt = u'%s\\%s'% (index, new_name) print '[%d]O: %s' % ( x + 1, old_pt) print '[%d]N: %s' % ( x + 1, new_pt) else: errors.append(item) if errors and errors != []: print 'Not Match:' logs({ 'index': index, 'title': 'Not Match', 'error_paths': errors, }) for i, item in enumerate(errors): print item, '|', if i % 5 == 4: print '' print '' return correct #---------------------------------------------------------------------- def manage(index): """ 程序组织块 """ file_filter = batch_rename_test(index) do_choice = get_input_result( word = 'Do with this(y / n)', choice = ['y', 'n'] ) if do_choice == 'y': batch_rename(index, dirs= file_filter) print 'Finished !' if __name__ == '__main__': path = WORKING_PATH manage(index = path)