python 学习分享-实战篇类 Fabric 主机管理程序开发

# 类 Fabric 主机管理程序开发：
# 1. 运行程序列出主机组或者主机列表
# 2. 选择指定主机或主机组
# 3. 选择让主机或者主机组执行命令或者向其传输文件（上传/下载）
# 4. 充分使用多线程或多进程
# 5. 不同主机的用户名密码、端口可以不同
import paramiko,threading,pickle,os

Base_path = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
config_path = Base_path +'/core/config'
#主机配置信息管理
list_host = [{'host':'192.168.117.236','port':22,'user':'ubuntu1','password':'ubuntu1','group':1},
             {'host':'192.168.117.237','port':22,'user':'ubuntu2','password':'ubuntu2','group':1},
             {'host':'192.168.117.239','port':22,'user':'ubuntu3','password':'ubuntu3','group':2},
             {'host':'192.168.117.240','port':22,'user':'aaa','password':'bbb','group':2}]
def pickle_dump(path,file):
    #存入配置文件
    with open(path,'wb') as f:
        pickle.dump(file,f)
#pickle_dump(config_path,list_host)
def pickle_load(path):
    #取出配置文件
    with open(path,'rb') as f:
        list_host_new = pickle.load(f)
    return list_host_new
# list_host_new = pickle_load(config_path)
# print(list_host_new)

class Paramiko_sshd(object):
    #sshd类
    def __init__(self,host,port,user,password):
        self.host = host
        self.port = port
        self.user = user
        self.password = password

    def sshd_command(self,command):
        #ssh远程主机执行命令，并返回结果
        try:
            ssh_client = paramiko.SSHClient()
            ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
            ssh_client.connect(self.host,self.port,self.user,self.password)
            std_in,std_out,std_err = ssh_client.exec_command(command)
            print(self.host.center(30,'*'))
            print(command.upper().center(30,'='))
            for line in std_out:
                print(line.strip('\n'))
            ssh_client.close()
        except Exception as e:
            print(e)
    def sshd_upload_file(self,server_path,local_path):
        #上传文件
        try:
            t = paramiko.Transport((self.host,self.port))
            t.connect(username=self.user,password=self.password)
            sftp = paramiko.SFTPClient.from_transport(t)
            sftp.put(local_path,server_path)
            t.close()
            print(self.host.center(30, '*'))
            print('上传成功')
        except Exception as e:
            print(e)
    def sshd_down_file(self,server_path,local_path):
        #下载文件
        try:
            t = paramiko.Transport((self.host, self.port))
            t.connect(username=self.user, password=self.password)
            sftp = paramiko.SFTPClient.from_transport(t)
            sftp.get(server_path, local_path)
            t.close()
            print(self.host.center(30, '*'))
            print('下载成功')
        except Exception as e:
            print(e)
def help():
    #帮助信息
    command_option = input('''
    *************请选择***********
    1.执行命令
    2.上传文件
    3.下载文件
    4.返回
    5.退出程序
    ''').strip()
    return command_option

def review():
    while 1:
        #展示界面
        list_host_new = pickle_load(config_path)
        host_group_1 = []
        host_group_2 = []
        for line in list_host_new:
            if line['group'] == 1:
                host_group_1.append(line)
            else:
                host_group_2.append(line)
        print('group-1'.center(30,'='))
        for line_1 in host_group_1:
            print('host:',line_1['host'],'port:',line_1['port'])
        print('group-2'.center(30,'='))
        for line_2 in host_group_2:
            print('host:',line_2['host'],'port:',line_2['port'])

        group_select = input('请输入您要操作的主机组：\033[1;35m group-1 \033[0m | \033[1;35m group-2 \033[0m '
                             '或输入\033[1;35mexit\033[0m退出程序 ==>').strip()
        if group_select in ['group-1','group-2']:
            if group_select == 'group-1':     #增加中间变量，来确定是对组一操作还是对组二进行操作
                group_sure = host_group_1
            else:
                group_sure = host_group_2
            while 1:
                command_option = help()
                if command_option == '':
                    command = input('请输入执行的命令：').strip()
                    for line_3 in group_sure:
                        sshd = Paramiko_sshd(line_3['host'],line_3['port'],line_3['user'],line_3['password'])
                        t = threading.Thread(target=sshd.sshd_command,args=(command,))
                        t.start()
                        t.join()

                elif command_option == '':
                    server_path = input('请输入上传到的服务器目录：').strip()
                    local_path = input('请输入本地文件目录：').strip()
                    for line_3 in group_sure:
                        sshd = Paramiko_sshd(line_3['host'],line_3['port'],line_3['user'],line_3['password'])
                        t = threading.Thread(target=sshd.sshd_upload_file,args=(server_path,local_path,))
                        t.start()
                        t.join()

                elif command_option == '':
                    server_path = input('请输入下载的服务器目录文件路径：').strip()
                    local_path = input('请输入存放到本地的路径：').strip()
                    for line_3 in group_sure:
                        sshd = Paramiko_sshd(line_3['host'],line_3['port'],line_3['user'],line_3['password'])
                        t = threading.Thread(target=sshd.sshd_down_file,args=(server_path,local_path,))
                        t.start()
                        t.join()
                elif command_option == '':
                    break
                elif command_option == '':
                    exit()
                else:
                    print('%s为非法命令，请重新输入！'%command_option)
        elif group_select == 'exit':
            break
        else:
            print('%s为非法命令，请重新输入！'%group_select)

if __name__ =='__main__':
    review()