UnboundLocalError排查：file_format未赋值报错及代码优化建议

错误核心原因 #

file_format变量在load_files函数中存在未被所有代码路径赋值的情况。比如函数内通过if/elif分支为变量赋值，但当所有分支条件都不满足时，变量未被初始化，后续代码直接引用它就会触发UnboundLocalError。

举个典型的错误代码场景：

def load_files(file_path):
    if file_path.endswith('.csv'):
        file_format = 'csv'
    elif file_path.endswith('.parquet'):
        file_format = 'parquet'
    # 若文件后缀既不是csv也不是parquet，file_format未被赋值
    parse_file(file_format)  # 此处触发错误

代码优化建议 #

1. 给变量设置初始默认值 #

在函数开头先为file_format赋值（比如None），确保所有代码路径下变量都有定义，后续可通过判断值是否合法来处理异常情况：

def load_files(file_path):
    file_format = None  # 初始化默认值
    if file_path.endswith('.csv'):
        file_format = 'csv'
    elif file_path.endswith('.parquet'):
        file_format = 'parquet'
    
    if not file_format:
        raise ValueError(f"无法识别文件格式：{file_path}")
    parse_file(file_format)

2. 完善条件分支，覆盖所有场景 #

添加else分支处理不支持的文件格式，主动抛出明确的异常，避免变量未赋值：

def load_files(file_path):
    if file_path.endswith('.csv'):
        file_format = 'csv'
    elif file_path.endswith('.parquet'):
        file_format = 'parquet'
    else:
        raise ValueError(f"不支持的文件类型：{file_path}，仅支持csv/parquet格式")
    parse_file(file_format)

3. 提前校验输入合法性 #

在调用load_files函数前，先校验文件后缀是否符合要求，从源头避免进入无赋值的分支：

# 主程序 run_presc_pipeline.py 中
allowed_formats = ('.csv', '.parquet')
file_path = "xxx/xxx.xxx"
if not file_path.endswith(allowed_formats):
    raise ValueError(f"文件格式不合法，仅支持{allowed_formats}")
load_files(file_path)

4. 增加异常捕获机制 #

在调用load_files的地方添加异常捕获，便于快速定位问题并给出友好提示：

# 主程序 run_presc_pipeline.py 中
try:
    load_files(file_path)
except UnboundLocalError as e:
    print(f"加载文件出错：{e}，请检查文件格式是否支持")
except ValueError as e:
    print(f"加载文件出错：{e}")

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