ZIP模塊--用於壓縮多個文件或目錄

import os
import shutil


def copy_and_zip(file_list, dst_folder_name):
   '''
   批量複製文件到指定文件夾,而後把指定文件夾的內容壓縮成ZIP而且刪掉該文件夾
   :param file_list: 文件或文件夾
   :param dst_folder_name: 目標壓縮文件的名稱
   :return:
   '''
   source = os.path.join(os.getcwd(), dst_folder_name)
   if os.path.isfile(source+".zip"):
       print('aaa')
       os.remove(source+".zip")  # 刪除 dst_folder_name 文件,避免數據重複
   for item in file_list:
       copy_file(item, dst_folder_name)

   # 這裏我把輸出文件的路徑選到了程序根目錄下

   print(source)
   shutil.make_archive(source, "zip", source)
   shutil.rmtree(source) # 刪除


def copy_file(srcfile, filename):
   '''
   把文件或文件夾複製到指定目錄中
   :param srcfile: 文件或者文件夾的絕對路徑
   :param filename: 指定目錄
   :return:
   '''
   dstfile = os.path.abspath(os.getcwd())
   # 這裏我把輸出文件的路徑選到了程序根目錄下
   folder_name = dstfile + os.sep + filename + os.sep
   if not os.path.isfile(srcfile):
       last_name = os.path.basename(srcfile)
       destination_name = folder_name + last_name
       shutil.copytree(srcfile, destination_name)
       print("copy %s -> %s" % (srcfile, destination_name))
   else:
       fpath, fname = os.path.split(folder_name)  # 分離文件名和路徑
       if not os.path.exists(fpath):
           os.makedirs(fpath)  # 建立路徑
       shutil.copy2(srcfile, folder_name)  # 移動文件
       print("copy %s -> %s" % (srcfile, folder_name))

class ZIP:
   def __init__(self,file_list:list,folder_name):
       copy_and_zip(file_list, folder_name)


if __name__ == '__main__':
   # 壓縮後文件路徑到程序根目錄下
   # file_list爲字典,能夠是多個文件或目錄
   file_list = [r"F:\Django\xqyunpan\data\101.jpg",r"F:\Django\xqyunpan\data\build.txt"]
   ZIP(file_list,"dst_folder_name") # 注意:這個壓縮後的文件名不要隨意更改,不然刪除自身文件後自負!!
相關文章
相關標籤/搜索