如何在Python中遍历同目录下的链式文件？

要处理这种链式文件遍历，核心是跟踪已访问的文件，避免循环导致无限执行。以下是具体实现思路和代码：

核心步骤 #

• 从指定的起始文件开始（比如A.txt）
• 每次读取当前文件的内容，获取下一个文件名
• 检查下一个文件是否已访问过：如果是，说明循环闭合，终止遍历；如果不是，继续
• 处理文件读取时的异常（比如文件不存在、内容为空等）

示例代码 #

def traverse_chain(start_file):
    current_file = start_file
    visited = set()
    
    while current_file not in visited:
        visited.add(current_file)
        print(f"访问文件: {current_file}")
        
        # 读取文件内容获取下一个文件名
        try:
            with open(current_file, 'r') as f:
                # 读取并去除换行符/空白字符
                next_file = f.read().strip()
        except FileNotFoundError:
            print(f"错误：文件 {current_file} 不存在")
            break
        except Exception as e:
            print(f"读取 {current_file} 时出错: {e}")
            break
        
        # 检查下一个文件是否为空
        if not next_file:
            print(f"文件 {current_file} 内容为空，链式遍历终止")
            break
        
        current_file = next_file
    
    # 遍历结束后检查是否形成循环
    if current_file in visited:
        print(f"遍历完成，已形成循环链: {' → '.join(visited)} → {current_file}")

# 调用示例，起始文件为A.txt
traverse_chain("A.txt")

关键说明 #

• 使用set存储已访问文件，集合的成员检查是O(1)，效率更高
• 读取文件时用strip()去除多余的换行符或空格，避免因格式问题找不到下一个文件
• 加入异常处理，防止因文件损坏、缺失导致程序崩溃
• 遍历终止条件：遇到已访问过的文件（循环闭合）、文件不存在、文件内容为空

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