网络编程与并发-批量主机管理工具开发

批量主机管理工具开发

要求

1. 实现批量命令执行、文件分发

core.py

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
__author__ = "Elijah"
__date__ = "2017/10/8 13:56"

import sys

sys.path.append('../conf')
import settings
import paramiko
from multiprocessing import Process

class operation_client(object):
# 远程操作主机
def __init__(self, host, port, username, password, cmd):
self.host = host
self.port = port
self.username = username
self.password = password
self.cmd = cmd

def run(self):
'''
起线程连接远程主机后调用
:return:
'''
cmd_str = self.cmd.split()[0]
if hasattr(self, cmd_str):  # 反射 eg:调用upload方法
getattr(self, cmd_str)()
else:
# setattr(x,'y',v)is  equivalent  to   ``x.y=v''
setattr(self, cmd_str, self.command)
getattr(self, cmd_str)()  # 调用command方法，执行批量命令处理

def command(self):
"""批量命令处理"""
ssh = paramiko.SSHClient()  # 创建ssh对象
# 允许连接不在know_hosts文件中的主机
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect(hostname=self.host, port=self.port, username=self.username, password=self.password)
stdin, stdout, stderr = ssh.exec_command(self.cmd)
result = stdout.read()
print("%s".center(30, "-") % self.host)
print(result.decode())
ssh.close()

def upload(self):
"""上传文件"""
filename = self.cmd.split()[1]  # 要上传的文件
transport = paramiko.Transport((self.host, self.port))
transport.connect(username=self.username, password=self.password)
sftp = paramiko.SFTPClient.from_transport(transport)
sftp.put(filename, filename.split('\\')[-1])
print(
d878
'上传文件：' + str(filename) + ' 成功!')
transport.close()

def show_host_list():
"""通过选择分组显示主机名与IP"""
num_list = []
for index, key in enumerate(settings.msg_dic):
num_list.append(key)
print('主机组编号：' + str(index + 1), '\n\t\t主机组名：' + str(key), '\n\t\t主机数量：' + str(len(settings.msg_dic[key])))
while True:
choose_host_list = input("请输入批量操作的主机组编号(或者输入q退出)：>>>").strip()
if choose_host_list.lower() == 'q':
break
host_dic = settings.msg_dic.get(num_list[int(choose_host_list) - 1])
if host_dic:
for key in host_dic:
print(key, host_dic[key]["IP"])
return host_dic
else:
print("对不起！您输入的主机组编号有误！\n")
continue

def interactive(host_dic):
'''
根据选择的分组主机多线程批量操作
:param host_dic:
:return:
'''
str_info = '''
批量主机管理工具功能说明：
1、批量文件上传：upload file_path
例如：upload C:\\Users\\YG\\Desktop\\file.txt

2、批量操作主机：df -h 、ls 、pwd ......
'''
print(str_info)
thread_list = []
while True:
command = input("请输入操作命令(或者输入q退出)：>>>").strip()
if command.lower() == 'q':
print('感谢使用批量主机管理工具！再见！')
break
elif command:
for key in host_dic:
host, port, username, password = host_dic[key]["IP"], host_dic[key]["port"], host_dic[key]["username"], \
host_dic[key]["password"]
func = operation_client(host, port, username, password, command)  # 实例化类
t = Process(target=func.run)  # 起线程
t.start()
thread_list.append(t)
for t in thread_list:
t.join()  # 主线程等待子线程执行完毕
else:
continue

def run():
host_dic = show_host_list()
interactive(host_dic)

settings.py

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
__author__ = "Elijah"
__date__ = "2017/10/8 13:57"

msg_dic = {
"group1":
{
"client_1": {"IP": "192.168.1.2", "username": "root", "password": "root", "port": 22},
},
"group2":
{
"client_1": {"IP": "192.168.1.3", "username": "root", "password": "root", "port": 22},
"client_2": {"IP": "192.168.1.4", "username": "root", "password": "root", "port": 22},
},
}

main.py

#!/usr/bin/env python
#_*_ coding:utf-8 _*_
__author__ = "Elijah"
__date__ = "2017/10/8 13:57"

import sys
sys.path.append('../src')
import core

if __name__ == "__main__":
core.run()