Python Zipfile 實戰：Write vs Writestr —— 硬碟搬運工與記憶體魔術師; zf.writestr(zinfo_or_arcname, data) #zinfo_or_arcname: ZipInfo | str ; data: bytes | str

import zipfile
import os
import time

# 1. 【環境準備】先在硬碟上建立一個真實檔案
dummy_filename = "disk_file.txt"
with open(dummy_filename, "w", encoding="utf-8") as f:
    f.write("我是硬碟上真實存在的檔案內容。")

output_zip = "demo_archive.zip"

print(f"正在建立壓縮檔: {output_zip} ...\n")

with zipfile.ZipFile(output_zip, 'w', zipfile.ZIP_DEFLATED) as zf:
    
    # ==========================================
    # 場景 A：使用 write (硬碟搬運工)
    # ==========================================
    print("[A] 執行 write: 將硬碟檔案搬入 Zip")
    # 語法: zf.write(實體路徑, 壓縮檔內的名稱)
    zf.write(dummy_filename, arcname="folder_a/real_file.txt")
    
    
    # ==========================================
    # 場景 B：使用 writestr (記憶體魔術師)
    # ==========================================
    print("[B] 執行 writestr: 將變數內容直接寫入 Zip")
    
    # 假設這是我們動態產生的資料 (例如修改過的 XML)
    dynamic_content = "這是存在於記憶體中的字串，硬碟上找不到我。"
    
    # 語法: zf.writestr(壓縮檔內的名稱, 資料內容)
    zf.writestr("folder_b/memory_note.txt", dynamic_content)
    
    
    # ==========================================
    # 場景 C： DOCX 分割場景 (讀取 -> 寫入)
    # ==========================================
    print("[C] 執行 writestr (進階): 模擬從另一個 Zip 複製資料")
    # 這裡我們模擬「讀取」剛才寫入的資料 (bytes)，然後用新名字寫回去
    # 這就是 zfout.writestr(item, zfin.read(item)) 的原理
    
    # 1. 取得資料 (Bytes)
    # (在您的程式中，這是 zfin.read(filename))
    image_data_in_memory = b'\x89PNG\r\n\x1a\n...fake_image_data...' 
    
    # 2. 直接寫入，不落地
    zf.writestr("word/media/image1.png", image_data_in_memory)

print("\n完成！請檢查生成的 demo_archive.zip")

# 清理測試用的暫存檔
os.remove(dummy_filename)

# zfin 是來源 Zip (舊 docx)
# zfout 是目標 Zip (新 docx)

# 從來源讀取資料 (變成記憶體中的 bytes)
data = zfin.read(item.filename) 

# 直接寫入目標 (不需要存成暫存檔)
zfout.writestr(item, data)

        # --- Step E: 單次寫入 (Single Pass Write) ---
        # 若有需要更新的 .rels 內容 (new_doc_rels_content) 或有要刪除的孤兒檔案 (orphans)：
        # 1. 開啟兩個 Zip：來源 (zfin) 與 暫存目標 (zfout)。
        # 2. 遍歷來源 Zip 的每個檔案：
        #    - 若是 document.xml.rels 且有新內容：寫入修改後的 XML (new_doc_rels_content)。
        #    - 若是孤兒檔案 (orphans)：跳過不寫入 (即刪除)。
        #    - 其他檔案：直接從來源複製到目標。
        # 3. 最後用暫存檔覆蓋原始檔案。
        if new_doc_rels_content is not None or orphans:
            with tempfile.NamedTemporaryFile(suffix='.docx', delete=False) as tmpzip:
                tmp_zip_path = tmpzip.name
            
            with zipfile.ZipFile(copied_docx_path, 'r') as zfin, \
                 zipfile.ZipFile(tmp_zip_path, 'w', zipfile.ZIP_DEFLATED) as zfout:
                
                for item in zfin.infolist():
                    # item 如: 
                    #<ZipInfo filename='word/_rels/document.xml.rels' compress_type=deflate file_size=121173 compress_size=6317>
                    if item.filename == doc_rels_path and new_doc_rels_content is not None:
                        # doc_rels_path = 'word/_rels/document.xml.rels'
                        zfout.writestr(doc_rels_path, new_doc_rels_content) # type: ignore
                        continue
                    
                    if item.filename in orphans:
                        continue
                    
                    zfout.writestr(item, zfin.read(item.filename))
            
            shutil.move(tmp_zip_path, copied_docx_path)

# 這是最完美的寫法：
# 1. 用 ZipInfo (item) 保留了檔案屬性與時間。
# 2. 用 read() 讀取 bytes 避免了編碼問題。
# 3. 用 writestr() 避免了硬碟 I/O。
zfout.writestr(item, zfin.read(item.filename))