最新需要做一个小工具,让协作部门能够获取到服务器上的一些资源讯息,因为工具是pyqt写的所以牵扯到用python链接linux的问题,这里记录一下一些碰到的问题
上网搜了很多,发现paramiko实现远程SSH链接非常的方便,作为python的第三方库安装也很方便
pip install paramiko
ssh = paramiko.SSHClient()
首先通过paramiko创建SSHClient类,SSHClient类封装了链接SSH和发送命令等方法
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
设置在连接到没有已知主机密钥的服务器时使用的策略,有三种情况,一般直接用Auto,会将远程主机的秘钥添加到known_hosts文件中
ssh.connect(ip, port, username, password)
通过connect链接
cmd = "python /work/tools/build_project.py"
stdin, stdout, stderr = ssh.exec_command(cmd)
我们通过exec_command方法在链接成功后向主机发送命令,有三个返回参数
stdin, stdout, stderr = ssh.exec_command(cmd)
successResult = stdout.read()
errorResult = stdout.read()
可以通过read的方法获取到linux控制台的输出信息
在实际运用的时候,碰到了不少问题,如下:
这两个问题在实际运行的时候会造成很大的问题,比如下面的情况:
cd /work
git pull
cd tools
python test.py
同时执行多条命令的时候,因为git pull是一个非瞬时完成的命令,所有后边的命令都没有生效,并且read也拿不到控制台的信息,想了想可以通过延时调用的方式去初始,time.sleep(1),这个样子,但是很不优雅,因为时间是不固定的,这样做也很不保险
搜索查找了一下,找到了相关问题的解决方式,在exec_command返回参数stdout内有一个channel类,channel可以内可以接收远端主机的状态,就是exit_status_ready(),当远端主机执行结束后,exit_status_ready()就是true,如此就方便处理了
stdin, stdout, stderr = ssh.exec_command(s_cmd, get_pty=True)
result = ""
while not stdout.channel.exit_status_ready():
result = result + stdout.readline()
if stdout.channel.exit_status_ready():
break
如上述代码所示,在发送命令后,在while循环去判断远端注意执行状态,当执行状态结束时,返还所有接收到的信息
下述为完整代码演示, 为了方便执行多条命令,这里直接调用shell脚本
#!/bin/bash
project_path="/works"
cd $project_path
# 清理
git clean -f
git checkout *
# 拉代码
git pull
cd tools
python build_android.py
import paramiko
ssh = paramiko.SSHClient()
ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
ssh.connect('192.168.x.xxx', 22, "root", 123456)
s_cmd = "sh /work/tools/ssh/build_android.sh"
stdin, stdout, stderr = ssh.exec_command(s_cmd, get_pty=True)
result = ""
while not stdout.channel.exit_status_ready():
result = result + stdout.readline()
if stdout.channel.exit_status_ready():
break
print(result)
如上述代码所示,两个脚本的配合,就可以实现多命令的远程调用和回执,另外建议ssh的操作都放到子线程里去处理,这样就不会卡着主线程了
https://github.com/KingSun5
若是觉得博主的文章写的不错,不妨关注一下博主,点赞一下博文,另博主能力有限,若文中有出现什么错误的地方,欢迎各位评论指摘。
本文属于原创文章,转载请著名作者出处并置顶!!