获取对应文件夹和图片名称

古城微笑少年丶 2023-06-19 06:58 1阅读 0赞

运行程序获取文件夹名称(一级目录、二级目录)和文件名

**目标** 文件结构是三层结构，从中把一级目录、二级目录、文件名获取到  
效果如下：

![list\_dir][list_dir]

# _*_coding:utf-8
    import os
    import re
    import time
    from openpyxl import Workbook
    from openpyxl.styles import Font
    
    def get_file(root_path,suffix): 
        global fileNum
        global dirNum
        try:
            for dir_name in os.listdir(root_path):
                exts = suffix.split(' ')
                #获取目录或文件的路径
                file_path = os.path.join(root_path,dir_name)
                #判断路径为文件还是路径
                if os.path.isdir(file_path):
                    #递归获取所有文件和目录的路径
                    fileNum += 1
                    # file.write('#'+'\n') 
                    get_file(file_path,suffix)  
                else:
                    for ext in exts:
                        #根据后缀名判断文件类别
                        if(dir_name.endswith(ext)):
                            dirNum += 1
                            this_dirNum = str(len([lists for lists in os.listdir(root_path)]))
                            fileName_1dir = file_path.split('\\')[-3]
                            fileName_2dir = file_path.split('\\')[-2]
                            file.write(fileName_1dir + '\t' + fileName_2dir + '\t' + this_dirNum + '\t'+ dir_name + '\n')
        except FileNotFoundError:
            print('地址输入错误，请检查地址'.center(20,'-'))
            time.sleep(1.2)
    
    def xls(root_path, outfile):
        #生成Excel文件
        wb = Workbook()
        ws = wb.active
        ws.title = '文件名'
        #设置表头
        ws.row_dimensions[1].font = Font(bold=True,size=20)
        ws.column_dimensions['A'].width = 85.0
        ws.column_dimensions['B'].width = 95.0
        ws.column_dimensions['C'].width = 15.0
        ws.column_dimensions['D'].width = 30.0
        first_row = []
        datas = []
        with open(outfile,encoding='utf-8') as f:
            is_first_row = True
            for line in f:
                line = line[:-1]
                #存储第一行
                if is_first_row:
                    is_first_row = False
                    first_row = line.split('\t')
                    continue
                #存储第二行至max_row
                datas.append(line.split('\t'))
        ws.append(first_row)
        for data in datas:
            ws.append(data)
        wb.save(root_path + '\\dir_list.xlsx')
    
    def output(root_path):
        print('The total number of files in '+root_path+' is:',fileNum)
        print('The total number of directories in '+ '(' + suffix + ')' +' is:',dirNum)
        print('文件写入完成'.center(20,'-'))
        time.sleep(1.2)
    
    def main():
        get_file(root_path,suffix)
        file.close()
        xls(root_path, outfile)
        output(root_path)
    
    if __name__ == "__main__":
        root_path = input('请输入图片所在地址：')
        suffix = '.jpg'
        outfile = root_path + '\\dir_list.txt'
        fileNum = 0
        dirNum = 0
        seq = '一级目录','二级目录','文件数量','文件名'
        file = open(outfile,'w',encoding='utf-8')
        file.write('\t'.join(seq) + '\n')
        main()

[list_dir]: https://img-blog.csdnimg.cn/20200305112828361.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzM4Njg5Mzk1,size_16,color_FFFFFF,t_70