Python编写合并字典并实现敏感目录的小脚本
0x00起因
接触网安快一年了，爱收集一些字典啊敏感目录文件啊什么的。收集多了难免有重复，并且有的也需要合并使用方便。自己就写了一个小小的脚步用来完成这个目的。
0x01代码
第一次发表文章，希望大家多提意见。抱拳了！
代码我就贴出来了
importosimporttime
importargparse
importoptparse
defTime(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')
foriinf:
fil=os.path.join(path,i)#dir是绝对路径
ifos.path.isdir(fil):#如果fil是目录加入到D里
D.write(fil)
__O__(fil)#继续执行当前函数
ifos.path.isfile(fil):#如果是文件，加入到F里
F.write(fil)
D.close()
F.close()
deftype_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',]#合并的字典列表
foriindict_li:
di=[]
forhinfile:#合并给的路径及文件名
fn=os.path.join(path,h)
path_list.append(fn)
forjinpath_list:#循环需要打开的文件列表
print('文件为：',j)
withopen(j,'r')asf:
date=f.readlines()
print('-t当前执行文件为：'+j,'数量：',len(date))
foriindate:#读取每一行
af,aq=os.path.splitext(i)#分裂成路径及后缀名
aq=aq.lower()
ifi!='\n':#如果没有换行符会加上
if'\n'notini:
i=i+'\n'
aq+='\n'
ifaq=='.php\n'andinotindi['new_php.txt']:#判读后缀名并通过后缀名放到相应的字典中，同时判读字典对应的文件列表中是否存在，如果是False则加入进去
di['new_php.txt'].append(i)
elifaq=='.asp\n'andinotindi['new_asp.txt']:
di['new_asp.txt'].append(i)
elifaq=='.aspx\n'andinotindi['new_aspx.txt']:
di['new_aspx.txt'].append(i)
elifaq=='.jsp\n'andinotindi['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'))andinotindi['new_mdb.txt']:
di['new_mdb.txt'].append(i)
elifinotindi['new_dir.txt']:
di['new_dir.txt'].append(i)
f.close()
except(UnicodeDecodeError)ase:
print("字典文件中有错误请检查是否是字典",e)
di1,fi=os.path.split(path)
print('分类完毕......准备写入')
forkeyindi:#写入到文件中
o=open(key,'w')
foriindi[key]:
o.write(i)
Time(n,time.time())
defdir_txt(path):#传进去路径，返回来的是路径下的后缀为txt的文件。如果没有则是当前路径。
dir_txt=[]
a=os.listdir(path)
foriina:
dd,ff=os.path.splitext(i)
ifff=='.txt':
dir_txt.append(i)
returndir_txt
defpassword(file,path):
try:
path_list=[]
password_list=[]
foriinfile:
pathl=os.path.join(path,i)
path_list.append(pathl)
print("开始去除重复.....")
foriinpath_list:#重复每一个给定的密码文件
F=open(i,'r')
txt=F.readlines()
F.close
forjintxt:#读取每一个密码文件，并且判读是否存在于password_list列表，如不存在则放进列表
if'\n'notinj:
j+='\n'
ifjnotinpassword_list:
password_list.append(j)
print("完成，并且准备写入。")
file=open('Password.txt','w')#打开即将写入的Password.txt，把passowrd_list列表数据全部写入进去
foriinpassword_list:
file.write(i)
file.close
print("运行完毕。")
except:
print("请检查文件是否存在，路径是否正确。")
defOptparse():
try:
parser=argparse.ArgumentParser(
"""把字典分类：new_php.txt,new_asp.txt,new_aspx.txt,new_jsp.txt,new_mdb.txt,new_dir.
-t-pathC:\\x\\x\*
-t-pathC:\\x\\x-filexxx.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()
ifargs.t:#判读t是否真
ifargs.file==None:
path,file=os.path.split(args.path)
dir__txt=dir_txt(path)
type_txt(dir__txt,path)
elifargs.file!=None:
list_file=args.file.split(',')
type_txt(list_file,args.path)
elifargs.p:#判读p是否真
ifargs.file==None:
path,file=os.path.split(args.path)
dir__txt=dir_txt(path)
password(dir__txt,path)
elifargs.file!=None:
list_file=args.file.split(',')
password(list_file,args.path)
else:
print("错误请检查输入。")
except(KeyboardInterrupt):
print('以退出')
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。