Python中操控ssh和sftp

1 Reply

在Python(其实任何语言都是)中操控ssh执行远程命令是一个很麻烦的事情……

首先要突破ssh密码的非交互模式，我之前一直用sshpass拼接各种复杂的字符串。

然后是之后的执行命令只能执行一行，或者是很长的字符串，拼接起来很麻烦。

Python中有一个很活跃的包ssh，它fork自大名鼎鼎的Paramiko，后者是经典的ssh模块，不过作者不再维护了。

1、安装

wget http://pypi.python.org/packages/source/s/ssh/ssh-1.7.14.tar.gz

tar -xzvf ssh-1.7.14.tar.gz

cd ssh-1.7.14

python2.7 setup.py install

上面的安装最好在联网状态下进行，因为它会自动下载依赖包pycrypto。

2、ssh用法：设置缺失Key、登录、执行命令、获取结果

import ssh

# New SSHClient

client = ssh.SSHClient()

# Default accept unknown keys

client.set_missing_host_key_policy(ssh.AutoAddPolicy())

# Connect

client.connect("xxx.coder4.com", port=22, username="xxxx", password="xxxx")

# Execute shell remotely

stdin, stdout, stderr = client.exec_command("ls -alh")

stdout.read()

注意：一定要获取stdout/stderr，然后read，否则远程命令很有可能无法执行完毕。

如果你想执行后台命令，看这里。

输出结果如下：

total 64K

drwxr-xr-x 7 liheyuan liheyuan 4.0K Dec 19 16:31 .

drwxr-xr-x 3 root     root     4.0K Oct 18  2011 ..

-rw------- 1 liheyuan liheyuan 7.5K Mar 20 12:16 .bash_history

-rw-r--r-- 1 liheyuan liheyuan  220 Oct 18  2011 .bash_logout

-rw-r--r-- 1 liheyuan liheyuan 3.1K Nov 28  2011 .bashrc

drwx------ 2 liheyuan liheyuan 4.0K Oct 18  2011 .cache

drwxr-xr-x 3 liheyuan liheyuan 4.0K Oct 18  2011 .config

-rw-r--r-- 1 liheyuan liheyuan  675 Oct 18  2011 .profile

drwx------ 2 liheyuan liheyuan 4.0K Dec 19 16:09 .ssh

drwxr-xr-x 2 liheyuan liheyuan 4.0K Dec 19 16:25 .vim

-rw------- 1 root     root     6.8K Dec 19 16:31 .viminfo

drwxr-xr-x 3 liheyuan liheyuan 4.0K Nov 25  2011 code

-rwxr--r-- 1 liheyuan liheyuan 1.1K Dec 19 16:11 firewall.sh

-rw------- 1 liheyuan liheyuan 1.9K Oct 18  2011 settings.json

lrwxrwxrwx 1 liheyuan liheyuan   28 Oct 18  2011 transmission -> /var/lib/transmission-daemon

3、sftp用法

当拿到一个SSHClient后，我们除了执行命令，还可以开启一个sftp的session，用于传输文件、创建文件夹等等。

# Open sftp session

sftp = client.open_sftp()

# Make a dir

sftp.mkdir('abc')

# Down file from remote to local 这个可能会抛出异常，如果失败的话

sftp.get('firewall.sh', '/tmp/firewall.sh')

# Upload file from local to remote 也可能会抛出异常

sftp.put('/tmp/firewall.sh', 'firewall_new.sh')

最终结果如下：

drwxr-xr-x 3 liheyuan liheyuan 4.0K Jun  1 07:18 abc

-rwxr--r-- 1 liheyuan liheyuan 1.1K Dec 19 16:11 firewall.sh

-rw-r--r-- 1 liheyuan liheyuan 1.1K Jun  1 07:20 firewall_new.sh

总结：用ssh模块开发Linux服务器集群管理、远程部署，绝对是神器啊……