90_ 发表于 2015-11-16 15:54:14

web信息收集脚本

From:Joseph

#! /usr/bin/python
# -*- coding: utf-8 -*-
# Joseph

import os
import re
import urllib
def getFileName(rootpath,path):
    ''' 获取指定目录下的所有指定后缀的文件名 '''

    f_list = os.listdir(path)
    for i in f_list:
      # os.path.splitext():分离文件名与扩展名
      if os.path.splitext(i) == '.php' or os.path.splitext(i) == '.txt' or os.path.splitext(i) == '.xls' or os.path.splitext(i) == '.rar':
                Absolutepath=urllib.quote(path+i)
                scandir=re.sub(rootpath,"",Absolutepath)
                print urllib.unquote("/"+scandir)
def processDirectory(rootpath,dirname):
      Filelist=os.listdir(dirname)
      getFileName(rootpath,dirname)
      for files in Filelist:
                if os.path.isdir(dirname+files):
                        processDirectory(rootpath,dirname+files+"/")
                        pass
                pass
def main(path):
      rootpath=urllib.quote(path)
      processDirectory(rootpath,path)
      pass

if __name__ == '__main__':
      scandir=raw_input('Please input startdir:')
      main(scandir+"/")

Mrlin 发表于 2015-11-16 18:47:14

balisong 发表于 2015-11-16 20:00:24

续爷这个脚本只是一个雏形,期待更多的人来添加功能

贯之 发表于 2015-11-16 22:58:25

前排瞻仰大牛!!!

吸血鬼 发表于 2015-11-17 21:00:27

学习一下 看看是怎么写的
页: [1]
查看完整版本: web信息收集脚本