Python编写合并字典并实现敏感目录的小脚本

作者:不成大佬不改名 时间:2023-05-28 18:36:29 

0x00 起因

接触网安快一年了,爱收集一些字典啊敏感目录文件啊什么的。收集多了难免有重复,并且有的也需要合并使用方便。自己就写了一个小小的脚步用来完成这个目的。

0x01 代码

第一次发表文章,希望大家多提意见。抱拳了!

代码我就贴出来了


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),'小时',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是绝对路径
   if os.path.isdir(fil): #如果fil是目录加入到D里
     D.write(fil)
     __O__(fil) #继续执行当前函数
   if os.path.isfile(fil): #如果是文件,加入到F里
     F.write(fil)
 D.close()
 F.close()

def type_txt(file,path): #传入路径,文件名(列表)
 try:
   n = time.time()
   print(' 准备执行!')
   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: #循环需要打开的文件列表
     print(' 文件为:',j)
     with open(j, 'r') as f:
       date = f.readlines()

print(' -t 当前执行文件为: ' + j,'数量:',len(date))
       for i in date:     #读取每一行
         af, aq = os.path.splitext(i)        #分裂成路径及后缀名
         aq = aq.lower()
         if i != '\n':               #如果没有换行符会加上
           if '\n' not in i:
             i = i + '\n'
             aq+='\n'
           if aq == '.php\n' and i not in di['new_php.txt']:            #判读后缀名并通过后缀名放到相应的字典中,同时判读字典对应的文件列表中是否存在,如果是False则加入进去
             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("字典文件中有错误请检查是否是字典",e)
 di1,fi = os.path.split(path)
 print(' 分类完毕......准备写入')
 for key in di:                 #写入到文件中
   o = open(key,'w')
   for i in di[key]:
     o.write(i)
 Time(n, time.time())

def dir_txt(path):                                   #传进去路径,返回来的是路径下的后缀为txt的文件。如果没有则是当前路径。
 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(" 开始去除重复.....")
   for i in path_list:      #重复每一个给定的密码文件
     F = open(i,'r')
     txt = F.readlines()
     F.close
     for j in txt:    #读取每一个密码文件,并且判读是否存在于password_list列表,如不存在则放进列表
       if '\n' not in j:
         j+='\n'
       if j not in password_list:
         password_list.append(j)
   print(" 完成,并且准备写入。")
   file = open('Password.txt','w')    #打开即将写入的Password.txt,把passowrd_list列表数据全部写入进去
   for i in password_list:
     file.write(i)
   file.close
   print(" 运行完毕。")
 except :
   print(" 请检查文件是否存在,路径是否正确。")

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参数是互相排斥的,并不能一起使用。
   group.add_argument('-t',action='store_true',help='后台目录字典合并',)
   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(" 错误请检查输入。")
 except (KeyboardInterrupt):
   print(' 以退出')

来源:https://bbs.ichunqiu.com/thread-49190-1-1.html

标签:Python,合并字典,敏感目录
0
投稿

猜你喜欢

  • django 使用全局搜索功能的实例详解

    2023-01-26 05:56:56
  • Python如何使用字符打印照片

    2023-06-12 09:20:34
  • 数据库设计三大范式简析

    2024-01-13 10:33:10
  • PHP脚本监控Nginx 502错误并自动重启php-fpm

    2023-09-26 00:11:45
  • 关于matlab图像滤波详解(二维傅里叶滤波)

    2022-04-16 19:51:35
  • Python 实现引用其他.py文件中的类和类的方法

    2023-08-02 10:38:01
  • Python的迭代器和生成器

    2022-04-29 17:56:09
  • 梅尔频率倒谱系数(mfcc)及Python实现

    2021-08-11 11:06:39
  • python工具快速为音视频自动生成字幕(使用说明)

    2021-04-14 15:15:26
  • Python3从零开始搭建一个语音对话机器人的实现

    2023-11-15 05:52:12
  • 表格头固定而列可滚动的效果

    2020-08-11 21:28:41
  • CentOS 6.5中安装Python 3.6.2的方法步骤

    2022-10-11 09:13:43
  • Python 数据可视化pyecharts的使用详解

    2021-07-07 20:29:55
  • vue-loader和webpack项目配置及npm错误问题的解决

    2024-04-29 13:10:33
  • php和asp利用Shell.Application来执行程序的代码

    2024-04-29 13:58:02
  • 利用django-suit模板添加自定义的菜单、页面及设置访问权限

    2023-01-13 02:09:09
  • golang时间、时区、格式的使用方法

    2024-02-02 07:46:06
  • 基于python实现图书管理系统

    2022-11-28 05:46:42
  • 超详细注释之OpenCV操作图像平移转换

    2022-08-14 19:29:38
  • Oracle + mybatis实现对数据的简单增删改查实例代码

    2024-01-27 06:14:53
  • asp之家 网络编程 m.aspxhome.com