79 lines
2.5 KiB
Python
79 lines
2.5 KiB
Python
from __future__ import annotations
|
||
|
||
"""将 .docx 文档转换为 Markdown。
|
||
|
||
实现思路：
|
||
- 使用 mammoth 将 docx 转为 HTML（对 Word 样式有较好兼容）
|
||
- 再使用 markdownify 将 HTML 转为 Markdown
|
||
- 可选导出文档内图片到 assets 目录，并在 Markdown 中引用相对路径
|
||
"""
|
||
|
||
import argparse
|
||
import os
|
||
from pathlib import Path
|
||
|
||
|
||
def main() -> int:
|
||
"""命令行入口：执行 docx -> md 转换。"""
|
||
parser = argparse.ArgumentParser()
|
||
parser.add_argument("docx_path")
|
||
parser.add_argument("md_path")
|
||
parser.add_argument("--assets-dir", default=None)
|
||
args = parser.parse_args()
|
||
|
||
docx_path = Path(args.docx_path).expanduser().resolve()
|
||
md_path = Path(args.md_path).expanduser().resolve()
|
||
assets_dir = Path(args.assets_dir).expanduser().resolve() if args.assets_dir else None
|
||
|
||
if not docx_path.exists() or not docx_path.is_file():
|
||
raise FileNotFoundError(str(docx_path))
|
||
|
||
import mammoth
|
||
from markdownify import markdownify as md
|
||
|
||
image_index = 0
|
||
|
||
def _convert_image(image):
|
||
"""将 docx 内嵌图片写入 assets 目录，并返回 Markdown 可用的相对路径。"""
|
||
nonlocal image_index
|
||
if assets_dir is None:
|
||
# 不导出图片时，返回空 src，避免把图片内容直接内联到 Markdown
|
||
return {"src": ""}
|
||
assets_dir.mkdir(parents=True, exist_ok=True)
|
||
ext = _guess_image_extension(image.content_type)
|
||
image_index += 1
|
||
name = f"image_{image_index:03d}{ext}"
|
||
target = assets_dir / name
|
||
with target.open("wb") as f:
|
||
f.write(image.read())
|
||
rel = os.path.relpath(target, md_path.parent)
|
||
rel = rel.replace("\\", "/")
|
||
return {"src": rel}
|
||
|
||
result = mammoth.convert_to_html(docx_path, convert_image=mammoth.images.img_element(_convert_image))
|
||
html = result.value
|
||
markdown = md(html, heading_style="ATX", bullets="-")
|
||
|
||
md_path.parent.mkdir(parents=True, exist_ok=True)
|
||
md_path.write_text(markdown, encoding="utf-8")
|
||
return 0
|
||
|
||
|
||
def _guess_image_extension(content_type: str) -> str:
|
||
"""根据图片的 MIME 类型推断文件扩展名。"""
|
||
mapping = {
|
||
"image/png": ".png",
|
||
"image/jpeg": ".jpg",
|
||
"image/jpg": ".jpg",
|
||
"image/gif": ".gif",
|
||
"image/bmp": ".bmp",
|
||
"image/tiff": ".tiff",
|
||
"image/webp": ".webp",
|
||
"image/svg+xml": ".svg",
|
||
}
|
||
return mapping.get(content_type, "")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
raise SystemExit(main())
|