理发店收银管理系统 python、sqlite3、pyqt5

给姐姐家店写的一个，功能比较简单，结合gpt
功能包含：次卡和充值卡，可以查剩余次数、以及查找消费记录
···后期会把sqlite3转到mysql，换成springboot的一个项目
1.使用技术：

Python、sqlite3、PyQt5
2.页面

3.sqlite3

一共三张数据表：
user_card : 次卡数据
user_store ：储蓄卡
consumption：消费记录表
消费记录为了省事，包含了所有的记录，
4.代码（主要部分）

python 复制代码
import re
import sys
from PyQt5.QtWidgets import QApplication, QWidget, QLabel, QLineEdit, QPushButton, QVBoxLayout, QHBoxLayout, \
    QTableWidget, QTableWidgetItem, QHeaderView, QMessageBox, QDialog, QFormLayout, QInputDialog, QMainWindow
from PyQt5.QtGui import QColor, QPalette, QImage, QPixmap, QBrush, QPainter
from PyQt5.QtCore import Qt, QDate, QDateTime
import sqlite3


class BackgroundWidget(QWidget):
    def __init__(self, parent=None):
        super().__init__(parent)
        # self.image_path = "image/background.png"
        self.image_path = "image/panda.png"

    def paintEvent(self, event):
        painter = QPainter(self)
        painter.setRenderHint(QPainter.Antialiasing)

        # 绘制背景图片
        background_image = QImage(self.image_path)
        background_image = background_image.scaled(self.size(), Qt.IgnoreAspectRatio, Qt.SmoothTransformation)
        painter.drawImage(0, 0, background_image)

        # 绘制半透明遮罩层
        opacity = 0.6
        overlay_color = QColor(0, 0, 0, int(255 * opacity))
        painter.setCompositionMode(QPainter.CompositionMode_DestinationOver)
        painter.fillRect(self.rect(), overlay_color)

        super().paintEvent(event)


class ConsumeCard(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("理发店收银系统")
        self.resize(1200, 800)

        # 创建数据库连接
        self.connection = sqlite3.connect("db/barbershop.db")
        self.cursor = self.connection.cursor()

        # 创建界面控件
        self.id_label = QLabel("ID:")
        self.id_input = QLineEdit()
        self.name_label = QLabel("姓名:")
        self.name_input = QLineEdit()
        self.tel_label = QLabel("电话:")
        self.tel_input = QLineEdit()
        self.remain_label = QLabel("次数:")
        self.remain_input = QLineEdit()
        self.remark_label = QLabel("备注:")
        self.remark_input = QLineEdit()
        # self.discount_label = QLabel("折扣:")
        # self.discount_input = QLineEdit()

        self.create_user_button = QPushButton("新建用户")
        self.search_label = QLabel("查询(ID或电话):")
        self.search_input = QLineEdit()
        self.search_button = QPushButton("查询")
        self.clear_button = QPushButton("清除")
        self.recharge_button = QPushButton("充值")
        self.consume_button = QPushButton("消费")
        # self.record_button = QPushButton("消费记录")
        self.consumption_record_button = QPushButton("消费记录")

        self.user_table = QTableWidget()

        # 设置布局
        form_layout = QFormLayout()
        # form_layout.addRow(self.id_label, self.id_input)
        form_layout.addRow(self.name_label, self.name_input)
        form_layout.addRow(self.tel_label, self.tel_input)
        form_layout.addRow(self.remain_label, self.remain_input)
        form_layout.addRow(self.remark_label, self.remark_input)
        # form_layout.addRow(self.discount_label, self.discount_input)

        button_layout = QHBoxLayout()
        button_layout.addWidget(self.create_user_button)
        button_layout.addWidget(self.search_label)
        button_layout.addWidget(self.search_input)
        button_layout.addWidget(self.search_button)
        button_layout.addWidget(self.clear_button)
        button_layout.addWidget(self.recharge_button)
        button_layout.addWidget(self.consume_button)
        button_layout.addWidget(self.consumption_record_button)

        main_layout = QVBoxLayout()
        main_layout.addLayout(form_layout)
        main_layout.addLayout(button_layout)
        main_layout.addWidget(self.user_table)

        self.setLayout(main_layout)

        # 绑定信号与槽函数
        self.create_user_button.clicked.connect(self.create_user)
        self.search_button.clicked.connect(self.search_user)
        self.clear_button.clicked.connect(self.clear)
        self.recharge_button.clicked.connect(self.recharge)
        self.consume_button.clicked.connect(self.consume)
        # self.record_button.clicked.connect(self.show_consumption_records)
        self.consumption_record_button.clicked.connect(self.show_consumption_record)


        # 初始化用户表格
        self.init_user_table()

        # 当前选中的用户信息
        self.current_user = None

        background_widget = BackgroundWidget()
        background_widget.setLayout(main_layout)
        self.setCentralWidget(background_widget)

        self.setAttribute(Qt.WA_TranslucentBackground)  # 设置窗口背景透明

        # 设置样式
        self.setStyleSheet(
            
        )

    def set_background_image(self, image_path):
        background_image = QImage(image_path)
        background_image = background_image.scaled(self.size(), Qt.IgnoreAspectRatio, Qt.SmoothTransformation)
        palette = QPalette()
        palette.setBrush(QPalette.Background, QBrush(background_image))
        self.setPalette(palette)

    def init_user_table(self):
        self.user_table.setColumnCount(5)
        self.user_table.setHorizontalHeaderLabels(["ID", "姓名", "电话", "剩余次数", "备注"])
        self.user_table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.user_table.setSelectionBehavior(QTableWidget.SelectRows)
        self.user_table.setEditTriggers(QTableWidget.NoEditTriggers)
        self.user_table.cellClicked.connect(self.show_user_info)

        self.update_user_table()

    def update_user_table(self):
        self.user_table.clearContents()
        self.cursor.execute("SELECT id, name, tel, remain, remark FROM user_card")
        users = self.cursor.fetchall()
        self.user_table.setRowCount(len(users))
        for i, user in enumerate(users):
            for j in range(len(user)):
                item = QTableWidgetItem(str(user[j]))
                item.setTextAlignment(Qt.AlignCenter)
                self.user_table.setItem(i, j, item)

    def show_user_info(self, row):
        self.id_input.setText(self.user_table.item(row, 0).text())
        self.name_input.setText(self.user_table.item(row, 1).text())
        self.tel_input.setText(self.user_table.item(row, 2).text())
        self.remain_input.setText(self.user_table.item(row, 3).text())
        self.remark_input.setText(self.user_table.item(row, 4).text())
        # self.discount_input.setText(self.user_table.item(row, 5).text())

        # 存储选中的用户信息
        self.current_user = {
            'id': self.user_table.item(row, 0).text(),
            'name': self.user_table.item(row, 1).text(),
            'tel': self.user_table.item(row, 2).text(),
            'remain': self.user_table.item(row, 3).text(),
            'remark': self.user_table.item(row, 4).text(),
            # 'discount': self.user_table.item(row, 5).text()
        }

    def create_user(self):
        name = self.name_input.text()
        tel = self.tel_input.text()
        remain = self.remain_input.text()
        remark = self.remark_input.text()
        # discount = self.discount_input.text()

        if not name or not tel or not remain:
            QMessageBox.warning(self, "错误", "请填写完整用户信息")
            return

        try:
            remain = int(remain)
            # discount = float(discount)
        except ValueError:
            QMessageBox.warning(self, "错误", "次数必须是数字")
            return

        # 验证手机号格式
        if not self.validate_phone_number(tel):
            QMessageBox.warning(self, "错误", "手机号格式错误")
            return

        # 检查是否已经存在该手机号的用户
        self.cursor.execute("SELECT id FROM user_card WHERE tel = ?", (tel,))
        existing_user = self.cursor.fetchone()
        if existing_user:
            QMessageBox.warning(self, "错误", "已存在该用户手机号")
            return


        current_datetime = QDateTime.currentDateTime()
        formatted_datetime = current_datetime.toString(Qt.ISODate)
        formatted_datetime = formatted_datetime[:10] + " " + formatted_datetime[11:]
        self.cursor.execute("INSERT INTO consumption_record (name, tel, consumption, balance, consumption_time, remark) "
                            "VALUES (?, ?, ?, ?, ?, ?)",
                            (name, tel, remain, remain, formatted_datetime, remark))
        self.cursor.execute("INSERT INTO user_card (name, tel, remain, remark) "
                            "VALUES (?, ?, ?, ?)",
                            (name, tel, remain, remark))
        self.connection.commit()
        self.update_user_table()
        self.clear()

        QMessageBox.information(self, "成功", "用户创建成功")

    # 验证手机号
    def validate_phone_number(self, phone_number):
        pattern = r"^1[3-9]\d{9}$"
        return re.match(pattern, phone_number) is not None

    def search_user(self):
        keyword = self.search_input.text()
        self.cursor.execute(
            "SELECT id, name, tel, remain, remark  FROM user_card WHERE id LIKE ? OR name like ? OR tel LIKE ?",
            (f"%{keyword}%", f"%{keyword}%", f"%{keyword}%"))
        users = self.cursor.fetchall()
        if len(users) == 0:
            QMessageBox.warning(self, "错误", "未找到用户")
            return

        self.user_table.clearContents()
        self.user_table.setRowCount(len(users))
        for i, user in enumerate(users):
            for j in range(len(user)):
                item = QTableWidgetItem(str(user[j]))
                item.setTextAlignment(Qt.AlignCenter)
                self.user_table.setItem(i, j, item)

    def clear(self):
        self.id_input.clear()
        self.name_input.clear()
        self.tel_input.clear()
        self.remain_input.clear()
        self.remark_input.clear()
        # self.discount_input.clear()
        self.user_table.clearSelection()

        # 重置当前选中的用户信息
        self.current_user = None

    def recharge(self):
        if self.current_user is None:
            QMessageBox.warning(self, "错误", "请先选择用户")
            return

        id = self.current_user['id']
        name = self.current_user['name']
        tel = self.current_user['tel']
        remain = self.current_user['remain']
        remark = self.current_user['remark']

        # amount, ok = QInputDialog.getDouble(self, "充值", "请输入充值次数:", decimals=2)
        amount, ok = QInputDialog.getInt(self, "充值", "请输入充值次数:")
        if ok and amount > 0:
            self.cursor.execute("UPDATE user_card SET remain = remain + ? WHERE id = ?", (amount, id))
            try:
                remain = int(remain)
                amount = int(amount)
            except ValueError:
                QMessageBox.warning(self, "错误", "余额和折扣必须是数字")
                return
            current_datetime = QDateTime.currentDateTime()
            formatted_datetime = current_datetime.toString(Qt.ISODate)
            formatted_datetime = formatted_datetime[:10] + " " + formatted_datetime[11:]
            self.cursor.execute("INSERT INTO consumption_record (name, tel, consumption, balance, consumption_time, remark) "
                                "VALUES (?, ?, ?, ?, ?, ?)",
                                (name, tel, amount, remain + amount, formatted_datetime, remark))
            self.connection.commit()
            self.update_user_table()
            self.clear()
            QMessageBox.information(self, "成功", "充值成功")

        # 重置当前选中的用户信息
        self.current_user = None

    def consume(self):
        if self.current_user is None:
            QMessageBox.warning(self, "错误", "请先选择用户")
            return

        id = self.current_user['id']
        name = self.current_user['name']
        tel = self.current_user['tel']
        remain = self.current_user['remain']
        remark = self.current_user['remark']

        # amount, ok = QInputDialog.getDouble(self, "消费", "请输入消费金额:", decimals=2)
        amount, ok = QInputDialog.getInt(self, "消费", "请输入消费金额:")
        if ok and amount > 0:
            # self.cursor.execute("SELECT discount FROM user_card WHERE id = ?", (id,))
            # discount = self.cursor.fetchone()[0]

            try:
                remain = int(remain)
                # discount = float(discount)
                amount = int(amount)
            except ValueError:
                QMessageBox.warning(self, "错误", "余额和折扣必须是数字")
                return

            # amount *= discount
            if remain < amount:
                QMessageBox.warning(self, "错误", "余额不足")
                return

            self.cursor.execute("UPDATE user_card SET remain = remain - ? WHERE id = ?", (amount, id))
            current_datetime = QDateTime.currentDateTime()
            formatted_datetime = current_datetime.toString(Qt.ISODate)
            formatted_datetime = formatted_datetime[:10] + " " + formatted_datetime[11:]
            self.cursor.execute(
                "INSERT INTO consumption_record (name, tel, consumption, balance, consumption_time, remark) "
                "VALUES (?, ?, ?, ?, ?, ?)",
                (name, tel, -amount, remain - amount, formatted_datetime, remark))
            self.connection.commit()
            self.update_user_table()
            self.clear()
            QMessageBox.information(self, "成功", "消费成功")

        # 重置当前选中的用户信息
        self.current_user = None


    def show_consumption_record(self):
        if self.current_user is None:
            QMessageBox.warning(self, "错误", "请先选择用户")
            return

        tel = self.current_user['tel']
        self.cursor.execute(
            "SELECT id, name, tel, consumption, balance, consumption_time, remark FROM consumption_record WHERE tel = ?",
            (tel,))
        records = self.cursor.fetchall()

        dialog = QDialog(self)
        dialog.setWindowTitle("消费记录")
        dialog.resize(1400, 600)

        table = QTableWidget(dialog)
        table.setColumnCount(6)
        table.setHorizontalHeaderLabels(["姓名", "电话", "消费", "剩余次数", "消费时间", "备注"])
        table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        table.setSelectionBehavior(QTableWidget.SelectRows)
        table.setEditTriggers(QTableWidget.NoEditTriggers)
        table.setRowCount(len(records))

        for i, record in enumerate(records):
            for j in range(1, len(record)):
                item = QTableWidgetItem(str(record[j]))
                item.setTextAlignment(Qt.AlignCenter)
                table.setItem(i, j - 1, item)

        main_layout = QVBoxLayout()
        main_layout.addWidget(table)
        dialog.setLayout(main_layout)

        dialog.exec_()

    # def get_consumption_records(self, phone_number):
    #     self.cursor.execute("SELECT * FROM consumption_record WHERE tel=?",
    #                         (phone_number,))
    #     return self.cursor.fetchall()

    def closeEvent(self, event):
        self.connection.close()


if __name__ == '__main__':
    app = QApplication(sys.argv)
    cashier = ConsumeCard()
    cashier.show()
    sys.exit(app.exec_())