如何让'print（）'，'os.system（）'和'subprocess.run（）'输出显示在控制台和日志文件中？

问题描述

最初，我有一个简单的程序将整个输出打印到控制台。

初始代码仅在控制台中显示输出

import os, subprocess

print("1. Before")
os.system('ver')                            
subprocess.run('whoami')        
print('\n2. After')

控制台输出

1. Before

Microsoft Windows [Version 10]
user01

2. After

然后，我决定在日志文件（log.txt）上复制一份，同时保持原始输出到控制台。

所以，这是新代码。

import os, subprocess, sys

old_stdout = sys.stdout
log_file = open("log.txt","w")
sys.stdout = log_file

print("1. Before")          # This appear in message.log only, but NOT in console
os.system('ver')            # This appear in console only, but NOT in message.log
subprocess.run('whoami')    # This appear in console only, but NOT in message.log
print('\n2. After')         # This appear in message.log only, but NOT in console

sys.stdout = old_stdout
log_file.close()

不幸的是，这并没有像预期的那样真正奏效。 某些输出仅显示在控制台上（ os.system('ver')和os.system('ver') subprocess.run('whoami') ），而print()函数仅显示在log.txt文件中，而不再显示在控制台中。

控制台输出

Microsoft Windows [Version 10]
user01

在log.txt文件中输出

1. Before

2. After

我希望在console和log.txt文件中获得类似的输出。 这可能吗？ 我的新代码出了什么问题？ 请让我知道如何解决这个问题。

控制台和log.txt文件中的所需输出

1. Before

Microsoft Windows [Version 10]
user01

2. After

系统没有任何魔术，你的代码需要区别对待stdout和stderr等文件指针 。 例如， stdout是文件指针之一，您可以在下面执行：

log_file_pointer = open('log.txt', 'wt')
print('print_to_fp', file=log_file_pointer)
# Note: the print function will actually call log_file_pointer.write('print_to_fp')

根据您的要求，您希望使魔术函数在单行中处理多个文件指针，您需要一个包装函数如下：

def print_fps(content, files=[]):
    for fi in files:
        print(content, file=fi)
# the argument `file` of print does zero magic, it can only handle one file pointer once. 

然后，你可以让魔法现在发生（在屏幕和文件中输出。）

import sys

log_file_pointer = open('log.txt', 'wt')
print_fps('1. Before', files=[log_file_pointer, sys.stdout])
print_fps('\n2. After', files=[log_file_pointer, sys.stdout])

完成print部分后，让我们继续进行系统调用。 在操作系统中运行任何命令，您将获得默认系统文件指针的返回： stdout和stderr 。 在python3中，您可以通过subprocess.Popen以字节为单位获取这些结果。 在代码下运行时，你想要的应该是stdout的结果。

import subprocess

p = subprocess.Popen("whoami", shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
stdout, stderr = p.communicate()

# stdout: b'user01'
# stdout: b''

再一次，您可以调用上面编写的包装函数，并在stdout和目标file_pointer中生成输出。

print_fps(stdout, files=[log_file_pointer, sys.stdout])

最后，结合上面的所有代码。 （另外还有一个方便的功能。）

import subprocess, sys

def print_fps(content, files=[]):
    for fi in files:
        print(content, file=fi)

def get_stdout(command):
    p = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    stdout, stderr = p.communicate()
    # Note: Original idea is to return raw stdout 
    # return stdout
    # Based on the scenario of the @Sabrina, the raw bytes of stdout needs decoding in utf-8 plus replacing newline '\r\n' to be pure
    return stdout.decode().replace('\r\n', '')

log_file_pointer = open('log.txt', 'wt')
print_fps('1. Before', files=[log_file_pointer, sys.stdout])
print_fps(get_stdout('ver'), files=[log_file_pointer, sys.stdout])
print_fps(get_stdout('whoami'), files=[log_file_pointer, sys.stdout])
print_fps('\n2. After', files=[log_file_pointer, sys.stdout])

• 注意：因为Popen的输出以字节为单位，您可能需要进行解码才能删除b'' 。 您可以运行stdout.decode（）将字节解码为utf-8解码的str。*