Python 装饰器使用场景疑问

[superbai]

背景：一个脚本里有 2 个函数，里面都执行了比较危险的操作(删除数据)，所以我想在执行前加一个确认的动作。打算用装饰器来实现，但是把装饰器加到 2 个函数上之后，执行一个函数需要确认两次...网上搜索了下，装饰器不是随着函数调用执行的。所以这种问题应该用装饰器解决么，还是我使用的姿势不对？ 样例代码：

def confirm(func):
	def wrapper():
    	# do confirm action
        return func
    return wrapper()
    

@confirm
def delete_1():
	# do something


@confirm
def delete_2():
	# do something
    
    
if __name__ == '__main__':
	delete_1()

这种场景下，confirm action 会执行两次

[fzzff]

装饰器最终返回一个函数引用,你的返回值就不对

[Tracy1997]

return wrapper

[Takamine]

感觉用切面做这个感觉不太合适还是在这两个方法前面再抽一层出来做确认功能吧。

[Takamine]

@Takamine 直接把要运行的函数当参数丢进去。

[superbai]

@fzzff #1
@Tracy1997 #2 感谢，已经可以了~~太粗心了

[superbai]

@Takamine #4 那感觉和装饰器一样....？

[frostming]

def confirm(func):
def wrapper():
# do confirm action
return func()
return wrapper

你装饰器的返回搞错了

[sikariba]

def confirm(func):
def wrapper():
user_input = input('yes or no?')
if user_input == 'yes':
return func()
return wrapper


@confirm
def delete_1():
print('delete 1')


@confirm
def delete_2():
print('delete 2')


if __name__ == '__main__':
delete_1()

[Takamine]

@superbai 嗯阿，实现上差不多，只是自己拿业务做个代理，不用注解，呸，装饰器。

[vkhsyj]

装饰器应该返回的是装饰函数,也就是你的 wrapper, 你这样返回 wrapper()的结果, 等于在脚本运行时就会需要确认,而不是在你函数调用时确认了

[superbai]

@Takamine #9 哈哈，我也是总想着注解，写装饰器写起来不太适应😂
@vkhsyj #10 感谢，对装饰器理解不深，搞错了。