from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *

class AwardGenerator(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("獎(jiǎng)狀生成器")
        self.setFixedSize(900, 600)
        
        # 中央控件
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        
        # 主布局
        main_layout = QHBoxLayout(central_widget)
        
        # 左側(cè)控制面板
        left_panel = QWidget()
        left_layout = QFormLayout(left_panel)
        
        self.name_input = QLineEdit()
        self.award_input = QLineEdit()
        self.date_input = QDateEdit()
        self.date_input.setDate(QDate.currentDate())
        
        left_layout.addRow("獲獎(jiǎng)人姓名：", self.name_input)
        left_layout.addRow("獎(jiǎng)項(xiàng)名稱(chēng)：", self.award_input)
        left_layout.addRow("獲獎(jiǎng)日期：", self.date_input)
        
        self.generate_btn = QPushButton("生成獎(jiǎng)狀")
        self.generate_btn.clicked.connect(self.generate_certificate)
        left_layout.addRow(self.generate_btn)
        
        # 右側(cè)預(yù)覽區(qū)
        self.preview_area = QLabel()
        self.preview_area.setFixedSize(500, 400)
        self.preview_area.setStyleSheet("border: 1px solid #ccc; background: white;")
        self.preview_area.setAlignment(Qt.AlignCenter)
        
        main_layout.addWidget(left_panel)
        main_layout.addWidget(self.preview_area)

def setup_styles(self):
    # 主窗口背景
    self.setStyleSheet("""
        QMainWindow {
            background-color: #f5f0e8;
        }
        QPushButton {
            background-color: #c0392b;
            color: white;
            border: none;
            padding: 8px 15px;
            border-radius: 5px;
            font-size: 14px;
        }
        QPushButton:hover {
            background-color: #e74c3c;
        }
        QLineEdit, QDateEdit {
            padding: 5px;
            border: 1px solid #ddd;
            border-radius: 3px;
        }
    """)

# 預(yù)覽區(qū)域的樣式
self.preview_area.setStyleSheet("""
    border: 3px solid #d4af37;
    border-radius: 15px;
    background: white;
""")

def generate_certificate(self):
    # 獲取用戶(hù)輸入
    name = self.name_input.text()
    award = self.award_input.text()
    date = self.date_input.date().toString("yyyy年MM月dd日")
    
    if not name or not award:
        QMessageBox.warning(self, "提示", "請(qǐng)?zhí)顚?xiě)完整信息")
        return
    
    # 創(chuàng)建空白圖片
    image = QImage(800, 600, QImage.Format_ARGB32)
    image.fill(Qt.white)
    
    painter = QPainter(image)
    painter.setRenderHint(QPainter.Antialiasing)
    
    # 繪制金色邊框
    pen = QPen(QColor(212, 175, 55))
    pen.setWidth(8)
    painter.setPen(pen)
    painter.drawRect(20, 20, 760, 560)
    
    # 繪制裝飾花紋（簡(jiǎn)單版：四個(gè)角加圓點(diǎn)）
    painter.setBrush(QBrush(QColor(212, 175, 55)))
    for x, y in [(30, 30), (770, 30), (30, 570), (770, 570)]:
        painter.drawEllipse(x-5, y-5, 10, 10)
    
    # 設(shè)置標(biāo)題字體
    title_font = QFont("華文行書(shū)", 48, QFont.Bold)
    painter.setFont(title_font)
    painter.drawText(image.rect(), Qt.AlignTop | Qt.AlignHCenter, "榮譽(yù)證書(shū)")
    
    # 設(shè)置正文字體
    text_font = QFont("微軟雅黑", 16)
    painter.setFont(text_font)
    
    # 繪制正文
    text = f"茲證明 {name} 同志："
    painter.drawText(image.rect().adjusted(0, 120, 0, 0), 
                     Qt.AlignHCenter, text)
    
    award_text = f"榮獲{2025}年度 “{award}” 榮譽(yù)稱(chēng)號(hào)"
    painter.drawText(image.rect().adjusted(0, 180, 0, 0), 
                     Qt.AlignHCenter, award_text)
    
    # 特此表彰，以資鼓勵(lì)
    painter.drawText(image.rect().adjusted(0, 280, 0, 0), 
                     Qt.AlignHCenter, "特此表彰，以資鼓勵(lì)")
    
    # 落款
    painter.drawText(image.rect().adjusted(-100, 450, 0, 0), 
                     Qt.AlignRight, f"某某科技有限公司\n{date}")
    
    painter.end()
    
    # 顯示到預(yù)覽區(qū)
    pixmap = QPixmap.fromImage(image)
    scaled_pixmap = pixmap.scaled(self.preview_area.size(), 
                                   Qt.KeepAspectRatio, 
                                   Qt.SmoothTransformation)
    self.preview_area.setPixmap(scaled_pixmap)
    
    # 保存圖片
    self.current_image = image

def batch_generate(self):
    # 選擇Excel文件
    file_path, _ = QFileDialog.getOpenFileName(
        self, "選擇Excel文件", "", "Excel文件(*.xlsx *.xls)"
    )
    if not file_path:
        return
    
    # 讀取數(shù)據(jù)
    import pandas as pd
    df = pd.read_excel(file_path)
    
    # 創(chuàng)建保存目錄
    output_dir = QFileDialog.getExistingDirectory(self, "選擇保存目錄")
    if not output_dir:
        return
    
    # 批量生成
    for idx, row in df.iterrows():
        name = row['姓名']
        award = row['獎(jiǎng)項(xiàng)']
        date = row.get('日期', QDate.currentDate().toString("yyyy-MM-dd"))
        
        # 調(diào)用生成函數(shù)
        image = self.create_certificate_image(name, award, date)
        image.save(f"{output_dir}/{name}_獎(jiǎng)狀.png")
    
    QMessageBox.information(self, "完成", f"已生成{len(df)}張獎(jiǎng)狀")

def keyPressEvent(self, event):
    if event.key() == Qt.Key_Return or event.key() == Qt.Key_Enter:
        self.generate_certificate()

if not name or not award:
    QMessageBox.warning(self, "提示", "請(qǐng)?zhí)顚?xiě)完整信息")
    return

progress = QProgressDialog("正在生成獎(jiǎng)狀...", "取消", 0, total, self)
for i, row in df.iterrows():
    if progress.wasCanceled():
        break
    progress.setValue(i)
    # 生成獎(jiǎng)狀...