A-14-OS模块练习

高渐飞

1. #coding: UTF-8
   
2. 
   
3. #知识点总结
   
4. #1.os模块即操作系统模块，详细文档http://bbs.fishc.org/thread-45512-1-1.html
   
5. #2.
   
6. 
   
7. 
   
8. #实操练习
   
9. #1.查询文件夹下各种文件类型的个数
   
10. #思路-利用os.listdir得到指定文件夹下各文件的名称，
    
11. #之后通过循环，查找特定文件扩展名是否在已得到的文件名中，有则加一。最后输出即可。
    
12. #优点:代码简单，思路清晰；缺点:需提前指定文件扩展名，未指定的无法监测其个数
    
13. import os
    
14. 
    
15. file_name = os.listdir('E:\\test')
    
16. print(file_name)
    
17. 
    
18. txt = 0
    
19. png = 0
    
20. py = 0
    
21. docx = 0
    
22. doc = 0
    
23. 
    
24. for each_file in file_name:
    
25.     if '.txt' in each_file:
    
26.         txt += 1
    
27.     elif '.png' in each_file:
    
28.         png += 1
    
29.     elif '.py' in each_file:
    
30.         py += 1
    
31.     elif '.docx' in each_file:
    
32.         docx += 1
    
33.     elif '.' not in each_file:
    
34.         doc += 1
    
35. 
    
36. print('txt 是 %d, png 是 %d, py 是 %d, docx 是 %d, doc 是 %d' % (txt,png,py,docx,doc))
    
37. 
    
38. 
    
39. #2.改进版-查询文件夹下各种文件类型的个数
    
40. #思路-通过os.listdir得到文件名称，再对文件名称进行遍历，分离出名称和扩展名
    
41. #若扩展名在file_type字典中，其value值赋1；否则原value+1；之后进行下一次循环直到遍历结束
    
42. #最后根据键值即文件扩展扩展名，遍历字典file_type输出value即可
    
43. import  os
    
44. 
    
45. def  seek_extension(path):
    
46.     file_type = {}         #建立空字典，有key和value两部分
    
47.     file_names = os.listdir(path)            #列举指定目录中的文件名，如a.txt,b.py,c.txt
    
48.     for  each_file in file_names:
    
49.         file_name_extension = os.path.splitext(each_file)  #分离文件名与扩展名，返回(f_name, f_extension)元组
    
50.         if  file_name_extension[1] not in file_type:     #file_name_extension[1]此处即f_extension。
    
51.                                                          #file_name_extension非元组嵌套元组，每次都是(name,ext...)
    
52.             file_type[file_name_extension[1]] = 1    #例如file_name_extension[1]为txt,字典通过键值key(可为字符串)来精确索引
    
53.         else:
    
54.             file_type[file_name_extension[1]] += 1
    
55. 
    
56.     for eachkey in file_type.keys():
    
57.         print("该文件夹下共有类型为【%s】的文件 %d 个" % ( eachkey, file_type[eachkey] ) )
    
58. 
    
59. path = input("输入要查找的绝对路径： ")
    
60. seek_extension(path)
    
61. 
    
62. 
    
63. #3.查询指定文件夹下各文件的大小
    
64. #思路-通过os.listdir获得指定目录下文件名，遍历文件，其中通过os.path.getsize即可获得文件大小。
    
65. #通过字典来存储文件名和其占用内存大小，遍历输出即可
    
66. import os
    
67. 
    
68. def  file_size_calc(path):
    
69.     file = os.listdir(path)     #得到文件名构成列表如['a.txt','b.py']
    
70.     file_size = {}              #key为文件名，value为文件大小，如{'asd.txt':141,'b.py':409}
    
71.     for  each_file in file:
    
72.         file_size[each_file] = os.path.getsize(path + '\\' + each_file)     #其中path为绝对路径，返回文件大小，单位字节
    
73. 
    
74.     for  each_key in file_size.keys():
    
75.         print('%s 【%d bytes】' % ( each_key, file_size[each_key] ) )
    
76. 
    
77. 
    
78. path = input("请输入要查找位置的绝对路径： ")
    
79. while not isabs(path)
    
80. {
    
81.     path = input("请输入要查找位置的绝对路径： ")
    
82. }
    
83. file_size_calc(path)
    
84. 
    
85. 
    
86. #4.将指定目录下指定格式的文件保存的文件中
    
87. import os
    
88. 
    
89. doc_path = input('请输入待查找的初始目录:')
    
90. while not os.path.isdir(doc_path):              #判断指定路径是否存在且是一个目录
    
91.     doc_path = input('输入错误,请重新输入待查找的初始目录:')
    
92. 
    
93. save_path=input('请输入文件保存目录:')
    
94. while not os.path.isdir(save_path):              #判断指定路径是否存在且是一个目录
    
95.     save_path = input('输入错误,请重新输入文件保存目录:')
    
96. 
    
97. doc_walk = os.walk(doc_path)   #遍历doc_path路径下所有的子目录，返回一个三元组：(路径, [包含目录], [包含文件])
    
98. 
    
99. file_path = []   #建立列表，存储文件地址
    
100. 
     
101. for each_part in doc_walk:
     
102.     for each_file in each_part[2]:
     
103.         if '.avi' or '.mp4' or '.rmvb' in each_file.lower():
     
104.             file_path.append(os.path.join(each_part[0],each_file) + '\n' * 2)  #将each_part[0],each_file组合成一个路径名
     
105.                                                                                #将路径添加到列在file_path中
     
106. print(file_path)
     
107. 
     
108. f = open(save_path+'\\content2.txt','w',encoding = 'GBK')
     
109. f.writelines(file_path)
     
110. f.close()
     
111. 
     
112. 
     
113. #5.没看懂，以后再说
     
114. 
     
115. import os
     
116. 
     
117. def  pos_in_line(line, key):
     
118.     pos = []
     
119.     begin = line.find(key)
     
120. 
     
121.     while  begin != -1:
     
122.         pos.append(begin + 1)
     
123.         begin = line.find(key, begin + 1)
     
124. 
     
125.     return pos
     
126. 
     
127. def  pos_in_file(file, key):
     
128.     dict_key = dict()
     
129.     count = 0
     
130.     target_file = open(file)
     
131. 
     
132.     for  each_line in target_file:
     
133.         count += 1
     
134.         if  key in each_line:
     
135.             pos = pos_in_line(each_line, key)
     
136.             dict_key[count] = pos
     
137. 
     
138.     target_file.close()
     
139.     return dict_key
     
140. 
     
141. def  print_key(dict_key):
     
142.     keys = dict_key.keys()
     
143.     keys = sorted(keys)
     
144.     for  each_key in keys:
     
145.         print("关键字出现在第 %s 行， 第 %s 个位置" % ( each_key, dict_key[each_key] ) )
     
146. 
     
147. 
     
148. def  search_key(path, key):
     
149.     os.chdir(path)
     
150. 
     
151.     for  each_file in os.listdir(os.curdir):
     
152.         if  os.path.isdir(each_file):
     
153.             search_key(each_file, key)
     
154.             os.chdir(os.pardir)
     
155.         if  os.path.isfile(each_file):
     
156.             if  os.path.splitext(each_file)[1] == '.txt':
     
157.                 dict_key = pos_in_file(each_file, key)
     
158.                 if  dict_key != {}:
     
159.                     print('==================================================================================')
     
160.                     print("在文件【%s】中找到关键字【%s】" %( (os.getcwd() + os.sep + each_file), key ) )
     
161.                     print_key(dict_key)
     
162. 
     
163. 
     
164. path = input("请输入要查找的路径： ")
     
165. key = input("请输入要查找的关键字： ")
     
166. search_key(path, key)

复制代码

软糖别太软

请教一下：

#2.改进版-查询文件夹下各种文件类型的个数

这个地方的path是统一的绝对路径吗？