自动化运维 Python Paramiko库的使用

Python自动化运维 Paramiko库的使用

简介

Paramiko是用Python语言写的一个库，遵循SSH2协议，支持以加密和认证的方式，进行远程服务器的连接。paramiko支持Linux, Solaris, BSD, MacOS X, Windows等平台通过SSH从一个平台连接到另外一个平台。利用该模块，可以方便的进行ssh连接和sftp协议进行sftp文件传输

paramiko包含两个核心组件：SSHClient和SFTPClient。

• SSHClient的作用类似于Linux的ssh命令，是对SSH会话的封装，该类封装了传输(Transport)，通道(Channel)及SFTPClient建立的方法(open_sftp)，通常用于执行远程命令。
• SFTPClient的作用类似与Linux的sftp命令，是对SFTP客户端的封装，用以实现远程文件操作，如文件上传、下载、修改文件权限等操作。

这两个组件实现了常见ssh客户端的一些功能, 通过Paramiko编写脚本任务, 就可以实现自动化运维

使用

在命令行下输入下面命令即可安装

pip install paramiko 

密码远程连接

单个连接

import paramiko 
##1.创建一个ssh对象 
client = paramiko.SSHClient() 
#2.解决问题:如果之前没有，连接过的ip，会出现选择yes或者no的操作， 
##自动选择yes 
client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
#3.连接服务器 
client.connect(hostname='172.25.254.31', 
 port=22, 
 username='root', 
 password='westos') 
#4.执行操作 
stdin,stdout, stderr = client.exec_command('hostname') 
#5.获取命令执行的结果 
result=stdout.read().decode('utf-8') 
print(result) 
#6.关闭连接 
client.close() 

批量远程密码连接

from paramiko.ssh_exception import NoValidConnectionsError 
from paramiko.ssh_exception import AuthenticationException 
def connect(cmd,hostname,port=22,username='root',passwd='westos'): 
 import paramiko 
 ##1.创建一个ssh对象 
 client = paramiko.SSHClient() 
 #2.解决问题:如果之前没有，连接过的ip，会出现选择yes或者no的操作， 
 ##自动选择yes 
 client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
 #3.连接服务器 
 try: 
 client.connect(hostnamehostname=hostname, 
 portport=port, 
 usernameusername=username, 
 password=passwd) 
 print('正在连接主机%s......'%(hostname)) 
 except NoValidConnectionsError as e: ###用户不存在时的报错 
 print("连接失败") 
 except AuthenticationException as t: ##密码错误的报错 
 print('密码错误') 
 else: 
 #4.执行操作 
 stdin,stdout, stderr = client.exec_command(cmd) 
 #5.获取命令执行的结果 
 result=stdout.read().decode('utf-8') 
 print(result) 
 #6.关闭连接 
 finally: 
 client.close() 
with open('ip.txt') as f: #ip.txt为本地局域网内的一些用户信息 
 for line in f: 
 lineline = line.strip() ##去掉换行符 
 hostname,port,username,passwd= line.split(':') 
 print(hostname.center(50,'*')) 
 connect('uname', hostname, port,username,passwd) 

基于公钥密钥连接

import paramiko 
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException 
def connect(cmd, hostname, port=22, user='root'): 
 client = paramiko.SSHClient()  
 private_key = paramiko.RSAKey.from_private_key_file('id_rsa') 
 ###id_rsa为本地局域网密钥文件 
 client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
 try: 
 client.connect(hostnamehostname=hostname, 
 portport=port, 
 userusername=user, 
 pkey=private_key 
 ) 
 stdin, stdout, stderr = client.exec_command(cmd) 
 except NoValidConnectionsError as e: 
 print("连接失败") 
 except AuthenticationException as e: 
 print("密码错误") 
 else: 
 result = stdout.read().decode('utf-8') 
 print(result) 
 finally: 
 client.close() 
for count in range(254): 
 host = '172.25.254.%s' %(count+1) 
 print(host.center(50, '*')) 
 connect('uname', host) 

sftp文件操作

上传文件

import paramiko 
#获取Transport实例 
tran = paramiko.Transport("172.25.254.31",22) 
#连接SSH服务端 
tran.connect(username = "root", password = "westos") 
#获取SFTP实例 
sftp = paramiko.SFTPClient.from_transport(tran) 
#设置上传的本地/远程文件路径 
localpath="passwd.html" ##本地文件路径 
remotepath="/home/kiosk/Desktop/fish" ##上传对象保存的文件路径 
#执行上传动作 
sftp.put(localpath,remotepath) 
tran.close() 

下载文件

import paramiko 
#获取SSHClient实例 
client = paramiko.SSHClient() 
client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
#连接SSH服务端 
client.connect("172.25.254.31",username="root",password="westos") 
#获取Transport实例 
tran = client.get_transport() 
#获取SFTP实例 
sftp = paramiko.SFTPClient.from_transport(tran) 
remotepath='/home/kiosk/Desktop/fish' 
localpath='/home/kiosk/Desktop/fish' 
sftp.get(remotepath, localpath) 
client.close() 

密钥的上传和下载

import paramiko 
private_key = paramiko.RSAKey.from_private_key_file('id_rsa') 
tran = paramiko.Transport('172.25.254.31',22) 
tran.connect(username='root',password='westos') 
#获取SFTP实例 
sftp = paramiko.SFTPClient.from_transport(tran) 
remotepath='/home/kiosk/Desktop/fish8' 
localpath='/home/kiosk/Desktop/fish1' 
sftp.put(localpath,remotepath) 
sftp.get(remotepath, localpath) 

paramiko的再封装

import os 
import paramiko 
from paramiko.ssh_exception import NoValidConnectionsError, AuthenticationException, SSHException 
class SshRemoteHost(object): 
 def __init__(self, hostname, port, user, passwd, cmd): 
 self.hostname = hostname 
 self.port = port 
 self.user = user 
 self.passwd = passwd 
 self.cmd = cmd 
 def run(self): 
 """默认调用的内容""" 
 # cmd hostname 
 # put 
 # get 
 cmd_str = self.cmd.split()[0] # cmd 
 # 类的反射，判断类里面是否可以支持该操作 
 if hasattr(self, 'do_'+ cmd_str): # do_cmd 
 getattr(self, 'do_'+cmd_str)() 
 else: 
 print("目前不支持该功能") 
 def do_cmd(self): 
 client = paramiko.SSHClient() 
 client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
 try: 
 client.connect(hostname=self.hostname, 
 port=self.port, 
 username=self.user, 
 password=self.passwd) 
 print("正在连接%s......." % (self.hostname)) 
 except NoValidConnectionsError as e: 
 print("连接失败") 
 except AuthenticationException as e: 
 print("密码错误") 
 else: 
 # 4. 执行操作 
 cmd = ''.join(self.cmd.split()[1:]) ##将输入的后面的取出，作为 
 stdin, stdout, stderr = client.exec_command(cmd) 
 # 5.获取命令执行的结果 
 result = stdout.read().decode('utf-8') 
 print(result) 
 finally: 
 # 6.关闭连接 
 client.close() 
 def do_put(self): 
 ###put /tmp/passwd ###将本地的/tmp/passwd上传到远端/tmp/passwd 
 print('正在上传...') 
 try: 
 #获取Transport实例 
 tran = paramiko.Transport(self.hostname,int(self.port)) ##由于端口为整形，而我们用split方法得到的是str 
 #连接SSH服务端 
 tran.connect(username = self.user, password = self.passwd) 
 except SSHException as e: 
 print('连接失败') 
 else: 
 #获取SFTP实例 
 sftp = paramiko.SFTPClient.from_transport(tran) 
 newCmd = self.cmd.split()[1:] 
 if len(newCmd) == 2: 
 #设置上传的本地/远程文件路径 
 localpath=newCmd[0] 
 remotepath=newCmd[1] 
 #执行上传动作 
 sftp.put(localpath,remotepath) 
 print('%s文件上传到%s主机的%s文件成功' %(localpath,self.hostname,remotepath)) 
 else: 
 print('上传文件信息错误') 
 tran.close() 
 def do_get(self): 
 print('正在下载...') 
 try: 
 # 获取Transport实例 
 tran = paramiko.Transport(self.hostname, int(self.port)) ##由于端口为整形，而我们用split方法得到的是str 
 # 连接SSH服务端 
 tran.connect(username=self.user, password=self.passwd) 
 except SSHException as e: 
 print('连接失败') 
 else: 
 # 获取SFTP实例 
 sftp = paramiko.SFTPClient.from_transport(tran) 
 newCmd = self.cmd.split()[1:] 
 if len(newCmd) == 2: 
 # 设置下载的本地/远程文件路径 
 localpath = newCmd[1] 
 remotepath = newCmd[0] 
 # 执行上传动作 
 sftp.get( remotepath,localpath) 
 print('%s主机的%s文件下载到%s文件成功' % (self.hostname,remotepath,localpath)) 
 else: 
 print('上传文件信息错误') 
 tran.close() 
import paramiko 
import os 
# 1.选择操作的主机组:eg:mysql,web,ftp 
groups=[file.rstrip('.conf') for file in os.listdir('conf')] 
print("主机组显示：".center(50,'*')) 
for group in groups: 
 print('\t',group) 
choiceGroup = input("选择批量操作的主机组(eg:mysql):") 
##2.根据选择的主机组，显示包含的主机IP/主机名 
# 1).打开文件conf/choiceGroup.conf 
# 2).依次读取文件每一行 
# 3).只拿出 
print("主机组包含的主机：".center(50,'*')) 
with open('conf/%s.conf' %(choiceGroup)) as f: 
 for line in f: 
 print(line.split(':')[0]) 
 f.seek(0,0) ##把指针移动到文件最开始 
 hostinfos = [line.strip() for line in f.readlines()] 
###3.让用户确认信息，选择需要批量执行的命令； 
## -cmd shell 命令 
## -put 本地文件 远程文件 
## -get 远程文件 本地文件 
print("批量执行脚本".center(50,"*")) 
while True: 
 cmd = input('>>:').strip() 
 if cmd : 
 if cmd == 'exit' or cmd == "quit": 
 print("执行完毕，正在退出") 
 break 
 for info in hostinfos: 
 host,port,user,passwd = info.split(':') 
 clientObj = SshRemoteHost(host,port,user,passwd,cmd) 
 clientObj.run()