Python 2.7字典推导无法访问内部函数strip_dirt的原因咨询

问题现象 #

你在Python 2.7.12中碰到了一个挺坑的作用域问题：在外部函数validate_available_links内定义了嵌套函数strip_dirt，在字典推导式里调用这个嵌套函数时触发NameError: global name 'strip_dirt' is not defined，但在函数其他位置（比如pdb调试时直接调用）却能正常访问它。

你的代码片段：

def validate_available_links(self, link_dict, sub_element=None):
    def strip_dirt(key_val):
        #TODO: properly strip HTML encoded characters
        key_val = key_val.replace("\u2039", "<")
        key_val = key_val.replace("\u203a", ">")
        return key_val
    # Scrape all the links from the current webpage and verify
    # it with the provided link dictionary.
    if sub_element is None:
        act_links = self.suiteContext['browser'].webdriver.find_elements_by_tag_name('a')
    else:
        act_links = sub_element.find_elements_by_tag_name('a')
    result = {strip_dirt(lnk.text): lnk.get_attribute('href') for lnk in act_links}
    diff_keys = set(result.keys()) - set(link_dict.keys())
    diff_values = set(result.values()) - set(link_dict.values())
    self.tear_down_hook()
    for l_text, l_url in link_dict.iteritems():
        self.cfg.logger.info("[VALIDATION] Verify Link text [{}] and URL [{}]."
        .format(l_text, l_url))
    import pdb; pdb.set_trace()

调试时的异常场景：

(Pdb) result = {strip_dirt(lnk.text): lnk.get_attribute('href') for lnk in act_links}
*** NameError: global name 'strip_dirt' is not defined
(Pdb) strip_dirt
<function strip_dirt at 0x0651BBB0>
(Pdb) strip_dirt('x')
'x'
(Pdb) {strip_dirt(lnk.text): lnk.get_attribute('href') for lnk in act_links}
*** NameError: global name 'strip_dirt' is not defined

原因分析 #

这是Python 2.x版本中推导式（包括字典推导、列表推导、生成器表达式）的作用域设计缺陷：

• 在Python 2里，推导式会创建一个独立的作用域，但这个作用域不会继承当前外部函数的本地作用域，而是直接去全局作用域查找变量。
• 而strip_dirt是定义在validate_available_links内部的嵌套函数，属于外部函数的本地作用域，推导式找不到它，就会抛出“全局名称未定义”的错误。
• 这个问题在Python 3中已经被修复，Python 3的推导式会正确继承外部函数的本地作用域，不会出现这个报错。

解决方案 #

针对Python 2.7的环境，你可以用以下两种方式解决：

1. 用普通循环替代字典推导式

把字典推导改成显式的for循环构建字典，这样就能正常访问嵌套函数，可读性也更高：

result = {}
for lnk in act_links:
    cleaned_text = strip_dirt(lnk.text)
    result[cleaned_text] = lnk.get_attribute('href')

2. 将嵌套函数赋值给本地变量（可选）

虽然不如循环直观，但你也可以先把strip_dirt赋值给一个本地变量，让推导式能找到它：

strip_func = strip_dirt
result = {strip_func(lnk.text): lnk.get_attribute('href') for lnk in act_links}

内容的提问来源于stack exchange，提问作者Chuk Ultima