Question

我有一个函数和类装饰器，将它们更改为单例，但文档字符串有点丢弃。转换函数返回一个具有自己的docstring的类对象，该对象应该被覆盖。我该如何解决这个问题？

def singleton(cls):
    return cls()

def singletonfunction(func):
    return func()

@singletonfunction
def Bit():
    """A 1-bit BitField; must be enclosed in a BitStruct"""
    return BitsInteger(1)

检查Bit会产生BitInteger文档字符串，而不是函数的文档字符串。

Answer 1

让我们假设原始版本是这样的：

class BitsInteger:
    """BitsInteger docstring"""
    def __init__(self, num):
        pass

def singleton(cls):
    return cls()

def singletonfunction(func):
    return func()

@singletonfunction
def Bit():
    """A 1-bit BitField; must be enclosed in a BitStruct"""
    return BitsInteger(1)

b = Bit
print("b.__doc__ :", b.__doc__)

使用Python3.5运行它会得到输出：

b.__doc_ : BitsInteger docstring

这可能不是您所期望的。当我们使用python -i original.py时，我们可以看看这里真正发生的事情：

>>> vars()
{'__name__': '__main__', '__builtins__': <module 'builtins' (built-in)>, 'b': <__main__.BitsInteger object at 0x7ff05d2ae3c8>, '__spec__': None, 'singletonfunction': <function singletonfunction at 0x7ff05d2b40d0>, 'singleton': <function singleton at 0x7ff05d30cd08>, '__cached__': None, 'BitsInteger': <class '__main__.BitsInteger'>, '__loader__': <_frozen_importlib.SourceFileLoader object at 0x7ff05d2eb4a8>, '__package__': None, 'Bit': <__main__.BitsInteger object at 0x7ff05d2ae3c8>, '__doc__': None}
>>>

正如您所看到的，b实际上属于BitsInteger

类型

原因是因为你写的时候真正发生的是：

@singleton_function
def Bit():
    """Bit docstring"""
    ...
    return BitsInteger(1)

你真的只是为此获得语法糖：

def Bit():
    """Bit docstring"""
    ...
    return BitsInteger(1)
Bit = singleton_function(Bit)

在这种情况下，Bit是singleton_function的返回值，实际上是BitsInteger。我发现当你剥离语法糖时，它会更清楚这里发生了什么。

如果您希望装饰者不会改变文档字符串，我建议您使用wrapt

import wrapt

class BitsInteger:
    """BitsInteger docstring"""
    def __init__(self, num):
        pass

def singleton(cls):
    return cls()

@wrapt.decorator
def singletonfunction(func):
    return func()

@singletonfunction
def Bit():
    """A 1-bit BitField; must be enclosed in a BitStruct"""
    return BitsInteger(1)

b = Bit
print("b.__doc_ :", b.__doc__)

输出：

b.__doc_ : A 1-bit BitField; must be enclosed in a BitStruct

现在，当您查看Vars()时，您会看到'b': <FunctionWrapper at 0x7f9a9ac42ba8 for function at 0x7f9a9a9e8c80>不再是BitInteger类型。我个人喜欢使用包装，因为我得到了我想要的东西。实现该功能并覆盖所有边缘情况需要付出努力，我知道包装已经过充分测试并且按预期工作。

以编程方式更改类对象的docstring

1 个答案: