-
Notifications
You must be signed in to change notification settings - Fork 65
/
Copy pathinstall-brook.py
197 lines (182 loc) · 6.83 KB
/
install-brook.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
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
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
#coding=utf-8
#。——————————————————————————————————————————
#。
#。 install-brook.py
#。
#。 @Time : 2018/10/30 下午5:54
#。 @Author : capton
#。 @Software: PyCharm
#。 @Blog : http://ccapton.cn
#。 @Github : https://github.com/ccapton
#。 @Email : chenweibin1125@foxmail.com
#。__________________________________________
from __future__ import print_function
from __future__ import division
import platform
import ssl
import sys
import os
SERVICE_TYPE_BROOK = 0
SERVICE_TYPE_SS = 1
SERVICE_TYPE_SOCKS5 = 2
# 判断Python执行版本
python_version = sys.version
if python_version.startswith('2.'):
python_version = '2'
elif python_version.startswith('3.'):
python_version = '3'
# 获取html网页源代码字符串
def get_html_source(url):
html_source = ''
try:
try:
if python_version == '3':
import urllib.request as req
context = ssl._create_unverified_context()
f = req.urlopen(url,context=context)
else:
import urllib as req
f = req.urlopen(url)
html_source = f.read()
f.close()
except KeyboardInterrupt:
print('\n取消请求')
except Exception:
print('请求错误，请重试')
return html_source
# 获取brook其github主页的所有release下载链接
def match_brook_release_list():
url = 'https://github.com/txthinking/brook'
if python_version == '2':
source = get_html_source(url)
else:
source = get_html_source(url).decode(encoding='utf-8')
import re
# 正则匹配所有可用下载链接
result = re.findall('https://github.com/txthinking/brook/releases/download/.+"', source)
link_list = []
for raw_link in result:
link = raw_link[:len(raw_link) - 1]
link_list.append(link)
return link_list
# 根据下载链接列表，返回一个包含不同操作系统(架构)列表的对象
def brook_release_json(releaseLinkList):
brook_release_list = []
import os
for link in releaseLinkList:
released = {}
released['url']=link
released['name']=os.path.basename(link)
brook_release_list.append(released)
return brook_release_list
# def has_service_start(service_type=SERVICE_TYPE_BROOK):
# result = os.popen('ps aux | grep brook').read()
# try:
# global brook_pid,ss_pid,socks5_pid
# if service_type == SERVICE_TYPE_BROOK:
# brook_pid = match_pid(result, service_type)
# elif service_type == SERVICE_TYPE_SS:
# ss_pid = match_pid(result, service_type)
# elif service_type == SERVICE_TYPE_SOCKS5:
# socks5_pid = match_pid(result, service_type)
# except Exception:
# if service_type == SERVICE_TYPE_BROOK:brook_pid = ''
# elif service_type == SERVICE_TYPE_SS:ss_pid = ''
# elif service_type == SERVICE_TYPE_SOCKS5:socks5_pid = ''
# started = False
# if service_type == SERVICE_TYPE_BROOK:
# if str(result).find(' servers -l') != -1:
# started = True
# elif service_type == SERVICE_TYPE_SS:
# if str(result).find(' ssservers -l') != -1:
# started = True
# elif service_type == SERVICE_TYPE_SOCKS5:
# if str(result).find(' socks5 -l') != -1:
# started = True
# return started
#
# def stop_service(service_type=SERVICE_TYPE_BROOK):
# has_service_start(service_type)
# try:
# global brook_pid,ss_pid
# if service_type == SERVICE_TYPE_BROOK:
# if brook_pid != '':
# os.system('kill ' + brook_pid)
# elif service_type == SERVICE_TYPE_SS:
# if ss_pid != '':
# os.system('kill ' + ss_pid)
# elif service_type == SERVICE_TYPE_SOCKS5:
# if socks5_pid != '':
# os.system('kill ' + socks5_pid)
# finally:
# pass
#
# def has_brook_start():
# return has_service_start(SERVICE_TYPE_BROOK)
#
#
# def stop_brook():
# has_brook_start()
# stop_service(SERVICE_TYPE_BROOK)
# 下载Brook程序，先保存一个缓存文件再删除原brook文件、将刚下载的缓存文件改名为brook、授予执行权限
def download_brook(url,is_exe=False):
print(' 开始下载brook ' + url)
brook_name = 'brook'
command = 'curl -o '+os.path.join(sys.path[0], 'brook_temp')+' -L ' + url
code = os.system(command)
if code != 0:
print('')
print(' 下载brook错误，请重新运行本程序')
os.system('rm -rf '+os.path.join(sys.path[0], 'brook_temp'))
return
if not is_exe:
command2 = 'rm -rf '+os.path.join(sys.path[0], brook_name)+' && mv '+os.path.join(sys.path[0], 'brook_temp')+\
' '+os.path.join(sys.path[0], brook_name)
os.system(command2)
command3 = 'chmod +x '+os.path.join(sys.path[0], brook_name)
os.system(command3)
else:
brook_name = 'brook.exe'
os.remove(os.path.join(sys.path[0], brook_name))
os.rename(os.path.join(sys.path[0], 'brook_temp'), os.path.join(sys.path[0], brook_name))
print(" brook下载完毕!保存在："+os.path.join(sys.path[0],brook_name))
# 匹配MacOS系统下载链接
def is_mac():
brook_list = brook_release_json(match_brook_release_list())
for brook in brook_list:
if 'darwin'in brook['name'] :
download_brook(brook['url'])
break
# 匹配不同Linux系统的下载链接
def is_linux(arch):
brook_list = brook_release_json(match_brook_release_list())
for brook in brook_list:
if str(brook['name']).endswith('brook') and arch == 'x86_64':
download_brook(brook['url'])
break
elif 'linux' in brook['name'] and arch == 'x86' and '386' in brook['name']:
download_brook(brook['url'])
break
elif 'linux' in brook['name'] and arch in brook['name']:
download_brook(brook['url'])
break
else:
download_brook(brook_list[0]['url'])
break
# 判断操作系统，并执行下载任务
def guest_platform():
sys_name = platform.system()
#machine_name = platform.machine().lower()
if 'Darwin' == sys_name:
is_mac()
elif 'Linux' == sys_name:
arch = os.popen('uname -m').read()
arch = arch[:len(arch) - 1]
is_linux(arch)
elif 'Windows' == sys_name:
print('暂不支持Windows平台,请期待作者完成')
#isWindows(is_upgrade,machine_name)
else:
print('暂不支持此平台')
if __name__ == "__main__":
guest_platform()