背景:一个脚本里有 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 会执行两次
1
fzzff 2019-08-06 17:15:33 +08:00 1
装饰器最终返回一个函数引用,你的返回值就不对
|
2
Tracy1997 2019-08-06 17:20:59 +08:00 1
return wrapper
|
3
Takamine 2019-08-06 17:23:19 +08:00
感觉用切面做这个感觉不太合适还是在这两个方法前面再抽一层出来做确认功能吧。
|
7
frostming 2019-08-06 17:32:16 +08:00 1
def confirm(func):
def wrapper(): # do confirm action return func() return wrapper 你装饰器的返回搞错了 |
8
sikariba 2019-08-06 17:34:27 +08:00 1
|
10
vkhsyj 2019-08-06 18:06:11 +08:00 1
装饰器应该返回的是装饰函数,也就是你的 wrapper, 你这样返回 wrapper()的结果, 等于在脚本运行时就会需要确认,而不是在你函数调用时确认了
|