Auto-Finance/app/fill_consumable_doc.py

"""
将 invoice_summary.csv 填入「易耗品、出库单.doc」表格。

仅写入表格数据单元格，保留原模板字体、边框与版式。
"""

from __future__ import annotations

import argparse
import re
import shutil
from pathlib import Path

from . import get_logger
from .bot import load_invoice_data
from .config import load_config

log = get_logger("fill_consumable_doc")

CONSUMABLE_DOC_FILENAME = "易耗品、出库单.doc"

# Word COM 常量
WD_CHARACTER = 1

# 表格统一字体：宋体、五号（10.5 磅）
TABLE_FONT_NAME = "宋体"
TABLE_FONT_SIZE = 10.5


def _split_name_spec(left: str) -> tuple[str, str]:
    m = re.search(r"(\S+一批)\s*$", left)
    if m:
        return m.group(1), left[: m.start()].strip()
    parts = left.split(" ", 1)
    if len(parts) == 2:
        return parts[0], parts[1]
    return left, ""


def parse_spec_model(spec: str) -> dict[str, str]:
    spec = (spec or "").strip()
    if " 个 " not in spec:
        return {
            "product_name": spec,
            "spec": "",
            "unit": "",
            "qty": "",
            "unit_price": "",
        }

    left, right = spec.split(" 个 ", 1)
    product_name, model_spec = _split_name_spec(left.strip())
    tokens = right.split()

    qty = ""
    unit_price = ""
    if len(tokens) >= 4 and re.fullmatch(r"\d+(?:\.\d+)?", tokens[0]):
        qty, unit_price = tokens[0], tokens[1]
    elif tokens and re.fullmatch(r"\d+(?:\.\d+)?", tokens[0]):
        qty, unit_price = "1", tokens[0]

    return {
        "product_name": product_name,
        "spec": model_spec,
        "unit": "个",
        "qty": qty,
        "unit_price": unit_price,
    }


def _format_money(value: str | float) -> str:
    """单价、金额：固定保留两位小数。"""
    if value is None or value == "":
        return ""
    try:
        num = float(value)
    except (TypeError, ValueError):
        return str(value)
    return f"{num:.2f}"


def _format_cn_date(date_str: str) -> str:
    if not date_str:
        return ""
    parts = date_str.replace("-", "/").split("/")
    if len(parts) != 3:
        return date_str
    y, m, d = parts[0], str(int(parts[1])), str(int(parts[2]))
    return f"{y}年{m}月{d}日"


def _apply_font(rng) -> None:
    """将范围字体设为宋体五号（含数字与英文）。"""
    font = rng.Font
    font.Name = TABLE_FONT_NAME
    font.NameFarEast = TABLE_FONT_NAME
    font.NameAscii = TABLE_FONT_NAME
    font.NameOther = TABLE_FONT_NAME
    font.NameBi = TABLE_FONT_NAME
    font.Size = TABLE_FONT_SIZE


def _set_cell_value(cell, text: str) -> None:
    """写入单元格正文（不含末尾单元格标记）。"""
    rng = cell.Range
    rng.MoveEnd(WD_CHARACTER, -1)
    rng.Text = "" if text is None else str(text)
    _apply_font(rng)


def _normalize_table_font(tbl) -> None:
    """填写完成后统一整张表的字体。"""
    for row in tbl.Rows:
        for cell in row.Cells:
            rng = cell.Range
            rng.MoveEnd(WD_CHARACTER, -1)
            _apply_font(rng)


def _replace_date_in_doc(doc, new_date: str) -> None:
    """仅替换表头段落中的日期文字，不改动段落其余部分。"""
    if not new_date:
        return
    try:
        para = doc.Paragraphs(3)
    except Exception:
        return
    rng = para.Range
    text = rng.Text.replace("\r", "").replace("\x07", "")
    m = re.search(r"\d{4}年\d{1,2}月\d{1,2}日", text)
    if not m:
        return
    start = rng.Start + m.start()
    end = rng.Start + m.end()
    doc.Range(Start=start, End=end).Text = new_date


def fill_consumable_doc(
    csv_path: str | Path,
    doc_path: str | Path,
    config: dict | None = None,
    backup: bool = True,
) -> Path:
    csv_path = Path(csv_path)
    doc_path = Path(doc_path)
    if config is None:
        config = load_config()
    invoices = load_invoice_data(str(csv_path), config)

    if backup:
        bak = doc_path.with_suffix(doc_path.suffix + ".bak")
        shutil.copy2(doc_path, bak)

    import win32com.client

    word = win32com.client.Dispatch("Word.Application")
    word.Visible = False
    word.DisplayAlerts = 0
    doc = word.Documents.Open(str(doc_path.resolve()))

    try:
        if invoices:
            cn_date = _format_cn_date(invoices[0].get("invoice_date", ""))
            _replace_date_in_doc(doc, cn_date)

        tbl = doc.Tables(1)
        storage = config.get("consumable_storage", "躬行楼 C205")

        for i, inv in enumerate(invoices):
            row_idx = i + 2
            if row_idx > tbl.Rows.Count:
                break

            parsed = parse_spec_model(inv.get("spec_model", ""))
            amount = _format_money(inv.get("total_amount", ""))
            unit_price = _format_money(parsed["unit_price"])
            qty = parsed["qty"]
            if qty and qty.isdigit():
                qty = str(int(qty))

            values = [
                str(inv.get("seq", i + 1)),
                parsed["product_name"],
                parsed["spec"],
                parsed["unit"],
                qty,
                unit_price,
                amount,
                "",  # 购货人签字 — 保持空白
                storage,
                "",  # 领用人签字 — 保持空白
                "",  # 备注 — 保持空白，避免撑破版式
            ]

            for col_idx, val in enumerate(values, start=1):
                _set_cell_value(tbl.Cell(row_idx, col_idx), val)

        _normalize_table_font(tbl)

        doc.Save()
    finally:
        doc.Close()
        word.Quit()

    return doc_path


def fill_consumable_from_template(
    csv_path: str | Path,
    template_path: str | Path,
    output_path: str | Path,
    config: dict | None = None,
) -> Path:
    """从模板复制并填写出库单（Web 会话每次从模板重新生成）。"""
    template_path = Path(template_path)
    output_path = Path(output_path)
    if not template_path.exists():
        raise FileNotFoundError(f"出库单模板不存在: {template_path}")
    shutil.copy2(template_path, output_path)
    return fill_consumable_doc(csv_path, output_path, config=config, backup=False)


def main() -> None:
    root = Path(__file__).resolve().parents[1]
    parser = argparse.ArgumentParser(description="将发票 CSV 填入易耗品出库单")
    parser.add_argument("--csv", default=str(root / "invoice_summary.csv"))
    parser.add_argument("--doc", default=str(root / "易耗品、出库单.doc"))
    parser.add_argument("--config", default=str(root / "config.json"))
    parser.add_argument("--no-backup", action="store_true")
    args = parser.parse_args()

    cfg = None
    if Path(args.config).exists():
        import json

        with open(args.config, encoding="utf-8") as f:
            cfg = {**load_config(), **json.load(f)}
    out = fill_consumable_doc(args.csv, args.doc, config=cfg, backup=not args.no_backup)
    print(f"已填写并保存: {out}")


if __name__ == "__main__":
    main()