Chuyển hướng stdout sang không có gì

Tôi có một dự án lớn bao gồm đủ số lượng lớn các mô-đun, mỗi mô-đun in một cái gì đó đến đầu ra tiêu chuẩn. Bây giờ khi dự án đã tăng kích thước, không có lớn. của printbáo cáo in nhiều vào ra std mà đã có những chương trình đáng kể chậm hơn.

Vì vậy, bây giờ tôi muốn quyết định trong thời gian chạy có hay không in bất cứ thứ gì lên thiết bị xuất chuẩn. Tôi không thể thay đổi các mô-đun vì có rất nhiều trong số chúng. (Tôi biết rằng tôi có thể chuyển hướng thiết bị xuất chuẩn đến một tệp nhưng ngay cả điều này cũng khá chậm.)

Vì vậy, câu hỏi của tôi là làm thế nào để tôi chuyển hướng thiết bị xuất chuẩn thành không có nghĩa là làm thế nào để tôi printtuyên bố không làm gì?

# I want to do something like this.
sys.stdout = None         # this obviously will give an error as Nonetype object does not have any write method.

Hiện tại ý tưởng duy nhất tôi có là tạo một lớp có phương thức ghi (không làm gì) và chuyển hướng thiết bị xuất chuẩn sang một thể hiện của lớp này.

class DontPrint(object):
    def write(*args): pass

dp = DontPrint()
sys.stdout = dp

Có một cơ chế sẵn có trong python cho điều này? Hoặc có một cái gì đó tốt hơn thế này?

Đa nền tảng:

import os
import sys
f = open(os.devnull, 'w')
sys.stdout = f

Trên Windows:

f = open('nul', 'w')
sys.stdout = f

Trên Linux:

f = open('/dev/null', 'w')
sys.stdout = f

Một cách hay để làm điều này là tạo một bộ xử lý bối cảnh nhỏ mà bạn bọc các bản in của mình. Sau đó, bạn chỉ cần sử dụng là ở chế độ withđể tắt tất cả đầu ra.

Con trăn 2:

import os
import sys
from contextlib import contextmanager

@contextmanager
def silence_stdout():
    old_target = sys.stdout
    try:
        with open(os.devnull, "w") as new_target:
            sys.stdout = new_target
            yield new_target
    finally:
        sys.stdout = old_target

with silence_stdout():
    print("will not print")

print("this will print")

Python 3,4+:

Python 3.4 có bộ xử lý bối cảnh như tích hợp sẵn này, vì vậy bạn chỉ cần sử dụng bối cảnh như thế này:

import contextlib

with contextlib.redirect_stdout(None):
    print("will not print")

print("this will print")

Chạy mã này chỉ in dòng đầu ra thứ hai, không in dòng đầu tiên:

$ python test.py
this will print

Điều này hoạt động đa nền tảng (Windows + Linux + Mac OSX) và sạch hơn các câu trả lời khác.

Nếu bạn ở python 3,4 trở lên, có một giải pháp đơn giản và an toàn bằng thư viện chuẩn:

import contextlib

with contextlib.redirect_stdout(None):
  print("This won't print!")

(ít nhất là trên hệ thống của tôi) có vẻ như việc viết cho os.devnull nhanh hơn khoảng 5 lần so với việc viết cho lớp DontPrint, tức là

#!/usr/bin/python
import os
import sys
import datetime

ITER = 10000000
def printlots(out, it, st="abcdefghijklmnopqrstuvwxyz1234567890"):
   temp = sys.stdout
   sys.stdout = out
   i = 0
   start_t = datetime.datetime.now()
   while i < it:
      print st
      i = i+1
   end_t = datetime.datetime.now()
   sys.stdout = temp
   print out, "\n   took", end_t - start_t, "for", it, "iterations"

class devnull():
   def write(*args):
      pass


printlots(open(os.devnull, 'wb'), ITER)
printlots(devnull(), ITER)

đã cho đầu ra sau:

<open file '/dev/null', mode 'wb' at 0x7f2b747044b0> 
   took 0:00:02.074853 for 10000000 iterations
<__main__.devnull instance at 0x7f2b746bae18> 
   took 0:00:09.933056 for 10000000 iterations

Nếu bạn đang ở trong môi trường Unix (bao gồm Linux), bạn có thể chuyển hướng đầu ra sang /dev/null:

python myprogram.py > /dev/null

Và đối với Windows:

python myprogram.py > nul

Còn cái này thì sao:

from contextlib import ExitStack, redirect_stdout
import os

with ExitStack() as stack:
    if should_hide_output():
        null_stream = open(os.devnull, "w")
        stack.enter_context(null_stream)
        stack.enter_context(redirect_stdout(null_stream))
    noisy_function()

Này sử dụng các tính năng trong contextlib module để giấu đầu ra của bất cứ lệnh bạn đang cố gắng để chạy, tùy thuộc vào kết quả của should_hide_output(), và sau đó phục hồi hành vi đầu ra sau khi chức năng đó là thực hiện hoạt động.

Nếu bạn muốn ẩn đầu ra lỗi tiêu chuẩn, sau đó nhập redirect_stderrtừ contextlibvà thêm một dòng nói stack.enter_context(redirect_stderr(null_stream)).

Nhược điểm chính là nó chỉ hoạt động trong Python 3.4 và các phiên bản mới hơn.

Lớp của bạn sẽ hoạt động tốt (ngoại trừ write()tên phương thức - nó cần được gọi write(), chữ thường). Chỉ cần chắc chắn rằng bạn lưu một bản sao của sys.stdoutmột biến khác.

Nếu bạn đang sử dụng * NIX, bạn có thể làm được sys.stdout = open('/dev/null'), nhưng điều này ít mang tính di động hơn so với việc lăn lớp của chính bạn.

Bạn chỉ có thể chế giễu nó.

import mock

sys.stdout = mock.MagicMock()

Tại sao bạn không thử điều này?

sys.stdout.close()
sys.stderr.close()

sys.stdout = None

Nó là OK cho print()trường hợp. Nhưng nó có thể gây ra lỗi nếu bạn gọi bất kỳ phương thức nào của sys.stdout, vd sys.stdout.write().

Có một lưu ý trong tài liệu :

Trong một số điều kiện stdin, stdout và stderr cũng như các giá trị ban đầu stdin , stdout và stderr có thể là Không có. Nó thường là trường hợp cho các ứng dụng GUI của Windows không được kết nối với bảng điều khiển và các ứng dụng Python bắt đầu bằng pythonw.

Bổ sung cho câu trả lời của iFreilicht - nó hoạt động cho cả python 2 & 3.

import sys

class NonWritable:
    def write(self, *args, **kwargs):
        pass

class StdoutIgnore:
    def __enter__(self):
        self.stdout_saved = sys.stdout
        sys.stdout = NonWritable()
        return self

    def __exit__(self, *args):
        sys.stdout = self.stdout_saved

with StdoutIgnore():
    print("This won't print!")