您的位置：首页 > 脚本大全 > > 正文

python批量创建字典（Python编写合并字典并实现敏感目录的小脚本）

更多时间：2022-01-20 00:32:58 类别：脚本大全浏览量：599

python批量创建字典

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

0x00 起因

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

0x01 代码

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

代码我就贴出来了

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

95

96

97

98

99

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149 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 合并字典敏感目录

上一篇：python学生管理系统的思路（python实现学员管理系统）

下一篇：ubuntu下mysql安装教程（Ubuntu 20.04 安装和配置MySql5.7的详细教程）

您可能感兴趣

python批量创建字典（Python编写合并字典并实现敏感目录的小脚本）

python批量创建字典

热门推荐

排行榜