【粉丝真实需求】外贸办公 Python 自动化・提单水印批量实战 6/6

import osimport shutilimport zipfileimport win32com.clientimport subprocess# ======================== 【公众号读者仅需修改这里2行】 ========================# 存放所有待处理提单doc的文件夹绝对路径WORK_FOLDER = r"C:\Users\Desktop\new_project"# 水印新旧文本（复制艺术字内完整文字，避免换行空格匹配失败）OLD_WATERMARK = "Non-Negotiable"NEW_WATERMARK = "Original"# ============================================================================# 临时缓存文件夹（程序自动创建删除，无需手动操作）TEMP_DOCX_CACHE = "temp_doc_cache"def kill_winword_process():    """    【WPS兼容核心补丁】替代Word.Quit()，强制清理后台Office进程    解决WPS COM无Quit属性、进程残留锁死文件问题    """    try:        # 强制终止所有WINWORD后台进程        subprocess.call(            ["taskkill", "/f", "/im", "WINWORD.EXE"],            stdout=subprocess.PIPE,            stderr=subprocess.PIPE        )    except Exception:        # 无进程时静默跳过，不抛报错        passdef single_doc_to_docx(source_doc_path, output_docx_path):    """    步骤1：单个老式.doc 中转转为 .docx    仅做格式转换，不修改任何单据内容、排版样式    """    # 初始化COM对象    word_app = win32com.client.Dispatch("Word.Application")    word_app.Visible = False    # 后台静默运行，不弹出Word窗口    word_app.DisplayAlerts = 0   # 屏蔽弹窗警告    run_success = False    try:        # 只读打开源doc文件，保护原始单据        doc_file = word_app.Documents.Open(os.path.abspath(source_doc_path))        # FileFormat=16 → 标准docx格式        doc_file.SaveAs(os.path.abspath(output_docx_path), FileFormat=16)        doc_file.Close()        run_success = True        print(f"✅ 中转格式转换完成：{os.path.basename(source_doc_path)}")    except Exception as err:        print(f"❌ 转换失败 {os.path.basename(source_doc_path)}：{str(err)}")    finally:        # 释放COM对象内存        del word_app        # 强制清理后台进程        kill_winword_process()    return run_successdef xml_replace_watermark(docx_input, docx_output):    """    步骤2：docx底层XML全局替换水印（稳定100%捕获页眉艺术字）    纯标准库zip操作，无COM图形读取，彻底规避Shapes属性报错    """    temp_unzip_dir = "tmp_unzip_buffer"    # 清理上次解压残留    if os.path.exists(temp_unzip_dir):        shutil.rmtree(temp_unzip_dir)    os.mkdir(temp_unzip_dir)    # 1. 解压docx压缩包    with zipfile.ZipFile(docx_input, "r") as zip_reader:        zip_reader.extractall(temp_unzip_dir)    # 2. 遍历全部XML文件，全局替换水印文本    for root_path, _, file_names in os.walk(temp_unzip_dir):        for file_name in file_names:            if file_name.endswith(".xml"):                full_file_path = os.path.join(root_path, file_name)                # 读取XML文本                with open(full_file_path, "r", encoding="utf-8") as f:                    xml_content = f.read()                # 替换水印文字                xml_content = xml_content.replace(OLD_WATERMARK, NEW_WATERMARK)                # 写回修改后的XML                with open(full_file_path, "w", encoding="utf-8") as f:                    f.write(xml_content)    # 3. 重新打包生成修改后的docx    with zipfile.ZipFile(docx_output, "w", zipfile.ZIP_DEFLATED) as zip_writer:        for root_path, _, file_names in os.walk(temp_unzip_dir):            for file_name in file_names:                full_file_path = os.path.join(root_path, file_name)                # 计算压缩包内相对路径                relative_path = os.path.relpath(full_file_path, temp_unzip_dir)                zip_writer.write(full_file_path, relative_path)    # 清空解压缓存    shutil.rmtree(temp_unzip_dir)    return Truedef single_docx_to_doc(source_docx_path, output_final_doc_path):    """    步骤3：修改好的docx 转回老式原生.doc格式交付客户    FileFormat=0 锁定Word97-2003 .doc，杜绝字体加宽、水印偏移    """    word_app = win32com.client.Dispatch("Word.Application")    word_app.Visible = False    word_app.DisplayAlerts = 0    run_success = False    try:        docx_file = word_app.Documents.Open(os.path.abspath(source_docx_path))        # FileFormat=0 → 老式doc格式        docx_file.SaveAs(os.path.abspath(output_final_doc_path), FileFormat=0)        docx_file.Close()        run_success = True    except Exception as err:        print(f"❌ 转回doc失败 {os.path.basename(source_docx_path)}：{str(err)}")    finally:        del word_app        kill_winword_process()    return run_successdef batch_auto_process_all_bl():    """    【批量主调度函数】整篇代码闭环核心：自动遍历文件夹所有待处理提单    1. 自动过滤后缀.doc，跳过已带「成品_」前缀的文件，防止二次重复处理    2. 单文件失败隔离，批量不会整体崩溃    3. 统计成功/失败总数，结束输出汇总报表    """    # 初始化临时缓存文件夹    if os.path.exists(TEMP_DOCX_CACHE):        shutil.rmtree(TEMP_DOCX_CACHE)    os.mkdir(TEMP_DOCX_CACHE)    success_count = 0    fail_count = 0    print("==================== 批量提单替换流水线启动 ====================")    # 遍历目标文件夹全部文件    for file_name in os.listdir(WORK_FOLDER):        # 筛选规则：只处理原始doc，跳过成品、docx、其他格式文件        if file_name.lower().endswith(".doc") and not file_name.startswith("成品_"):            origin_full_path = os.path.join(WORK_FOLDER, file_name)            # 定义各阶段临时文件路径            temp_mid_docx = os.path.join(TEMP_DOCX_CACHE, file_name.replace(".doc", ".docx"))            modified_docx = os.path.join(TEMP_DOCX_CACHE, f"mod_{file_name.replace('.doc', '.docx')}")            # 最终输出成品doc，带前缀隔离源文件            final_export_doc = os.path.join(WORK_FOLDER, f"成品_{file_name}")            print(f"\n----- 正在处理单据：{file_name} -----")            # 流水线三步执行            step1_ok = single_doc_to_docx(origin_full_path, temp_mid_docx)            if not step1_ok:                fail_count += 1                continue            step2_ok = xml_replace_watermark(temp_mid_docx, modified_docx)            step3_ok = single_docx_to_doc(modified_docx, final_export_doc)            if step3_ok:                success_count += 1                print(f"✅ 单据 {file_name} 全部处理完成，输出：成品_{file_name}")            else:                fail_count += 1    # 运行结束清理缓存    shutil.rmtree(TEMP_DOCX_CACHE)    kill_winword_process()    # 批量任务汇总报表    print("\n==================== 批量处理全部结束 汇总 ====================")    print(f"✅ 成功处理单据数量：{success_count} 份")    print(f"❌ 处理失败单据数量：{fail_count} 份")    print(f"📁 成品文件统一保存在文件夹内，名称前缀为【成品_】，原始源文件无任何改动")# 程序入口启动if __name__ == "__main__":    batch_auto_process_all_bl()

今日鸡汤分享：没有那么多天赋异禀，优秀的人总是在努力翻山越岭！