将（sys.）stdout重定向到日志记录（python3）

Question

概述：两个脚本，一个用于日志记录（ddlog），另一个使用paramiko进行ssh连接和命令执行（ddlogin）。记录器工作正常，并为我连接到的每个主机创建日志文件，但是标准输入和标准输出的处理程序不会触发日志事件。目前我得到以下io错误。我试过修改日志记录.StreamHandler（）与系统标准输出但这似乎也行不通。在

错误：

Exception in thread Thread-1:
Traceback (most recent call last):
  File "/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/threading.py", line 916, in _bootstrap_inner
    self.run()
  File "ddlogin.py", line 89, in run
    main(host)
  File "ddlogin.py", line 60, in main
    line = sys.stdout.read().splitlines(4096)
io.UnsupportedOperation: not readable

动态日志：

^{pr2}$

D登录：

import argparse, ddlog, os, paramiko, sys
from threading import Thread
from queue import Queue

def main(hostlist):
    global username, password, commands

    try:
        # Filepath, logger, and console handler
        logfile = os.getcwd() + '//logs/' + hostlist + '.log'
        ddlog.setup(logfile)

        ssh = paramiko.SSHClient()
        ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
        try:
            ssh.connect(hostlist, username=username, password=password)
        except paramiko.AuthenticationException:
            ddlog.event("[-] Failed to Authenticate.",'error')
            pass
        except paramiko.SSHException:
            ddlog.event("[-] An SSH exception was raised. Check clients ssh settings.",'error') #, exc_info=True)
            pass
        except TimeoutError:
            ddlog.event("[-] host is not reachable.",'error')
            pass

        host = hostlist.strip('\n')
        ddlog.event('[+] successfully connected to ' + host,'info')

        for command in commands:
            stdin, stdout, stderr = ssh.exec_command(command)
            stdin.close()
            line = sys.stdout.read().splitlines(4096)
            ddlog.event('[+] ' + host + line.strip().decode('ascii'),'info')

    except keyboardInterrupt:
        sys.exit('[-] Process aborted by user input')

if __name__ == "__main__":
        # system arguments: ddlogin.py -u admin -p password -l 192.168.1.1 -c dmesg 'uname -a'
        parser = argparse.ArgumentParser()
        parser.add_argument('-u', '--username', action='store', dest='username', help='The ssh username.')
        parser.add_argument('-p', '--password', action='store', dest='password', help='The ssh password.')
        parser.add_argument('-l', '--hostlist', nargs='+',action='store', dest='hostlist', help='List of devices to interact with.')
        parser.add_argument('-c', '--commands', nargs='+',action='store', dest='commands', help='An exact list of commands to run.')
        #parser.add_argument('-o', '--output', nargs='+',action='store', dest='commands', help='TBD...')
        args = parser.parse_args()
        username = args.username
        password = args.password
        hostlist = args.hostlist
        commands = args.commands


        class ParallelConnections(Thread):
            def __init__(self, queue):
                Thread.__init__(self)
                self.queue = queue

            def run(self):
                while True:
                    host = self.queue.get()
                    main(host)
                    self.queue.task_done()

        thread_count = 4
        queue = Queue()

        for i in range(thread_count):
            Parallel = ParallelConnections(queue)
            Parallel.daemon = True
            Parallel.start()

        for host in hostlist:
            queue.put(host)

        queue.join()

Answer 1

感谢this wonderful post我找到了一个答案，并构建了下面的解决方案。希望其他面临日志问题的人会阅读这篇文章。在

  for command in commands:
        stdin, stdout, stderr = ssh.exec_command(command, get_pty=True)
        lines = iter(stdout.readline, "")
        for line in lines:
            ddlog.event('[+] '+line,'info')
        stdin.close()