Monkey-patchingclass.__str__使用装饰器不起作用
我正在尝试__str__通过以下装饰器修补类的特殊方法：
def str_patcher(klass):
def decorator(*args, **kwargs):
def __str__(self):
items = {key: value for key, value in self.__dict__.items() if not key.startswith("_")}
return f"{type(self).__qualname__}({items})"
klass.__str__ = __str__
return klass
return decorator
然后，我做了一个MyClass装饰的类str_patcher：
@str_patcher
class MyClass:
def __init__(self, **kwargs):
for key, value in kwargs.items():
setattr(self, key, value)
但是当我打印出来时MyClass(a=1, b=2, c=3)，我希望看到，MyClass({a: 1, b: 2, c: 3})但我却看到了<class __main__.MyClass>。
为什么是这样？
回答
你更深一层。以下就足够了：
def str_patcher(klass):
def __str__(self):
items = {key: value for key, value in self.__dict__.items() if not key.startswith("_")}
return f"{type(self).__qualname__}({items})"
klass.__str__ = __str__
return klass
然后：
@str_patcher
class Foo: pass
foo = Foo()
foo.bar = "baz"
print(foo)
# Foo({'bar': 'baz'})
请注意，您使用了参数化装饰器的模式。如果您想为每个装饰类添加动态内容，这将很有用，但您仍然需要对其进行一些更改，例如：
def str_patcher(**kwargs):
def decorator(klass):
def __str__(self):
items = {key: value for key, value in self.__dict__.items() if not key.startswith("_")}
items.update(kwargs)
return f"{type(self).__qualname__}({items})"
klass.__str__ = __str__
return klass
return decorator
@str_patcher(funny="stuff")
class Foo: pass
foo = Foo()
foo.bar = "baz"
print(foo)
# Foo({'bar': 'baz', 'funny': 'stuff'})
THE END
二维码