Python 装饰器使用场景疑问

2019-08-06 17:06:49 +08:00
 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 会执行两次

2002 次点击
所在节点    Python
11 条回复
fzzff
2019-08-06 17:15:33 +08:00
装饰器最终返回一个函数引用,你的返回值就不对
Tracy1997
2019-08-06 17:20:59 +08:00
return wrapper
Takamine
2019-08-06 17:23:19 +08:00
感觉用切面做这个感觉不太合适还是在这两个方法前面再抽一层出来做确认功能吧。
Takamine
2019-08-06 17:27:13 +08:00
@Takamine 直接把要运行的函数当参数丢进去。
superbai
2019-08-06 17:30:48 +08:00
@fzzff #1
@Tracy1997 #2 感谢,已经可以了~~太粗心了
superbai
2019-08-06 17:31:02 +08:00
@Takamine #4 那感觉和装饰器一样....?
frostming
2019-08-06 17:32:16 +08:00
def confirm(func):
def wrapper():
# do confirm action
return func()
return wrapper

你装饰器的返回搞错了
sikariba
2019-08-06 17:34:27 +08:00
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
2019-08-06 17:43:11 +08:00
@superbai 嗯阿,实现上差不多,只是自己拿业务做个代理,不用注解,呸,装饰器。
vkhsyj
2019-08-06 18:06:11 +08:00
装饰器应该返回的是装饰函数,也就是你的 wrapper, 你这样返回 wrapper()的结果, 等于在脚本运行时就会需要确认,而不是在你函数调用时确认了
superbai
2019-08-07 12:58:22 +08:00
@Takamine #9 哈哈,我也是总想着注解,写装饰器写起来不太适应😂
@vkhsyj #10 感谢,对装饰器理解不深,搞错了。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/589561

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX