Python3壓縮和解壓縮的實(shí)現(xiàn)方法

這篇文章主要為大家展示了Python3壓縮和解壓縮的實(shí)現(xiàn)方法，內(nèi)容簡(jiǎn)而易懂，條理清晰，希望能夠幫助大家解決疑惑，下面讓小編帶大家一起來(lái)研究并學(xué)習(xí)一下“Python3壓縮和解壓縮的實(shí)現(xiàn)方法”這篇文章吧。

python可以做什么

Python是一種編程語(yǔ)言，內(nèi)置了許多有效的工具，Python幾乎無(wú)所不能，該語(yǔ)言通俗易懂、容易入門(mén)、功能強(qiáng)大，在許多領(lǐng)域中都有廣泛的應(yīng)用，例如最熱門(mén)的大數(shù)據(jù)分析，人工智能，Web開(kāi)發(fā)等。

一、說(shuō)明

壓縮和解壓縮是日常常用的操作，不管是windows上圖形界面的操作，還是linux上用命令來(lái)進(jìn)行壓縮解壓縮，總的而言都還是比較方便的。

但用代碼來(lái)實(shí)現(xiàn)就沒(méi)做過(guò)，近期也得實(shí)現(xiàn)代碼壓縮與解壓縮操作，所以就抽時(shí)間來(lái)研究一下。

二、zip文件壓縮和解壓縮實(shí)現(xiàn)

import os
import zipfile

# 函數(shù)功能是zip_file_list所有文件，和zip_dir_list所有目錄下的所有文件，被壓縮到一個(gè)zip_file_name的壓縮文件中
def my_zip_function(zip_file_name, zip_file_list=[], zip_dir_list=[]):
  # 壓縮文件最后需要close，為了方便我們直接用with
  with zipfile.ZipFile(zip_file_name, "w") as zip_obj:
    # 壓縮文件
    for tmp_file in zip_file_list:
      zip_obj.write(tmp_file)
    # 壓縮目錄
    for tmp_dir in zip_dir_list:
      # zipfile沒(méi)有直接壓縮目錄的功能，要壓縮目錄只能遍歷目錄一個(gè)一個(gè)文件壓。
      for root, dirs, files in os.walk(tmp_dir):
        # 如果想要目錄為空時(shí)仍將該目錄壓縮進(jìn)去，該目錄也要壓縮一遍；反之請(qǐng)將以下行注釋掉
        zip_obj.write(root)
        for tmp_file in files:
          # 拼接文件完整目錄，不然只用文件名代碼找不到文件
          tmp_file_path = os.path.join(root, tmp_file)
          zip_obj.write(tmp_file_path)


# 函數(shù)功能是遍歷壓縮文件中的所有文件
def my_traversal_zip_function(zip_file_name):
  with zipfile.ZipFile(zip_file_name, "r") as zip_obj:
    # 返回結(jié)果是一個(gè)ZipInfo列表
    # 如果在壓縮時(shí)顯示壓縮目錄，則目錄也作為一個(gè)單獨(dú)的ZipInfo呈現(xiàn)在列表中；反之則沒(méi)有目錄的ZipInfo
    all_file_list = zip_obj.infolist()
    for tmp_file in all_file_list:
      print(tmp_file.filename)
      # 還可以在不解壓的情況下直接讀取文件的內(nèi)容
      # 可以通過(guò)ZipInfo.is_dir()來(lái)區(qū)分是文件還是目錄
      # if not tmp_file.is_dir():
      #   with zip_obj.open(tmp_file) as zip_fd:
      #     print(zip_fd.read())

# 函數(shù)的功能是將壓縮文件直接解壓
def my_unzip_function(zip_file_name, path="."):
  with zipfile.ZipFile(zip_file_name, "r") as zip_obj:
    zip_obj.extractall(path=path)

if __name__ == "__main__":
  zip_file_name = "test_zip.zip"
  # 自己在測(cè)試時(shí)要先自行創(chuàng)建好要壓縮的文件和目錄
  zip_file_list = ["test_tar_file1.txt", "test_tar_file2.txt"]
  zip_dir_list = ["test_tar_dir"]
  my_zip_function(zip_file_name, zip_file_list, zip_dir_list)
  my_traversal_zip_function(zip_file_name)
  # my_unzip_function(zip_file_name, path=".")

三、tar文件的壓縮與解壓縮實(shí)現(xiàn)

除了直接的.tar文件，還包括.tar.gz/.tar.bz2/.tar.xz等格式文件的壓縮與解壓縮實(shí)現(xiàn)。

import os
import tarfile

# Python學(xué)習(xí)交流群：778463939
# 函數(shù)功能是tar_file_list所有文件，和tar_dir_list所有目錄下的所有文件，被壓縮到一個(gè)tar_file_name的壓縮文件中
def my_tar_function(tar_file_name, tar_file_list=[], tar_dir_list=[], model="w"):
  # 本來(lái)也應(yīng)該是tarfile.TarFile(tar_file_name, model)來(lái)創(chuàng)建的，但TarFile不支持"r:gz"等擴(kuò)展形式
  # 壓縮文件最后需要close，為了方便我們直接用with
  with tarfile.open(tar_file_name, model) as tar_obj:
    # 壓縮文件
    for tmp_file in tar_file_list:
      tar_obj.add(tmp_file)
    # 壓縮目錄。和zipfile相比tarfile允許直接壓縮目錄，而不需要去遍歷目錄一個(gè)個(gè)文件壓
    for tmp_dir in tar_dir_list:
      tar_obj.add(tmp_dir)


# 函數(shù)功能是遍歷壓縮文件中的所有文件
def my_traversal_tar_function(tar_file_name, model="r"):
  with tarfile.open(tar_file_name, model) as tar_obj:
    # 返回結(jié)果是一個(gè)TarInfo列表
    all_file_list = tar_obj.getmembers()
    for tmp_file in all_file_list:
      print(tmp_file.name)
      # 還可以在不解壓的情況下直接讀取文件的內(nèi)容
      # 可以通過(guò)TarInfo.isdir()來(lái)區(qū)分是文件還是目錄
      # if not tmp_file.isdir():
      #   # 相當(dāng)于zip的open，并不會(huì)把文件給解壓出來(lái)
      #   tar_fd = tar_obj.extractfile(tmp_file)
      #   print(tar_fd.read())


# 函數(shù)的功能是將壓縮文件直接解壓
def my_untar_function(tar_file_name, path=".", model="r"):
  with tarfile.open(tar_file_name, model) as tar_obj:
    tar_obj.extractall(path=path)


if __name__ == "__main__":
  # 自己在測(cè)試時(shí)要先自行創(chuàng)建好要壓縮的文件和目錄
  tar_file_list = ["test_tar_file1.txt", "test_tar_file2.txt"]
  tar_dir_list = ["test_tar_dir"]
  tar_file_name = "test_tar.tar"
  # 在.tar基礎(chǔ)上，tarfile還支持gz/bz2/xz的壓縮，只要在原來(lái)打開(kāi)模式的基礎(chǔ)上使用:或|接上壓縮方法即可，如"r:gz"
  # 特別的，如果是讀取文件，可以使用"r:*"來(lái)指示嘗試以任意格式讀取
  open_model = "w"
  # open_model = "w:gz"
  my_tar_function(tar_file_name, tar_file_list, tar_dir_list, model=open_model)
  open_model = "r"
  # open_model = "r:*"
  my_traversal_tar_function(tar_file_name, model=open_model)
  # open_model = "r:*"
  # my_untar_function(tar_file_name, path=".", model=open_model)

以上就是關(guān)于“Python3壓縮和解壓縮的實(shí)現(xiàn)方法”的內(nèi)容，如果改文章對(duì)你有所幫助并覺(jué)得寫(xiě)得不錯(cuò)，勞請(qǐng)分享給你的好友一起學(xué)習(xí)新知識(shí)，若想了解更多相關(guān)知識(shí)內(nèi)容，請(qǐng)多多關(guān)注億速云行業(yè)資訊頻道。