python通过SSH登陆linux并操作

用的昨天刚接触到的库，在windows下通过paramiko来登录linux系统并执行了几个命令，基本算是初试成功，后面会接着学习的。

代码：

>>> import paramiko

>>> ssh = paramiko.SSHClient()
>>> ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
>>> ssh.connect(‘10.104.6.8‘,username = ‘xiaopeng‘,password=‘******‘)
>>> cmd = ‘cd‘    #进入用户目录home
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> cmd = ‘ls >test‘  #管道，ls命名的输出到文件test里面
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> cmd = ‘cat test‘  # 显示test的内容，即ls命名的结果
>>> stdin,stdout,stderr = ssh.exec_command(cmd)
>>> print stdout.readlines()   #结果 汉字用的字符显示
[‘code\n‘, ‘Desktop\n‘, ‘order.cpp\n‘, ‘python\n‘, ‘test\n‘, ‘\xe5\x85\xac\xe5\x85\xb1\xe7\x9a\x84\n‘, ‘\xe6\xa8\xa1\xe6\x9d\xbf\n‘, ‘\xe8\xa7\x86\xe9\xa2\x91\n‘, ‘\xe5\x9b\xbe\xe7\x89\x87\n‘, ‘\xe6\x96\x87\xe6\xa1\xa3\n‘, ‘\xe4\xb8\x8b\xe8\xbd\xbd\n‘, ‘\xe9\x9f\xb3\xe4\xb9\x90\n‘]

这里我们都会发现，使用exec_command(‘cd dirname‘)时并不会切换目录，execute_command() 是a single session，每次执行完后都要回到缺省目录。所以可以 .execute_command(‘cd  /var; pwd‘)。

python里面的paramiko模块，用这个实现ssh登录更加简单。看下面的代码：

1. #-*- coding: utf-8 -*-  
2. #!/usr/bin/python   
3. import paramiko  
4. import threading  
5. def ssh2(ip,username,passwd,cmd):  
6.     try:  
7.         ssh = paramiko.SSHClient()  
8.         ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())  
9.         ssh.connect(ip,22,username,passwd,timeout=5)  
10.         for m in cmd:  
11.             stdin, stdout, stderr = ssh.exec_command(m)  
12. #           stdin.write("Y")   #简单交互，输入 ‘Y’   
13.             out = stdout.readlines()  
14.             #屏幕输出  
15.             for o in out:  
16.                 print o,  
17.         print ‘%s\tOK\n‘%(ip)  
18.         ssh.close()  
19.     except :  
20.         print ‘%s\tError\n‘%(ip)  
21. if __name__==‘__main__‘:  
22.     cmd = [‘cal‘,‘echo hello!‘]#你要执行的命令列表  
23.     username = ""  #用户名  
24.     passwd = ""    #密码  
25.     threads = []   #多线程  
26.     print "Begin......" 
27.     for i in range(1,254):  
28.         ip = ‘192.168.1.‘+str(i)  
29.         a=threading.Thread(target=ssh2,args=(ip,username,passwd,cmd))   
30.         a.start() 

上面的程序还是有些技巧的：

1.利用多线程，同时发出登录请求，同时去连接电脑，这样速度快很多，我试了一下，如果不用多线程，直接一个一个挨着执行的话，大约5～10秒钟才能对一台电脑操作完，具体时间要根据命令的来决定，如果是软件安装或者卸载时间要更长一些。这样下来怎么也要一二十分钟，用多线程后就快多了，所有的命令执行完用了不到2分钟！

2.最好用root用户登录，因为安装或者卸载软件的时候如果用普通用户又会提示输入密码，这样又多了一次交互，处理起来就比较麻烦！安装软件时apt-get install xxx 最好加上“-y”参数，因为有时安装或删除软件时提示是否继续安装或卸载，这又是一次自动交互！加上那个参数后就没有人机交互了。

3. 循环时循环所有ip，因为计算机的ip是路由器自动分配的，保险起见，最好全部都执行，保证没有遗漏的主机

4.远端执行命令时如果有交互，可以这样用 stdin.write("Y")来完成交互，“Y”就是输入“Y”。

5.把所有的命令放到一个列表里面，遍历列表可以依次执行列表里面的命令

6.为了更好的进行控制，最好在电脑上提前把root用户打开，装好ssh服务器并让其开机自动执行。

ssh = paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

ssh.connect(‘10.111.43.18‘,22,‘root‘,‘12345‘,timeout=5)

stdin, stdout, stderr = ssh.exec_command(‘cd /root/epoll/‘)

stdin, stdout, stderr = ssh.exec_command(‘pwd‘)

上边的代码输出应该是 /root/epoll/，但结果却是 /root ，即使用root登陆的缺省目录

原因是exec_command为单个会话，执行完成之后会回到登录时的缺省目录

修改为这样执行结果则为预期的 /root/epoll 目录

stdin, stdout, stderr = ssh.exec_command(‘cd /root/epoll/;pwd‘)

最近闲着学习python，看到有个paramiko模块，貌似很强大，学着写了个一个小程序，可以直接在window下登录到linux，执行并返回执行结果。

  下面直接贴代码：

下面看下效果，呵呵：

paramiko模块是基于Python实现的ssh远程安全链接，实现命令远程、文件传输、SSH代理等功能。本文详细介绍了paramiko模块常用的方法，且实用价值非常高，稍作修改就能做出一个实用的运维工具。

安装

letong@me:~$ sudo pip install paramiko

简单的远程执行命令实例

#!/usr/bin/env python
import paramiko
hostname=‘127.0.0.1‘
username=‘root‘
password=‘123456‘
port=2222

ssh=paramiko.SSHClient()
ssh.load_system_host_keys()
ssh.connect(hostname=hostname,port=port,username=username,password=password)
stdin,stdout,stderr=ssh.exec_command(‘ls /‘)
print stdout.read()
ssh.close()

SSHClient类

SSHClient类是SSH服务会话的高级表示，封装了传输、通道以及SFTPClient的校验、建立方法，通常用于执行命令。

connect方法

connect(self, hostname, port=22, username=None, password=None, pkey=None, Key_filename=None, timeout=None, allow_agent=True, look_for_keys=True, compress=False)

参数说明：

hostname，连接目标的主机地址 port，连接目标的端口，默认22 username，用户名 password，用户密码 pkey，私钥方式用户验证 key_filename，私钥文件名 timeout，超时时间 allow_agent，是否允许ssh代理 look_for_keys，是否允许搜索私钥文件 compress，打开时是否压缩

exec_command方法

exec_command(self, command, bufzise=-1)

参数说明：

command，执行的命令 bufsize，文件缓冲区大小，-1不限制

load_system_host_keys方法

load_system_host_keys(self, filename=None)

参数说明：

filename，指定远程主机公钥文件

set_missing_host_key_policy方法

ssh=paramiko.SSHClient()

ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())

参数说明：

AutoAddPolicy，自动添加主机名及密钥到本地并保存，不依赖load_system_host_keys()配置 RejectPolicy，自动拒绝未知主机名和密钥，依赖load_system_host_keys()配置 WarnningPolicy，功能与AutoAddPolicy相同，但是未知主机会报警

SFTPClient类

根据SSH传输协议的sftp会话，实现远程文件上传、下载等操作。

from_transport方法

from_transport(cls, t)

参数说明：

t，一个已通过验证的传输对象

例：

>>> import paramiko >>> a = paramiko.Transport((“127.0.0.1″,2222)) >>> a.connect(username=”root”, password=’123456′) >>> sftp = paramiko.SFTPClient.from_transport(a)

put方法

put(self, localpath, remotepath, callback=None, confirm=True)

参数说明：

localpath，上传源文件的本地路径 remotepath，目标路径 callback，获取接收与总传输字节数 confirm，上传完毕后是否调用stat()方法，以便确认文件大小

例：

>>> localpath=’ftp-test.log’ >>> remotepath=’/data/ftp-test.log’ >>> sftp.put(localpath,remotepath)

get方法

get(self, remotepath, localpath, callback=None)

参数说明：

remotepath，需要下载的远程文件 localpath，本地存储路径 callback，同put方法

其他方法

mkdir，用于创建目录 remove，删除目录 rename，重命名 stat，获取文件信息 listdir，获取目录列表