import os

# 需要排除的目录 & 文件（你可以自定义）
EXCLUDED_DIRS = {".git", ".idea", ".venv", "build", "dist", "__pycache__", "desktop", "logs", "models", "documents"}
EXCLUDED_FILES = {".DS_Store", "Thumbs.db"}

def print_directory_tree(start_path=".", indent=""):
    try:
        files = sorted(os.listdir(start_path))
    except PermissionError:
        return

    files = [f for f in files if f not in EXCLUDED_FILES]  # 过滤不需要的文件
    dirs = [d for d in files if os.path.isdir(os.path.join(start_path, d)) and d not in EXCLUDED_DIRS]
    files = [f for f in files if os.path.isfile(os.path.join(start_path, f))]

    for index, file in enumerate(dirs + files):
        path = os.path.join(start_path, file)
        is_last = index == len(dirs + files) - 1
        prefix = "└── " if is_last else "├── "
        print(indent + prefix + file)

        if os.path.isdir(path):
            next_indent = indent + ("    " if is_last else "│   ")
            print_directory_tree(path, next_indent)

if __name__ == "__main__":
    print_directory_tree("..")