如何遍历python目录并在根目录处停止

import os

def traverse_directory(directory):
    for root, dirs, files in os.walk(directory):
        print(f"Current Directory: {root}")

        # 判断是否为根目录
        if root == directory:
            print("Reached the root directory. Stop traversal.")
            break

        # 遍历文件夹
        for directory in dirs:
            print(f"Directory: {os.path.join(root, directory)}")

        # 遍历文件
        for file in files:
            print(f"File: {os.path.join(root, file)}")

# 遍历当前目录（即根目录）
traverse_directory(".")