unzip/file.py

import json
import os
from log import logger, log_info
from data_dict import ext_name_list, know_zip_ext_name, know_ext_name, handle_zip_ext_name, clear_list


class Files:

    def __init__(self, root_path):
        self.root_path = root_path

    # 目标根目录，返回根目录下全部文件夹的列表
    def get_root_folder_list(self):
        return list(os.walk(self.root_path))[0][1]

    # 目标根目录，返回根目录下全部文件夹的列表
    @staticmethod
    def get_folder_dict(_path):
        data_dict = dict()
        for d in os.walk(_path):
            data_dict[d[0]] = dict(folders=d[1], files=d[2])
        return data_dict

    # 目标目录， 获取此目录下的全部文件，未经整理
    @staticmethod
    def get_all_files(_path):
        file_list = []
        for item in os.walk(_path):
            if len(item[2]) > 0:
                for file_name in item[2]:
                    file_list.append(os.path.join(item[0], file_name))
        return file_list

    # 目标文件， 重命名文件
    @staticmethod
    def file_rename(_org, _target):
        os.rename(_org, _target)

    # 目标文件， 获取文件扩展名， 返回 文件路径+文件名， 扩展名
    @staticmethod
    def get_ext_name(_file):
        base_name = os.path.splitext(_file)[0]
        ext_name = os.path.splitext(_file)[1].lower()
        return base_name, ext_name

    # 目标文件， 将错误的扩展名换成正确的，执行重命名操作，返回新的文件名列表
    def change_to_know_name(self, _file):
        base_file_name, ext_file_name = self.get_ext_name(_file)
        if ext_file_name in know_zip_ext_name:
            return _file
        for unknown_ext, know_ext in ext_name_list.items():
            if ext_file_name == unknown_ext:
                new_file_name = base_file_name + know_ext
                self.file_rename(_file, new_file_name)
                return new_file_name
        return _file

    # 目标文件， 将文件分类整理返回字典格式文件列表
    def clear_files(self, files_list):
        new_file_list = dict(handle_zip=[], zip=[], others=[], unknown=[])
        for _f in files_list:
            new = self.change_to_know_name(_f)
            _, ext_new = self.get_ext_name(new)
            if ext_new in handle_zip_ext_name:
                new_file_list['handle_zip'].append(new)
                new_file_list['zip'].append(new)
            elif ext_new in know_zip_ext_name:
                new_file_list['zip'].append(new)
            elif ext_new in know_ext_name:
                new_file_list['others'].append(new)
            else:
                new_file_list['unknown'].append(new)
        return new_file_list

    # 获取可删除文件列表
    def get_del_files(self, _path):
        del_list = []
        for _f in self.get_all_files(_path):
            if os.path.basename(_f) in clear_list['file_name']:
                del_list.append(_f)
            if os.path.splitext(_f)[1] in clear_list['ext_name']:
                del_list.append(_f)
        return del_list

    # 目标文件， 将文件按照目录整理
    def collection_files(self, _path):
        _all_files = self.get_all_files(_path)
        files_dict = dict(path_files={}, path=[], files=_all_files)
        for _f in _all_files:
            file_path = os.path.dirname(_f)
            if file_path not in files_dict['path_files'].keys():
                files_dict['path'].append(file_path)
                files_dict['path_files'][file_path] = [_f]
            else:
                files_dict['path_files'][file_path].append(_f)
        return files_dict

    def get_file_dict(self, _path):
        for p in os.walk(_path):
            pass


    def move_files(self, _root_path):
        move_file_list = []
        _files = self.collection_files(_root_path)
        if not _files['files']:
            log_info('文件夹为空')
        else:
            if len(_files['files']) == 1:  # 单文件处理
                source = _files['files'][0]
                target = os.path.join(_root_path, os.path.basename(source))
                move_file_list.append((source, target))
            elif len(_files['path']) == 1:  # 只有一个目录下有多文件处理
                for _f in _files['files']:
                    source = _f
                    target = os.path.join(_root_path, os.path.basename(source))
                    move_file_list.append((source, target))
            else:  # 有多个目录，每个目录下有一个或多个文件
                temp_data = dict()
                for _p in _files['path']:
                    for _f in _files['path_files'][_p]:
                        temp_data[_p] = dict(current=_p, valid='', name=os.path.basename(_f))

    # 寻找不重复的基础路径
    # 传入的字典格式 {文件路径：{current: 当前路径，valid: 基础路径之后的部分, name: 文件名}}
    # def get_base_dir(self, file_list):
    #     base_dir = dict()
    #     temp = []
    #     for _f in file_list.keys():
    #         current = base_dir[_f]['current']
    #         base_dir[_f]['current'] = os.path.dirname(current)
    #         base_dir[_f]['valid'] = os.path.basename(current) + '/' + base_dir[_f]['valid']
    #         temp.append(base_dir[_f]['valid'])
    #     if len(set(temp)) == len(temp):
    #         return base_dir
    #     else:
    #         self.get_base_dir(base_dir)

    def get_base_dir(self, _path):
        files_dict = self.get_folder_dict(_path)

        # for _p in check_list:
        #     if len(files_dict[_p]['folders']) > 1 and len(files_dict[_p]['files']) == 0:
        #         for _f in files_dict[_p]['folders']:
        #             os.path.join()
        #     elif len(files_dict[_path]['folders']) > 1 and len(files_dict[_path]['files']) > 0:
        #         pass
        #     elif len(files_dict[_path]['folders']) == 1 and len(files_dict[_path]['files']) == 0:
        #         pass
        #     elif len(files_dict[_path]['folders']) == 1 and len(files_dict[_path]['files']) > 0:
        #         pass
        #     elif len(files_dict[_path]['folders']) == 0 and len(files_dict[_path]['files']) > 0:
        #         pass
        #     else:
        #         pass
    # 删除文件
    @staticmethod
    def del_file(path_list):
        for path in path_list:
            os.system(f'del "{path}"')
            # logger.info(f"删除文件成功： {path}")
            log_info(f"删除文件成功： {path}")


if __name__ == '__main__':
    root = r"F:\Temp\sjry\hj"
    files = Files(root)
    # print(files.get_root_folder_list())
    # print(files.get_all_files(r"F:\Temp\sjry\hj\35316"))
    # files.file_rename('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.123', 'F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.222')
    # print(files.get_ext_name('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.222'))
    # print(files.change_to_know_name('F:\\Temp\\sjry\\hj\\35316\\22\\新建 文本文档.win'))
    # for f in files.get_root_folder_list():
    #     all_file = files.get_all_files(os.path.join(root, f))
    #     print(files.clear_files(all_file))
    # print(files.collection_files(r"F:\Temp\sjry\hj\35316"))
    print(json.dumps(files.get_folder_dict(r'F:\Temp\test\12345')))