4.python 系统批量运维管理器之paramiko模块

时间：2023-01-02 21:54:17

 

 

paramiko

paramiko是ssh服务最经常使用的模块，遵循SSH2协议，支持以加密和认证的方式，进行远程服务器的连接。

paramiko实现ssh2不外乎两个角度：SSH客户端与服务端

  • SSHClient：包装了Channel，Transport，SFTPClient。
  • Channel：是一种Socket类，一种安全的SSH传输通道。
  • Transport：是一种加密的会话（但是这样一个对象的Session并未建立），并且创建了一个加密的tunnels，这个tunnels叫做Channel。
  • Session：是Client与Server保持连接的对象，用Connect()，start_client()，start_server()开始会话。

具体请参考paramiko的库文档：http://docs.paramiko.org/en/2.0/index.html

使用pip install paramiko安装，失败的话，建议使用yum安装：yum install python-paramiko

代码实例：

paramiko远程密码连接：

import paramiko  

#实例化SSHClient
client = paramiko.SSHClient() 

#创建ssh连接日志文件(只保留前一次连接的详细日志，以前的日志会自动被覆盖)
paramiko.util.log_to_file('paramiko.log')

#自动添加策略，保存服务器的主机名和密钥信息，如果不添加，不在know_hosts文件中记录的主机将无法连接
client.set_missing_host_key_policy(paramiko.AutoAddPolicy())

#连接SSH服务端，以用户名和密码进行认证
client.connect(hostname='127.0.0.1', port=22, username='root', password='123456') 

#打开一个Channel并执行命令
stdin,stdout,stderr = client.exec_command(command)

#打印执行结果
result=stdout.read().decode('utf-8') 
print(result)

#关闭SSHClient连接
client.close()

SSHClient()里面有一个transport变量，这个是用于获取连接的，因此我们也可以单独获取到transport变量，然后执行连接操作：

import paramiko

#创建一个通道 transport = paramiko.Transport(('127.0.0.1', 22))

#用户名密码连接服务器 transport.connect(username='root', password='123456')

#实例化SSHClient ssh = paramiko.SSHClient()
ssh._transport = transport stdin, stdout, stderr = ssh.exec_command(command) print(stdout.read().decode('utf-8'))

#关闭连接 transport.close()

paramiko使用密钥连接

import paramiko 
# 配置私人密钥文件位置 private = paramiko.RSAKey.from_private_key_file('/Users/zjz/.ssh/id_rsa') 
#实例化SSHClient client = paramiko.SSHClient() 
#自动添加策略，保存服务器的主机名和密钥信息，如果不添加，那么不再本地know_hosts文件中记录的主机将无法连接 client.set_missing_host_key_policy(paramiko.AutoAddPolicy()) 
#连接SSH服务端，以用户名和密码进行认证 client.connect(hostname='127.0.0.1',port=22,username='root',pkey=private) 
# 打开一个Channel并执行命令 stdin, stdout, stderr = client.exec_command(command) # stdout 为正确输出，stderr为错误输出，同时是有1个变量有值 
# 打印执行结果 print(stdout.read().decode('utf-8')) 
# 关闭SSHClient client.close()

使用transport封装使用密钥连接：

import paramiko 
# 配置私人密钥文件位置 private = paramiko.RSAKey.from_private_key_file('/Users/zjz/.ssh/id_rsa') 
# 创建一个通道 transport = paramiko.Transport(('127.0.0.1', 22)) transport.connect(username='root', pkey=private_key) 
#实例化SSHClient client = paramiko.SSHClient() client._transport = transport 
# 打开一个Channel并执行命令 stdin, stdout, stderr = client.exec_command('df -h ') # stdout 为正确输出，stderr为错误输出，同时是有1个变量有值 
# 打印执行结果 print(stdout.read().decode('utf-8')) 
# 关闭SSHClient transport.close()

SFTPClient实现上传文件

import paramiko 
# 获取Transport实例 tran = paramiko.Transport(('127.0.0.1', 22))
 # 连接SSH服务端，使用password tran.connect(username="root", password='123456') 
# 或使用 # 配置私人密钥文件位置 private = paramiko.RSAKey.from_private_key_file('/Users/zjz/.ssh/id_rsa') 
# 连接SSH服务端，使用pkey指定私钥 tran.connect(username="root", pkey=private) 
# 获取SFTP实例 sftp = paramiko.SFTPClient.from_transport(tran) 
# 设置上传的本地/远程文件路径 localpath = "/Users/1.txt" remotepath = "/tmp/1.txt" 
# 执行上传动作 sftp.put(localpath, remotepath) 
#如果远程主机有这个文件则返回一个对象，负责抛出异常
try:
 sftp.file(remotepath)
 print("上传成功")
except IOError:
 print("上传失败")
finally: 
# 关闭连接 tran.close()

SFTPClinet实现下载文件

import paramiko
import os 
# 获取Transport实例 tran = paramiko.Transport(('127.0.0.1', 22)) 
# 连接SSH服务端，使用password tran.connect(username="root", password='123456') 
# 或使用 # 配置私人密钥文件位置 private = paramiko.RSAKey.from_private_key_file('/Users/zjz/.ssh/id_rsa') 
# 连接SSH服务端，使用pkey指定私钥 tran.connect(username="root", pkey=private) 
# 获取SFTP实例 sftp = paramiko.SFTPClient.from_transport(tran) 
# 设置下载的本地/远程文件路径 localpath = "/Users/1.txt" remotepath = "/tmp/1.txt" 
#判断远程服务器是否有这个文件
try: sftp.file(remotepath)
 #执行下载动作 sftp.get(remotepath, localpath) except IOError as e:
 print(remotepath+"remote file no exit!")
 exit()
finally:
 #关闭连接
 sftp.close()
if os.path.isfile(localpath):
 print("下载成功")
else:
 print("下载失败")

SSHClient也可以实现交互式终端

核心代码：

# 打开一个通道 channel = trans.open_session() 
# 获取终端 channel.get_pty() 
# 激活终端，这样就可以登录到终端了，就和我们用类似于xshell登录系统一样 channel.invoke_shell()

重要的是select模块的使用，select会监听socket或者文件描述符的I/O状态变化，并返回变化的socket或者文件描述符的对象

r_list,w_list,x_list = select.select(rlist, wlist, xlist, timeout])

参数详解：

  • rlist：list类型，监听其中的socket或者文件描述符是否变为可读状态，返回那些可读的socket或者文件描述符组成的list。
  • wlist：list类型，监听其中的socket或者文件描述符是否变为可写状态，返回那些可写的socket或者文件描述符组成的list。
  • xlist：list类型，监听其中的socket或者文件描述符是否出错，返回那些出错的socket或者文件描述符组成的list。
  • timeout：设置select的超时时间，设置为None代表永远不会超时，即阻塞。

详细模块用法见：https://blog.csdn.net/weixin_43367828/article/details/84640103

xshell功能实现

import paramiko 
import os 
import select 
import sys 

# 建立一个socket 
trans = paramiko.Transport(('127.0.0.1', 22)) 

# 启动一个客户端 
trans.start_client() 

# 如果使用rsa密钥登录的话 
''' default_key_file = os.path.join(os.environ['HOME'], '.ssh', 'id_rsa') prikey = paramiko.RSAKey.from_private_key_file(default_key_file) trans.auth_publickey(username='super', key=prikey) ''' 

# 如果使用用户名和密码登录 
trans.auth_password(username='root', password='123456') 

# 打开一个通道 
channel = trans.open_session() 

# 获取终端 
channel.get_pty() 

# 激活终端，这样就可以登录到终端了，就和我们用类似于xshell登录系统一样 
channel.invoke_shell() 

# 下面就可以执行你所有的操作，用select实现 
# 对输入终端sys.stdin和 通道进行监控, 
# 当用户在终端输入命令后，将命令交给channel通道，这个时候sys.stdin就发生变化，select就可以感知 
# channel的发送命令、获取结果过程其实就是一个socket的发送和接受信息的过程 

while True: 
 readlist, writelist, errlist = select.select([channel, sys.stdin,], [], []) 
 # 如果是用户输入命令了,sys.stdin发生变化 
 if sys.stdin in readlist: 
 # 获取输入的内容 
 input_cmd = sys.stdin.read(1) 
 # 将命令发送给服务器 
 channel.sendall(input_cmd) 
 # 服务器返回了结果,channel通道接受到结果,发生变化 select感知到 
 if channel in readlist: 
 # 获取结果 
 result = channel.recv(1024) 
 # 断开连接后退出 
 if len(result) == 0: 
 print("\r\n**** EOF **** \r\n") 
 break 
 # 输出到屏幕 
 sys.stdout.write(result.decode()) 
 sys.stdout.flush() 
# 关闭通道 
channel.close() 
# 关闭链接 
trans.close()

SSHClient实现堡垒机模式下的命令执行

         堡垒机在一定程度上提升了运营安全级别，但同时也提高了日常运营成本，作为管理的中转设备，任何针对业务服务器的管理请求都会经过此节点，比如SSH协议，首先运维人员在办公电脑通过SSH协议登陆堡垒机，再通过堡垒机SSH跳转到所有的业务服务器进行维护操作。

         可以利用paramiko的invoke_shell机制实现通过堡垒机实现服务器操作，原理是SSHClient.connect到堡垒机后开启一个新的SSH会话（session），通过新的会话运行 "sshuser@IP" 去实现远程命令执行。

SFTPClient实现堡垒机模式下的远程文件上传

        实现堡垒机模式下的文件上传，原理是通过paramiko的SFTPClient将文件从办公设备上传至堡垒机指定的临时目录，如 /tmp。再通过SSHClient的invoke_shell方法开启ssh会话，执行scp命令，将 /tmp下的指定文件复制到目标业务服务器上。

具体代码实现请参考：http://www.cnblogs.com/charliedaifu/p/10098848.html

 

sftp是安全文件传输协议，提供一种安全的加密方法，sftp是SSH的一部分，SFTPClient类实现了sftp客户端，通过已建立的SSH通道传输文件，与其他的操作，如下：

方法 描述
sftp.getcwd() 返回当前工作目录
sftp.chdir(path) 改变工作目录
sftp.chmod(path, mode) 修改权限
sftp.chown(path, uid, gid) 设置属主属组
sftp.close() 关闭sftp
sftp.file(filename, mode=’r’, bufsize=-1) 读取文件
sftp.from_transport(s) 创建SFTP客户端通道
sftp.listdir(path=’.’) 列出目录，返回一个列表
sftp.listdir_attr(path=’.’) 列出目录，返回一个SFTPAttributes列表
sftp.mkdir(path, mode=511) 创建目录
sftp.normalize(path) 返回规范化path
sftp.open(filename, mode=’r’, bufsize=-1) 在远程服务器打开文件
sftp.put(localpath, remotepath, callback=None) localpath文件上传到远程服务器remotepath
sftp.get(remotepath, localpath, callback=None) 从远程服务器remotepath拉文件到本地localpath
sftp.readlink(path) 返回一个符号链接目标
sftp.remove(path) 删除文件
sftp.rename(oldpath, newpath) 重命名文件或目录
sftp.rmdir(path) 删除目录
sftp.stat(path) 返回远程服务器文件信息（返回一个对象的属性）
sftp.truncate(path, size) 截取文件大小
sftp.symlink(source, dest) 创建一个软链接（快捷方式）
sftp.unlink(path) 删除软链接

 

参考连接：

https://blog.csdn.net/songfreeman/article/details/50920767

https://blog.csdn.net/qq_29778641/article/details/82186438

https://blog.csdn.net/forever_wen/article/details/82556154

标签：

相关文章