wbzxz 发表于 2024-7-12 14:21:15

我想在装饰器函数中写个参数进去,但是没有成功

def outer(inner_fun):
    def call_func():
      print('这是装饰函数的开始')
      inner_fun()
      print('这是装饰函数的结束')
    return call_func


def my_fun(s):
    print(f'{s}--这是实际函数执行的内容')


a = outer(my_fun)
a(1)

我想在装饰器函数中,写一个参数进去,但是没有成功,报错啦,应该怎么写才对呢,谢谢各位大佬。

某一个“天” 发表于 2024-7-12 15:12:18

我想到的一种办法
def outer(inner_fun, sth):
    def call_func():
      print('这是装饰函数的开始')
      inner_fun(sth)
      print('这是装饰函数的结束')
    return call_func


def my_fun(s):
    print(f'{s}--这是实际函数执行的内容')


a = outer(my_fun, 1)
a()

某一个“天” 发表于 2024-7-12 15:23:49

当然也可以用语法糖
def xxx(sth):
    def outer(inner_fun):
      def call_func():
            print('这是装饰函数的开始')
            inner_fun(sth)
            print('这是装饰函数的结束')
      return call_func
    return outer

@xxx(sth = 1)
def my_fun(s):
    print(f'{s}--这是实际函数执行的内容')


my_fun()

smallwh 发表于 2024-7-12 16:11:18

outer(my_fun)实际上返回的是call_func
只需向call_func函数传参,就可以实现参数传入inner_fun【在本例中即my_fun】
def outer(inner_fun):
    def call_func(s):
      print('这是装饰函数的开始')
      inner_fun(s)
      print('这是装饰函数的结束')
    return call_func


def my_fun(s):
    print(f'{s}--这是实际函数执行的内容')


a = outer(my_fun)
a(1)

wbzxz 发表于 2024-7-12 19:46:08

smallwh 发表于 2024-7-12 16:11
outer(my_fun)实际上返回的是call_func
只需向call_func函数传参,就可以实现参数传入inner_fun【在本例中 ...

哦,那我想直接把函数传给inner_fun是无法实现的,只有通过call_func函数进行中转? 对吗?

zxbin1 发表于 2024-7-12 20:22:15

a(1)
a是outer函数的返回值
也是嵌套函数outer中的call_func
a == call_func
参数应该写在call_func之中
为了防止参数个数不同
call_func(*b)可变参数打包
inner_fun(*b)对多个参数解包引用给my_fun(n,m)

def outer(inner_fun):
    def call_func(*b):
      print('这是装饰函数的开始')
      inner_fun(*b)
      print('这是装饰函数的结束')
    return call_func


def my_fun(n,m):
    print(f'{n},{m}--这是实际函数执行的内容-my_fun')

def he_fun(s):
    print(f'{s}--这是实际函数执行的内容-he_fun')

a = outer(my_fun)
b = outer(he_fun)
a(1,2)
a(2,3)
b(4)
b(5)

print("-----------------------------------")
#错误,没有参数s
b()

上方是演示内容
你要的

def outer(inner_fun):
    def call_func(*b):
      print('这是装饰函数的开始')
      inner_fun(*b)
      print('这是装饰函数的结束')
    return call_func


def my_fun(s):
    print(f'{s}--这是实际函数执行的内容')

a = outer(my_fun)
a(1)

smallwh 发表于 2024-7-12 20:49:27

wbzxz 发表于 2024-7-12 19:46
哦,那我想直接把函数传给inner_fun是无法实现的,只有通过call_func函数进行中转? 对吗?

对。
楼上已经给出了可变参数的处理方式,可以适应参数个数不同的情况
下面的形式还能适应关键字参数。基本上就可以适应所有的函数了
def outer(inner_fun):
    def call_func(*args,**kwargs):
      print('这是装饰函数的开始')
      inner_fun(*args,**kwargs)
      print('这是装饰函数的结束')
    return call_func

wbzxz 发表于 2024-7-12 20:50:24

smallwh 发表于 2024-7-12 20:49
对。
楼上已经给出了可变参数的处理方式,可以适应参数个数不同的情况
下面的形式还能适应关键字参数。 ...

非常感谢您的解答,谢谢您。
页: [1]
查看完整版本: 我想在装饰器函数中写个参数进去,但是没有成功