快捷導(dǎo)航

Python解析壓縮包內(nèi)部文件的后綴名并分類存放

更新時間：2024年12月24日 08:48:33 作者：一晌小貪歡

這篇文章主要為大家詳細介紹了如何使用Python解析壓縮包內(nèi)部文件的后綴名并分類存放,文中的示例代碼講解詳細,感興趣的小伙伴可以跟隨小編一起學習一下

1、背景

我們有很多壓縮包文件，假設(shè)壓縮包的種類只有【.7z】，【.rar】，【.zip】，壓縮包里面有一個文件，文件可以是不同的格式(后綴)，我們希望通過壓縮包內(nèi)的文件后綴，自動創(chuàng)建文件夾，讓后將壓縮包進行批量移動

如下圖：每一個壓縮包內(nèi)都有一個文件(不同類型\也可以相同類型)

預(yù)期結(jié)果，如下圖

2、庫的安裝

庫	用途	安裝
py7zr	壓縮包.7z操作	pip install py7zr -i https://pypi.tuna.tsinghua.edu.cn/simple/
rarfile	壓縮包.rar操作	pip install rarfile -i https://pypi.tuna.tsinghua.edu.cn/simple/
zipfile	壓縮包.rar操作	內(nèi)置庫無需安裝
os	獲取絕對路徑	內(nèi)置庫無需安裝

3、核心代碼

if zipfile.is_zipfile(archive_path):
    with zipfile.ZipFile(archive_path, 'r') as zip_ref:
        for file_info in zip_ref.infolist():
            ext = os.path.splitext(file_info.filename)[1].lower()
            if ext:
                extensions.add(ext[1:])  # 去掉點號

elif rarfile.is_rarfile(archive_path):
    with rarfile.RarFile(archive_path) as rar_ref:
        for file_info in rar_ref.infolist():
            ext = os.path.splitext(file_info.filename)[1].lower()
            if ext:
                extensions.add(ext[1:])

elif py7zr.is_7zfile(archive_path):
    with py7zr.SevenZipFile(archive_path, mode='r') as z7_ref:
        for filename in z7_ref.getnames():
            ext = os.path.splitext(filename)[1].lower()
            if ext:
                extensions.add(ext[1:])

4、完整代碼

# -*- coding: UTF-8 -*-
'''
@Project ：測試 
@File    ：main.py
@IDE     ：PyCharm 
@Author  ：一晌小貪歡(278865463@qq.com)
@Date    ：2024/12/23 15:32 
'''


import os
import zipfile
import py7zr
import rarfile
import shutil

# 設(shè)置源文件夾和目標文件夾
source_folder = '壓縮包數(shù)據(jù)源'
target_folder = '分類文件夾'

# 檢查目標文件夾是否存在，不存在則創(chuàng)建
if not os.path.exists(target_folder):
    os.makedirs(target_folder)


def create_folder(folder_name):
    folder_path = os.path.join(target_folder, folder_name)
    if not os.path.exists(folder_path):
        os.makedirs(folder_path)
    return folder_path


def get_archive_extensions(archive_path):
    extensions = set()

    try:
        if zipfile.is_zipfile(archive_path):
            with zipfile.ZipFile(archive_path, 'r') as zip_ref:
                for file_info in zip_ref.infolist():
                    ext = os.path.splitext(file_info.filename)[1].lower()
                    if ext:
                        extensions.add(ext[1:])  # 去掉點號

        elif rarfile.is_rarfile(archive_path):
            with rarfile.RarFile(archive_path) as rar_ref:
                for file_info in rar_ref.infolist():
                    ext = os.path.splitext(file_info.filename)[1].lower()
                    if ext:
                        extensions.add(ext[1:])

        elif py7zr.is_7zfile(archive_path):
            with py7zr.SevenZipFile(archive_path, mode='r') as z7_ref:
                for filename in z7_ref.getnames():
                    ext = os.path.splitext(filename)[1].lower()
                    if ext:
                        extensions.add(ext[1:])
    except Exception as e:
        print(f"讀取文件 {archive_path} 時出錯: {str(e)}")
        return set()

    return extensions


# 主程序
for filename in os.listdir(source_folder):
    file_path = os.path.join(source_folder, filename)

    if os.path.isfile(file_path):
        # 獲取壓縮包內(nèi)所有文件的擴展名
        extensions = get_archive_extensions(file_path)

        if extensions:
            # 如果只有一種文件類型，移動到對應(yīng)文件夾
            if len(extensions) == 1:
                ext = extensions.pop()
                dest_folder = create_folder(ext)
                dest_path = os.path.join(dest_folder, filename)
                shutil.move(file_path, dest_path)
                print(f"移動 {filename} 到 {ext} 文件夾")
            else:
                # 如果有多種文件類型，移動到 mixed 文件夾
                dest_folder = create_folder('mixed')
                dest_path = os.path.join(dest_folder, filename)
                shutil.move(file_path, dest_path)
                print(f"移動 {filename} 到 mixed 文件夾 (包含文件類型: {', '.join(extensions)})")
        else:
            print(f"跳過文件: {filename} (無法讀取或沒有有效文件)")

到此這篇關(guān)于Python解析壓縮包內(nèi)部文件的后綴名并分類存放的文章就介紹到這了,更多相關(guān)Python解析文件后綴名內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: