elc-invoice-engine-cli 0.1.0__py3-none-any.whl

cli/config.py

@@ -0,0 +1,28 @@
+import os
+from pathlib import Path
+from dotenv import load_dotenv
+
+# 从工程根目录的 .env.local → .env 依次加载
+_root = Path(__file__).parent.parent
+load_dotenv(_root / ".env.local")
+load_dotenv(_root / ".env")
+
+BASE_URL: str = os.getenv("BASE_URL", "http://localhost:12007/xm-demo").rstrip("/")
+REQUEST_TIMEOUT: int = int(os.getenv("REQUEST_TIMEOUT", "30"))
+
+SSO_CONFIG: dict = {
+    "app_id":      os.getenv("SSO_APP_ID", "sso"),
+    "secret":      os.getenv("SSO_SECRET", ""),
+    "domain":      os.getenv("SSO_DOMAIN", "kingdee-fpy"),
+    "mobile":      os.getenv("SSO_MOBILE", ""),
+    "username":    os.getenv("SSO_USERNAME", ""),
+    "work_number": os.getenv("SSO_WORK_NUMBER", ""),
+    "org_num":     os.getenv("SSO_ORG_NUM", ""),
+}
+
+DEFAULT_HEADERS: dict = {
+    "Content-Type":   "application/json",
+    "Accept-Language": "zh-CN",
+    "Authorization":  "",
+    "X-Company-ID":   os.getenv("X_COMPANY_ID", ""),
+}

cli/export/excel_writer.py

@@ -0,0 +1,110 @@
+"""Excel 模版填写：将映射后的发票数据写入 openpyxl workbook。"""
+from __future__ import annotations
+
+from pathlib import Path
+from openpyxl import load_workbook
+
+
+def write_excel(
+    invoices: list[dict],
+    template_path: str,
+    output_path: str,
+    mapping: dict,
+) -> int:
+    """
+    将发票数据按映射表写入 Excel 模版。
+
+    invoices: 每个元素包含：
+      - _header: {col_name: value}  表头信息字段
+      - _lines:  [{col_name: value}, ...]  行项目字段
+    mapping: load_mapping() 返回的映射表
+    返回写入的发票数量。
+    """
+    wb = load_workbook(template_path)
+    header_ws = wb["表头信息"]
+    line_ws   = wb["商品详情"]
+
+    # 读取模版列顺序（第1行英文列名）
+    header_cols = [c.value for c in header_ws[1]]
+    line_cols   = [c.value for c in line_ws[1]]
+
+    header_row = 3  # 第1行英文列名，第2行中文说明，第3行起写数据
+    line_row   = 3
+
+    for inv in invoices:
+        flat   = inv.get("_header", {})
+        lines  = inv.get("_lines", [])
+
+        # 写表头信息行
+        for col_idx, col_name in enumerate(header_cols, 1):
+            if col_name:
+                header_ws.cell(row=header_row, column=col_idx, value=flat.get(col_name))
+        header_row += 1
+
+        # 写商品详情行
+        for line in lines:
+            for col_idx, col_name in enumerate(line_cols, 1):
+                if col_name:
+                    line_ws.cell(row=line_row, column=col_idx, value=line.get(col_name))
+            line_row += 1
+
+    Path(output_path).parent.mkdir(parents=True, exist_ok=True)
+    wb.save(output_path)
+    return len(invoices)
+
+
+def apply_mapping(invoice_json: dict, xml_fields: dict, mapping: dict) -> dict:
+    """
+    将一条发票的 JSON + XML 字段按映射表合并为 {col_name: value} 扁平字典。
+    返回 {"_header": {...}, "_lines": [{...}, ...]}
+    """
+    header_map = mapping.get("header_mapping", {})
+    line_map   = mapping.get("line_mapping", {})
+
+    # 表头字段
+    header_flat: dict = {}
+    for col, rule in header_map.items():
+        source = rule.get("source")
+        if source == "json":
+            header_flat[col] = _get_nested(invoice_json, rule.get("path", ""))
+        elif source == "xml":
+            header_flat[col] = xml_fields.get(col)
+        # source == "none" 留空
+
+    # 行项目字段
+    lines_out: list[dict] = []
+    raw_lines = invoice_json.get("lines", [])
+    for raw_line in raw_lines:
+        line_flat: dict = {}
+        for col, rule in line_map.items():
+            source = rule.get("source")
+            if source == "json":
+                line_flat[col] = _get_nested(raw_line, rule.get("path", ""))
+            elif source == "xml":
+                line_flat[col] = xml_fields.get(col)
+        # invoiceNo 注入到每行
+        line_flat["*invoiceNo"] = invoice_json.get("invoiceNo")
+        lines_out.append(line_flat)
+
+    return {"_header": header_flat, "_lines": lines_out}
+
+
+def _get_nested(obj: dict, path: str) -> object:
+    """支持点分隔路径，如 'taxInfo.0.taxRate'。"""
+    if not path:
+        return None
+    parts = path.split(".")
+    cur = obj
+    for p in parts:
+        if cur is None:
+            return None
+        if isinstance(cur, list):
+            try:
+                cur = cur[int(p)]
+            except (ValueError, IndexError):
+                return None
+        elif isinstance(cur, dict):
+            cur = cur.get(p)
+        else:
+            return None
+    return cur

cli/export/mapping_store.py

@@ -0,0 +1,46 @@
+"""映射表本地持久化：读写 ~/.elc/mappings/<template_md5>.json。"""
+from __future__ import annotations
+
+import hashlib
+import json
+from datetime import datetime
+from pathlib import Path
+
+MAPPINGS_DIR = Path.home() / ".elc" / "mappings"
+
+
+def _md5(template_path: str) -> str:
+    return hashlib.md5(Path(template_path).read_bytes()).hexdigest()
+
+
+def load_mapping(template_path: str) -> dict | None:
+    """读取缓存映射表；模版内容变了（MD5 不同）返回 None。"""
+    MAPPINGS_DIR.mkdir(parents=True, exist_ok=True)
+    key = _md5(template_path)
+    path = MAPPINGS_DIR / f"{key}.json"
+    if not path.exists():
+        return None
+    try:
+        return json.loads(path.read_text())
+    except Exception:
+        return None
+
+
+def save_mapping(template_path: str, mapping: dict) -> None:
+    """保存映射表，附带元信息。"""
+    MAPPINGS_DIR.mkdir(parents=True, exist_ok=True)
+    key = _md5(template_path)
+    payload = {
+        "template_path": str(Path(template_path).resolve()),
+        "template_md5":  key,
+        "created_at":    datetime.now().isoformat(timespec="seconds"),
+        **mapping,
+    }
+    (MAPPINGS_DIR / f"{key}.json").write_text(
+        json.dumps(payload, ensure_ascii=False, indent=2)
+    )
+
+
+def mapping_exists(template_path: str) -> bool:
+    key = _md5(template_path)
+    return (MAPPINGS_DIR / f"{key}.json").exists()

cli/export/xml_parser.py

@@ -0,0 +1,50 @@
+"""KDUBL XML 解析：按映射表中的 XPath 批量提取字段。"""
+from __future__ import annotations
+
+from lxml import etree
+
+NSMAP = {
+    "cbc": "urn:oasis:names:specification:ubl:schema:xsd:CommonBasicComponents-2",
+    "cac": "urn:oasis:names:specification:ubl:schema:xsd:CommonAggregateComponents-2",
+    "ext": "urn:oasis:names:specification:ubl:schema:xsd:CommonExtensionComponents-2",
+}
+
+
+def _xpath_text(root: etree._Element, xpath: str) -> str | None:
+    try:
+        nodes = root.xpath(xpath, namespaces=NSMAP)
+        if not nodes:
+            return None
+        node = nodes[0]
+        return node.text.strip() if hasattr(node, "text") and node.text else str(node).strip()
+    except Exception:
+        return None
+
+
+def extract_fields(xml_bytes: bytes, xpath_map: dict[str, str]) -> dict[str, str | None]:
+    """按 xpath_map 批量提取字段，返回 {col_name: value}。"""
+    root = etree.fromstring(xml_bytes)
+    return {col: _xpath_text(root, xpath) for col, xpath in xpath_map.items()}
+
+
+def flatten_xml_fields(xml_bytes: bytes) -> list[dict]:
+    """提取 XML 中所有叶子节点的 XPath + 值，用于 AI 推断映射时的上下文。"""
+    root = etree.fromstring(xml_bytes)
+    results = []
+
+    def _walk(node: etree._Element, path: str) -> None:
+        tag = etree.QName(node.tag).localname if node.tag else ""
+        current_path = f"{path}/{tag}" if path else tag
+        # 属性
+        for attr_name, attr_val in node.attrib.items():
+            attr_local = etree.QName(attr_name).localname if "{" in attr_name else attr_name
+            results.append({"xpath": f"{current_path}[@{attr_local}]", "value": attr_val})
+        # 文本值
+        text = (node.text or "").strip()
+        if text:
+            results.append({"xpath": current_path, "value": text})
+        for child in node:
+            _walk(child, current_path)
+
+    _walk(root, "")
+    return results

cli/main.py

@@ -0,0 +1,29 @@
+"""CLI 主入口。"""
+from __future__ import annotations
+
+import typer
+from cli.commands import describe, exchange, export, invoice, invoice_doc, login, party, product, tax
+
+app = typer.Typer(
+    name="elc",
+    help="ELC Invoice Engine CLI — V2 接口命令行工具",
+    no_args_is_help=True,
+)
+
+# 认证
+app.add_typer(login.app,        name="login")
+
+# 上下文（AI 使用）
+app.add_typer(describe.app,     name="describe")
+
+# 核心业务
+app.add_typer(invoice.app,      name="invoice-request")
+app.add_typer(invoice_doc.app,  name="invoice")
+app.add_typer(party.app,        name="party")
+app.add_typer(product.app,      name="product")
+app.add_typer(tax.app,          name="tax")
+app.add_typer(exchange.app,     name="exchange")
+app.add_typer(export.app,       name="export")
+
+if __name__ == "__main__":
+    app()

cli/output.py

@@ -0,0 +1,44 @@
+"""输出格式化：table / json / raw。"""
+from __future__ import annotations
+
+import json as _json
+import typer
+from rich.console import Console
+from rich.table import Table
+
+console = Console()
+
+
+def print_json(data: dict | list) -> None:
+    console.print_json(_json.dumps(data, ensure_ascii=False, indent=2))
+
+
+def print_table(rows: list[dict], columns: list[str] | None = None) -> None:
+    if not rows:
+        typer.echo("(无数据)")
+        return
+    cols = columns or list(rows[0].keys())
+    t = Table(show_header=True, header_style="bold cyan")
+    for c in cols:
+        t.add_column(c)
+    for row in rows:
+        t.add_row(*[str(row.get(c, "")) for c in cols])
+    console.print(t)
+
+
+def print_result(status: int, body: dict, output: str = "table",
+                 rows_key: str = "data", columns: list[str] | None = None) -> None:
+    """统一打印 V2 响应：output 为 'table'、'json' 或 'raw'。"""
+    if output == "json":
+        print_json(body)
+        return
+    if status >= 400:
+        console.print(f"[red]HTTP {status}[/red]  {body}")
+        return
+    data = body.get(rows_key, body)
+    if output == "table" and isinstance(data, list):
+        print_table(data, columns)
+    elif output == "table" and isinstance(data, dict):
+        print_table([data], columns)
+    else:
+        console.print(data)

cli/session.py

@@ -0,0 +1,36 @@
+"""懒加载 ApiClient：优先读取本地缓存 token，过期才重新登录。"""
+from __future__ import annotations
+
+import typer
+from rich.console import Console
+from cli.token_store import load_token, save_token
+from cli.auth import fetch_token
+from cli.client import ApiClient
+
+_console = Console()
+_client: ApiClient | None = None
+
+
+def get_client() -> ApiClient:
+    global _client
+    if _client is not None:
+        return _client
+
+    # 优先读本地缓存
+    token = load_token()
+    if not token:
+        try:
+            token, expires_in = fetch_token()
+            save_token(token, expires_in)
+        except Exception as e:
+            _console.print(f"[red]认证失败：{e}[/red]")
+            _console.print("[yellow]提示：请先运行 elc login 配置凭证[/yellow]")
+            raise typer.Exit(1)
+
+    _client = ApiClient(token=token)
+    return _client
+
+
+def reset_client() -> None:
+    global _client
+    _client = None

cli/token_store.py

@@ -0,0 +1,35 @@
+"""Token 本地持久化：读写 ~/.elc/token.json。"""
+from __future__ import annotations
+
+import json
+import time
+from pathlib import Path
+
+_TOKEN_FILE = Path.home() / ".elc" / "token.json"
+
+
+def save_token(access_token: str, expires_in: int) -> None:
+    _TOKEN_FILE.parent.mkdir(parents=True, exist_ok=True)
+    payload = {
+        "access_token": access_token,
+        "expires_at": int(time.time()) + expires_in - 60,  # 提前 60s 过期
+    }
+    _TOKEN_FILE.write_text(json.dumps(payload, indent=2))
+
+
+def load_token() -> str | None:
+    """读取本地 token；过期或不存在返回 None。"""
+    if not _TOKEN_FILE.exists():
+        return None
+    try:
+        payload = json.loads(_TOKEN_FILE.read_text())
+        if time.time() < payload.get("expires_at", 0):
+            return payload["access_token"]
+    except Exception:
+        pass
+    return None
+
+
+def clear_token() -> None:
+    if _TOKEN_FILE.exists():
+        _TOKEN_FILE.unlink()