图：
现在我们需要安装额外的依赖包：
pip install qt-material qdarkstyle
主要改动包括：
- 添加了菜单栏，包含主题切换菜单
- 提供了四种主题类型：
- 默认主题（系统原生样式）
- Fusion主题（亮色）
- Qt-Material主题（dark_teal、light_blue、dark_blue）
- QDarkStyle主题（暗色）
代码：
main.py
import sys
import os
from datetime import datetime
import tempfile
import webbrowser
from PySide6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,QHBoxLayout, QLabel, QComboBox, QLineEdit,QPushButton, QFrame, QScrollArea, QMessageBox,QFileDialog, QDateEdit, QMenuBar, QMenu, QStyle,QStyleFactory)
from PySide6.QtCore import Qt, Signal, QDate, QTimer
from PySide6.QtGui import QPalette, QColor
from database import Database
from dialogs import EditComponentDialog
from qt_material import apply_stylesheet, list_themes
import qdarkstyleclass ConfigRow(QWidget):# 添加自定义信号amount_changed = Signal()def __init__(self, database, component_id, component_name, parent=None):super().__init__(parent)self.database = databaseself.component_id = component_idself.component_name = component_nameself.setup_ui()def setup_ui(self):layout = QHBoxLayout(self)layout.setContentsMargins(5, 2, 5, 2)# 配件名称name_label = QLabel(self.component_name)name_label.setFixedWidth(100)layout.addWidget(name_label)# 配置1self.model1_combo = QComboBox()self.model1_combo.setMinimumWidth(200)self.model1_combo.currentIndexChanged.connect(self.update_price1)layout.addWidget(self.model1_combo)self.qty1_edit = QLineEdit('1')self.qty1_edit.setFixedWidth(60)self.qty1_edit.textChanged.connect(self.calculate_amount1)layout.addWidget(self.qty1_edit)self.price1_edit = QLineEdit('0')self.price1_edit.setFixedWidth(80)self.price1_edit.textChanged.connect(self.calculate_amount1)layout.addWidget(self.price1_edit)self.amount1_label = QLabel('0.00')self.amount1_label.setFixedWidth(100)layout.addWidget(self.amount1_label)# 配置2self.model2_combo = QComboBox()self.model2_combo.setMinimumWidth(200)self.model2_combo.currentIndexChanged.connect(self.update_price2)layout.addWidget(self.model2_combo)self.qty2_edit = QLineEdit('1')self.qty2_edit.setFixedWidth(60)self.qty2_edit.textChanged.connect(self.calculate_amount2)layout.addWidget(self.qty2_edit)self.price2_edit = QLineEdit('0')self.price2_edit.setFixedWidth(80)self.price2_edit.textChanged.connect(self.calculate_amount2)layout.addWidget(self.price2_edit)self.amount2_label = QLabel('0.00')self.amount2_label.setFixedWidth(100)layout.addWidget(self.amount2_label)# 加载型号数据self.load_models()def load_models(self):# 加载配置1的型号self.model1_combo.clear()self.model1_combo.addItem('', None) # 添加空选项models1 = self.database.get_models(self.component_id, 'option1')for model_id, model_name, price in models1:self.model1_combo.addItem(model_name, (model_id, price))# 加载配置2的型号self.model2_combo.clear()self.model2_combo.addItem('', None) # 添加空选项models2 = self.database.get_models(self.component_id, 'option2')for model_id, model_name, price in models2:self.model2_combo.addItem(model_name, (model_id, price))def update_price1(self, index):data = self.model1_combo.currentData()if data:_, price = dataself.price1_edit.setText(str(price))else:self.price1_edit.setText('0')def update_price2(self, index):data = self.model2_combo.currentData()if data:_, price = dataself.price2_edit.setText(str(price))else:self.price2_edit.setText('0')def calculate_amount1(self):try:qty = float(self.qty1_edit.text() or 0)price = float(self.price1_edit.text() or 0)amount = qty * priceself.amount1_label.setText(f'{amount:.2f}')self.amount_changed.emit() # 发送信号except ValueError:self.amount1_label.setText('0.00')self.amount_changed.emit() # 发送信号def calculate_amount2(self):try:qty = float(self.qty2_edit.text() or 0)price = float(self.price2_edit.text() or 0)amount = qty * priceself.amount2_label.setText(f'{amount:.2f}')self.amount_changed.emit() # 发送信号except ValueError:self.amount2_label.setText('0.00')self.amount_changed.emit() # 发送信号def get_data(self):"""获取行数据"""return {'component_name': self.component_name,'config1': {'model': self.model1_combo.currentText(),'qty': self.qty1_edit.text(),'price': self.price1_edit.text(),'amount': self.amount1_label.text()},'config2': {'model': self.model2_combo.currentText(),'qty': self.qty2_edit.text(),'price': self.price2_edit.text(),'amount': self.amount2_label.text()}}class MainWindow(QMainWindow):def __init__(self):super().__init__()self.database = Database()self.current_theme = 'qt_material_dark_teal' # 默认主题self.setup_menu() # 在setup_ui之前设置菜单self.setup_ui()def setup_menu(self):"""设置菜单栏"""menubar = self.menuBar()# 主题菜单theme_menu = menubar.addMenu('主题')# 原始样式default_action = theme_menu.addAction('默认主题')default_action.triggered.connect(lambda: self.change_theme('default'))# Fusion主题fusion_light = theme_menu.addAction('Fusion主题')fusion_light.triggered.connect(lambda: self.change_theme('fusion_light'))# Qt-Material主题qt_material_menu = theme_menu.addMenu('Qt-Material主题')material_themes = ['dark_teal.xml', 'light_blue.xml', 'dark_blue.xml']for theme in material_themes:action = qt_material_menu.addAction(theme.replace('.xml', ''))action.triggered.connect(lambda checked, t=theme: self.change_theme(f'qt_material_{t}'))# QDarkStyle主题dark_action = theme_menu.addAction('QDarkStyle暗色')dark_action.triggered.connect(lambda: self.change_theme('qdarkstyle_dark'))def change_theme(self, theme_name):"""切换主题"""self.current_theme = theme_nameapp = QApplication.instance()if theme_name == 'default':# 恢复默认样式app.setStyleSheet("")app.setStyle('Windows')elif theme_name.startswith('fusion'):# 设置Fusion主题app.setStyle('Fusion')# 恢复亮色调色板app.setPalette(app.style().standardPalette())elif theme_name.startswith('qt_material'):# 设置Qt-Material主题theme_file = theme_name.replace('qt_material_', '')apply_stylesheet(app, theme=theme_file)elif theme_name.startswith('qdarkstyle'):# 设置QDarkStyle主题app.setStyleSheet(qdarkstyle.load_stylesheet(qt_api='pyside6'))def setup_ui(self):self.setWindowTitle('电脑配置单')self.setMinimumSize(1200, 800)# 创建中央部件central_widget = QWidget()self.setCentralWidget(central_widget)layout = QVBoxLayout(central_widget)# 标题title_label = QLabel('电脑配置单')title_label.setAlignment(Qt.AlignCenter)layout.addWidget(title_label)# 日期date_layout = QHBoxLayout()date_layout.addWidget(QLabel('日期：'))# 创建日期选择器self.date_edit = QDateEdit()self.date_edit.setDisplayFormat('yyyy-MM-dd')self.date_edit.setCalendarPopup(True) # 允许弹出日历选择self.date_edit.setDate(QDate.currentDate()) # 设置当前日期date_layout.addWidget(self.date_edit)# 添加同步系统时间按钮sync_button = QPushButton('同步系统时间')sync_button.clicked.connect(self.sync_system_time)date_layout.addWidget(sync_button)date_layout.addStretch()layout.addLayout(date_layout)# 创建表头header = QWidget()header_layout = QHBoxLayout(header)header_layout.setContentsMargins(5, 2, 5, 2)labels = ['配件名称', '型号[1]', '数量[1]', '单价[1]', '金额[1]','型号[2]', '数量[2]', '单价[2]', '金额[2]']widths = [100, 200, 60, 80, 100, 200, 60, 80, 100]for label, width in zip(labels, widths):lbl = QLabel(label)lbl.setFixedWidth(width)header_layout.addWidget(lbl)layout.addWidget(header)# 创建滚动区域scroll = QScrollArea()scroll.setWidgetResizable(True)scroll.setFrameShape(QFrame.NoFrame)layout.addWidget(scroll)# 创建配置行容器self.rows_widget = QWidget()self.rows_layout = QVBoxLayout(self.rows_widget)scroll.setWidget(self.rows_widget)# 创建配置行self.config_rows = []components = self.database.get_components()for component_id, component_name in components:row = ConfigRow(self.database, component_id, component_name)self.rows_layout.addWidget(row)self.config_rows.append(row)# 创建合计区域total_frame = QFrame()total_frame.setFrameShape(QFrame.Box)total_layout = QHBoxLayout(total_frame)# 配置1合计total1_layout = QVBoxLayout()total1_layout.addWidget(QLabel('配置1合计：'))self.total1_label = QLabel('0.00')total1_layout.addWidget(self.total1_label)total_layout.addLayout(total1_layout)# 配置2合计total2_layout = QVBoxLayout()total2_layout.addWidget(QLabel('配置2合计：'))self.total2_label = QLabel('0.00')total2_layout.addWidget(self.total2_label)total_layout.addLayout(total2_layout)layout.addWidget(total_frame)# 创建按钮区域button_layout = QHBoxLayout()export_button = QPushButton('导出配置')export_button.clicked.connect(self.export_config)print_button = QPushButton('打印配置')print_button.clicked.connect(self.print_config)edit_button = QPushButton('编辑配件数据')edit_button.clicked.connect(self.edit_components_data)button_layout.addWidget(export_button)button_layout.addWidget(print_button)button_layout.addWidget(edit_button)layout.addLayout(button_layout)# 设置定时器更新合计for row in self.config_rows:row.amount_changed.connect(self.update_totals) # 连接新的信号def update_totals(self):"""更新合计金额"""total1 = 0total2 = 0for row in self.config_rows:try:total1 += float(row.amount1_label.text())total2 += float(row.amount2_label.text())except ValueError:continueself.total1_label.setText(f'{total1:.2f}')self.total2_label.setText(f'{total2:.2f}')def sync_system_time(self):"""同步系统时间"""self.date_edit.setDate(QDate.currentDate())def export_config(self):"""导出配置到Excel文件"""try:import openpyxlfrom openpyxl.styles import Alignment, Font, Border, Side# 创建工作簿wb = openpyxl.Workbook()ws = wb.activews.title = "电脑配置单"# 设置列宽ws.column_dimensions['A'].width = 15ws.column_dimensions['B'].width = 30ws.column_dimensions['C'].width = 8ws.column_dimensions['D'].width = 10ws.column_dimensions['E'].width = 12ws.column_dimensions['F'].width = 30ws.column_dimensions['G'].width = 8ws.column_dimensions['H'].width = 10ws.column_dimensions['I'].width = 12# 写入标题ws['A1'] = "电脑配置单"ws.merge_cells('A1:I1')ws['A1'].font = Font(size=14, bold=True)ws['A1'].alignment = Alignment(horizontal='center')# 修改日期获取方式ws['A2'] = f"日期：{self.date_edit.date().toString('yyyy-MM-dd')}"ws.merge_cells('A2:I2')# 写入表头headers = ['配件名称', '型号[1]', '数量[1]', '单价[1]', '金额[1]','型号[2]', '数量[2]', '单价[2]', '金额[2]']for col, header in enumerate(headers, 1):cell = ws.cell(row=3, column=col)cell.value = headercell.font = Font(bold=True)cell.alignment = Alignment(horizontal='center')# 写入数据for row_idx, config_row in enumerate(self.config_rows, 4):data = config_row.get_data()ws.cell(row=row_idx, column=1).value = data['component_name']ws.cell(row=row_idx, column=2).value = data['config1']['model']ws.cell(row=row_idx, column=3).value = data['config1']['qty']ws.cell(row=row_idx, column=4).value = data['config1']['price']ws.cell(row=row_idx, column=5).value = data['config1']['amount']ws.cell(row=row_idx, column=6).value = data['config2']['model']ws.cell(row=row_idx, column=7).value = data['config2']['qty']ws.cell(row=row_idx, column=8).value = data['config2']['price']ws.cell(row=row_idx, column=9).value = data['config2']['amount']# 写入合计total_row = len(self.config_rows) + 4ws.cell(row=total_row, column=1).value = "合计"ws.cell(row=total_row, column=5).value = f"配置1：{self.total1_label.text()}"ws.cell(row=total_row, column=9).value = f"配置2：{self.total2_label.text()}"# 设置边框thin_border = Border(left=Side(style='thin'),right=Side(style='thin'),top=Side(style='thin'),bottom=Side(style='thin'))for row in ws.iter_rows(min_row=3, max_row=total_row, min_col=1, max_col=9):for cell in row:cell.border = thin_bordercell.alignment = Alignment(horizontal='center')# 保存文件file_path, _ = QFileDialog.getSaveFileName(self,"保存配置单",f"电脑配置单_{datetime.now().strftime('%Y%m%d_%H%M%S')}.xlsx","Excel Files (*.xlsx)")if file_path:wb.save(file_path)QMessageBox.information(self, "成功", "配置已成功导出到Excel文件！")except Exception as e:QMessageBox.critical(self, "错误", f"导出失败：{str(e)}")def print_config(self):"""生成打印预览HTML并在浏览器中打开"""try:html_content = f"""<!DOCTYPE html><html><head><meta charset="UTF-8"><title>电脑配置单</title><style>body {{ font-family: Arial, sans-serif; margin: 20px; }}table {{ width: 100%; border-collapse: collapse; margin-top: 20px; }}th, td {{ border: 1px solid black; padding: 8px; text-align: center; }}th {{ background-color: #f2f2f2; }}.total-row {{ background-color: #f9f9f9; font-weight: bold; }}@media print {{button {{ display: none; }}}}</style></head><body><h2 style="text-align: center;">电脑配置单</h2><p>日期：{self.date_edit.date().toString('yyyy-MM-dd')}</p><table><tr><th>配件名称</th><th>型号[1]</th><th>数量[1]</th><th>单价[1]</th><th>金额[1]</th><th>型号[2]</th><th>数量[2]</th><th>单价[2]</th><th>金额[2]</th></tr>"""# 添加数据行for row in self.config_rows:data = row.get_data()html_content += f"""<tr><td>{data['component_name']}</td><td>{data['config1']['model']}</td><td>{data['config1']['qty']}</td><td>{data['config1']['price']}</td><td>{data['config1']['amount']}</td><td>{data['config2']['model']}</td><td>{data['config2']['qty']}</td><td>{data['config2']['price']}</td><td>{data['config2']['amount']}</td></tr>"""# 添加合计行html_content += f"""<tr class="total-row"><td>合计</td><td colspan="3"></td><td>配置1：{self.total1_label.text()}</td><td colspan="3"></td><td>配置2：{self.total2_label.text()}</td></tr></table><button onclick="window.print()" style="margin-top: 20px;">打印</button></body></html>"""# 创建临时HTML文件with tempfile.NamedTemporaryFile('w', delete=False, suffix='.html', encoding='utf-8') as f:f.write(html_content)temp_path = f.name# 在浏览器中打开webbrowser.open('file://' + temp_path)except Exception as e:QMessageBox.critical(self, "错误", f"打印预览失败：{str(e)}")def edit_components_data(self):"""打开编辑配件数据对话框"""dialog = EditComponentDialog(self.database, self)if dialog.exec_():# 重新加载所有配置行的型号数据for row in self.config_rows:row.load_models()if __name__ == '__main__':app = QApplication(sys.argv)window = MainWindow()window.show()sys.exit(app.exec()) dialogs.py
from datetime import datetime
import openpyxl
from openpyxl.styles import Font, Alignment, Border, Side
from PySide6.QtWidgets import *
from PySide6.QtCore import Qt
from PySide6.QtGui import QIntValidator, QDoubleValidatorclass ComponentDialog(QDialog):def __init__(self, db, component=None, parent=None):super().__init__(parent)self.db = dbself.component = componentself.setupUi()if component:self.load_data()class CustomerDialog(QDialog):def __init__(self, db, customer=None, parent=None):super().__init__(parent)self.db = dbself.customer = customerself.setupUi()if customer:self.load_data()class QuotationDialog(QDialog):def __init__(self, db, quotation_id=None, parent=None):super().__init__(parent)self.db = dbself.quotation_id = quotation_idself.details = [] # 初始化明细列表self.setupUi()if quotation_id:self.load_quotation()def export_to_excel(self):"""导出报价单到Excel"""try:# 检查是否有报价单IDif not hasattr(self, 'quotation_id') or not self.quotation_id:# 如果是新建的报价单，先保存self.save_quotation()if not hasattr(self, 'quotation_id') or not self.quotation_id:raise Exception("请先保存报价单")# 获取报价单数据quotation = self.db.get_quotation(self.quotation_id)if not quotation:raise Exception("报价单不存在")print(f"导出报价单数据: {quotation}")# 获取报价单明细details = self.db.get_quotation_details(self.quotation_id)if not details:raise Exception("报价单没有明细数据")print(f"导出报价单明细: {details}")# 创建Excel工作簿wb = openpyxl.Workbook()ws = wb.activews.title = "报价单"# 设置列宽ws.column_dimensions['A'].width = 15ws.column_dimensions['B'].width = 40ws.column_dimensions['C'].width = 10ws.column_dimensions['D'].width = 15ws.column_dimensions['E'].width = 15# 标题样式title_font = Font(name='宋体', size=14, bold=True)header_font = Font(name='宋体', size=11, bold=True)normal_font = Font(name='宋体', size=11)# 对齐方式center_align = Alignment(horizontal='center', vertical='center')right_align = Alignment(horizontal='right', vertical='center')# 边框样式thin_border = Border(left=Side(style='thin'),right=Side(style='thin'),top=Side(style='thin'),bottom=Side(style='thin'))# 写入标题ws.merge_cells('A1:E1')ws['A1'] = "报价单"ws['A1'].font = title_fontws['A1'].alignment = center_align# 写入基本信息customer_name = quotation[8] if len(quotation) > 8 and quotation[8] else "未知客户"quotation_date = quotation[2] if len(quotation) > 2 and quotation[2] else datetime.now().strftime("%Y-%m-%d")valid_days = quotation[4] if len(quotation) > 4 and quotation[4] else 7status = quotation[6] if len(quotation) > 6 and quotation[6] else "待确认"notes = quotation[5] if len(quotation) > 5 and quotation[5] else ""ws['A2'] = "客户名称："ws['B2'] = customer_namews['D2'] = "报价日期："ws['E2'] = quotation_datews['A3'] = "有效期："ws['B3'] = f"{valid_days}天"ws['D3'] = "状态："ws['E3'] = status# 写入表头headers = ['序号', '配件名称', '数量', '单价', '小计']for col, header in enumerate(headers, 1):cell = ws.cell(row=4, column=col)cell.value = headercell.font = header_fontcell.alignment = center_aligncell.border = thin_border# 写入明细row = 5total_amount = 0for i, detail in enumerate(details, 1):try:component_name = detail[7] if len(detail) > 7 and detail[7] else "未知配件"quantity = int(detail[3]) if len(detail) > 3 and detail[3] is not None else 0unit_price = float(detail[4]) if len(detail) > 4 and detail[4] is not None else 0.0subtotal = float(detail[5]) if len(detail) > 5 and detail[5] is not None else 0.0ws.cell(row=row, column=1, value=i).alignment = center_alignws.cell(row=row, column=2, value=component_name)ws.cell(row=row, column=3, value=quantity).alignment = center_alignws.cell(row=row, column=4, value=f"¥{unit_price:.2f}").alignment = right_alignws.cell(row=row, column=5, value=f"¥{subtotal:.2f}").alignment = right_align# 添加边框for col in range(1, 6):ws.cell(row=row, column=col).border = thin_borderws.cell(row=row, column=col).font = normal_fonttotal_amount += subtotalrow += 1except Exception as e:print(f"处理明细行时出错: {str(e)}, detail={detail}")continue# 写入合计ws.merge_cells(f'A{row}:D{row}')ws.cell(row=row, column=1, value="合计：").alignment = right_alignws.cell(row=row, column=5, value=f"¥{total_amount:.2f}").alignment = right_align# 写入备注if notes:row += 1ws.merge_cells(f'A{row}:E{row}')ws[f'A{row}'] = f"备注：{notes}"# 保存文件filename = f"报价单_{datetime.now().strftime('%Y%m%d%H%M%S')}.xlsx"wb.save(filename)QMessageBox.information(self, "成功", f"报价单已导出到：{filename}")except Exception as e:print(f"导出报价单失败: {str(e)}")QMessageBox.warning(self, "错误", f"导出报价单失败: {str(e)}")class SupplierDialog(QDialog):def __init__(self, db, supplier=None, parent=None):super().__init__(parent)self.db = dbself.supplier = supplierself.setupUi()if supplier:self.load_data() class EditComponentDialog(QDialog):def __init__(self, database, parent=None):super().__init__(parent)self.database = databaseself.setup_ui()self.load_data()def setup_ui(self):"""设置UI界面"""self.setWindowTitle('编辑配件数据')self.setMinimumSize(800, 600)# 创建主布局layout = QVBoxLayout(self)# 创建树形视图self.tree = QTreeWidget()self.tree.setHeaderLabels(['型号', '价格', '配置方案'])self.tree.itemSelectionChanged.connect(self.on_selection_changed)layout.addWidget(self.tree)# 创建编辑区域edit_layout = QVBoxLayout()# 配件选择part_layout = QHBoxLayout()part_layout.addWidget(QLabel('配件:'))self.part_combo = QComboBox()self.part_combo.currentIndexChanged.connect(self.on_part_changed)part_layout.addWidget(self.part_combo)edit_layout.addLayout(part_layout)# 配置方案选择option_layout = QHBoxLayout()option_layout.addWidget(QLabel('配置方案:'))self.option1_radio = QRadioButton('配置1')self.option2_radio = QRadioButton('配置2')self.option1_radio.setChecked(True)option_layout.addWidget(self.option1_radio)option_layout.addWidget(self.option2_radio)edit_layout.addLayout(option_layout)# 型号和价格输入input_layout = QHBoxLayout()input_layout.addWidget(QLabel('型号:'))self.model_edit = QLineEdit()input_layout.addWidget(self.model_edit)input_layout.addWidget(QLabel('价格:'))self.price_edit = QLineEdit()input_layout.addWidget(self.price_edit)edit_layout.addLayout(input_layout)# 按钮button_layout = QHBoxLayout()add_button = QPushButton('添加/更新')add_button.clicked.connect(self.add_item)delete_button = QPushButton('删除')delete_button.clicked.connect(self.delete_item)button_layout.addWidget(add_button)button_layout.addWidget(delete_button)edit_layout.addLayout(button_layout)layout.addLayout(edit_layout)def load_data(self):"""加载数据"""# 加载配件列表components = self.database.get_components()self.part_combo.clear()for component_id, name in components:self.part_combo.addItem(name, component_id)self.update_tree()def update_tree(self):"""更新树形视图"""self.tree.clear()component_id = self.part_combo.currentData()if component_id is None:return# 获取配置1的型号models1 = self.database.get_models(component_id, 'option1')for model_id, model_name, price in models1:item = QTreeWidgetItem([model_name, str(price), '配置1'])item.setData(0, Qt.UserRole, model_id)self.tree.addTopLevelItem(item)# 获取配置2的型号models2 = self.database.get_models(component_id, 'option2')for model_id, model_name, price in models2:item = QTreeWidgetItem([model_name, str(price), '配置2'])item.setData(0, Qt.UserRole, model_id)self.tree.addTopLevelItem(item)def on_part_changed(self, index):"""配件选择改变时的处理"""self.update_tree()def on_selection_changed(self):"""选择项改变时的处理"""items = self.tree.selectedItems()if items:item = items[0]self.model_edit.setText(item.text(0))self.price_edit.setText(item.text(1))if item.text(2) == '配置1':self.option1_radio.setChecked(True)else:self.option2_radio.setChecked(True)def add_item(self):"""添加或更新项目"""try:component_id = self.part_combo.currentData()model_name = self.model_edit.text().strip()price = float(self.price_edit.text())option_type = 'option1' if self.option1_radio.isChecked() else 'option2'if not model_name:QMessageBox.warning(self, '警告', '请输入型号')returnself.database.add_model(component_id, model_name, price, option_type)self.update_tree()self.model_edit.clear()self.price_edit.clear()except ValueError:QMessageBox.warning(self, '错误', '价格必须是数字')def delete_item(self):"""删除项目"""items = self.tree.selectedItems()if not items:QMessageBox.warning(self, '警告', '请选择要删除的项目')returnif QMessageBox.question(self, '确认', '确定要删除选中的项目吗？') == QMessageBox.Yes:for item in items:model_id = item.data(0, Qt.UserRole)self.database.delete_model(model_id)self.update_tree() database.py
import sqlite3
from datetime import datetime
import osclass Database:def __init__(self):self.db_file = os.path.join(os.path.dirname(__file__), 'computer_config.db')self.init_database()def init_database(self):"""初始化数据库表"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()# 创建配件类型表cursor.execute('''CREATE TABLE IF NOT EXISTS components (id INTEGER PRIMARY KEY AUTOINCREMENT,name TEXT NOT NULL UNIQUE)''')# 创建配件型号表cursor.execute('''CREATE TABLE IF NOT EXISTS models (id INTEGER PRIMARY KEY AUTOINCREMENT,component_id INTEGER,model_name TEXT NOT NULL,price REAL NOT NULL,option_type TEXT NOT NULL,FOREIGN KEY (component_id) REFERENCES components (id),UNIQUE(component_id, model_name, option_type))''')# 初始化基础配件数据base_components = ['CPU', '主板', '内存', '硬盘', 'SSD固态盘', '显卡', '机箱', '电源', '显示器', '键鼠套装', '键盘', '鼠标', '散热器', '音箱', '光存储']for component in base_components:cursor.execute('INSERT OR IGNORE INTO components (name) VALUES (?)', (component,))conn.commit()def get_components(self):"""获取所有配件类型"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()cursor.execute('SELECT id, name FROM components')return cursor.fetchall()def get_models(self, component_id, option_type):"""获取指定配件的型号列表"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()cursor.execute('''SELECT id, model_name, price FROM models WHERE component_id = ? AND option_type = ?''', (component_id, option_type))return cursor.fetchall()def add_model(self, component_id, model_name, price, option_type):"""添加或更新配件型号"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()cursor.execute('''INSERT OR REPLACE INTO models (component_id, model_name, price, option_type)VALUES (?, ?, ?, ?)''', (component_id, model_name, price, option_type))conn.commit()def delete_model(self, model_id):"""删除配件型号"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()cursor.execute('DELETE FROM models WHERE id = ?', (model_id,))conn.commit()def update_model_price(self, model_id, new_price):"""更新配件型号价格"""with sqlite3.connect(self.db_file) as conn:cursor = conn.cursor()cursor.execute('UPDATE models SET price = ? WHERE id = ?', (new_price, model_id))conn.commit()