如何在函数声明中向 python 函数对象添加自定义属性？-解网

问：

我希望创建一个注释丰富的代码库。我记得 Python 中的函数是对象，因此它们允许任意属性。例如：

def foo(x:int, y:int)->int:
    '''This is a doc string'''
    return x**2 + y**2

foo.__notes__ = {
    'Dependencies':['x', 'y']
}

foo.__notes__

>>> {'Dependencies': ['x', 'y']}

我希望从函数定义中设置属性。关于如何实现这一目标的任何想法？__notes__

编辑：正如评论中指出的那样，这个属性是一个糟糕的名称。更合适。__notes___notes

蟒

@PaulCornelius - 我承认其中一部分是纯粹的好奇心，看看它是否可以做到。我打算用它来支持工具提示、别名、映射数据依赖关系、对缺失值的假设等。我可以为它使用“更大”的数据结构（例如自定义类），但认为尝试新的东西可能会很有趣。我现在在字符串中有一些东西，我通过正则表达式提取它们 - 这是为了取代它。此外，我可以嵌入非字符串对象，例如小熊猫 DataFrames。__doc__

答：

6赞 chepner 12/7/2021 #1

定义函数后，可以使用装饰器附加注释。

def add_note(k, v):
    def _(f):
        if not hasattr(f, '_notes'):
            f._notes = {}
        f._notes[k] = v
        return f
    return _


@add_note('Dependencies', ['x', 'y'])
def foo(x: int, y: int) -> int:
    return x**2 + y**2

顺便说一句，dunder 名称（like ）是为实现保留的;你不应该发明你自己的。只需使用一个普通的“私人”名称，前缀为一个 .__notes___


def adder1(x:int = 1, y: int = 2, z: int =3 )->int:
    '''This is a doc string'''
    return x**2 + y**2

def decorate(**kwds):
    """ assign kwds as attributes to the function object """
    def actual_decorator(func):
        for k, v in kwds.items():
            setattr(func, k, v)
        return func
    return actual_decorator

depends = {"x","y"}
@decorate(_note=f"depends:{depends}", depends=depends)
def adder2(x:int = 1, y: int = 2, z: int =3 )->int:
    # a function can totally refer to its name 
    # and use annotations to dynamically adjust its behavior
    depends = adder2.depends
    #sum up dependencies
    li = []
    for key in depends:
        v = locals()[key] ** 2
        li.append(v)

    return sum(li)        

def adder3(x:int = 1, y: int = 2, z: int =3 )->int:
    """ no note until function execution so this won't allow tools to inspect the module"""
    _note = getattr(adder3,"_note",None)
    if not _note:
        adder3._note = "depends x,y"
    return x**2 + y**2

def adder4(x:int = 1, y: int = 2, z: int =3 )->int:
    """there is no guarantee that referring by its own name works"""
    print("\n\nadder4.name:", adder4.__name__)
    return (x**2 + y**2) * 1000

adder4_saved = adder4
#swap name
adder4 = adder2 

for func in [adder1,adder2,adder3,adder4_saved]:
    print(f"\n{func.__name__} [note(before)={getattr(func,'_note',None)}] -> {func()} [note(after)={getattr(func,'_note',None)}]")

输出：


adder1 [note(before)=None] -> 5 [note(after)=None]

adder2 [note(before)=depends:{'y', 'x'}] -> 5 [note(after)=depends:{'y', 'x'}]
adder3 [note(before)=None] -> 5 [note(after)=depends x,y]


adder4.name: adder2

adder4 [note(before)=None] -> 5000 [note(after)=None]

上一个：有没有办法将 Azure 容器应用内存限制提高到 4GB 以上？

下一个：如何安排一些 python 脚本在 Windows PC 上定期运行？

如何在函数声明中向 python 函数对象添加自定义属性？

How can you add custom attributes to a python function object within the function declaration?

评论

评论

输出：