|
1
2
3
4
5
6
7
|
1
.主机批量管理工具功能
2
.设计框架
3
.实现：数据库信息与程序源代码
4
.实战演示
5
.程序的不足
6
.在写程序过程中的经验教训
7
.往后的改进思路
|
|
1
2
3
4
5
6
7
8
9
|
mysql> show databases;
+--------------------+
| Database |
+--------------------+
| information_schema |
| manager_system |
| mysql |
+--------------------+
3
rows
in
set
(
0.03
sec)
|
|
1
2
3
4
5
6
7
8
9
10
|
mysql>
use
manager_system
mysql> show tables;
+--------------------------+
| Tables_in_manager_system |
+--------------------------+
| manager1_server |
| manager2_server |
| users |
+--------------------------+
3
rows
in
set
(
0.00
sec)
|
|
1
2
3
4
5
6
7
8
9
10
|
mysql> describe users;
+-----------+------------------+------+-----+---------+----------------+
| Field | Type |
Null
| Key | Default | Extra |
+-----------+------------------+------+-----+---------+----------------+
| id |
int
(
10
) unsigned | NO | PRI | NULL | auto_increment |
| username | char(
20
) | NO | | NULL | |
| password | char(
20
) | NO | | NULL | |
| real_name | char(
20
) | NO | | NULL | |
+-----------+------------------+------+-----+---------+----------------+
4
rows
in
set
(
0.01
sec)
|
|
1
2
3
4
5
6
7
8
|
mysql> select * from users;
+----+----------+----------+-----------+
| id | username | password | real_name |
+----+----------+----------+-----------+
|
1
| manager1 |
123456
| zhangsan |
|
2
| manager2 |
123456
| lisi |
+----+----------+----------+-----------+
2
rows
in
set
(
0.00
sec)
|
|
1
2
3
4
5
6
7
8
9
10
|
mysql>
use
manager_system
mysql> show tables;
+--------------------------+
| Tables_in_manager_system |
+--------------------------+
| manager1_server |
| manager2_server |
| users |
+--------------------------+
3
rows
in
set
(
0.00
sec)
|
|
1
2
3
4
5
6
7
8
9
10
11
12
|
mysql> describe manager1_server;
+-------------+------------------+------+-----+---------+----------------+
| Field | Type |
Null
| Key | Default | Extra |
+-------------+------------------+------+-----+---------+----------------+
| id |
int
(
10
) unsigned | NO | PRI | NULL | auto_increment |
| ip | char(
20
) | NO | | NULL | |
| username | char(
20
) | NO | | NULL | |
| password | char(
20
) | NO | | NULL | |
| port |
int
(
11
) | NO | | NULL | |
| server_type | char(
20
) | NO | | NULL | |
+-------------+------------------+------+-----+---------+----------------+
6
rows
in
set
(
0.00
sec)
|
|
1
2
3
4
5
6
7
8
|
mysql> select * from manager1_server;
+----+---------------+-----------+----------+-------+-------------+
| id | ip | username | password | port | server_type |
+----+---------------+-----------+----------+-------+-------------+
|
1
|
192.168
.
1.124
| oldboy |
123456
|
22
| DNS Server |
|
2
|
192.168
.
1.134
| yonghaoye |
123456
|
52113
| DHCP Server |
+----+---------------+-----------+----------+-------+-------------+
2
rows
in
set
(
0.00
sec)
|
|
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
|
import
MySQLdb,os,paramiko,sys,time
from multiprocessing
import
Process,Pool
#数据库连接类
class
Connect_mysql:
conn = MySQLdb.connect(host =
'localhost'
, user =
'root'
,passwd =
'123456'
, db =
'manager_system'
, port =
3306
)
cur = conn.cursor()
def __init__(self,username,password=
'NULL'
):
self.username = username
self.password = password
#contect to the login table
def login_check(self): #连接管理系统账号信息数据库并验证用户名密码信息
try
:
self.cur.execute(
"select * from users where username = '%s' and password = '%s'"
% (self.username,self.password))
qur_result = self.cur.fetchall() #
return
the tuple
if
qur_result == (): #database
do
not have
this
user
return
0
else
:
return
1
#database has
this
user
self.cur.close()
self.conn.close()
except MySQLdb.Error,e:
print
'\033[31;1mMysql Error Msg:%s\033[0m'
% e
#contect to the server table
def return_server(self): #连接用户主机列表数据库并返回表信息
self.cur.execute(
"select * from %s_server"
% self.username)
qur_result = self.cur.fetchall()
return
qur_result
def ssh_run(host_info,cmd,sysname): #批量远程命令执行程序
ip,username,password,port= host_info[
1
],host_info[
2
],host_info[
3
],host_info[
4
]
date = time.strftime(
'%Y_%m_%d'
)
date_detial = time.strftime(
'%Y_%m_%d %H:%M:%S'
)
f = file(
'./log/%s_%s_record.log'
% (sysname,date),
'a+'
) #操作日志记录，记录程序所有目录的/log目录里
try
:
s.connect(ip,
int
(port),username,password,timeout=
5
)
stdin,stdout,stderr = s.exec_command(cmd)
cmd_result = stdout.read(),stderr.read()
print
'\033[32;1m-------------%s--------------\033[0m'
% ip
for
line
in
cmd_result:
print line,
print
'\033[32;1m-----------------------------\033[0m'
except:
log =
"Time:%s | Type:%s | Detial:%s | Server:%s | Result:%s\n"
% (date_detial,
'cmd batch'
,cmd,ip,
'failed'
)
f.write(log)
f.close()
print
'\033[31;1mSomething is wrong of %s\033[0m'
% ip
else
:
log =
"Time:%s | Type:%s | Detial:%s | Server:%s | Result:%s\n"
% (date_detial,
'cmd batch'
,cmd,ip,
'success'
)
f.write(log)
f.close()
return
1
def distribute_file(host_info,file_name,sysname): #批量文件分发函数
ip,username,password,port = host_info[
1
],host_info[
2
],host_info[
3
],
int
(host_info[
4
])
date = time.strftime(
'%Y_%m_%d'
)
date_detial = time.strftime(
'%Y_%m_%d %H:%M:%S'
)
f = file(
'./log/%s_%s_record.log'
% (sysname,date),
'a+'
) #日志记录
try
:
t = paramiko.Transport((ip,port))
t.connect(username=username,password=password)
sftp = paramiko.SFTPClient.from_transport(t)
sftp.put(file_name,
'/tmp/%s'
% file_name)
t.close()
except:
log =
"Time:%s | Type:%s | Detial:%s | Server:%s | Result:%s\n"
% (date_detial,
'distribute file'
,file_name,ip,
'failed'
)
f.write(log)
f.close()
print
'\033[31;1mSomething is wrong of %s\033[0m'
% ip
else
:
log =
"Time:%s | Type:%s | Detial:%s | Server:%s | Result:%s\n"
% (date_detial,
'distribute file'
,file_name,ip,
'success'
)
f.write(log)
f.close()
print
"\033[32;1mDistribute '%s' to %s Successfully!\033[0m"
% (file_name,ip)
os.system(
'clear'
)
print
'\033[32;1mWelcome to the Manager System!\033[0m'
while
True: #程序主程序
username = raw_input(
'Username:'
).strip()
password = raw_input(
'Password:'
).strip()
if
len(username) <=
3
or len(password) <
6
:
print
'\033[31;1mInvalid username or password!\033[0m'
continue
#Begin to login
p = Connect_mysql(username,password)
mark = p.login_check()
if
mark ==
0
: #login failed
print
'\033[31;1mUsername or password wrong!Please try again!\033[0m'
elif mark ==
1
: #login success
print
'\033[32;1mLogin Success!\033[0m'
print
'The server list are as follow:'
#seek
for
the server list managed by the system user
p = Connect_mysql(username)
server_list = p.return_server()
for
server
in
server_list:
print
'%s:%s'
% (server[
5
],server[
1
])
while
True:
print
''
'What
do
you want to
do
? #程序主菜单
1
.Execute the command batch.
2
.Distribute file(s) batch.
3
.Exit.
''
'
choice = raw_input(
'\033[32;1mYour choice:\033[0m'
).strip()
if
'1'
<= choice <=
'4'
:pass
else
:
continue
#Execute the command batch.
if
choice ==
'1'
: #批量执行命令程序块
s = paramiko.SSHClient() #调用Paramiko模块
s.load_system_host_keys()
s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
p = Pool(processes=
3
) #设定进程池数据
result_list = []
while
True:
cmd = raw_input(
'\033[32;0mEnter the command(or quit to quit):\033[0m'
)
if
cmd ==
'quit'
:
break
for
h
in
server_list:
result_list.append(p.apply_async(ssh_run,[h,cmd,username])) #the usename
is
system name
#调用相关功能函数，并执行多进程并发
for
res
in
result_list:
res.
get
()
s.close()
#Distribute file(s) batch.
elif choice ==
'2'
: #批量分发文件程序块
s = paramiko.SSHClient() #调用Paramiko模块
s.load_system_host_keys()
s.set_missing_host_key_policy(paramiko.AutoAddPolicy())
p = Pool(processes=
3
)
result_list = [] #save the suanfa that come from the apply_async
while
True:
file_name = raw_input(
'The file you want to distribute(or quit to quit):'
).strip()
start = time.time()
if
file_name ==
'quit'
:
break
file_chcek = os.path.isfile(file_name)
log_list = []
if
file_chcek == False:
print
'\033[31;1mThe file does not exist or it is a directory!\033[0m'
continue
for
h
in
server_list:
result_list.append(p.apply_async(distribute_file,[h,file_name,username])) #the list save the suanfa
for
res
in
result_list:
res.
get
() #run the suanfa
end = time.time()
print
'\033[31;1mCost time:%ss\033[0m'
% str(end - start)
s.close()
#Exit the system
elif choice ==
'3'
: #退出系统
sys.exit(
'\033[32;1mWelcome to use our system!\033[0m'
)
|
