gaokao-volunteer-system/scripts/gaokao-visual-report-v2.py

"""
高考志愿填报可视化报告生成器 V2.0
支持：雷达图、热力图、对比表、风险检测
新增：HTML导出、PDF导出、精美样式
"""

from datetime import datetime

# 尝试导入可选依赖
try:
    from jinja2 import Template

    HAS_JINJA = True
except ImportError:
    HAS_JINJA = False
    print("警告: jinja2 未安装，HTML模板功能将受限")

try:
    from weasyprint import HTML  # type: ignore[import-untyped]

    HAS_WEASYPRINT = True
except ImportError:
    HAS_WEASYPRINT = False
    print("警告: weasyprint 未安装，PDF导出功能不可用")


def generate_student_radar(student_profile):
    """
    生成考生画像雷达图数据
    """
    scores = {
        "兴趣匹配度": student_profile.get("interest_match", 0),
        "能力匹配度": student_profile.get("ability_match", 0),
        "就业匹配度": student_profile.get("employment_match", 0),
        "家庭适配度": student_profile.get("family_match", 0),
    }

    # 综合得分计算
    weighted_score = (
        scores["兴趣匹配度"] * 0.3
        + scores["能力匹配度"] * 0.3
        + scores["就业匹配度"] * 0.25
        + scores["家庭适配度"] * 0.15
    )

    # 生成ASCII雷达图
    radar_chart = f"""
┌─────────────────────────────────────────────────────────────────┐
│                    🎯 {student_profile.get("name", "考生")} 画像雷达图                          │
├─────────────────────────────────────────────────────────────────┤
│                                                                  │
│                        兴趣匹配度                                │
│                             {scores["兴趣匹配度"]:>3}/10                             │
│                              │                                   │
│              {min(scores["能力匹配度"], 10):>2}/10 ─────────┼───────── {min(scores["就业匹配度"], 10):>2}/10          │
│                能力匹配度     │     就业匹配度                    │
│                               │                                   │
│                         【{weighted_score:.1f}/10】                           │
│                         综合得分                                │
│                               │                                   │
│                      {scores["家庭适配度"]:>3}/10                                   │
│                        家庭适配度                                │
│                                                                  │
├─────────────────────────────────────────────────────────────────┤
│  评分说明：                                                      │
│  • 兴趣匹配：基于霍兰德职业兴趣测试结果                        │
│  • 能力匹配：学科成绩与专业要求的匹配度                        │
│  • 就业匹配：专业就业前景评估                                  │
│  • 家庭适配：经济条件、地域要求的符合度                        │
└─────────────────────────────────────────────────────────────────┘
"""
    return radar_chart, weighted_score, scores


def generate_school_comparison(volunteer_list):
    """
    生成院校对比决策表
    """
    table = """
┌──────────────────────────────────────────────────────────────────────┐
│                    🎯 志愿方案可视化对比                              │
├────────────┬────────────┬──────────┬─────────┬─────────┬───────────┤
│   志愿类型  │    院校    │   专业   │ 录取概率│ 匹配指数│  推荐指数 │
├────────────┼────────────┼──────────┼─────────┼─────────┼───────────┤
"""

    for idx, vol in enumerate(volunteer_list, 1):
        v_type = vol.get("type", "稳")
        emoji = {"冲": "🔴", "稳": "🟡", "保": "🟢"}.get(v_type, "⚪")

        prob_bar = "█" * int(vol.get("probability", 0) / 10) + "░" * (
            10 - int(vol.get("probability", 0) / 10)
        )
        match_score = vol.get("match_score", 0)
        stars = "⭐" * int(match_score / 20) + "☆" * (5 - int(match_score / 20))

        table += f"""│ {emoji} {v_type:>2} {idx:>2} │ {vol.get("school", "待定"):<10} │ {vol.get("major", "待定"):<8} │ {prob_bar} │   {match_score:>3}   │ {stars} │
├────────────┼────────────┼──────────┼─────────┼─────────┼───────────┤
"""

    table += """│                                                                  │
│ 图例：                                                           │
│ 录取概率 █████ 90%+  ████░ 80%+  ███░░ 60%+  ██░░░ 40%+  █░░░░ <20%│
│ 匹配指数  90-100 完美  80-90 高度  70-80 中度  <70 需谨慎          │
└──────────────────────────────────────────────────────────────────┘
"""
    return table


def generate_major_heatmap(majors):
    """
    生成专业匹配度热力图
    """
    heatmap = """
┌──────────────────────────────────────────────────────────────────────┐
│              🔥 兴趣-专业匹配度热力图                                 │
├──────────────────────────────────────────────────────────────────────┤
│                                                                      │
"""

    for major in majors:
        name = major.get("name", "")
        score = major.get("match_score", 0)

        # 生成热力条
        if score >= 90:
            bar = "█" * 20
            status = "强烈推荐"
        elif score >= 80:
            bar = "█" * 17 + "░" * 3
            status = "推荐选择"
        elif score >= 70:
            bar = "█" * 14 + "░" * 6
            status = "可以考虑"
        elif score >= 60:
            bar = "█" * 11 + "░" * 9
            status = "谨慎考虑"
        elif score >= 40:
            bar = "█" * 8 + "░" * 12
            status = "不太建议"
        else:
            bar = "█" * 5 + "░" * 15
            status = "不推荐"

        heatmap += f"│  {name:<12} {bar}  {score:>3}%  {status:<12} │\n"

    heatmap += """│                                                                      │
│ 热力指数：████ 90-100%  ████ 80-90%  ███░ 70-80%  ██░░ 60-70%        │
│            █░░░ 30-60%  ░░░░ <30% (不推荐)                          │
└──────────────────────────────────────────────────────────────────────┘
"""
    return heatmap


def detect_risks(student_profile, volunteer_list):
    """
    智能风险检测
    """
    risks = []

    # 检查位次差距
    for vol in volunteer_list:
        if vol.get("type") == "冲":
            if vol.get("probability", 0) < 30:
                risks.append({
                    "level": "warning",
                    "item": f"{vol.get('school')}录取概率过低",
                    "desc": f"录取概率仅{vol.get('probability')}%，需要增加相近备选",
                })

    # 检查学科匹配
    weak_subjects = student_profile.get("weak_subjects", [])
    for vol in volunteer_list:
        required = vol.get("required_subjects", [])
        for subj in required:
            if subj in weak_subjects:
                risks.append({
                    "level": "danger",
                    "item": f"{vol.get('school')}-{vol.get('major')}学科不匹配",
                    "desc": f"该专业需要{subj}，但考生{subj}为弱项",
                })

    # 检查梯度合理性
    types_count = {"冲": 0, "稳": 0, "保": 0}
    for vol in volunteer_list:
        v_type = vol.get("type", "稳")
        types_count[v_type] = types_count.get(v_type, 0) + 1

    total = len(volunteer_list)
    if total > 0:
        if types_count["保"] / total < 0.2:
            risks.append({
                "level": "danger",
                "item": "保底志愿不足",
                "desc": f"保底志愿仅占{types_count['保'] / total * 100:.0f}%，建议至少30%",
            })

    # 生成风险报告
    risk_report = """
┌──────────────────────────────────────────────────────────────────────┐
│              🚦 志愿填报风险检测报告                                  │
├──────────────────────────────────────────────────────────────────────┤
│                                                                      │
"""

    danger_list = [r for r in risks if r["level"] == "danger"]
    warning_list = [r for r in risks if r["level"] == "warning"]

    if danger_list:
        risk_report += (
            "│  🔴 高风险项目（必须修改）：                                        │\n"
        )
        for risk in danger_list:
            risk_report += f"│     ✗ {risk['item']:<30}                            │\n"
            risk_report += f"│       → {risk['desc']:<50}                │\n"
            risk_report += "│                                                                      │\n"

    if warning_list:
        risk_report += (
            "│  🟡 中风险项目（建议调整）：                                        │\n"
        )
        for risk in warning_list:
            risk_report += f"│     ⚠ {risk['item']:<30}                            │\n"
            risk_report += f"│       → {risk['desc']:<50}                │\n"
            risk_report += "│                                                                      │\n"

    if not risks:
        risk_report += (
            "│  🟢 恭喜！未检测到高风险项目，当前方案可以安全填报               │\n"
        )

    risk_report += """│                                                                      │
└──────────────────────────────────────────────────────────────────────┘
"""
    return risk_report, risks


def generate_html_report(student_data, volunteer_list, output_file=None):
    """
    生成HTML可视化报告
    """
    if not HAS_JINJA:
        print("错误: 需要安装 jinja2 才能生成HTML报告")
        print("运行: pip3 install --user jinja2")
        return None

    # 计算雷达图数据
    radar_data = {
        "interest": student_data.get("interest_match", 0),
        "ability": student_data.get("ability_match", 0),
        "employment": student_data.get("employment_match", 0),
        "family": student_data.get("family_match", 0),
    }
    radar_data["average"] = sum(radar_data.values()) / 4

    # HTML模板
    html_template = """<!DOCTYPE html>
<html lang="zh-CN">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>高考志愿填报方案 - {{ name }}</title>
    <script src="https://cdn.jsdelivr.net/npm/chart.js"></script>
    <style>
        * { margin: 0; padding: 0; box-sizing: border-box; }
        body {
            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, "Noto Sans SC", sans-serif;
            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
            min-height: 100vh;
            padding: 20px;
            color: #333;
        }
        .container {
            max-width: 1200px;
            margin: 0 auto;
            background: white;
            border-radius: 20px;
            box-shadow: 0 20px 60px rgba(0,0,0,0.3);
            overflow: hidden;
        }
        .header {
            background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
            color: white;
            padding: 40px;
            text-align: center;
        }
        .header h1 { font-size: 2.5em; margin-bottom: 10px; }
        .header .subtitle { font-size: 1.2em; opacity: 0.9; }
        .info-grid {
            display: grid;
            grid-template-columns: repeat(auto-fit, minmax(200px, 1fr));
            gap: 20px;
            padding: 30px;
            background: #f8f9fa;
        }
        .info-card {
            background: white;
            padding: 20px;
            border-radius: 12px;
            box-shadow: 0 2px 10px rgba(0,0,0,0.1);
        }
        .info-card .label { color: #666; font-size: 0.9em; margin-bottom: 5px; }
        .info-card .value { font-size: 1.5em; font-weight: bold; color: #667eea; }
        .section {
            padding: 40px;
            border-bottom: 1px solid #eee;
        }
        .section:last-child { border-bottom: none; }
        .section-title {
            font-size: 1.8em;
            color: #333;
            margin-bottom: 20px;
            display: flex;
            align-items: center;
            gap: 10px;
        }
        .chart-container {
            background: #f8f9fa;
            padding: 30px;
            border-radius: 12px;
            margin: 20px 0;
        }
        .school-table {
            width: 100%;
            border-collapse: collapse;
            margin: 20px 0;
        }
        .school-table th, .school-table td {
            padding: 15px;
            text-align: left;
            border-bottom: 1px solid #eee;
        }
        .school-table th {
            background: #667eea;
            color: white;
        }
        .school-table tr:hover { background: #f8f9fa; }
        .badge {
            display: inline-block;
            padding: 5px 15px;
            border-radius: 20px;
            font-size: 0.85em;
            font-weight: bold;
        }
        .badge-red { background: #ffebee; color: #c62828; }
        .badge-yellow { background: #fff3e0; color: #ef6c00; }
        .badge-green { background: #e8f5e9; color: #2e7d32; }
        .heatmap { display: flex; flex-direction: column; gap: 10px; }
        .heat-item {
            display: flex;
            align-items: center;
            gap: 15px;
            padding: 15px;
            background: white;
            border-radius: 8px;
            box-shadow: 0 2px 5px rgba(0,0,0,0.1);
        }
        .heat-bar {
            flex: 1;
            height: 20px;
            background: linear-gradient(90deg, #ff6b6b 0%, #ffd93d 50%, #6bcf7f 100%);
            border-radius: 10px;
            position: relative;
        }
        .heat-marker {
            position: absolute;
            top: -5px;
            width: 30px;
            height: 30px;
            background: #667eea;
            border-radius: 50%;
            transform: translateX(-50%);
            box-shadow: 0 2px 10px rgba(102,126,234,0.5);
        }
        .risk-list { list-style: none; }
        .risk-item {
            padding: 15px;
            margin: 10px 0;
            border-radius: 8px;
            display: flex;
            align-items: center;
            gap: 10px;
        }
        .risk-danger { background: #ffebee; border-left: 4px solid #c62828; }
        .risk-warning { background: #fff3e0; border-left: 4px solid #ef6c00; }
        .risk-safe { background: #e8f5e9; border-left: 4px solid #2e7d32; }
        .footer {
            background: #333;
            color: white;
            padding: 30px;
            text-align: center;
        }
        @media print {
            body { background: white; }
            .container { box-shadow: none; }
        }
    </style>
</head>
<body>
    <div class="container">
        <div class="header">
            <h1>🎓 高考志愿填报方案</h1>
            <p class="subtitle">{{ name }} | {{ province }} | {{ score }}分 | 位次: {{ rank }}</p>
            <p style="margin-top: 10px; opacity: 0.8;">生成时间: {{ generated_at }}</p>
        </div>
        
        <div class="info-grid">
            <div class="info-card">
                <div class="label">考生姓名</div>
                <div class="value">{{ name }}</div>
            </div>
            <div class="info-card">
                <div class="label">高考总分</div>
                <div class="value">{{ score }}分</div>
            </div>
            <div class="info-card">
                <div class="label">全省位次</div>
                <div class="value">{{ rank }}</div>
            </div>
            <div class="info-card">
                <div class="label">综合匹配</div>
                <div class="value">{{ radar.average }}/10</div>
            </div>
        </div>
        
        <div class="section">
            <h2 class="section-title">📊 考生画像分析</h2>
            <div class="chart-container">
                <canvas id="radarChart" width="400" height="300"></canvas>
            </div>
        </div>
        
        <div class="section">
            <h2 class="section-title">🎯 志愿方案对比</h2>
            <table class="school-table">
                <thead>
                    <tr>
                        <th>类型</th>
                        <th>院校</th>
                        <th>专业</th>
                        <th>录取概率</th>
                        <th>匹配度</th>
                    </tr>
                </thead>
                <tbody>
                    {% for vol in volunteers %}
                    <tr>
                        <td>
                            {% if vol.type == '冲' %}
                            <span class="badge badge-red">🔴 冲</span>
                            {% elif vol.type == '稳' %}
                            <span class="badge badge-yellow">🟡 稳</span>
                            {% else %}
                            <span class="badge badge-green">🟢 保</span>
                            {% endif %}
                        </td>
                        <td>{{ vol.school }}</td>
                        <td>{{ vol.major }}</td>
                        <td>
                            <div style="background: #eee; border-radius: 10px; overflow: hidden;">
                                <div style="background: linear-gradient(90deg, #ff6b6b, #ffd93d, #6bcf7f); 
                                            height: 20px; width: {{ vol.probability }}%;
                                            display: flex; align-items: center; justify-content: center;
                                            color: white; font-size: 0.8em; font-weight: bold;">
                                    {{ vol.probability }}%
                                </div>
                            </div>
                        </td>
                        <td>{{ vol.match_score }}分</td>
                    </tr>
                    {% endfor %}
                </tbody>
            </table>
        </div>
        
        <div class="section">
            <h2 class="section-title">🔥 专业匹配度热力图</h2>
            <div class="heatmap">
                {% for major in majors %}
                <div class="heat-item">
                    <span style="width: 120px; font-weight: bold;">{{ major.name }}</span>
                    <div class="heat-bar">
                        <div class="heat-marker" style="left: {{ major.match_score }}%;"></div>
                    </div>
                    <span style="width: 60px; text-align: right; font-weight: bold;">{{ major.match_score }}%</span>
                </div>
                {% endfor %}
            </div>
        </div>
        
        <div class="section">
            <h2 class="section-title">🚦 风险检测报告</h2>
            {% if risks %}
            <ul class="risk-list">
                {% for risk in risks %}
                <li class="risk-item {% if risk.level == 'danger' %}risk-danger{% elif risk.level == 'warning' %}risk-warning{% else %}risk-safe{% endif %}">
                    {% if risk.level == 'danger' %}✗{% elif risk.level == 'warning' %}⚠{% else %}✓{% endif %}
                    <div>
                        <strong>{{ risk.item }}</strong>
                        <p style="color: #666; margin-top: 5px;">{{ risk.desc }}</p>
                    </div>
                </li>
                {% endfor %}
            </ul>
            {% else %}
            <div class="risk-item risk-safe">
                ✓ <strong>恭喜！当前方案未检测到高风险，可以安全填报</strong>
            </div>
            {% endif %}
        </div>
        
        <div class="footer">
            <p>© 2026 高考志愿填报智能系统</p>
            <p style="margin-top: 10px; opacity: 0.7;">本报告仅供参考，请以官方信息为准</p>
        </div>
    </div>
    
    <script>
        // 雷达图
        const ctx = document.getElementById('radarChart').getContext('2d');
        new Chart(ctx, {
            type: 'radar',
            data: {
                labels: ['兴趣匹配', '能力匹配', '就业匹配', '家庭适配'],
                datasets: [{
                    label: '匹配度',
                    data: [{{ radar.interest }}, {{ radar.ability }}, {{ radar.employment }}, {{ radar.family }}],
                    backgroundColor: 'rgba(102, 126, 234, 0.2)',
                    borderColor: 'rgba(102, 126, 234, 1)',
                    borderWidth: 2,
                    pointBackgroundColor: 'rgba(102, 126, 234, 1)',
                    pointSize: 6
                }]
            },
            options: {
                scales: {
                    r: {
                        beginAtZero: true,
                        max: 100,
                        ticks: {
                            stepSize: 20
                        }
                    }
                },
                plugins: {
                    legend: { display: false }
                }
            }
        });
    </script>
</body>
</html>"""

    template = Template(html_template)

    # 准备数据
    majors = [
        {"name": vol["major"], "match_score": vol["match_score"]}
        for vol in volunteer_list
    ]
    _, risks = detect_risks(student_data, volunteer_list)

    html_content = template.render(
        name=student_data.get("name", "考生"),
        province=student_data.get("province", "未知"),
        score=student_data.get("score", 0),
        rank=student_data.get("rank", 0),
        radar=radar_data,
        volunteers=volunteer_list,
        majors=majors,
        risks=risks,
        generated_at=datetime.now().strftime("%Y年%m月%d日 %H:%M"),
    )

    # 保存HTML
    if output_file is None:
        output_file = f"/tmp/gaokao_report_{student_data.get('name', 'unknown')}_{datetime.now().strftime('%Y%m%d_%H%M%S')}.html"

    with open(output_file, "w", encoding="utf-8") as f:
        f.write(html_content)

    print(f"✓ HTML报告已生成: {output_file}")
    return output_file


def generate_pdf_from_html(html_file, pdf_file=None):
    """
    将HTML转换为PDF
    """
    if not HAS_WEASYPRINT:
        print("错误: 需要安装 weasyprint 才能生成PDF")
        print("运行: pip3 install --user weasyprint")
        return None

    if pdf_file is None:
        pdf_file = html_file.replace(".html", ".pdf")

    try:
        HTML(filename=html_file).write_pdf(pdf_file)
        print(f"✓ PDF报告已生成: {pdf_file}")
        return pdf_file
    except Exception as e:
        print(f"✗ PDF生成失败: {e}")
        return None


def generate_visual_report(student_data, volunteer_list, output_format="all"):
    """
    生成完整可视化报告（支持多种格式）

    参数:
        student_data: 考生数据
        volunteer_list: 志愿列表
        output_format: 输出格式 ('md', 'html', 'pdf', 'all')

    返回:
        生成的文件路径列表
    """
    results = []
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    base_name = f"gaokao_report_{student_data.get('name', 'unknown')}_{timestamp}"

    # 1. 生成Markdown报告
    if output_format in ["md", "all"]:
        md_content = f"""# 高考志愿填报方案报告

**生成时间**: {datetime.now().strftime("%Y年%m月%d日 %H:%M")}  
**考生姓名**: {student_data.get("name", "考生")}  
**考生省份**: {student_data.get("province", "未知")}  
**高考总分**: {student_data.get("score", 0)}分  
**全省位次**: {student_data.get("rank", 0)}名  

---

## 一、考生画像雷达图

"""
        radar_chart, weighted_score, scores = generate_student_radar(student_data)
        md_content += radar_chart
        md_content += f"\n**综合匹配指数**: {weighted_score:.1f}/10\n\n"

        md_content += "## 二、志愿方案可视化对比\n\n"
        md_content += generate_school_comparison(volunteer_list)

        md_content += "## 三、专业匹配度热力图\n\n"
        majors = [
            {"name": vol["major"], "match_score": vol["match_score"]}
            for vol in volunteer_list
        ]
        md_content += generate_major_heatmap(majors)

        md_content += "## 四、风险检测报告\n\n"
        risk_report, _ = detect_risks(student_data, volunteer_list)
        md_content += risk_report

        md_file = f"/tmp/{base_name}.md"
        with open(md_file, "w", encoding="utf-8") as f:
            f.write(md_content)
        print(f"✓ Markdown报告已生成: {md_file}")
        results.append(md_file)

    # 2. 生成HTML报告
    html_file = None
    if output_format in ["html", "pdf", "all"]:
        html_file = f"/tmp/{base_name}.html"
        html_file = generate_html_report(student_data, volunteer_list, html_file)
        if html_file:
            results.append(html_file)

    # 3. 生成PDF报告
    if output_format in ["pdf", "all"] and html_file:
        pdf_file = f"/tmp/{base_name}.pdf"
        pdf_file = generate_pdf_from_html(html_file, pdf_file)
        if pdf_file:
            results.append(pdf_file)

    return results


# 示例使用
if __name__ == "__main__":
    # 示例考生数据
    student = {
        "name": "李明",
        "province": "浙江省",
        "score": 612,
        "rank": 15230,
        "interest_match": 85,
        "ability_match": 90,
        "employment_match": 88,
        "family_match": 95,
        "weak_subjects": ["化学", "语文"],
    }

    # 示例志愿列表
    volunteers = [
        {
            "school": "浙江大学",
            "major": "计算机类",
            "type": "冲",
            "probability": 35,
            "match_score": 95,
            "required_subjects": ["数学", "物理"],
        },
        {
            "school": "杭州电子科技大学",
            "major": "计算机类",
            "type": "稳",
            "probability": 70,
            "match_score": 92,
            "required_subjects": ["数学", "物理"],
        },
        {
            "school": "浙江工业大学",
            "major": "软件工程",
            "type": "稳",
            "probability": 80,
            "match_score": 88,
            "required_subjects": ["数学"],
        },
        {
            "school": "浙江理工大学",
            "major": "软件工程",
            "type": "保",
            "probability": 95,
            "match_score": 82,
            "required_subjects": ["数学"],
        },
    ]

    print("=" * 60)
    print("高考志愿填报可视化报告生成器 V2.0")
    print("=" * 60)
    print()

    # 生成所有格式
    files = generate_visual_report(student, volunteers, output_format="all")

    print()
    print("=" * 60)
    print("生成完成！文件列表：")
    print("=" * 60)
    for f in files:
        print(f"  • {f}")
    print()
    print("提示：")
    print("  • HTML文件可直接在浏览器中打开")
    print("  • PDF文件可直接打印或分享")
    print("  • Markdown文件可编辑和转换")