#!/usr/bin/env python3
"""
chartjs-reporter: generate_report.py
将结构化数据转换为自包含的 Chart.js HTML 可视化报告。
用法：
python generate_report.py --title "标题" --data '{"kpis":[...],"charts":[...]}' --output report.html
或直接 import build_report 函数使用。
"""
import argparse
import json
import sys
from datetime import datetime
from pathlib import Path
# ── 默认调色板 ──────────────────────────────────────────────────────────────
PALETTE = [
"#38bdf8", "#818cf8", "#34d399", "#fbbf24", "#fb923c",
"#f472b6", "#a78bfa", "#4ade80", "#60a5fa", "#facc15",
]
KPI_COLORS = {
"blue": ("var(--accent)", "#1e293b"),
"green": ("var(--green)", "#1e293b"),
"yellow": ("var(--yellow)", "#1e293b"),
"purple": ("var(--accent2)", "#1e293b"),
"red": ("var(--red)", "#1e293b"),
}
# ── KPI 卡片 HTML ────────────────────────────────────────────────────────────
def _render_kpis(kpis: list) -> str:
if not kpis:
return ""
cards = []
for kpi in kpis:
color_var, _ = KPI_COLORS.get(kpi.get("color", "blue"), KPI_COLORS["blue"])
cards.append(f"""
{kpi.get('label', '')}
{kpi.get('value', '')}
{kpi.get('sub', '')}
""")
return f'{"".join(cards)}\n
'
# ── 单张图表 JS 配置 ──────────────────────────────────────────────────────────
def _chart_js_config(chart: dict, idx: int) -> str:
ctype = chart.get("type", "bar")
labels = json.dumps(chart.get("labels", []), ensure_ascii=False)
title = chart.get("title", f"图表{idx+1}")
raw_datasets = chart.get("datasets", [])
# 处理水平柱状图别名
index_axis = ""
if ctype == "horizontalBar":
ctype = "bar"
index_axis = 'indexAxis: "y",'
# 为每个 dataset 自动注入颜色
datasets_js_parts = []
for i, ds in enumerate(raw_datasets):
color = PALETTE[i % len(PALETTE)]
label = json.dumps(ds.get("label", ""), ensure_ascii=False)
data = json.dumps(ds.get("data", []))
if ctype in ("doughnut", "pie"):
# 多色背景
bg_colors = json.dumps([PALETTE[j % len(PALETTE)] for j in range(len(ds.get("data", [])))])
ds_str = f"""{{
label: {label},
data: {data},
backgroundColor: {bg_colors},
borderWidth: 2,
borderColor: "#1e293b"
}}"""
elif ctype == "line":
fill = str(ds.get("fill", True)).lower()
ds_str = f"""{{
label: {label},
data: {data},
borderColor: "{color}",
backgroundColor: "{color}22",
tension: 0.4,
fill: {fill},
pointRadius: 4,
pointBackgroundColor: "{color}"
}}"""
else: # bar / grouped bar
ds_str = f"""{{
label: {label},
data: {data},
backgroundColor: "{color}",
borderRadius: 6
}}"""
datasets_js_parts.append(ds_str)
datasets_js = ",\n ".join(datasets_js_parts)
# y轴 tick 格式（数值大于 999 时加 k）
y_tick = ""
if ctype not in ("doughnut", "pie"):
y_tick = "callback: v => v >= 1000 ? (v/1000).toFixed(0)+'k' : v,"
scales_block = ""
if ctype not in ("doughnut", "pie"):
scales_block = f"""
scales: {{
x: {{ grid: {{ color: "#1e293b" }}, ticks: {{ color: "#94a3b8" }} }},
y: {{ grid: {{ color: "#334155" }}, ticks: {{ color: "#94a3b8", {y_tick} }} }}
}},"""
legend_display = "false" if len(raw_datasets) <= 1 and ctype not in ("doughnut", "pie") else "true"
legend_pos = '"right"' if ctype in ("doughnut", "pie") else '"top"'
return f"""new Chart(document.getElementById('chart{idx}'), {{
type: '{ctype}',
data: {{
labels: {labels},
datasets: [{datasets_js}]
}},
options: {{
responsive: true,
maintainAspectRatio: false,
{index_axis}
{scales_block}
plugins: {{
legend: {{ display: {legend_display}, position: {legend_pos}, labels: {{ color: "#94a3b8", font: {{ size: 12 }} }} }},
title: {{ display: false }}
}}
}}
}});"""
# ── 图表区 HTML ──────────────────────────────────────────────────────────────
def _render_charts(charts: list) -> str:
if not charts:
return ""
n = len(charts)
if n == 1:
cols = "1fr"
elif n % 3 == 0 and n >= 3:
cols = "1fr 1fr 1fr"
else:
cols = "1fr 1fr"
cards_html = []
for i, chart in enumerate(charts):
title = chart.get("title", f"图表{i+1}")
cards_html.append(f"""
""")
return f"""
{"".join(cards_html)}
"""
# ── 表格区 HTML ──────────────────────────────────────────────────────────────
def _render_tables(tables: list) -> str:
if not tables:
return ""
tables_html = []
for table in tables:
title = table.get("title", "")
columns = table.get("columns", [])
data = table.get("data", [])
highlight_col = table.get("highlight", "")
# 表头
header_cells = "".join([f"{col} | " for col in columns])
# 表体
rows_html = []
for row in data:
cells_html = []
for i, col in enumerate(columns):
value = row.get(col, "")
cell_html = f"{value} | "
# 高亮列添加样式
if highlight_col and col == highlight_col:
# 根据数值添加颜色
cell_html = f'{value} | '
cells_html.append(cell_html)
rows_html.append(f"{''.join(cells_html)}
")
tables_html.append(f"""
{title}
{header_cells}
{"".join(rows_html)}
""")
return f"""
{"".join(tables_html)}
"""
# ── 退货率级别判断 ─────────────────────────────────────────────────────────
def _get_rate_level(value: str) -> str:
"""根据退货率返回级别"""
try:
# 移除百分号并转为浮点数
num = float(value.replace('%', '').replace('万', '').strip())
if '万' in value:
# 处理金额情况
return 'normal'
if num >= 90:
return 'danger'
elif num >= 75:
return 'warning'
elif num >= 65:
return 'medium'
else:
return 'good'
except:
return 'normal'
# ── 主构建函数 ───────────────────────────────────────────────────────────────
def build_report(
title: str,
subtitle: str = "",
kpis: list = None,
charts: list = None,
tables: list = None,
footer: str = "",
) -> str:
kpis = kpis or []
charts = charts or []
tables = tables or []
footer = footer or f"由 chartjs-reporter 技能生成 · {datetime.now().strftime('%Y-%m-%d %H:%M')}"
kpi_html = _render_kpis(kpis)
chart_html = _render_charts(charts)
table_html = _render_tables(tables)
chart_scripts = "\n ".join(_chart_js_config(c, i) for i, c in enumerate(charts))
return f"""
{title}
📊 {title}
{subtitle}
{kpi_html}
{chart_html}
{table_html}
"""
# ── CLI 入口 ─────────────────────────────────────────────────────────────────
def main():
parser = argparse.ArgumentParser(description="生成 Chart.js HTML 可视化报告")
parser.add_argument("--title", required=True, help="报告标题")
parser.add_argument("--subtitle", default="", help="副标题")
parser.add_argument("--data", required=True, help='JSON 字符串，含 kpis 和 charts 字段')
parser.add_argument("--output", default="report.html", help="输出 HTML 文件路径")
parser.add_argument("--footer", default="", help="页脚文字")
args = parser.parse_args()
try:
data = json.loads(args.data)
except json.JSONDecodeError as e:
print(f"❌ --data 不是合法 JSON：{e}", file=sys.stderr)
sys.exit(1)
html = build_report(
title=args.title,
subtitle=args.subtitle,
kpis=data.get("kpis", []),
charts=data.get("charts", []),
footer=args.footer,
)
out = Path(args.output)
out.write_text(html, encoding="utf-8")
print(f"✅ 报告已生成：{out.resolve()}")
if __name__ == "__main__":
main()