Python目录搜索代码出现StopIteration错误：问题排查与修复

问题根源 #

• 环境变量未被解析：Python无法自动识别字符串中的$HOME，直接将"$HOME"作为路径传入，该路径实际不存在，导致os.walk返回空生成器，调用next()时触发StopIteration异常。
• 缺少路径合法性校验：当传入的路径不存在或不是目录时，os.walk不会抛出明确的路径错误，而是返回空生成器，此时next()因无迭代元素抛出StopIteration，原代码未做任何前置校验。

修复方案 #

1. 正确获取用户主目录：使用os.path.expanduser("~")或Path.home()（需转成字符串），这两种方式可跨平台正确解析用户主目录，替代硬编码的"$HOME"。
2. 添加路径有效性校验：在执行os.walk前，先检查路径是否存在且为目录，提前抛出清晰的错误提示，避免模糊的迭代异常。
3. 异常捕获优化：在主逻辑中添加异常捕获，让程序报错更友好，便于排查问题。

修正后的完整代码 #

import os
from pathlib import Path

def get_files(dir_str):
    # 校验路径是否存在
    if not os.path.exists(dir_str):
        raise ValueError(f"路径不存在：{dir_str}")
    # 校验路径是否为目录
    if not os.path.isdir(dir_str):
        raise ValueError(f"{dir_str} 不是有效目录")
    # 获取当前目录下的文件列表
    onlyfiles = next(os.walk(dir_str))[2] 
    return onlyfiles

# 方式1：用os模块解析用户主目录
input_files_path_str = os.path.expanduser("~")
# 方式2：用Path模块解析（需转成字符串）
# input_files_path_str = str(Path.home())

def main():
    try:
        input_files_list = get_files(input_files_path_str)
        print(input_files_list)
    except ValueError as e:
        print(f"执行错误：{e}")

if __name__=="__main__":
    main()

额外说明 #

• 原代码中导入的subprocess和time模块未被使用，可按需删除以精简代码。
• next(os.walk(dir_str))[2]仅能获取当前目录下的文件，若需遍历子目录的文件，需循环迭代os.walk的结果。

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