DIRECTORY TREE VISUALIZATION

#!/usr/bin/env python3
"""DIRECTORY TREE VISUALIZATION"""
import os, tempfile
temp = tempfile.gettempdir()
test_dir = os.path.join(temp, "tree_test")
os.makedirs(os.path.join(test_dir, "dir1/subdir1"))
os.makedirs(os.path.join(test_dir, "dir2"))
for f in ["file1.txt", "dir1/file2.txt", "dir1/subdir1/file3.txt"]:
    open(os.path.join(test_dir, f), 'w').write("test")
def print_tree(directory, prefix=""):
    items = sorted(os.listdir(directory))
    for i, item in enumerate(items):
        path = os.path.join(directory, item)
        is_last = i == len(items) - 1
        print(f"{prefix}{'└── ' if is_last else '├── '}{item}")
        if os.path.isdir(path):
            extension = "    " if is_last else "│   "
            print_tree(path, prefix + extension)
print("Directory Tree:")
print(f"{os.path.basename(test_dir)}/")
print_tree(test_dir)
import shutil
shutil.rmtree(test_dir)