import shutil,zipfile,os

class ToolModel(object):

    def dfs_get_zip_file(self,input_path, result, ignore=[]):

        '''

        递归目录

        :param input_path: 输入路径

        :param result: 列表

        :param ignore: 忽略文件或目录名

        :return:

        '''

        files = os.listdir(input_path)

        for file in files:

            filePath = input_path + '/' + file

            if file in ignore:

                continue

            if os.path.isdir(filePath):

                self.dfs_get_zip_file(filePath, result, ignore)

            else:

                result.append(filePath)

    def zip_path(self,input_path, output_path, ignore=[]):

        '''

        :param input_path:  输入路径   /app/adminkit

        :param output_path: 输出路径  /app/adminkit.zip

        :param ignore: 忽略文件或目录名

        :return:

        '''

        outdir = os.path.dirname(output_path)

        if not os.path.isdir(outdir):

            os.makedirs(outdir)

        f = zipfile.ZipFile(output_path, 'w', zipfile.ZIP_DEFLATED)

        filelists = []

        self.dfs_get_zip_file(input_path, filelists, ignore)

        for file in filelists:

            file = file.replace('\\', '/')

            input_path = input_path.replace('\\', '/')

            f.write(file, file.replace(input_path, ''))

        f.close()

        return output_path

    def unzip(self,filename: str,dirname):

        '''

        解压缩

        :param filename: 压缩文件名

        :param dirname: 解压缩输出目录

        :return:

        '''

        try:

            file = zipfile.ZipFile(filename)

            file.extractall(dirname)

            file.close()

            # 递归修复编码

            self.rename(dirname)

        except:

            print(f'{filename} unzip fail')

    def rename(self,pwd: str, filename=''):

        """压缩包内部文件有中文名, 解压后出现乱码，进行恢复"""

        path = f'{pwd}/{filename}'

        if os.path.isdir(path):

            for i in os.scandir(path):

                self.rename(path, i.name)

        newname = filename.encode('cp437').decode('gbk')

        os.rename(path, f'{pwd}/{newname}')

    def del_file(self,filepath):

        """

        删除指定路径下的所有文件和文件夹

        :param filepath: 路径

        :return:

        """

        del_list = os.listdir(filepath)

        for f in del_list:

            file_path = os.path.join(filepath, f)

            if os.path.isfile(file_path):

                os.remove(file_path)

            elif os.path.isdir(file_path):

                shutil.rmtree(file_path)