使用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还可以实现文件批量下载和批量上传,原理和批量执行命令基本一样。

以上就是使用python实现主机批量管理的详细内容,更多请关注 第一php社区 其它相关文章!

Posted in 未分类