python将目录下的所有文件写进excle表格中

人生苦短，还是用python好。
现在将.idea文件夹目录的所有文件都写进excle中。直接上图上代码！

效果如下：

上代码，复制即可运行，别忘了安装openpyxl

pip install openpyxl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

import os
from openpyxl import load_workbook
from openpyxl import Workbook

addr = "openpyxl.xlsx"
# 打开文件
if not os.path.exists(addr):
    # 实例化
    wb = Workbook()  # 如果文件不存在，则创建文件
    # 保存文件
    wb.save(addr)

wb = load_workbook(addr)  # 加载文件
# 读取表
ws = wb['Sheet']
# 第一行输入
ws['A1'] = '文件'
ws['B1'] = '路径'

while 1:
    print('''1、录入目录下的所有文件路径
2、退出程序''')
    while 1:
        try:
            b = float(input('请输入'))
            if b not in [1, 2]:
                raise ValueError
            break
        except:
            print('输入不正确，请重新输入')
    if b == 1:
        while 1:
            path = input('请输入您的路径(如：E:\cProject\)：').strip()
            try:
                row_max = ws.max_row          #获取当前Sheet表格的最大行
                for root, dirs, files in os.walk(rf'{path}'):
                    for file in files:
                        row_max += 1
                        #写入路径
                        ws.cell(row=row_max, column=2).value = os.path.join(root, file)
                        #写入文件名
                        ws.cell(row=row_max, column=1).value = file             
                wb.save(addr)
                print('录入完成')
            except:
                print('路径输入有误')
            else:
                break
    elif b == 2:
        print('成功退出程序')
        break