python实现主机批量管理

轩辕阁

　　在日常的运维工作中批量对主机的是很常见的，市面上也有许多主机批量管理的软件，但有时候这些软件并不能完全的满足我们的需求。python中刚好提供了关于主机批量管理的模块，今天就让我们来看看如何利用python实现主机批量管理
　　python提供主机批量管理的模块主要有三个paramiko、fabric与pexpect，今天我们主要说的是paramiko模块，paramiko模块是第三方模块
　　安装：pip install paramiko或者yum install python-paramiko如果都安装失败的话可以使用源码安装
　　源码包下载地址：https://github.com/paramiko/paramiko/archive/master.zip
　　常用操作：
　　ssh = paramiko.SSHClient()//实例化，ssh会话类
　　ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())   //第一次连接自动回答为yes
　　ssh.connect('服务器ip',端口,'用户名','密码')    //连接服务器
　　stdin,stdout,stderr = ssh.exec_command("命令")      //执行命令，标准输出，命令成功返回，命令失败返回
　　ssh.close()                        //关闭连接
　　上传、下载：
　　get_put = paramiko.Transport((ip,port))//服务器ip和端口(使用SFTP时使用)
　　get_put.connect(username="root", password="123.com")    //连接服务器，用户名和密码
　　sftp = paramiko.SFTPClient.from_transport(get_put)     //使用SFTP协议
　　sftp.put(需要上传的文件,上传文件保存的位置)              //上传
　　sftp.get(下载文件的位置, 下载保存的位置) //下载
　　主机批量管理：
　　#!/usr/bin/python
　　#-*- coding: utf-8 -*-
　　import paramiko
　　import os
　　def Connect(ip='127.0.0.1',Port=22,user='root',pwd='123.com'):
　　ssh = paramiko.SSHClient()
　　ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
　　ssh.connect(hostname=ip, port=Port, username=user, password=pwd)    #连接服务器
　　return ssh                                      #返回服务器句柄
　　def IP_ADD(ip):                      #判断用户输入的是一个地址还是地址范围
　　__ip = ip.split('--')
　　if len(__ip) == 2:                          #如果是地址范围
　　ip_add = __ip[0]                        #取出ip
　　ip_range = int(__ip[1])                 #取出ip的最大值
　　host_ip_split = ip_add.split('.')       #将ip以.分隔为4段
　　host_ip = int(host_ip_split.pop())      #取出ip的最小值
　　ip_3 = '.'.join(host_ip_split)+'.'   #将已经去除了ip的最小值的剩余部分重新拼接
　　host_add_range =  range(host_ip,ip_range+1)     #生成连接ip的范围
　　add_pool = []
　　for i in host_add_range:
　　add = ip_3+str(i)                    #拼接为合法的ip地址
　　add_pool.append(add)                 #生成地址池
　　return add_pool                          #返回地址池
　　elif len(__ip) == 1:                         #如果输入的为单个地址
　　valid_ip = __ip[0]
　　ip_value = valid_ip.split('.')[0:4]      #检测ip的合法性
　　[ int(i) for i in ip_value ]
　　return __ip                                 #返回地址
　　if __name__ == '__main__':
　　os.system('clear')
　　Fa = True
　　while Fa:
　　ip = raw_input('请输入一个ip地址或一个地址范围: ')
　　exclude_ip = raw_input('请输入要排除的地址: ')
　　try:
　　ip_list = IP_ADD(ip)           #将用户输入的ip交给函数处理
　　if exclude_ip.strip():          #判断用户是否要排除地址池中的地址
　　exclude_add = exclude_ip.split(',')   #定义排除格式
　　[ ip_list.remove(i) for i in exclude_add ]  #排除ip地址
　　Fa = False                               #退出循环
　　except:
　　print '\033[31m地址格式错误\033[0m'
　　print '''例：
　　127.0.0.1            对单台主机执行操作
　　127.0.0.1--254         对一个地址范围的主机执行操作
　　127.0.0.8,127.0.0.10     排除地址池中的地址
　　'''
　　continue
　　print '\033[31m输入script_exit退出程序\033[0m'
　　while True:
　　shell_comd = raw_input('请输入执行的命令: ')        #执行的命令
　　if shell_comd == 'script_exit':
　　break
　　for i in ip_list:                           #循环地址池
　　try:
　　comd = Connect(ip=i)                #连接服务器
　　stdin, stdout, stderr = comd.exec_command(shell_comd)   #执行命令
　　stderr = stderr.read()      #命令正确执行的结果
　　stdout = stdout.read()      #命令错误执行的命令
　　if stdout:
　　print '\033[31m++++++++++++%s+++++++++++\033[0m'%i
　　#那台服务器执行的命令
　　print stdout
　　else:
　　print '\033[31m++++++++++++%s+++++++++++\033[0m'%i
　　print stderr
　　except:
　　print '\033[31m%s连接失败\033[0m' %i
　　paramiko还可以实现文件批量下载和批量上传，原理和批量执行命令基本一样。