需求说明
1.在文件中取出所有远程主机IP地址
2. 在shell命令行中接受远程服务器IP地址文件、远程服务器密码以及在远程主机上执行的命令
3.通过多线程实现在所有的远程服务器上并发执行命令
安装paramiko
paramiko 遵循SSH2协议，支持以加密和认证的方式，进行远程服务器的连接，可以实现远程文件的上传，下载或通过ssh远程执行命令。
pip3 install paramiko
测试是否安装成功
import paramiko
源代码
#导入模块
import paramiko
import sys
import getpass
import threading
import os
#创建函数实现远程连接主机,服务器密码以及远程在主机上执行的命令的功能
def remote_comm(host,pwd,command):
#创建用于连接ssh服务器的实例
ssh=paramiko.SSHClient()
#设置自动添加主机秘钥
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
#连接ssh服务器,添加连接的主机,用户名,密码填好
ssh.connect(hostname=host,username='root',password=pwd)
#在ssh服务器上执行指定命令,返回三项类对象,分别是:输入/输出/错误
stdin,stdout,stderr=ssh.exec_command(command)
#读取输出
out =stdout.read()
#读取错误
error=stderr.read()
#如果有输出
if out:
#打印主机输出内容
print('[%s]OUT:\n%s'%(host,out.decode('utf8')))
#如果有错误
if error:
#打印主机错误信息
print('[%s]ERROR:\n%s'%(host,out.decode('utf8')))
#程序结束
ssh.close()
if __name__ == '__main__':
#设定sys.argv长度,确保remote_comm函数中;参数数量
if len(sys.argv)!=3:
print('Usage:%s ipaddr_file "command"'%sys.argv[0])
exit(1)
#判断命令行上输入如果不是文件,确保输入的是文件
if not os.path.isfile(sys.argv[1]):
print('NO such file:',sys.argv[1])
exit(2)
#fname为存储远程主机ip的文件,用sys.argv位置参数的方法,可以在执行脚本时再输入文件名,更为灵活
fname=sys.argv[1]
#command为在远程主机上执行的命令,用sys.argv方法,可以在执行脚本时再输入响应的命令,command为remote_comm函数的第三个参数
command=sys.argv[2]
#通过getpass输入远程服务器密码,pwd为remote_comm函数第二个参数
pwd=getpass.getpass()
#打开存有远程主机ip的文件
with open(fname)as fobj:
#将遍历文件将ip以列表形式存入ips,line.strip()可以去掉每行后的换行符\n
ips=[line.strip() for line in fobj]
#循环遍历文件,获取ip地址,ip为remote_comm函数第一个参数
for ip in ips:
#将读取到的IP地址作为remote_comm函数实际参数传递给函数,ips中有几个ip地址循环几次
#创建多线程
t=threading.Thread(target=remote_comm,args=(ip,pwd,command))
#启用多线程
t.start()
代码运行结果
先创建虚拟机,并将虚拟机的ip地址写入文件
这里以一台为例子,可以写多台
vim server_addr.txt
192.168.1.10
#查询虚拟机上root用户的信息
python3 sshcommit.py server_addr.txt "id root"
#输入链接虚拟机的root用户密码
Password:
[192.168.1.10]OUT:
#可以看到执行到的命令结果
uid=0(root) gid=0(root) groups=0(root)
#为虚拟机添加用户zjh
python3 sshcommit.py server_addr.txt "useradd zjh"
#输入虚拟机密码
Password:
#更改刚才创建的zjh用户的密码
python3 sshcommit.py server_addr.txt "echo zjh |passwd --stdin zjh"
Password:
[192.168.1.10]OUT:
#显示zjh用户的密码修改成功
Changing password for user zjh.
passwd: all authentication tokens updated successfully.
在虚拟机上检测是否有zjh用户
[root@zabbix ~]# id zjh
uid=1000(zjh) gid=1000(zjh) 组=1000(zjh)
#可以看到是存在的,证明脚本执行成功
错误检测
参数给少了效果如下:
python3 sshcommit.py server_addr.txt
Usage:sshcommit.py ipaddr_file "command"
可以显示出我们自定义的报错内容信息
参数给多了效果如下:
python3 sshcommit.py server_addr.txt id zjh
Usage:sshcommit.py ipaddr_file "command"
