licos-dev-sdk 0.3.2__tar.gz → 0.3.3__tar.gz

{licos_dev_sdk-0.3.2 → licos_dev_sdk-0.3.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: licos-dev-sdk
-Version: 0.3.2
+Version: 0.3.3
 Summary: LICOS Dev SDK - file generation and model capability clients
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4>=4.12

{licos_dev_sdk-0.3.2 → licos_dev_sdk-0.3.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "licos-dev-sdk"
-version = "0.3.2"
+version = "0.3.3"
 description = "LICOS Dev SDK - file generation and model capability clients"
 requires-python = ">=3.10"
 dependencies = [

licos_dev_sdk-0.3.3/src/licos_dev_sdk/chart.py

@@ -0,0 +1,114 @@
+"""Chart generation — bar, line, pie, scatter, heatmap via matplotlib."""
+
+from __future__ import annotations
+
+from functools import lru_cache
+
+import matplotlib
+matplotlib.use("Agg")  # Non-interactive backend
+from matplotlib import font_manager
+import matplotlib.pyplot as plt
+import numpy as np
+
+from ._utils import resolve_output_path
+
+
+_CJK_FONT_CANDIDATES = (
+    "Noto Sans CJK SC",
+    "Noto Sans SC",
+    "Microsoft YaHei",
+    "SimHei",
+    "WenQuanYi Zen Hei",
+    "Source Han Sans CN",
+)
+
+
+@lru_cache(maxsize=1)
+def _resolve_cjk_font() -> str:
+    installed_fonts = {font.name.casefold(): font.name for font in font_manager.fontManager.ttflist}
+    for candidate in _CJK_FONT_CANDIDATES:
+        installed_name = installed_fonts.get(candidate.casefold())
+        if installed_name:
+            return installed_name
+    choices = ", ".join(_CJK_FONT_CANDIDATES)
+    raise RuntimeError(
+        "Chinese chart font is unavailable. Install a CJK font before generating charts "
+        f"(supported families: {choices})."
+    )
+
+
+def create_chart(chart_type: str, data: dict, filename: str, *,
+                 output_dir: str | None = None, format: str = "png",
+                 title: str = "", width: int = 800, height: int = 600) -> str:
+    """Generate a chart image (PNG/SVG). Returns absolute path.
+
+    Args:
+        chart_type: "bar" | "line" | "pie" | "scatter" | "heatmap"
+        data: {
+            "labels": ["A", "B", "C"],
+            "datasets": [{"label": "Series 1", "values": [10, 20, 30]}]
+        }
+        For pie: only first dataset is used.
+        For heatmap: data["matrix"] = [[1,2],[3,4]], data["x_labels"], data["y_labels"]
+    """
+    path = resolve_output_path(filename, f".{format}", output_dir)
+    dpi = 100
+    font_name = _resolve_cjk_font()
+    chart_style = {
+        "font.family": "sans-serif",
+        "font.sans-serif": [font_name],
+        "axes.unicode_minus": False,
+    }
+    with matplotlib.rc_context(chart_style):
+        fig, ax = plt.subplots(figsize=(width / dpi, height / dpi), dpi=dpi)
+
+        labels = data.get("labels", [])
+        datasets = data.get("datasets", [])
+
+        if chart_type == "bar":
+            x = np.arange(len(labels))
+            n = len(datasets)
+            bar_width = 0.8 / max(n, 1)
+            for i, ds in enumerate(datasets):
+                offset = (i - n / 2 + 0.5) * bar_width
+                ax.bar(x + offset, ds["values"], bar_width, label=ds.get("label", ""))
+            ax.set_xticks(x)
+            ax.set_xticklabels(labels)
+            if n > 1:
+                ax.legend()
+
+        elif chart_type == "line":
+            for ds in datasets:
+                ax.plot(labels, ds["values"], marker="o", label=ds.get("label", ""))
+            if len(datasets) > 1:
+                ax.legend()
+
+        elif chart_type == "pie":
+            values = datasets[0]["values"] if datasets else []
+            ax.pie(values, labels=labels, autopct="%1.1f%%", startangle=90)
+            ax.axis("equal")
+
+        elif chart_type == "scatter":
+            for ds in datasets:
+                x_vals = ds.get("x", list(range(len(ds["values"]))))
+                ax.scatter(x_vals, ds["values"], label=ds.get("label", ""), alpha=0.7)
+            if len(datasets) > 1:
+                ax.legend()
+
+        elif chart_type == "heatmap":
+            matrix = np.array(data.get("matrix", [[]]))
+            im = ax.imshow(matrix, cmap="YlOrRd", aspect="auto")
+            fig.colorbar(im, ax=ax)
+            if "x_labels" in data:
+                ax.set_xticks(range(len(data["x_labels"])))
+                ax.set_xticklabels(data["x_labels"])
+            if "y_labels" in data:
+                ax.set_yticks(range(len(data["y_labels"])))
+                ax.set_yticklabels(data["y_labels"])
+
+        if title:
+            ax.set_title(title)
+        fig.tight_layout()
+        fig.savefig(str(path), format=format, bbox_inches="tight")
+        plt.close(fig)
+    return str(path.resolve())

{licos_dev_sdk-0.3.2 → licos_dev_sdk-0.3.3}/tests/test_document_spreadsheet.py

@@ -3,6 +3,7 @@ from __future__ import annotations
 import sys
 import tempfile
 import unittest
+import warnings
 from pathlib import Path
 
 sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))
@@ -11,10 +12,36 @@ from docx import Document
 from openpyxl import load_workbook
 from PIL import Image
 
-from licos_dev_sdk import create_docx, create_docx_from_template, create_xlsx_workbook
+from licos_dev_sdk import create_chart, create_docx, create_docx_from_template, create_xlsx_workbook
 
 
 class DocumentSpreadsheetTests(unittest.TestCase):
+    def test_create_chart_renders_chinese_text_without_missing_glyphs(self) -> None:
+        with tempfile.TemporaryDirectory() as tmp, warnings.catch_warnings(record=True) as caught:
+            warnings.simplefilter("always")
+            output = create_chart(
+                "bar",
+                {
+                    "labels": ["华东地区", "华中地区", "华南地区"],
+                    "datasets": [
+                        {"label": "预计收入（万元）", "values": [4800, 3200, 2600]},
+                        {"label": "项目ROI（%）", "values": [18, 23, 16]},
+                    ],
+                },
+                "chinese-chart",
+                output_dir=tmp,
+                title="重点行业客户投资回报分析",
+            )
+
+            self.assertTrue(Path(output).is_file())
+            self.assertGreater(Path(output).stat().st_size, 10_000)
+            missing_glyph_warnings = [
+                str(item.message)
+                for item in caught
+                if "Glyph" in str(item.message) and "missing from font" in str(item.message)
+            ]
+            self.assertEqual(missing_glyph_warnings, [])
+
     def test_create_docx_resolves_assets_from_explicit_base_path(self) -> None:
         with tempfile.TemporaryDirectory() as tmp:
             tmp_path = Path(tmp)

licos_dev_sdk-0.3.2/src/licos_dev_sdk/chart.py

@@ -1,80 +0,0 @@
-"""Chart generation — bar, line, pie, scatter, heatmap via matplotlib."""
-
-from __future__ import annotations
-
-import matplotlib
-matplotlib.use("Agg")  # Non-interactive backend
-import matplotlib.pyplot as plt
-import numpy as np
-
-from ._utils import resolve_output_path
-
-
-def create_chart(chart_type: str, data: dict, filename: str, *,
-                 output_dir: str | None = None, format: str = "png",
-                 title: str = "", width: int = 800, height: int = 600) -> str:
-    """Generate a chart image (PNG/SVG). Returns absolute path.
-
-    Args:
-        chart_type: "bar" | "line" | "pie" | "scatter" | "heatmap"
-        data: {
-            "labels": ["A", "B", "C"],
-            "datasets": [{"label": "Series 1", "values": [10, 20, 30]}]
-        }
-        For pie: only first dataset is used.
-        For heatmap: data["matrix"] = [[1,2],[3,4]], data["x_labels"], data["y_labels"]
-    """
-    path = resolve_output_path(filename, f".{format}", output_dir)
-    dpi = 100
-    fig, ax = plt.subplots(figsize=(width / dpi, height / dpi), dpi=dpi)
-
-    labels = data.get("labels", [])
-    datasets = data.get("datasets", [])
-
-    if chart_type == "bar":
-        x = np.arange(len(labels))
-        n = len(datasets)
-        bar_width = 0.8 / max(n, 1)
-        for i, ds in enumerate(datasets):
-            offset = (i - n / 2 + 0.5) * bar_width
-            ax.bar(x + offset, ds["values"], bar_width, label=ds.get("label", ""))
-        ax.set_xticks(x)
-        ax.set_xticklabels(labels)
-        if n > 1:
-            ax.legend()
-
-    elif chart_type == "line":
-        for ds in datasets:
-            ax.plot(labels, ds["values"], marker="o", label=ds.get("label", ""))
-        if len(datasets) > 1:
-            ax.legend()
-
-    elif chart_type == "pie":
-        values = datasets[0]["values"] if datasets else []
-        ax.pie(values, labels=labels, autopct="%1.1f%%", startangle=90)
-        ax.axis("equal")
-
-    elif chart_type == "scatter":
-        for ds in datasets:
-            x_vals = ds.get("x", list(range(len(ds["values"]))))
-            ax.scatter(x_vals, ds["values"], label=ds.get("label", ""), alpha=0.7)
-        if len(datasets) > 1:
-            ax.legend()
-
-    elif chart_type == "heatmap":
-        matrix = np.array(data.get("matrix", [[]]))
-        im = ax.imshow(matrix, cmap="YlOrRd", aspect="auto")
-        fig.colorbar(im, ax=ax)
-        if "x_labels" in data:
-            ax.set_xticks(range(len(data["x_labels"])))
-            ax.set_xticklabels(data["x_labels"])
-        if "y_labels" in data:
-            ax.set_yticks(range(len(data["y_labels"])))
-            ax.set_yticklabels(data["y_labels"])
-
-    if title:
-        ax.set_title(title)
-    fig.tight_layout()
-    fig.savefig(str(path), format=format, bbox_inches="tight")
-    plt.close(fig)
-    return str(path.resolve())