pip install Spire.Doc

from spire.doc import *
import os


def extract_images_from_single_doc(doc_path, output_folder):
    """
    從單個Word文檔中提取圖片并保存。
    :param doc_path: Word文檔的完整路徑。
    :param output_folder: 圖片保存的目錄。
    """
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    document = Document()
    document.LoadFromFile(doc_path)

    print(f"正在處理文檔: {os.path.basename(doc_path)}")

    image_count = 0
    # 遍歷文檔的所有節(jié)
    for i in range(document.Sections.Count):
        section = document.Sections.get_Item(i)
        # 遍歷節(jié)中的所有段落
        for j in range(section.Paragraphs.Count):
            paragraph = section.Paragraphs.get_Item(j)
            # 遍歷段落中的所有子元素
            for k in range(paragraph.ChildObjects.Count):
                doc_object = paragraph.ChildObjects.get_Item(k)
                if doc_object.DocumentObjectType == DocumentObjectType.Picture:
                    picture = doc_object if isinstance(doc_object, DocPicture) else None
                    if picture is not None:
                        image_bytes = picture.ImageBytes
                        # 構(gòu)建圖片文件名，包含原始文檔名和圖片索引
                        image_name = f"{os.path.splitext(os.path.basename(doc_path))[0]}_image_{image_count}.png"
                        image_path = os.path.join(output_folder, image_name)
                        with open(image_path, "wb") as f:
                            f.write(image_bytes)
                        print(f"  - 提取圖片: {image_name}")
                        image_count += 1

    document.Close()
    print(f"從 {os.path.basename(doc_path)} 中提取了 {image_count} 張圖片。")

# 示例用法
# doc_file = "path/to/your/document.docx"
# output_dir = "extracted_images"
# extract_images_from_single_doc(doc_file, output_dir)

from spire.doc import *


def batch_extract_images_from_docs(input_folder, output_base_folder):
    """
    批量從指定文件夾下的所有Word文檔中提取圖片。
    :param input_folder: 包含Word文檔的輸入目錄。
    :param output_base_folder: 所有提取圖片保存的根目錄，會為每個文檔創(chuàng)建子文件夾。
    """
    if not os.path.exists(output_base_folder):
        os.makedirs(output_base_folder)

    for filename in os.listdir(input_folder):
        if filename.endswith(".docx") or filename.endswith(".doc"):
            doc_path = os.path.join(input_folder, filename)

            # 為每個文檔創(chuàng)建一個獨立的輸出子文件夾
            doc_output_folder = os.path.join(output_base_folder, os.path.splitext(filename)[0])
            if not os.path.exists(doc_output_folder):
                os.makedirs(doc_output_folder)

            try:
                document = Document()
                document.LoadFromFile(doc_path)
                print(f"\n--- 正在處理文檔: {filename} ---")

                image_count = 0
                for i in range(document.Sections.Count):
                    section = document.Sections.get_Item(i)
                    for j in range(section.Paragraphs.Count):
                        paragraph = section.Paragraphs.get_Item(j)
                        for k in range(paragraph.ChildObjects.Count):
                            doc_object = paragraph.ChildObjects.get_Item(k)
                            if doc_object.DocumentObjectType == DocumentObjectType.Picture:
                                picture = doc_object if isinstance(doc_object, DocPicture) else None
                                if picture is not None:
                                    image_bytes = picture.ImageBytes
                                    image_name = f"image_{image_count}.png"  # 命名圖片，避免與不同文檔的圖片重名
                                    image_path = os.path.join(doc_output_folder, image_name)
                                    with open(image_path, "wb") as f:
                                        f.write(image_bytes)
                                    print(f"  - 提取圖片: {image_name}")
                                    image_count += 1

                document.Close()
                print(f"從 {filename} 中提取了 {image_count} 張圖片，保存至 {doc_output_folder}")

            except Exception as e:
                print(f"處理文檔 {filename} 時發(fā)生錯誤: {e}")
        else:
            print(f"跳過非Word文檔: {filename}")


# 示例用法
input_directory = "path/to/your/word_documents"  # 存放所有Word文檔的目錄
output_directory = "batch_extracted_images"  # 圖片提取后的總輸出目錄
# batch_extract_images_from_docs(input_directory, output_directory)

import hashlib

# ... (接上述批量處理代碼)

# 在保存圖片的代碼塊內(nèi)，可以加入哈希去重邏輯
# ...
                                if picture is not None:
                                    image_bytes = picture.ImageBytes
                                    # 計算圖片內(nèi)容的MD5哈希值
                                    image_hash = hashlib.md5(image_bytes).hexdigest()
                                    
                                    # 檢查是否已存在相同哈希值的圖片
                                    # 這里需要一個全局或文檔級別的set來存儲已保存的哈希值
                                    # For simplicity, let's assume we save all for now, 
                                    # but in a real scenario, you'd check a set of hashes.
                                    
                                    image_name = f"{image_hash}.png" # 使用哈希值作為文件名，方便去重
                                    image_path = os.path.join(doc_output_folder, image_name)
                                    
                                    if not os.path.exists(image_path): # 只有當圖片文件不存在時才寫入
                                        with open(image_path, "wb") as f:
                                            f.write(image_bytes)
                                        print(f"  - 提取并保存新圖片: {image_name}")
                                    else:
                                        print(f"  - 發(fā)現(xiàn)重復(fù)圖片，已跳過: {image_name}")
                                    image_count += 1
# ...