[已解决]含参函数如何使用装饰器

九歌当下

我先定义一个闭包

def logger(func):
    def call_func():
        print("----------")
        func()
        print("----------")
    return call_func

再给一个含参函数加上该闭包的装饰器

@logger
def sayhi(name):
    print("hi,{}".format(name))

然后函数sayhi()就没法用了

sayhi("小甲鱼")
Traceback (most recent call last):
  File "<pyshell#38>", line 1, in <module>
    sayhi("小甲鱼")
TypeError: logger.<locals>.call_func() takes 0 positional arguments but 1 was given

求帮助

最佳答案

月排行榜 / 总排行榜

isdkz

2023-3-5 14:10:07

一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而你给 sayhi 传参会报错，

所以就是要让 call_func 接收跟 sayhi 的参数个数一样的形参，并在 call_func 调用 sayhi 的时候把参数传递给 sayhi

故对你的代码修改如下：

def logger(func):
    def call_func(name):                         # 改了这里
        print("----------")
        func(name)                         # 改了这里
        print("----------")
    return call_func

@logger
def sayhi(name):
    print("hi,{}".format(name))

sayhi("小甲鱼")

跳转到最佳答案楼层

isdkz

一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而你给 sayhi 传参会报错，

所以就是要让 call_func 接收跟 sayhi 的参数个数一样的形参，并在 call_func 调用 sayhi 的时候把参数传递给 sayhi

故对你的代码修改如下：

def logger(func):
    def call_func(name):                         # 改了这里
        print("----------")
        func(name)                         # 改了这里
        print("----------")
    return call_func

@logger
def sayhi(name):
    print("hi,{}".format(name))

sayhi("小甲鱼")

九歌当下

isdkz 发表于 2023-3-5 14:10
一个函数在使用装饰器后会被替换成装饰器的返回值，

而装饰器的返回值 call_func 不接受任何参数，故而 ...

OK， 谢谢了

歌者文明清理员

def logger(func):
    def call_func(**args, **kwargs):                         # 改了这里
        print("----------")
        func(**args, **kwargs)                         # 改了这里
        print("----------")
    return call_func

@logger
def sayhi(name):
    print("hi,{}".format(name))

sayhi("小甲鱼")

这样就兼容所有函数了

九歌当下

歌者文明清理员 发表于 2023-3-5 15:05
这样就兼容所有函数了

真是太厉害了，没有想到收集参数能这么用

歌者文明清理员

九歌当下 发表于 2023-3-5 16:33
真是太厉害了，没有想到收集参数能这么用

可惜了我的最佳答案

九歌当下

歌者文明清理员 发表于 2023-3-5 16:40
可惜了我的最佳答案

下次看了得晚点在设最佳答案