主文件调用模块中DEF函数内IF条件定义的变量提示未定义怎么办？

问题原因 #

• 模块中用global声明的变量属于模块级全局变量，主文件直接写browser无法直接访问，必须通过模块名引用
• 若proxy_type不等于"type1"，browser变量根本不会被定义，必然触发NameError

解决方案 #

方案1：让函数返回浏览器实例（推荐）

这种方式更符合Python函数的设计逻辑，避免依赖全局变量，代码更清晰可控。

修改模块文件代码：

def set_proxy():
    if proxy_type == "type1":
        return webdriver.Chrome(CHROME_DIR)
    # 可根据需求添加其他代理类型的处理，或返回None
    return None

主文件代码：

from modules.proxy import set_proxy
browser = set_proxy()
# 先判断浏览器实例是否有效，避免调用get()时报错
if browser:
    browser.get("website.com")

方案2：正确引用模块的全局变量

如果一定要使用全局变量，需通过模块名访问，同时提前初始化变量避免条件不满足时变量不存在。

修改模块文件代码：

# 提前初始化变量，防止条件不满足时变量未定义
browser = None

def set_proxy():
    global browser
    if proxy_type == "type1":
        browser = webdriver.Chrome(CHROME_DIR)

主文件代码：

import modules.proxy
modules.proxy.set_proxy()
# 先确认变量已被正确初始化
if modules.proxy.browser:
    modules.proxy.browser.get("website.com")

方案3：将模块全局变量引入主命名空间

可以在主文件中明确导入模块的browser变量，但需注意这种方式导入的是变量引用，模块内变量重新赋值后主文件的引用不会自动更新，仅适用于简单场景。

主文件代码：

from modules.proxy import set_proxy, browser
set_proxy()
if browser:
    browser.get("website.com")

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