python: change sys.stdout print to custom print function

Question

Im пытается понять, как создать пользовательскую функцию печати. (с использованием Python 2.7)

import sys 
class CustomPrint(): 
    def __init__(self): 
     self.old_stdout=sys.stdout #save stdout 

    def write(self, text): 
     sys.stdout = self.old_stdout #restore normal stdout and print 
     print 'custom Print--->' + text 
     sys.stdout= self # make stdout use CustomPrint on next 'print' 
         # this is the line that trigers the problem 
         # how to avoid this?? 


myPrint = CustomPrint() 
sys.stdout = myPrint 
print 'why you make 2 lines??...' 

Код выше принтами это утешать:

>>> 
custom Print--->why you make 2 lines??... 
custom Print---> 

>>> 

и я хочу, чтобы напечатать только одну строку:

>>>  
1custom Print--->why you make 2 lines??... 
>>> 

Но не могу понять, как сделайте эту обычную работу печати, я понимаю, что есть какая-то рекурсия, которая запускает второй вывод на консоль (я использую self.write, чтобы назначить stdout для self.write!)

Как я могу сделать эту работу? или мой подход просто совершенно неправильный ...

Answer 1

Это не рекурсия. Случается, что ваша функция write вызывается дважды, один раз с текстом, который вы ожидаете, второй раз с '\n'. Попробуйте это:

import sys 
class CustomPrint(): 
    def __init__(self): 
     self.old_stdout=sys.stdout 

    def write(self, text): 
     text = text.rstrip() 
     if len(text) == 0: return 
     self.old_stdout.write('custom Print--->' + text + '\n') 

Что я делаю в приведенном выше коде, я добавить символ новой строки к тексту принятого в первом вызове, и убедитесь, что на второй вызов, сделанное заявление печати, один предназначен для печати новая строка, ничего не печатает.

Теперь попробуйте закомментировать первые две строки и посмотреть, что происходит:

def write(self, text): 
     #text = text.rstrip() 
     #if len(text) == 0: return 
     self.old_stdout.write('custom Print--->' + text + '\n') 

Answer 2

Как насчет того, чтобы делать from __future__ import print_function. Таким образом вы будете использовать функцию печати Python3 вместо инструкции print из Python2. После этого вы можете переопределить функцию печати:

def print(*args, **kwargs): 
    __builtins__.print("Custom--->", *args, **kwargs) 

Существует поймать, однако, вы должны будете начать использовать функцию печати.

Answer 3

Одним из решений может быть использовать менеджер контекста, если он локализован.

#!/usr/bin/env python 
from contextlib import contextmanager 
################################################################################ 
@contextmanager 
def no_stdout(): 
    import sys 
    old_stdout = sys.stdout 
    class CustomPrint(): 
     def __init__(self, stdout): 
      self.old_stdout = stdout 

     def write(self, text): 
      if len(text.rstrip()): 
       self.old_stdout.write('custom Print--->'+ text) 

    sys.stdout = CustomPrint(old_stdout) 

    try: 
     yield 
    finally: 
     sys.stdout = old_stdout 
################################################################################ 
print "BEFORE" 
with no_stdout(): 
    print "WHY HELLO!\n" 
    print "DING DONG!\n" 

print "AFTER" 

выше производит:

BEFORE 
custom Print--->WHY HELLO! 
custom Print--->DING DONG! 
AFTER 

код нужно будет прибирать особ. вокруг того, что класс должен делать WRT, устанавливая stdout обратно в то, что было.