Python Paramiko将命令输出到文件

时间:2015-09-13 08:04:04

标签: python logging paramiko

我正在编写一个脚本,它将获取服务器列表并运行service something restart ; service something status命令。 我正在使用paramiko SSHClient来实现这一点,具有以下功能:

def restart_service(node_name):
    print('='*30 + '  Starting to work on ' + node_name + '  ' + '='*30 + '\n')
    logging.info('Connecting to %s in order to restart %s...', node_name, service_name)
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.load_system_host_keys()
    ssh.connect(node_name)
    channel = ssh.get_transport().open_session()
    channel.exec_command(command)
    while True:
        if channel.exit_status_ready():
            break
        rl, wl, xl = select.select([channel], [], [], 0.0)
        if len(rl) > 0:
            print channel.recv(1024)
    ssh.get_transport().close()
    ssh.close()

我的问题是:如何设置paramiko logger以将我的命令输出写入日志?我不需要获取所有的调试信息,我只关心它,我将在文件中获得服务重启和服务状态命令的结果。

我设置的记录器配置是:

logging.basicConfig(filename='./log_restartService.log', level=logging.INFO, format='%(asctime)s  %(levelname)s: %(message)s')
logging.getLogger("paramiko").setLevel(logging.INFO)

我也尝试使用我发现的其他主题中建议的logger = paramiko.util.logging.getLogger(),但这也无济于事。

谢谢!

1 个答案:

答案 0 :(得分:1)

不要使用paramiko记录器。而是创建自己的。

import logging
logger = logging.getLogger(__name__)

def restart_service(node_name):
    print('='*30 + '  Starting to work on ' + node_name + '  ' + '='*30 + '\n')
    logging.info('Connecting to %s in order to restart %s...', node_name, service_name)
    ssh = paramiko.SSHClient()
    ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
    ssh.load_system_host_keys()
    ssh.connect(node_name)
    channel = ssh.get_transport().open_session()
    channel.exec_command(command)
    while True:
        if channel.exit_status_ready():
            break
        rl, wl, xl = select.select([channel], [], [], 0.0)
        if len(rl) > 0:
            # Log output
            logger.info(channel.recv(1024))
    ssh.get_transport().close()
    ssh.close()

通过这种方式,您可以精确控制您想要记录的内容以及哪些信息对您很重要