最新需要做一个小工具,让协作部门能够获取到服务器上的一些资源讯息,因为工具是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的操作都放到子线程里去处理,这样就不会卡着主线程了
Github