如何通过队列传递子进程标准输出？ 蟒蛇

Question

我有一个Logger类，其中包含以下内容：

class Logger():
    def __init__(self):
        self.terminal = sys.__stdout___
        self.log = open('logFile.log', 'w')
    def write(message):
        self.terminal.write(message)
        self.log.write(message)

具有以下内容的主体：

import Logger, sys
sys.stdout = Logger.Logger()
import File1
File1.aFunc()

在File1中：

def execSubProc(target,queue):
    target(queue)

q = multiprocess.Queue()
proc = multiprocess.Process(target=execSubProc, args=(testFunc,q))
proc.start()
proc.join()

在File2中：

def testFunc(queue)
#Some print statements
#I want to get these print statements in the queue for main process to read

好的，这是我的结构。 我想做的是从运行testFunc（）的子进程中获取标准输出，并将其放入队列中。 当程序返回main时，我想从队列中读取并将这些内容写入日志文件。

我不能只是再次在文件2中执行sys.stdout = Logger.Logger（），因为这只会覆盖电源日志文件。 如何捕获所有子进程stdout并将其放入队列？

Answer 1

一种解决方案是使用logging模块。
记录器在multiprocessing.util是不负责任的，可用于生成线程安全日志：

import time
import logging
import multiprocessing as mp
import multiprocessing.util as util
from sys import stdout as out


def runner(ids):
    log = util.get_logger()
    for i in range(10):
        time.sleep(.5)
        log.log(25, 'Process {} logging {}'.format(ids, i))

if __name__ == '__main__':
    # Setup the logger
    log = util.get_logger()
    log.getEffectiveLevel()
    # You can setup a file instead of stdout in the StreamHandler
    ch = logging.StreamHandler(out)
    ch.setLevel(25)
    ch.setFormatter(logging.Formatter('%(processName)s - %(message)s'))
    log.addHandler(ch)
    log.setLevel(25)

    p1 = mp.Process(target=runner, args=(1,), name="Process1")
    p2 = mp.Process(target=runner, args=(2,), name="Process2")

    p1.start()
    p2.start()

    time.sleep(2)
    log.log(25, 'Some main process logging meanwhile')

    p1.join()

级别> 20可以避免启动和停止日志。
如果参数out是打开的可写文件，则处理程序可以直接登录文件。
这避免了必须自己处理日志队列。 logging模块中还有许多其他高级功能。

使用multiprocessing模块中的记录器来获取线程安全日志很重要。