0x00 起因

站在用戶的角度思考問(wèn)題，與客戶深入溝通，找到龍?zhí)毒W(wǎng)站設(shè)計(jì)與龍?zhí)毒W(wǎng)站推廣的解決方案，憑借多年的經(jīng)驗(yàn)，讓設(shè)計(jì)與互聯(lián)網(wǎng)技術(shù)結(jié)合，創(chuàng)造個(gè)性化、用戶體驗(yàn)好的作品，建站類型包括：成都網(wǎng)站設(shè)計(jì)、做網(wǎng)站、企業(yè)官網(wǎng)、英文網(wǎng)站、手機(jī)端網(wǎng)站、網(wǎng)站推廣、域名注冊(cè)、網(wǎng)頁(yè)空間、企業(yè)郵箱。業(yè)務(wù)覆蓋龍?zhí)兜貐^(qū)。

接觸網(wǎng)安快一年了，愛(ài)收集一些字典啊敏感目錄文件啊什么的。收集多了難免有重復(fù)，并且有的也需要合并使用方便。自己就寫了一個(gè)小小的腳步用來(lái)完成這個(gè)目的。

0x01 代碼

第一次發(fā)表文章，希望大家多提意見(jiàn)。抱拳了！

代碼我就貼出來(lái)了

import osimport time
import argparse
import optparse

def Time(n,p):
  pat_time = p-n
  minute = pat_time/60
  hour = minute/60
  print(int(hour),'小時(shí)',int(minute%60),'分鐘',int(pat_time%60),'秒')

def __O__(path): #遍歷目錄，包括目錄里的子目錄
  f = os.listdir(path) #查看路徑下所有文件及目錄
  D = open('D.txt','w')
  F = open('F.txt','w')
  for i in f:
    fil = os.path.join(path,i) #dir是絕對(duì)路徑
    if os.path.isdir(fil): #如果fil是目錄加入到D里
      D.write(fil)
      __O__(fil) #繼續(xù)執(zhí)行當(dāng)前函數(shù)
    if os.path.isfile(fil): #如果是文件，加入到F里
      F.write(fil)
  D.close()
  F.close()


def type_txt(file,path): #傳入路徑，文件名（列表）
  try:
    n = time.time()
    print(' 準(zhǔn)備執(zhí)行!')
    path_list = []  #存放需要打開的文件
    di = {}
    date_list = []
    dict_li = ['new_php.txt', 'new_asp.txt', 'new_aspx.txt', 'new_jsp.txt', 'new_dir.txt', 'new_mdb.txt',] #合并的字典列表
    for i in dict_li:
      di = []
    for h in file:               #合并給的路徑及文件名
      fn = os.path.join(path, h)
      path_list.append(fn)
    for j in path_list: #循環(huán)需要打開的文件列表
      print(' 文件為：',j)
      with open(j, 'r') as f:
        date = f.readlines()

        print(' -t 當(dāng)前執(zhí)行文件為： ' + j,'數(shù)量：',len(date))
        for i in date:     #讀取每一行
          af, aq = os.path.splitext(i)        #分裂成路徑及后綴名
          aq = aq.lower()
          if i != '\n':               #如果沒(méi)有換行符會(huì)加上
            if '\n' not in i:
              i = i + '\n'
              aq+='\n'
            if aq == '.php\n' and i not in di['new_php.txt']:            #判讀后綴名并通過(guò)后綴名放到相應(yīng)的字典中，同時(shí)判讀字典對(duì)應(yīng)的文件列表中是否存在，如果是False則加入進(jìn)去
              di['new_php.txt'].append(i)
            elif aq == '.asp\n' and i not in di['new_asp.txt']:
              di['new_asp.txt'].append(i)
            elif aq == '.aspx\n' and i not in di['new_aspx.txt']:
              di['new_aspx.txt'].append(i)
            elif aq == '.jsp\n' and i not in di['new_jsp.txt'] :
              di['new_jsp.txt'].append(i)
            elif (aq == ('.ini\n' or '.xml\n' or '.cgi\n' or '.log\n' or '.db\n' or '.mdb\n' or '.sql\n' or '.xml\n' or '.htm\n' or '.html\n' or '.bak\n' or '.dbf\n' or '.zip\n' or '.rb\n' or '.js\n' or '.css\n' or '.html?\n')) and i not in di['new_mdb.txt']:
              di['new_mdb.txt'].append(i)
            elif i not in di['new_dir.txt'] :
              di['new_dir.txt'].append(i)
        f.close()
  except (UnicodeDecodeError) as e:
      print("字典文件中有錯(cuò)誤請(qǐng)檢查是否是字典",e)
  di1,fi = os.path.split(path)
  print(' 分類完畢......準(zhǔn)備寫入')
  for key in di:                 #寫入到文件中
    o = open(key,'w')
    for i in di[key]:
      o.write(i)
  Time(n, time.time())

def dir_txt(path):                                   #傳進(jìn)去路徑，返回來(lái)的是路徑下的后綴為txt的文件。如果沒(méi)有則是當(dāng)前路徑。
  dir_txt = []
  a = os.listdir(path)
  for i in a:
    dd, ff = os.path.splitext(i)
    if ff == '.txt':
      dir_txt.append(i)
  return dir_txt




def password(file,path):
  try:
    path_list = []
    password_list = []
    for i in file:
      pathl = os.path.join(path,i)
      path_list.append(pathl)
    print(" 開始去除重復(fù).....")
    for i in path_list:      #重復(fù)每一個(gè)給定的密碼文件
      F = open(i,'r')
      txt = F.readlines()
      F.close
      for j in txt:    #讀取每一個(gè)密碼文件，并且判讀是否存在于password_list列表，如不存在則放進(jìn)列表
        if '\n' not in j:
          j+='\n'
        if j not in password_list:
          password_list.append(j)
    print(" 完成，并且準(zhǔn)備寫入。")
    file = open('Password.txt','w')    #打開即將寫入的Password.txt，把passowrd_list列表數(shù)據(jù)全部寫入進(jìn)去
    for i in password_list:
      file.write(i)
    file.close
    print(" 運(yùn)行完畢。")
  except :
    print(" 請(qǐng)檢查文件是否存在，路徑是否正確。")

def Optparse():
  try:
    parser = argparse.ArgumentParser(
    """把字典分類：new_php.txt,new_asp.txt,new_aspx.txt,new_jsp.txt,new_mdb.txt,new_dir.
    
    
    
                     -t -path C:\\x\\x\*   
                     -t -path C:\\x\\x -file xxx.txt,xx.txt   """)

    group = parser.add_mutually_exclusive_group()
    group.add_argument('-p',action='store_true',help='密碼合并')           #-p 和-t參數(shù)是互相排斥的，并不能一起使用。
    group.add_argument('-t',action='store_true',help='后臺(tái)目錄字典合并',)
    parser.add_argument('-file', type=str, help='輸入的文件',required=False)
    parser.add_argument('-path',type=str,help='你所指定的目錄',required=False)
    args = parser.parse_args()

    if args.t:      #判讀t是否真
      if args.file == None:
        path,file = os.path.split(args.path)
        dir__txt = dir_txt(path)
        type_txt(dir__txt,path)
      elif args.file != None:
        list_file = args.file.split(',')
        type_txt(list_file,args.path)
    elif args.p:    #判讀p是否真
      if args.file == None:
        path, file = os.path.split(args.path)
        dir__txt = dir_txt(path)
        password(dir__txt,path)
      elif args.file != None:
        list_file = args.file.split(',')
        password(list_file,args.path)
    else:
      print(" 錯(cuò)誤請(qǐng)檢查輸入。")
  except (KeyboardInterrupt):
    print(' 以退出')

另外有需要云服務(wù)器可以了解下創(chuàng)新互聯(lián)scvps.cn，海內(nèi)外云服務(wù)器15元起步，三天無(wú)理由+7*72小時(shí)售后在線，公司持有idc許可證，提供“云服務(wù)器、裸金屬服務(wù)器、高防服務(wù)器、香港服務(wù)器、美國(guó)服務(wù)器、虛擬主機(jī)、免備案服務(wù)器”等云主機(jī)租用服務(wù)以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡(jiǎn)單易用、服務(wù)可用性高、性價(jià)比高”等特點(diǎn)與優(yōu)勢(shì)，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應(yīng)用場(chǎng)景需求。

本文標(biāo)題：Python編寫合并字典并實(shí)現(xiàn)敏感目錄的小腳本-創(chuàng)新互聯(lián)
文章地址：http://weahome.cn/article/gpjhp.html