python: file_handling 解决工作中出现的文件处理需求

最新推荐文章于 2025-09-04 16:51:32 发布

weixin_34021089

最新推荐文章于 2025-09-04 16:51:32 发布

阅读量78

点赞数

CC 4.0 BY-SA版权

文章标签： python

原文链接：http://www.cnblogs.com/Alex-Python-Waiter/archive/2012/07/09/2583631.html

本文介绍了一个Python脚本，用于在指定目录中搜索含有特定关键字的文件，并将找到的文件重命名为新的格式。此外，还介绍了如何利用Python标准库中的`shutil`和`copy`模块来实现文件操作和深拷贝。

 1 #!/usr/bin/python
 2 ## A file handling script, including functions that could search file by key
 3 ## and rename file by appending key value to the file names.
 4 
 5 import os
 6 import shutil
 7 import copy
 8 
 9 def get_INV_list(key_file):
10     """ read key_file and get key values into list"""
11     INV_file = open(key_file, 'r')
12     key_list = []
13     
14     for num in INV_file.readlines():
15         key_list.append([num[:-1]])
16     
17     INV_file.close()
18     return key_list
19 
20 def write_result_list(key_list,out_file):
21     """write updated key_list with reference into a file"""
22     result = open(out_file,'w')
23 
24     for key in key_list:
25         result.write('########'.join(key)+'\n')
26     
27     result.close()
28 
29 def search(key_list,search_dir):
30     """search key value in all files in search directory,
31        return a list with key reflect file name"""
32     
33     ##my original workaround
34     #result_list = []
35     #for item in key_list:
36     #    result_list.append(item[:])
37     ##A simpler code
38     #result_list = [item[:] for item in key_list]
39     ## or 
40     #result_list = [list(item) for item in key_list]
41     #best solution is copy.deepcopy()
42     result_list = copy.deepcopy(key_list)
43     
44     for file in os.listdir(search_dir):
45         search_file = open(search_dir+'/'+file,'r')
46         search_str = search_file.read()
47         search_file.close()
48         for key in result_list:
49             if key[0] == search_str[86:92]:    ##only check INV#
50             #if key[0] in search_str:
51                 key.append(file)
52     
53     return result_list
54 
55 def rename(result_list,input_dir,output_dir):
56     """ rename files in input directory and move to output directory,
57         return a list of file names those not moved."""
58     for item in result_list:
59         if len(item) > 1:
60             for filename in item[1:]:
61                 # move files to another folder and rename them
62                 # name format could be modified if necessary
63                 shutil.move(input_dir+'/'+filename,
64                         output_dir+'/'+filename+'.NZ'+item[0])
65 
66     return os.listdir(input_dir)
67 
68 xlist = get_INV_list('INV_Num')
69 ylist = search(xlist,'InputFiles')
70 rest_list = rename(ylist,'InputFiles','OutputFiles')
71 write_result_list(ylist,'resultList')