from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
import sys
import math

class Calculator(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("多功能計算器")
        self.setFixedSize(400, 500)
        
        # 顯示屏
        self.display = QLCDNumber()
        self.display.setDigitCount(15)
        self.display.setSegmentStyle(QLCDNumber.Flat)
        
        # 按鈕布局
        self.create_buttons()

def create_buttons(self):
    buttons = [
        ['7', '8', '9', '/'],
        ['4', '5', '6', '*'],
        ['1', '2', '3', '-'],
        ['0', '.', '=', '+'],
        ['C', '√', '^', 'AC']
    ]
    
    grid = QGridLayout()
    central_widget = QWidget()
    central_widget.setLayout(grid)
    self.setCentralWidget(central_widget)
    
    # 把顯示屏和按鈕網(wǎng)格拼在一起
    main_layout = QVBoxLayout()
    main_layout.addWidget(self.display)
    main_layout.addLayout(grid)
    central_widget.setLayout(main_layout)
    
    for row, row_buttons in enumerate(buttons):
        for col, btn_text in enumerate(row_buttons):
            btn = QPushButton(btn_text)
            btn.clicked.connect(self.on_button_clicked)
            grid.addWidget(btn, row, col)

def on_button_clicked(self):
    text = self.sender().text()
    
    if text.isdigit() or text == '.':
        # 數(shù)字或小數(shù)點，拼接到當(dāng)前輸入
        self.current_input += text
        self.display.display(self.current_input)
        
    elif text in ['+', '-', '*', '/']:
        # 運算符，保存當(dāng)前數(shù)字和運算符
        self.last_number = float(self.current_input)
        self.current_operator = text
        self.current_input = ''
        
    elif text == '=':
        # 計算結(jié)果
        current = float(self.current_input)
        if self.current_operator == '+':
            result = self.last_number + current
        elif self.current_operator == '-':
            result = self.last_number - current
        # ... 其他運算
        self.display.display(result)
        self.current_input = str(result)

def calculate_function(self, func_name):
    value = float(self.current_input)
    
    if func_name == 'sqrt':
        result = math.sqrt(value)
    elif func_name == 'sin':
        # 角度轉(zhuǎn)弧度
        result = math.sin(math.radians(value))  
    elif func_name == 'cos':
        result = math.cos(math.radians(value))
    elif func_name == 'log':
        result = math.log10(value)
    elif func_name == 'ln':
        result = math.log(value)
    # ...
    
    self.display.display(result)
    self.current_input = str(result)

def keyPressEvent(self, event):
    key = event.key()
    
    if key in [Qt.Key_0, Qt.Key_1, ..., Qt.Key_9]:
        self.current_input += chr(key)
    elif key == Qt.Key_Plus:
        self.set_operator('+')
    elif key == Qt.Key_Return or key == Qt.Key_Enter:
        self.calculate_result()
    elif key == Qt.Key_Backspace:
        self.current_input = self.current_input[:-1]
    elif key == Qt.Key_Escape:
        self.clear_all()
    
    self.display.display(self.current_input)

try:
    result = self.last_number / current
except ZeroDivisionError:
    QMessageBox.warning(self, "錯誤", "除數(shù)不能為零")
    return