"""批量主机管理接口"""

from 作业 import core

if __name__ == "__main__":

    core.run()

"""核心代码"""

from 作业 import settings

import paramiko

import threading,os

class REMOTE_HOST(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):

        """起线程连接远程主机后调用"""

        cmd_str = self.cmd.split()[0]

        if hasattr(self, cmd_str):  # 反射 eg:调用put方法

            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)

        print("%s".center(50, "-") % self.host)

        res, err = stdout.read(), stderr.read()

        result = res if res else err

        print(result.decode())

        ssh.close()

    def put(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)

        print("put sucesss")

        transport.close()

    def get(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.get(filename, filename)

        print("get sucesss")

        transport.close()

def show_host_list():

    """通过选择分组显示主机名与IP"""

    for index, key in enumerate(settings.msg_dic):

        print("%s\033[34m 主机组:%s \033[0m \033[33m 主机数量:%s\033[0m" % (index + 1, key, len(settings.msg_dic[key])))

    while True:

        choose_host_list = input(">>(请输入用户组编号如:group1): ").strip()

        host_dic = settings.msg_dic.get(choose_host_list)

        if host_dic:

            # print(host_dic)

            for key in host_dic:

                print(key, host_dic[key]["IP"])

            return host_dic

        else:

            print("NO exit this group!")

def interactive(choose_host_list):

    """根据选择的分组主机起多个线程进行批量交互"""

    thread_list = []

    while True:

        cmd = input(">>>").strip()

        if cmd:

            for key in choose_host_list:

                host, port, username, password = choose_host_list[key]["IP"], choose_host_list[key]["port"], \

                                                 choose_host_list[key]["username"], choose_host_list[key]["password"]

                func = REMOTE_HOST(host, port, username, password, cmd)  # 实例化类

                t = threading.Thread(target=func.run)  # 起线程

                t.start()

                thread_list.append(t)

            for t in thread_list:

                t.join()  # 主线程等待子线程执行完毕

        else:

            continue

def run():

    choose_host_list = show_host_list()

    interactive(choose_host_list)

"""配置文件"""

msg_dic = {

"group1":{    #分组1

"h1":{"IP":"192.168.50.134", "username":"root", "password":"521521", "port":22}

}}

#可以分多组我就一台测试机器

C:\Users\Administrator\PycharmProjects\s14\venv\Scripts\python.exe C:/Users/Administrator/PycharmProjects/s14/作业/main.py
1 主机组:group1   主机数量:1
>>(请输入用户组编号如:group1): group1
h1 192.168.50.134

>>>put from_windows.py

put sucesss

>>>

>>>ls

------------------------192.168.50.134------------------------

anaconda-ks.cfg

database_test

from_windows.py

install.log

install.log.syslog

m

oot

\root

tmp\from_windows.py

>>>