python--批量主机管理

1、要求

题目:简单主机批量管理工具
需求:
 1.主机分组
 2.主机信息配置文件用configparser解析
 3.可批量执行命令、发送文件，结果实时返回，执行格式如下 1.batch_run  -h h1,h2,h3   -g web_clusters,db_servers    -cmd  "df -h"　 2.batch_scp   -h h1,h2,h3   -g web_clusters,db_servers  -action put  -local test.py  -remote /tmp/　
4.主机用户名密码、端口可以不同 
5.执行远程命令使用paramiko模块 
6.批量命令需使用multiprocessing并发

2、实现
（1）文档结构

hostdetail是存放主机信息的
main是主程序
两个test1 2文件是用来测试上传下载文件的
（2）主要思路
主机信息要从文件里读出来，放在一个字典里，字典嵌套，以组分开
批量处理要用threading类，批量处理，先把需要的信息放在__init__中进行初始化，批量处理的逻辑放在run中，主程序中实例化线程，再start即可

import paramiko
import threading
import os

class batchthread(threading.Thread):
    def __init__(self,hostip,hostport,user,passwd,cmd):
        super(batchthread,self).__init__()
        self.ip=hostip
        self.port=hostport
        self.user=user
        self.passwd=passwd
        self.cmd=cmd
        return

    def run(self):
        #发送命令执行的代码
        ssh=paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        ssh.connect(hostname=self.ip,port=self.port,username=self.user,password=self.passwd)

        stdin,stdout,stderr=ssh.exec_command(self.cmd)

        result=stdout.read()#读出来的是byte，要decode才能看懂结果
        print(result.decode('utf-8'))

        ssh.close()
        return

class filebatch(threading.Thread):
    def __init__(self,hostip,hostport,user,passwd,file,targetfile):
        super(batchthread,self).__init__()
        self.ip=hostip
        self.port=hostport
        self.user=user
        self.passwd=passwd
        self.putfile=file
        self.targetfile=targetfile
        return

    def run(self):
        #建立对象，用户名密码连接，建立ftp对象，put上传，get下载
        transport=paramiko.Transport( (self.ip,self.port) )
        transport.connect(username=self.user,password=self.passwd)

        sftp=paramiko.SFTPClient.from_transport(transport)
        sftp.put(self.putfile,self.targetfile)
        transport.close()
        return

def chooseact():
    act=input("输入您要做的操作选项：\n1、批量命令\n2、上传文件")
    return

#用户输入名字和口令登陆本应用程序
appname=input("请输入用户名：")
kouling=input("请输入密码：")
if appname=='admin' and kouling=='admin':
    print("登陆成功")
    # 首先获取主机信息，信息存在文件里，要读出来，
    # 一个主机的信息，放在一个list中，按顺序 主机名、主机ip、port、用户名、密码
    # 根据它的属组，放在字典里，
    #字典设计：
    '''
    hostdic=
    {
        '组名1':[hostinfo1,hostinfo2.....]
        '组名2':[hostinfo3,hostinfo4.....]
    }
    hostinfo={
    主机名：主机ip，
    port：主机port，
    用户名：密码}
    '''
    rootdir = os.path.dirname(os.path.dirname(__file__))
    hostfiledir = os.path.join(rootdir, 'hostlist', 'hostdetail')

    hostdic={}
    with open(hostfiledir,'r') as fd:
        line=fd.readline()
        while line:
            list=line.split()
            hostinfo={
                'hostname':list[1],
                'hostip':list[2],
                'port':list[3],
                'user':list[4],
                'password':list[5]
            }
            if list[0] in hostdic:
                hostdic[list[0]].append(hostinfo)
            else:
                hostdic[list[0]]=[]
                hostdic[list[0]].append(hostinfo)
            line=fd.readline()
    # 到这里，文件里的主机信息已经存在dic里面了，后面的过程就是要读取相应的信息了

    #第一步显示主机信息
    for key in hostdic:
        print(key)
        i=0
        while i <len(hostdic[key]):
            print('\t'+hostdic[key][i]['hostname']+'\t'+hostdic[key][i]['hostip'])
            i=i+1

    #第二步 让用户选择要操作的组
    while True:
        groupchoose=input("请输入您要操作的主机组，输入组名（q退出）：")
        if groupchoose in hostdic:
            #操作对象hostdic[groupchoose]
            lenth=len(hostdic[groupchoose])#要操作的组里有多少台主机
            # 第三步 让用户选择是要批量处理命令，还是上传文件



            while True:
                actchoose=input("请输入您要做的操作（1、批量命令处理 2、上传文件 3、退出）：")
                if actchoose=='1':
                    cmd=input("请输入命令：")
                    #建立多线程，批量各自处理命令
                    j = 0
                    if j < lenth:
                        th = batchthread(hostdic[groupchoose][j]['hostip'], hostdic[groupchoose][j]['port'],
                                         hostdic[groupchoose][j]['user'], hostdic[groupchoose][j]['password'],cmd)
                        th.start()
                        j += 1
                elif actchoose=='2':
                    putfile=input("请输入要上传的文件路径：")
                    putfiletarget=input("请输入要上传到哪里：")
                    j=0
                    if j<lenth:
                        th=filebatch(hostdic[groupchoose][j]['hostip'], hostdic[groupchoose][j]['port'],
                                         hostdic[groupchoose][j]['user'], hostdic[groupchoose][j]['password'],putfile,putfiletarget)
                        th.start()
                        j+=1
                elif actchoose=='3':
                    exit()
                else:
                    print("输入无效，重新输入。")
                    continue
        elif groupchoose=='q':
            exit()
        else:
            print("输入无效，重新输入。")
            continue
else:
    print("登陆失败")
    exit()