-
-
Notifications
You must be signed in to change notification settings - Fork 670
/
DocumentGeneration.py
123 lines (116 loc) · 7.51 KB
/
DocumentGeneration.py
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
# !/usr/bin/env python
# -*- coding: utf-8 -*-
import os
import re
Path="/Users/ascotbe/code/Kernelhub"
window_success_list=[]
window_failure_list=[]
def WindowsConfig():
#中文配置的开头
chinese_begin="""let config = {\n title: 'Windows',\n home: 'Info.md',\n repo: 'Ascotbe/Kernelhub/Windows',\n nav: [\n {\n title: '简介', path: '/'\n },\n {\n title: '目录导航',type: 'dropdown', items: [\n {\n title: "有利用脚本(测试成功)",path: '/TestSuccess'\n },"""
#所有配置的结尾
config_end="""\n ]\n },\n ],\n tocVisibleDepth: 10,\n plugins: []\n};"""
#下表是黑名单，垃圾文件
reject_list=['.DS_Store', '.git','.idea', '.gitignore','LICENSE', 'Patch', 'README.CN.md', 'README.md', 'TestFailure', 'docs', 'DocumentGeneration.py','StencilPlate']
#英文配置的开头
english_begin="""let config = {\n title: 'Windows',\n home: 'Info.md',\n repo: 'Ascotbe/Kernelhub/Windows',\n nav: [\n {\n title: 'Introduction', path: '/'\n },\n {\n title: 'DirectoryNavigation',type: 'dropdown', items: [\n {\n title: "exploit script (successful test)",path: '/TestSuccess'\n },"""
#处理成功的编号
for i in os.listdir(Path+"/Windows"):
if i not in reject_list:
window_success_list.append(i)
#循环成功的编号和开头配置进行拼接
for x in sorted(window_success_list):
chinese_begin+="""\n {\n path: '/CN/"""+x+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/"""+x+"""/README.md'\n },"""
english_begin+="""\n {\n path: '/EN/"""+x+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/"""+x+"""/README_EN.md'\n },"""
#插入中间内容，未测试成功内容
chinese_begin+="""\n {\n title: '有利用脚本(测试未成功)', path: '/TestFailure'\n },"""
english_begin+="""\n {\n title: 'exploit script (test failure)', path: '/TestFailure'\n },"""
for i in os.listdir(Path+"/Windows/TestFailure"):
if i not in reject_list:
window_failure_list.append(i)
#写入windows测试失败漏洞
for xx in sorted(window_failure_list):
chinese_begin+="""\n {\n path: '/CN/"""+xx+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/TestFailure/"""+xx+"""/README.md'\n },"""
english_begin+="""\n {\n path: '/EN/"""+xx+"""', source: 'https://raw.githubusercontent.com/Ascotbe/Kernelhub/master/Windows/TestFailure/"""+xx+"""/README_EN.md'\n },"""
chinese_config_file=open(Path+"/docs/Windows/Docs/config.js","w+")
chinese_config_file.write(chinese_begin+config_end)
chinese_config_file.close()
english_config_file=open(Path+"/docs/Windows/EnglishDocs/config.js","w+")
english_config_file.write(english_begin+config_end)
english_config_file.close()
def WindowsMarkdown():
"""
从Windows目录中提取README内容中的《编号列表》内容进行拆分，判断那些是测试过的那些是没测试过的
把他们分成两个文件存放，然后再区分下语言
"""
common_vulnerabilities_and_exposures_list=[]
chinese_success_list=[]
chinese_failure_list=[]
english_success_list=[]
english_failure_list=[]
#打开Windows目录下面的文件读取内容
file=open(Path+"/Windows/README.md","r").read()
#提取《编号列表》的内容
date = re.findall(r'(?<=> Numbered list)([\w\W]+)(?=### Required environment)', file)[0].split("\n")
#进行处理拼接
for i in date:
if i != "":
common_vulnerabilities_and_exposures_list.append(i)
for x in common_vulnerabilities_and_exposures_list[2:]:
date = re.findall(r'(?<=\[)([\w\W]+)(?=\])', x)[0]
if date in window_success_list:
chinese_success_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/Docs/#/CN/"+date, x))
english_success_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/EnglishDocs/#/EN/"+date, x))
if date in window_failure_list:
chinese_failure_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/Docs/#/CN/"+date, x))
english_failure_list.append(re.sub(r'(?<=\()([\w\W]+)(?=\))', "http://kernelhub.ascotbe.com/Windows/EnglishDocs/#/EN/"+date, x))
chinese_success_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n"
chinese_failure_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n"
english_success_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n"
english_failure_date=common_vulnerabilities_and_exposures_list[0]+"\n"+common_vulnerabilities_and_exposures_list[1]+"\n"
for x1 in chinese_success_list:
chinese_success_date+=x1+"\n"
for x2 in chinese_failure_list:
chinese_failure_date+=x2+"\n"
for x3 in english_success_list:
english_success_date+=x3+"\n"
for x4 in english_failure_list:
english_failure_date+=x4+"\n"
#写入文件中
chinese_success_file=open(Path+"/docs/Windows/Docs/TestSuccess.md","w+")
chinese_success_file.write(chinese_success_date)
chinese_success_file.close()
chinese_failure_file=open(Path+"/docs/Windows/Docs/TestFailure.md","w+")
chinese_failure_file.write(chinese_failure_date)
chinese_failure_file.close()
english_success_file=open(Path+"/docs/Windows/EnglishDocs/TestSuccess.md","w+")
english_success_file.write(english_success_date)
english_success_file.close()
english_failure_file=open(Path+"/docs/Windows/EnglishDocs/TestFailure.md","w+")
english_failure_file.write(english_failure_date)
english_failure_file.close()
"""
处理Windows目录下面readme文件，没有测试成功的编号进行更新
"""
read_readme_english_file_data = open(Path + "/Windows/README.md", "r").read()
read_readme_chinese_file_data = open(Path + "/Windows/README.CN.md", "r").read()
header="\n\n| SecurityBulletin | | | |\n| ---------------- |---------------- | ---------------- | ---------------- |\n"
common_vulnerabilities_and_exposures=""
count=0
for i in sorted(window_failure_list):
count+=1
common_vulnerabilities_and_exposures += "| " + str(i)
if count>4:
header+=common_vulnerabilities_and_exposures+"|\n"
common_vulnerabilities_and_exposures=""
count=0
if count!=0:
header+=str(common_vulnerabilities_and_exposures)+" |\n"
#对内容进行替换
tmp_english = re.sub(r'(?<=and welcome to submit PR)([\w\W]+)(?=### Disclaimer)', header+"\n\n",read_readme_english_file_data)
tmp_chinese = re.sub(r'(?<=欢迎提交PR)([\w\W]+)(?=### 免责声明)', header+"\n\n",read_readme_chinese_file_data)
open(Path + "/Windows/README.md", "w+").write(tmp_english)
open(Path + "/Windows/README.CN.md", "w+").write(tmp_chinese)
if __name__ == '__main__':
WindowsConfig()
WindowsMarkdown()