มีcontextlib.redirect_stdout()
ฟังก์ชั่นใน Python 3.4:
from contextlib import redirect_stdout
with open('help.txt', 'w') as f:
with redirect_stdout(f):
print('it now prints to `help.text`')
มันคล้ายกับ:
import sys
from contextlib import contextmanager
@contextmanager
def redirect_stdout(new_target):
old_target, sys.stdout = sys.stdout, new_target # replace sys.stdout
try:
yield new_target # run some code with the replaced stdout
finally:
sys.stdout = old_target # restore to the previous value
ที่สามารถใช้ได้กับ Python เวอร์ชั่นก่อนหน้า รุ่นหลังไม่สามารถใช้ซ้ำได้ มันสามารถทำได้ถ้าต้องการ
ไม่เปลี่ยนเส้นทาง stdout ที่ระดับ file descriptors เช่น:
import os
from contextlib import redirect_stdout
stdout_fd = sys.stdout.fileno()
with open('output.txt', 'w') as f, redirect_stdout(f):
print('redirected to a file')
os.write(stdout_fd, b'not redirected')
os.system('echo this also is not redirected')
b'not redirected'
และ'echo this also is not redirected'
จะไม่ถูกเปลี่ยนเส้นทางไปยังoutput.txt
ไฟล์
หากต้องการเปลี่ยนเส้นทางที่ระดับตัวอธิบายไฟล์คุณos.dup2()
สามารถใช้:
import os
import sys
from contextlib import contextmanager
def fileno(file_or_fd):
fd = getattr(file_or_fd, 'fileno', lambda: file_or_fd)()
if not isinstance(fd, int):
raise ValueError("Expected a file (`.fileno()`) or a file descriptor")
return fd
@contextmanager
def stdout_redirected(to=os.devnull, stdout=None):
if stdout is None:
stdout = sys.stdout
stdout_fd = fileno(stdout)
# copy stdout_fd before it is overwritten
#NOTE: `copied` is inheritable on Windows when duplicating a standard stream
with os.fdopen(os.dup(stdout_fd), 'wb') as copied:
stdout.flush() # flush library buffers that dup2 knows nothing about
try:
os.dup2(fileno(to), stdout_fd) # $ exec >&to
except ValueError: # filename
with open(to, 'wb') as to_file:
os.dup2(to_file.fileno(), stdout_fd) # $ exec > to
try:
yield stdout # allow code to be run with the redirected stdout
finally:
# restore stdout to its previous value
#NOTE: dup2 makes stdout_fd inheritable unconditionally
stdout.flush()
os.dup2(copied.fileno(), stdout_fd) # $ exec >&copied
ตัวอย่างเดียวกันใช้ได้ทันทีถ้าstdout_redirected()
ใช้แทนredirect_stdout()
:
import os
import sys
stdout_fd = sys.stdout.fileno()
with open('output.txt', 'w') as f, stdout_redirected(f):
print('redirected to a file')
os.write(stdout_fd, b'it is redirected now\n')
os.system('echo this is also redirected')
print('this is goes back to stdout')
เอาต์พุตที่ก่อนหน้านี้พิมพ์บน stdout จะไปoutput.txt
ที่ตราบใดที่stdout_redirected()
ตัวจัดการบริบททำงานอยู่
หมายเหตุ: stdout.flush()
ไม่ต้องล้างข้อมูลบัฟเฟอร์ C stdio บน Python 3 โดยที่ I / O จะถูกนำไปใช้กับการโทรread()
/ write()
ระบบโดยตรง ในการล้างข้อมูลทั้งหมดของสตรีมเอาท์พุท C stdio คุณสามารถโทรได้libc.fflush(None)
อย่างชัดเจนหากส่วนขยาย C บางตัวใช้ I / O แบบอิง stdio:
try:
import ctypes
from ctypes.util import find_library
except ImportError:
libc = None
else:
try:
libc = ctypes.cdll.msvcrt # Windows
except OSError:
libc = ctypes.cdll.LoadLibrary(find_library('c'))
def flush(stream):
try:
libc.fflush(None)
stream.flush()
except (AttributeError, ValueError, IOError):
pass # unsupported
คุณสามารถใช้stdout
พารามิเตอร์เพื่อเปลี่ยนเส้นทางสตรีมอื่น ๆ ไม่เพียง แต่sys.stdout
เพื่อผสานsys.stderr
และsys.stdout
:
def merged_stderr_stdout(): # $ exec 2>&1
return stdout_redirected(to=sys.stdout, stdout=sys.stderr)
ตัวอย่าง:
from __future__ import print_function
import sys
with merged_stderr_stdout():
print('this is printed on stdout')
print('this is also printed on stdout', file=sys.stderr)
หมายเหตุ: stdout_redirected()
ผสม I / O ที่บัฟเฟอร์ ( sys.stdout
โดยปกติ) และ I / O ที่ไม่บัฟเฟอร์(การดำเนินการกับตัวให้คำอธิบายไฟล์โดยตรง) ระวังอาจจะมีบัฟเฟอร์ ปัญหา
หากต้องการตอบการแก้ไขของคุณ: คุณสามารถใช้python-daemon
daemonize สคริปต์ของคุณและใช้logging
โมดูล (ตามที่แนะนำ @ erikb85 ) แทนprint
ข้อความสั่งและเพียงแค่เปลี่ยนเส้นทาง stdout สำหรับสคริปต์ Python ที่รันมานานซึ่งคุณใช้งานnohup
อยู่ตอนนี้
script.p > file