python新手灰帽脚本练习---文件路径遍历收集

Posted 2021-01-02 i11USi0n

 

总体思路：

os.path.abspath(\'.\')      #获取当前路径

os.listdir                    #获取指定的文件夹包含的文件或文件夹的名字的列表

os.path.join                #用获取的文件名拼接路径

os.path.isdir               #判断路径是否为目录，用于递归

 1 # -*- coding:utf-8 -*-
 2 
 3 __author__=\'Dou—wei\'
 4 
 5 import os
 6 import sys
 7 
 8 file_name = sys.argv[1]                           #获取用户的终端输入以命名文件
 9 start_path = os.path.abspath(\'.\')                 #获取当前路径
10 host = os.path.join(start_path,file_name)         #拼接路径
11 f = open(host,\'w\')                                #在拼接的路径创建可写文件
12 
13 def search(path):
14     for x in os.listdir(path):                    # os.listdir(path)用于返回指定的文件夹包含的文件或文件夹的名字的列表
15         current_name = os.path.join(path,x)       #拼接路径
16         f.write(current_name)                     #写入包含文件名的路径
17         f.write(\'\\n\')                             #换行
18         if os.path.isdir(current_name):           #判断路径是否为目录
19             search(current_name)                  #递归执行，进一步探索
20 
21 def main():
22     try:
23         search(start_path)
24     except Exception,e:                           #输出错误信息
25         print e
26 
27 if __name__==\'__main__\':
28     main()                        

View Code

 

 

大致效果：