2311061111-lyt/autograde/generate_pdf_report.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
generate_pdf_report.py - PDF报告生成脚本
将评分结果和报告内容整合生成PDF文件
"""

import json
import argparse
import sys
import os
from datetime import datetime

# 尝试导入reportlab，如果不存在则显示错误
try:
    from reportlab.lib.pagesizes import A4
    from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
    from reportlab.lib.enums import TA_CENTER, TA_LEFT, TA_JUSTIFY
    from reportlab.platypus import (SimpleDocTemplate, Paragraph, Spacer, 
                                   Table, TableStyle, Image, PageBreak)
    from reportlab.lib import colors
    from reportlab.lib.units import cm
    REPORTLAB_AVAILABLE = True
    
except ImportError:
    REPORTLAB_AVAILABLE = False


def parse_args():    
    """解析命令行参数"""
    parser = argparse.ArgumentParser(description='PDF Report Generation Script')
    parser.add_argument('--score', '--grade', required=True, help='最终成绩文件')
    parser.add_argument('--report', required=True, help='REPORT.md文件路径')
    parser.add_argument('--frontend', required=True, help='FRONTEND.md文件路径')
    parser.add_argument('--out', required=True, help='输出PDF文件路径')
    parser.add_argument('--student-id', required=True, help='学生ID')
    parser.add_argument('--student-name', required=True, help='学生姓名')
    parser.add_argument('--class-id', '--class-name', help='班级ID/班级名称')
    parser.add_argument('--assignment-id', default='2311061111', help='作业ID')
    parser.add_argument('--images', help='图片目录路径')
    parser.add_argument('--commit-sha', help='提交SHA')
    return parser.parse_args()


def load_file_content(file_path: str) -> str:
    """加载文件内容"""
    with open(file_path, 'r', encoding='utf-8') as f:
        return f.read()


def load_json_file(file_path: str) -> dict:
    """加载JSON文件内容"""
    with open(file_path, 'r', encoding='utf-8') as f:
        return json.load(f)


def markdown_to_paragraphs(markdown_text: str) -> list:
    """将Markdown文本转换为ReportLab可处理的段落列表"""
    # 简单的Markdown处理，主要处理标题和段落
    paragraphs = []
    lines = markdown_text.split('\n')
    
    current_paragraph = ""
    for line in lines:
        line = line.strip()
        
        if not line:
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            continue
        
        # 处理标题
        if line.startswith('# '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('h1', line[2:]))
        elif line.startswith('## '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('h2', line[3:]))
        elif line.startswith('### '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('h3', line[4:]))
        elif line.startswith('#### '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('h4', line[5:]))
        # 处理列表
        elif line.startswith('- '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('list_item', line[2:]))
        elif line.startswith('* '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('list_item', line[2:]))
        elif line.startswith('1. '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('list_item', line[3:]))
        # 处理代码块
        elif line.startswith('```'):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('code_block_start', ''))
        elif paragraphs and paragraphs[-1][0] == 'code_block_start':
            if line.startswith('```'):
                paragraphs.append(('code_block_end', ''))
            else:
                paragraphs.append(('code_block', line))
        # 处理引用
        elif line.startswith('> '):
            if current_paragraph:
                paragraphs.append(('paragraph', current_paragraph))
                current_paragraph = ""
            paragraphs.append(('quote', line[2:]))
        # 处理粗体和斜体
        else:
            current_paragraph += ' ' + line if current_paragraph else line
    
    if current_paragraph:
        paragraphs.append(('paragraph', current_paragraph))
    
    return paragraphs


def generate_pdf_report(args, score_data, report_content, frontend_content):
    """生成PDF报告"""
    if not REPORTLAB_AVAILABLE:
        print("❌ ERROR: reportlab库未安装，无法生成PDF报告")
        print("请运行: pip install reportlab")
        return False
    
    # 创建PDF文档
    doc = SimpleDocTemplate(args.out, pagesize=A4)
    story = []
    
    # 获取样式
    styles = getSampleStyleSheet()
    
    # 自定义样式
    styles.add(ParagraphStyle(
        name='CustomHeading1',
        parent=styles['Heading1'],
        fontSize=24,
        alignment=TA_CENTER,
        textColor=colors.HexColor('#2c3e50'),
        spaceAfter=30
    ))
    
    styles.add(ParagraphStyle(
        name='CustomHeading2',
        parent=styles['Heading2'],
        fontSize=18,
        textColor=colors.HexColor('#34495e'),
        spaceAfter=15,
        spaceBefore=15
    ))
    
    styles.add(ParagraphStyle(
        name='CustomHeading3',
        parent=styles['Heading3'],
        fontSize=14,
        textColor=colors.HexColor('#7f8c8d'),
        spaceAfter=10
    ))
    
    styles.add(ParagraphStyle(
        name='CustomParagraph',
        parent=styles['Normal'],
        fontSize=12,
        leading=18,
        alignment=TA_JUSTIFY,
        textColor=colors.HexColor('#000000')
    ))
    
    styles.add(ParagraphStyle(
        name='CustomListItem',
        parent=styles['CustomParagraph'],
        leftIndent=20,
        bulletIndent=10
    ))
    
    styles.add(ParagraphStyle(
        name='CustomCentered',
        parent=styles['CustomParagraph'],
        alignment=TA_CENTER
    ))
    
    styles.add(ParagraphStyle(
        name='ScoreStyle',
        parent=styles['CustomCentered'],
        fontSize=36,
        textColor=colors.HexColor('#e74c3c'),
        fontWeight='bold',
        spaceAfter=20
    ))
    
    # 封面
    story.append(Spacer(1, 5*cm))
    story.append(Paragraph("作业成绩报告", styles['CustomHeading1']))
    story.append(Spacer(1, 2*cm))
    
    # 学生信息
    student_info = [
        ['学生姓名:', args.student_name],
        ['学生ID:', args.student_id],
        ['班级ID:', args.class_id],
        ['作业ID:', args.assignment_id],
        ['提交日期:', datetime.now().strftime('%Y-%m-%d')]
    ]
    
    table_style = TableStyle([
        ('ALIGN', (0, 0), (-1, -1), 'LEFT'),
        ('FONTNAME', (0, 0), (-1, -1), 'Helvetica'),
        ('FONTSIZE', (0, 0), (-1, -1), 12),
        ('BOTTOMPADDING', (0, 0), (-1, -1), 10),
        ('RIGHTPADDING', (0, 0), (0, -1), 20),
    ])
    
    info_table = Table(student_info, colWidths=[100, 200])
    info_table.setStyle(table_style)
    
    story.append(Spacer(1, 1*cm))
    story.append(info_table)
    
    # 最终成绩
    story.append(Spacer(1, 3*cm))
    story.append(Paragraph("最终成绩", styles['CustomHeading2']))
    story.append(Paragraph(f"{score_data['final_score']:.2f}/100", styles['ScoreStyle']))
    
    # 成绩等级
    if score_data['final_score'] >= 90:
        grade = "优秀"
        grade_color = colors.HexColor('#27ae60')
    elif score_data['final_score'] >= 80:
        grade = "良好"
        grade_color = colors.HexColor('#f39c12')
    elif score_data['final_score'] >= 70:
        grade = "中等"
        grade_color = colors.HexColor('#e67e22')
    elif score_data['final_score'] >= 60:
        grade = "及格"
        grade_color = colors.HexColor('#d35400')
    else:
        grade = "不及格"
        grade_color = colors.HexColor('#c0392b')
    
    story.append(Paragraph(f"等级: {grade}", 
        ParagraphStyle(name='GradeStyle', parent=styles['CustomCentered'], fontSize=20, textColor=grade_color)))
    
    story.append(PageBreak())
    
    # 成绩分布
    story.append(Paragraph("成绩分布", styles['CustomHeading2']))
    
    distribution_data = [
        ['部分', '得分', '权重', '占总分百分比']
    ]
    
    for part, data in score_data['breakdown'].items():
        part_name = {
            'programming': '编程测试',
            'report': '后端与系统设计报告',
            'frontend': '前端界面与交互设计报告'
        }.get(part, part)
        
        distribution_data.append([
            part_name,
            f"{data['score']:.2f}",
            f"{data['weight']*100:.0f}%",
            f"{data['percentage']:.2f}"
        ])
    
    distribution_table = Table(distribution_data, colWidths=[150, 60, 60, 100])
    distribution_table.setStyle(TableStyle([
        ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
        ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
        ('FONTSIZE', (0, 0), (-1, -1), 12),
        ('BACKGROUND', (0, 0), (-1, 0), colors.HexColor('#f8f9fa')),
        ('TEXTCOLOR', (0, 0), (-1, 0), colors.HexColor('#2c3e50')),
        ('BOTTOMPADDING', (0, 0), (-1, -1), 10),
        ('TOPPADDING', (0, 0), (-1, -1), 10),
        ('GRID', (0, 0), (-1, -1), 1, colors.HexColor('#dee2e6'))
    ]))
    
    story.append(Spacer(1, 1*cm))
    story.append(distribution_table)
    story.append(Spacer(1, 2*cm))
    
    # 编程测试详情
    story.append(Paragraph("编程测试详情", styles['CustomHeading2']))
    
    programming = score_data['details']['programming']
    programming_groups = programming.get('groups', [])
    
    if programming_groups:
        prog_data = [['测试组', '通过测试', '总测试数', '得分', '通过率']]
        
        for group in programming_groups:
            prog_data.append([
                group['name'],
                group['passed'],
                group['total'],
                f"{group['score']:.2f}",
                f"{group['passed']/group['total']*100:.1f}%"
            ])
        
        prog_table = Table(prog_data, colWidths=[100, 70, 70, 60, 70])
        prog_table.setStyle(TableStyle([
            ('ALIGN', (0, 0), (-1, -1), 'CENTER'),
            ('FONTNAME', (0, 0), (-1, 0), 'Helvetica-Bold'),
            ('FONTSIZE', (0, 0), (-1, -1), 12),
            ('BACKGROUND', (0, 0), (-1, 0), colors.HexColor('#f8f9fa')),
            ('TEXTCOLOR', (0, 0), (-1, 0), colors.HexColor('#2c3e50')),
            ('BOTTOMPADDING', (0, 0), (-1, -1), 8),
            ('TOPPADDING', (0, 0), (-1, -1), 8),
            ('GRID', (0, 0), (-1, -1), 1, colors.HexColor('#dee2e6'))
        ]))
        
        story.append(Spacer(1, 1*cm))
        story.append(prog_table)
    else:
        story.append(Paragraph("暂无编程测试详情", styles['CustomParagraph']))
    
    story.append(Spacer(1, 2*cm))
    story.append(PageBreak())
    
    # 后端与系统设计报告
    story.append(Paragraph("后端与系统设计报告", styles['CustomHeading2']))
    story.append(Spacer(1, 1*cm))
    
    report_paragraphs = markdown_to_paragraphs(report_content)
    in_code_block = False
    
    for para_type, content in report_paragraphs:
        if para_type == 'h1':
            story.append(Paragraph(content, styles['CustomHeading1']))
        elif para_type == 'h2':
            story.append(Paragraph(content, styles['CustomHeading2']))
        elif para_type == 'h3':
            story.append(Paragraph(content, styles['CustomHeading3']))
        elif para_type == 'paragraph':
            story.append(Paragraph(content, styles['CustomParagraph']))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'list_item':
            story.append(Paragraph(f'• {content}', styles['CustomListItem']))
            story.append(Spacer(1, 0.2*cm))
        elif para_type == 'quote':
            story.append(Paragraph(content, 
                ParagraphStyle(name='QuoteStyle', parent=styles['CustomParagraph'], 
                              leftIndent=20, rightIndent=20, 
                              borderLeftWidth=3, borderLeftColor=colors.HexColor('#bdc3c7'))))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'code_block_start':
            in_code_block = True
            story.append(Paragraph('```', styles['CustomParagraph']))
        elif para_type == 'code_block_end':
            in_code_block = False
            story.append(Paragraph('```', styles['CustomParagraph']))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'code_block' and in_code_block:
            story.append(Paragraph(content, 
                ParagraphStyle(name='CodeStyle', parent=styles['CustomParagraph'], 
                              fontName='Courier', fontSize=10, 
                              leftIndent=20, rightIndent=20)))
    
    story.append(PageBreak())
    
    # 前端界面与交互设计报告
    story.append(Paragraph("前端界面与交互设计报告", styles['CustomHeading2']))
    story.append(Spacer(1, 1*cm))
    
    frontend_paragraphs = markdown_to_paragraphs(frontend_content)
    in_code_block = False
    
    for para_type, content in frontend_paragraphs:
        if para_type == 'h1':
            story.append(Paragraph(content, styles['CustomHeading1']))
        elif para_type == 'h2':
            story.append(Paragraph(content, styles['CustomHeading2']))
        elif para_type == 'h3':
            story.append(Paragraph(content, styles['CustomHeading3']))
        elif para_type == 'paragraph':
            story.append(Paragraph(content, styles['CustomParagraph']))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'list_item':
            story.append(Paragraph(f'• {content}', styles['CustomListItem']))
            story.append(Spacer(1, 0.2*cm))
        elif para_type == 'quote':
            story.append(Paragraph(content, 
                ParagraphStyle(name='QuoteStyle', parent=styles['CustomParagraph'], 
                              leftIndent=20, rightIndent=20, 
                              borderLeftWidth=3, borderLeftColor=colors.HexColor('#bdc3c7'))))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'code_block_start':
            in_code_block = True
            story.append(Paragraph('```', styles['CustomParagraph']))
        elif para_type == 'code_block_end':
            in_code_block = False
            story.append(Paragraph('```', styles['CustomParagraph']))
            story.append(Spacer(1, 0.5*cm))
        elif para_type == 'code_block' and in_code_block:
            story.append(Paragraph(content, 
                ParagraphStyle(name='CodeStyle', parent=styles['CustomParagraph'], 
                              fontName='Courier', fontSize=10, 
                              leftIndent=20, rightIndent=20)))
    
    # 构建并保存PDF
    try:
        doc.build(story)
        return True
    except Exception as e:
        print(f"❌ PDF生成失败: {str(e)}")
        return False


def generate_text_report(args, score_data, report_content, frontend_content):
    """生成纯文本报告作为备选方案"""
    text_report = []
    text_report.append("=" * 80)
    text_report.append("                作业成绩报告                ")
    text_report.append("=" * 80)
    text_report.append("")
    
    # 学生信息
    text_report.append(f"学生姓名: {args.student_name}")
    text_report.append(f"学生ID: {args.student_id}")
    text_report.append(f"班级ID: {args.class_id}")
    text_report.append(f"作业ID: {args.assignment_id}")
    text_report.append(f"提交日期: {datetime.now().strftime('%Y-%m-%d')}")
    text_report.append("")
    
    # 最终成绩
    text_report.append("-" * 80)
    text_report.append(f"最终成绩: {score_data['final_score']:.2f}/100")
    
    # 成绩等级
    if score_data['final_score'] >= 90:
        grade = "优秀"
    elif score_data['final_score'] >= 80:
        grade = "良好"
    elif score_data['final_score'] >= 70:
        grade = "中等"
    elif score_data['final_score'] >= 60:
        grade = "及格"
    else:
        grade = "不及格"
    
    text_report.append(f"等级: {grade}")
    text_report.append("")
    
    # 成绩分布
    text_report.append("成绩分布:")
    text_report.append("| 部分 | 得分 | 权重 | 占总分百分比 |")
    text_report.append("|------|------|------|------------|")
    
    for part, data in score_data['breakdown'].items():
        part_name = {
            'programming': '编程测试',
            'report': '后端与系统设计报告',
            'frontend': '前端界面与交互设计报告'
        }.get(part, part)
        
        text_report.append(f"| {part_name} | {data['score']:.2f} | {data['weight']*100:.0f}% | {data['percentage']:.2f} |")
    
    text_report.append("")
    text_report.append("=" * 80)
    text_report.append("")
    
    # 保存文本报告
    text_report_path = args.out.replace('.pdf', '.txt')
    with open(text_report_path, 'w', encoding='utf-8') as f:
        f.write('\n'.join(text_report))
    
    print(f"✅ 已生成纯文本报告: {text_report_path}")
    return True


def main():
    """主函数"""
    args = parse_args()
    
    print("📁 加载评分数据...")
    score_data = load_json_file(args.score)
    
    print("📁 加载后端报告内容...")
    report_content = load_file_content(args.report)
    
    print("📁 加载前端报告内容...")
    frontend_content = load_file_content(args.frontend)
    
    print("📄 生成PDF报告...")
    pdf_success = generate_pdf_report(args, score_data, report_content, frontend_content)
    
    # 如果PDF生成失败，生成纯文本报告
    if not pdf_success:
        print("🔄 尝试生成纯文本报告...")
        generate_text_report(args, score_data, report_content, frontend_content)
    else:
        print(f"✅ PDF报告生成成功: {args.out}")
        
    # 生成一个简单的总结文件
    summary = {
        "student_info": {
            "name": args.student_name,
            "id": args.student_id,
            "class_id": args.class_id,
            "assignment_id": args.assignment_id
        },
        "final_score": score_data["final_score"],
        "report_path": args.out,
        "generated_at": datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
        "status": "completed"
    }
    
    summary_path = args.out.replace('.pdf', '_summary.json')
    with open(summary_path, 'w', encoding='utf-8') as f:
        json.dump(summary, f, ensure_ascii=False, indent=2)
    
    print(f"✅ 总结文件生成成功: {summary_path}")


if __name__ == '__main__':
    main()