-
Notifications
You must be signed in to change notification settings - Fork 2
/
check.py
58 lines (48 loc) · 1.79 KB
/
check.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
import paramiko, json, os
import multiprocessing
#读取服务器列表
def get_server_list():
with open('server.json', 'r', encoding='utf_8') as file:
server = json.load(file)
return server["server"]
#检查 Shadowsocks 服务状态
def check_shadowsocks_status(server,ssh):
print(server, ": shadowsocks status")
stdin, stdout, stderr = ssh.exec_command("/etc/init.d/shadowsocks status")
for str in stdout.readlines():
print(str)
if "ShadowsocksR is stopped" in str:
restart_shadowsocks(server, ssh)
#重启 Shadowsocks 服务
def restart_shadowsocks(server, ssh):
stdin, stdout, stderr = ssh.exec_command("/etc/init.d/shadowsocks restart")
print(server, ": shadowsocks restart complete")
#重启锐速
def restart_server_speeder(server, ssh):
stdin, stdout, stderr = ssh.exec_command("/serverspeeder/bin/serverSpeeder.sh restart")
for str in stdout.readlines():
print(str)
print(server, ": serverspeeder restart complete")
#建立服务器 SSHClient
def connect_server(server):
print("current pid :",os.getpid())
hostname = server["hostname"]
port = server["port"]
username = server["username"]
password = server["password"]
name = server["name"]
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname=hostname, port=port, username=username, password=password)
check_shadowsocks_status(name, ssh)
ssh.close()
if __name__=='__main__':
servers = get_server_list()
pool = multiprocessing.Pool()
cpus = multiprocessing.cpu_count()
for server in servers:
pool.apply_async(connect_server(server))
print('Waiting for all subprocesses done...')
pool.close()
pool.join()
print('All subprocesses done.')