python ssh模块设置超时_python paramiko模块中设置执行命令超时值的方法

使用paramiko工具对几百台设备进行管理，由于服务器本身或网络原因，有时返回值回不来，程序会一直等待，遇到这种情况时，就需要设置一个超时值。

paramiko模块中执行命令：

stdin, stdout , stderr = s.exec_command(command)

这个地方在模块中只有一个参数，paramiko默认在这个是并不能设置超时值。

其实paramiko本身是可以在这个地方设置超时值的，只是默认情况下是没有这个选项的，需要在paramiko的安装目录中修改他的源代码，让它支持，在代码中是有这个接口的。

之所以没有这个这个超时值，可能是因为开发方考虑有些有些命令可能执行的时间比较长，比如大文件的压缩等，需要很长的时间才能执行完，超时值如果设置的话，有可能会中断命令的执行，索性留下接口，并不设置超时值。

但是，用这个模块批量的去操作多台设备的话，有时超时值是很有必要的。

修改paramiko源代码方法：

找到C:\Python27\Lib\site-packages\paramiko目录，其中有个client.py文件，在其中找到这段代码： 代码示例:

#!/bin/

python

#site：www.#

#

def exec_command(self, command, bufsize=-1):

"""

Execute a command on the SSH server.  A new L{Channel} is opened and

the requested command is executed.  The command's input and output

streams are returned as python C{file}-like objects representing

stdin, stdout, and stderr.

@param command: the command to execute

@type command: str

@param bufsize: interpreted the same way as by the built-in C{file()} function in python

@type bufsize: int

@return: the stdin, stdout, and stderr of the executing command

@rtype: tuple(L{ChannelFile}, L{ChannelFile}, L{ChannelFile})

@raise SSHException: if the server fails to execute the command

"""

chan = self._transport.open_session()

chan.exec_command(command)

stdin = chan.makefile('wb', bufsize)

stdout = chan.makefile('rb', bufsize)

stderr = chan.makefile_stderr('rb', bufsize)

return stdin, stdout, stderr

修改为： 代码示例:

#!/bin/python

#site: www.#

#

def exec_command(self, command, bufsize=-1,timeout = None):

"""

Execute a command on the SSH server.  A new L{Channel} is opened and

the requested command is executed.  The command's input and output

streams are returned as python C{file}-like objects representing

stdin, stdout, and stderr.

@param command: the command to execute

@type command: str

@param bufsize: interpreted the same way as by the built-in C{file()} function in python

@type bufsize: int

@return: the stdin, stdout, and stderr of the executing command

@rtype: tuple(L{ChannelFile}, L{ChannelFile}, L{ChannelFile})

@raise SSHException: if the server fails to execute the command

"""

chan = self._transport.open_session()

if timeout is not None:

chan.settimeout(timeout)

chan.exec_command(command)

stdin = chan.makefile('wb', bufsize)

stdout = chan.makefile('rb', bufsize)

stderr = chan.makefile_stderr('rb', bufsize)

return stdin, stdout, stderr

主要修改了如下的两个地方：

1、def exec_command(self, command, bufsize=-1,timeout = None)定义时加一个timeout = None；

2、在chan = self._transport.open_session()下面添加一个判断

if timeout is not None:

chan.settimeout(timeout)

在使用paramiko模块执行命令时的代码：

stdin, stdout , stderr = s.exec_command(command, timeout=10)

有一个超时值，执行命令的超时时间为10s。