Python中操控ssh和sftp
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服务器集群管理、远程部署，绝对是神器啊……