需求
输入一个文件夹地址,打印出文件夹内部结构,即目录树
效果预览
输入:
请输入文件夹路径:
输出:
└── controller/
├── GoodsController.java
├── RecordController.java
├── RedisController.java
├── SeckillController.java
└── domain/
├── Goods.java
├── Record.java
└── mapper/
├── GoodsMapper.java
├── RecordMapper.java
├── Project4Application.java
└── service/
├── GoodsService.java
└── impl/
├── GoodsServiceImpl.java
├── RecordServiceImpl.java
├── RedisServiceImpl.java
├── SeckillServiceImpl.java
├── RecordService.java
├── RedisService.java
├── SeckillService.java
└── util/
├── CORSFilter.java
├── R.java
├── RedisConfig.java
├── ZookeeperConfig.java
├── ZooKeeperUtils.java
代码
import os
def print_directory_tree(path, prefix=''):
# 获取当前路径下的所有文件和文件夹
items = os.listdir(path)
# 遍历所有项
for item in items:
# 组合完整路径
item_path = os.path.join(path, item)
# 判断是文件还是文件夹
if os.path.isfile(item_path):
# 是文件,直接打印
print(f"{prefix}├── {item}")
elif os.path.isdir(item_path):
# 是文件夹,打印后递归调用打印其子目录
print(f"{prefix}└── {item}/")
print_directory_tree(item_path, prefix + ' ')
# 输入要打印的文件夹路径
folder_path = input("请输入文件夹路径: ")
# 调用函数打印文件目录树
print_directory_tree(folder_path)