Python 闭包使用中需要注意的地方
admin
2023-07-30 20:46:41
0

昨天正当我用十成一阳指功力戳键盘、昏天暗地coding的时候,正好被人问了一个问题,差点没收好功,洪荒之力侧漏震伤桌边的人,废话不多说,先上栗子(精简版,只为说明问题):

from functools import wraps
from time import sleep

def retry(attempts=3, wait=2):
    if attempts < 0 or attempts > 5:
        retry_times = 3
    else:
        retry_times = attempts
    if wait < 0 or wait > 5:
        retry_wait = 2
    else:
        retry_wait = after
    def retry_decorator(func):
        @wraps(func)
        def wrapped_function(*args, **kwargs):
            while retry_times > 0:
                try:
                    return func(*args, **kwargs)
                except :
                    sleep(retry_wait)
                    retry_times -= 1
        return wrapped_function
    return retry_decorator

简易版的retry装饰器,需要的变量被闭包完美捕捉,逻辑也挺简单明了。问的人说逻辑看着挺正常的,但就是一直报变量retry_times找不到(unresolved reference)的错误提示。这是为什么呢,相信看过撸主上篇文章正确使用Python可选参数(标题写的有点大有点扯,见谅)的童鞋应该很快能反应过来。没错仔细捋一下,这是一道送分题呢:闭包捕获的变量(retry_times,retry_wait)相当时引用的retry函数的局部变量,当在wrapped_function的局部作用于里面操作不可变类型的数据时,会生成新的局部变量,但是新生成的局部变量retry_times在使用时还没来得及初始化,因此会提示找不到变量;retry_wait相反能被好好的使用到。

python是duck-typing的编程语言,就算有warning照样跑,写个简单到极限的的函数,用一下装饰器,在wrapped_function逻辑里打个断点看一下各个变量的值也是很快能找到问题的(直接跑也能看到错误:UnboundLocalError: local variable \'retry_attempts\' referenced before assignment, 至少比warning msg有用):

@retry(7, 8)
def test():
    print 23333
    raise Exception(\'Call me exception 2333.\')

if __name__ == \'__main__\':
    test()

output: UnboundLocalError: local variable \'retry_times\' referenced before assignment

要解决这种问题也好办,用一个可变的容器把要用的不可变类型的数据包装一下就行了(说个好久没写C#代码记不太清楚完全不负责任的题外话,就像在C#.net里面,碰到闭包的时候,会自动生成一个混淆过名字的类然后把要被捕捉的值当作类的属性存着,这样在使用的时候就能轻松get,著名的老赵好像有一篇文章讲Lazy Evaluation的好像涉及到这个话题):

def retry(attempts=3, wait=2):
    temp_dict = {
        \'retry_times\': 3 if attempts < 0 or attempts > 5 else attempts,
        \'retry_wait\': 2 if wait < 0 or wait > 5 else wait
    }

    def retry_decorate(fn):
        @wraps(fn)
        def wrapped_function(*args, **kwargs):
            print id(temp_dict), temp_dict
            while temp_dict.get(\'retry_times\') > 0:
                try:
                    return fn(*args, **kwargs)
                except :
                    sleep(temp_dict.get(\'retry_wait\'))
                    temp_dict[\'retry_times\'] = temp_dict.get(\'retry_times\') - 1
                print id(temp_dict), temp_dict

        print id(temp_dict), temp_dict

        return wrapped_function

    return retry_decorate

@retry(7, 8)
def test():
    print 23333
    raise Exception(\'Call me exception 2333.\')

if __name__ == \'__main__\':
    test()

#output:
4405472064 {\'retry_wait\': 2, \'retry_times\': 3}
4405472064 {\'retry_wait\': 2, \'retry_times\': 3}
23333
4405472064 {\'retry_wait\': 2, \'retry_times\': 2}
23333
4405472064 {\'retry_wait\': 2, \'retry_times\': 1}
23333
4405472064 {\'retry_wait\': 2, \'retry_times\': 0}

从output中可以看到,用dict包装后,程序能够正常的工作,和预期的一致,其实我们也可以从函数的闭包的值再次确认:

>>> test.func_closure[1].cell_contents
{\'retry_wait\': 2, \'retry_times\': 2}

我是结尾,PEACE!

参考资料:

http://stackoverflow.com/questions/986006/how-do-i-pass-a-variable-by-reference

https://docs.python.org/2/faq/programming.html#how-do-i-write-a-function-with-output-parameters-call-by-reference

相关内容

热门资讯

Mobi、epub格式电子书如... 在wps里全局设置里有一个文件关联,打开,勾选电子书文件选项就可以了。
定时清理删除C:\Progra... C:\Program Files (x86)下面很多scoped_dir开头的文件夹 写个批处理 定...
500 行 Python 代码... 语法分析器描述了一个句子的语法结构,用来帮助其他的应用进行推理。自然语言引入了很多意外的歧义,以我们...
scoped_dir32_70... 一台虚拟机C盘总是莫名奇妙的空间用完,导致很多软件没法再运行。经过仔细检查发现是C:\Program...
65536是2的几次方 计算2... 65536是2的16次方:65536=2⁶ 65536是256的2次方:65536=256 6553...
小程序支付时提示:appid和... [Q]小程序支付时提示:appid和mch_id不匹配 [A]小程序和微信支付没有进行关联,访问“小...
pycparser 是一个用... `pycparser` 是一个用 Python 编写的 C 语言解析器。它可以用来解析 C 代码并构...
微信小程序使用slider实现... 众所周知哈,微信小程序里面的音频播放是没有进度条的,但最近有个项目呢,客户要求音频要有进度条控制,所...
Apache Doris 2.... 亲爱的社区小伙伴们,我们很高兴地向大家宣布,Apache Doris 2.0.0 版本已于...
python清除字符串里非数字... 本文实例讲述了python清除字符串里非数字字符的方法。分享给大家供大家参考。具体如下: impor...