如何在 "类 "对象中实现 "装饰器 "功能？

Question

我想建立一个抑制stdout和stderr的类。 我把它作为一个 with statement 但我想扩展这个功能，使其也可以作为一个装饰器，用来抑制函数的输出。 是否可以将所有的东西完全包含在类内，还是包装器需要是类外的函数？

我试图按照这些资源来做，但很难适应我的情况。

https:/stackabuse.pythons classmethod andstaticmethod-explained

如何以类的形式实现带参数的Python装饰器？

import os,sys, functools


class Suppress(object):
    def __init__(self, show_stdout=False, show_stderr=False):
        self.show_stdout = show_stdout
        self.show_stderr = show_stderr
        self.original_stdout = None
        self.original_stderr = None

    def __enter__(self):
        devnull = open(os.devnull, "w")

        # Suppress streams
        if not self.show_stdout:
            self.original_stdout = sys.stdout
            sys.stdout = devnull

        if not self.show_stderr:
            self.original_stderr = sys.stderr
            sys.stderr = devnull

    def __exit__(self, *args, **kwargs):
        # Restore streams
        if not self.show_stdout:
            sys.stdout = self.original_stdout

        if not self.show_stderr:
            sys.stderr = self.original_stderr


    def __call__(self, *args, **kwargs):
        def decorator(func):
            @functools.wraps(func)
            def wrapper(*args, **kwargs):
                with self(*args, **kwargs):
                    return func(*args, **kwargs)
            return wrapper
         return decorator



with Suppress(show_stdout=False, show_stderr=False):
    print("stdout", file=sys.stdout)
    print("stderr", file=sys.stderr)

我正试图将这个功能也作为一个装饰器。

@Suppress(show_stdout=True)
def f(x, y):
    print(x, file=sys.stdout)
    print(y, file=sys.stderr)
    return x*y
a = f(1,2)
# 1
# a = 2

我曾有一个类似的装饰器在工作 (尽管非常丑陋) ，用于一个 matplotlib 风格包装器。 然而，这必须使用一个外部函数，而不是在一个类中。

# # Decorators
# def stylize(style="seaborn-white"):
#     def decorator(func):
#         @functools.wraps(func)
#         def wrapper(*args, **kwargs):
#             with plt.style.context(style):
#                 return func(*args, **kwargs)
#         return wrapper
#     return decorator
# # Wrappers
# def subplots_wrapper(style="seaborn-white", *args, **kwargs):
#     @stylize(style)
#     def inner_wrapper(*args, **kwargs):
#         return plt.subplots(*args, **kwargs)
#     return inner_wrapper(*args, **kwargs)

Answer 1

如果你使用简单的 with self 无参数。

   def __call__(self, function):

        @functools.wraps(function)
        def decorated(*args, **kwargs):
            with self:
                return function(*args, **kwargs)
        return decorated

Answer 2

你可以通过组合不同的片段来定义它 contextlib. 简而言之，这使用 ExitStack 将请求的重定向上下文管理器组成一个单一的上下文管理器，并且 ContextDecorator 来使用产生的上下文管理器作为装饰器。

from contextlib import ExitStack, ContextDecorator, redirect_stdout, redirect_stderr
import sys
from os import devnull


class Suppress(ContextDecorator, ExitStack):
    def __init__(self, show_stdout=False, show_stderr=False, **kwargs):
        super().__init__(**kwargs)
        self.redirections = []
        if show_stdout:
            self.redirections.append(redirect_stdout)
        if show_stderr:
            self.redirections.append(redirect_stderr)

    def __enter__(self):
        rv = super().__enter__()

        if self.redirections:
            f = rv.enter_context(open(devnull, "w"))
            for r in self.redirections:
                rv.enter_context(r(f))
        return rv

__init__ 存储类 redirect_stdout 和 redirect_stderr 的，将在列表中进行必要的。

__enter__，如果请求了任何重定向，则打开相应的汇，并进行相应的重定向。作为一个退出栈。Suppress 会注意确保重定向结束，并将汇。