lingjing/gui/list.py

# -*- coding: utf-8 -*-
from PyQt6.QtCore import *
from PyQt6.QtWidgets import *
from PyQt6.QtGui import *
from qfluentwidgets import *
from .common.config import cfg, FEEDBACK_URL, HELP_URL, EXAMPLE_URL
from .common.icon import Icon
from .common.style_sheet import StyleSheet
from .common.signal_bus import signalBus
from demeter.core import *
# https://doc.qt.io/qt-6/qt.html
from qfluentwidgets.components.dialog_box.mask_dialog_base import MaskDialogBase
import gc

class List(ScrollArea):

    def __init__(self, title: str, subtitle: str, model : str, parent=None, button=None, addButton=True):
        super().__init__(parent = parent)
        self.box = 350
        self.base = QWidget(self)
        self.model = model;
        self.title = title;
        self.toolBar = ToolBar(title, subtitle, self, button, addButton)
        self.vBoxLayout = QVBoxLayout(self.base)

        self.setHorizontalScrollBarPolicy(Qt.ScrollBarPolicy.ScrollBarAlwaysOff)
        self.setViewportMargins(0, self.toolBar.height(), 0, 0)
        self.setWidget(self.base)
        self.setWidgetResizable(True)

        self.vBoxLayout.setSpacing(30)
        self.vBoxLayout.setAlignment(Qt.AlignmentFlag.AlignTop)
        self.vBoxLayout.setContentsMargins(36, 20, 36, 36)

        self.base.setObjectName('view')
        StyleSheet.LIST.apply(self)
        self.pageWidgets = []

    def addCard(self, title, widget, sourcePath: str, stretch = 0):
        self.widget = widget
        self.card = Card(title, widget, sourcePath, stretch, self.base)
        self.vBoxLayout.addWidget(self.card, 0, Qt.AlignmentFlag.AlignTop)
        return self.card

    def scrollToCard(self, index: int):
        """ scroll to example card """
        w = self.vBoxLayout.itemAt(index).widget()
        self.verticalScrollBar().setValue(w.y())

    def resizeEvent(self, e):
        super().resizeEvent(e)
        self.toolBar.resize(self.width(), self.toolBar.height())

    def initPage(self):
        show_page = 10
        total_page =  Demeter.config['page']['total']
        current_page = Demeter.config['page']['current']
        
        start_page = max(1, current_page - show_page // 2)
        end_page = min(total_page, current_page + show_page // 2)
        if end_page - start_page < show_page:
            if start_page == 1:
                end_page = min(total_page, show_page)
            else:
                start_page = max(1, end_page - show_page + 1)
        prev_page = max(1, current_page - 1)
        next_page = min(total_page, current_page + 1)
        layout = QHBoxLayout()
        widgets = []
        widgets.append(self.createPageLabel('<', prev_page))
        if start_page > 1:
            widgets.append(self.createPageLabel(1, 1))
            widgets.append(self.createPageLabel('...', 2))
        for v in range(start_page, end_page+1, 1):
            widgets.append(self.createPageLabel(v, v, current_page))
        if end_page < total_page:
            widgets.append(self.createPageLabel('...', v+1))
            widgets.append(self.createPageLabel(total_page, total_page))
        widgets.append(self.createPageLabel('>', next_page))
        page_info = QLabel(f"共 {Demeter.config['page']['totalNum']} 条")
        page_info.setObjectName('pageCurrent')
        widgets.append(page_info)
        for key, value in enumerate(self.pageWidgets):
            self.vBoxLayout.removeWidget(value)
        for key, value in enumerate(widgets):
            layout.addWidget(value)
        layout.setAlignment(Qt.AlignmentFlag.AlignVCenter | Qt.AlignmentFlag.AlignCenter)
        #layout.setContentsMargins(0, 0, 0, 0)
        layout.setSpacing(10)
        self.vBoxLayout.addLayout(layout)
        self.pageWidgets = widgets

    def createPageLabel(self, label, page, current_page = 0):
        label = QPushButton(str(label))
        if page:
            label.mousePressEvent = lambda event: self.initData(page)
        if current_page and current_page == page:
            label.setObjectName('pageCurrent')
        else:
            label.setObjectName('page')
        return label

    def initButton(self, config):
        if config:
            button = QWidget()
            layout = QHBoxLayout()
            for key, value in config.items():
                click = PushButton(key)
                click.setObjectName('operButton')
                layout.addWidget(click)
                click.clicked.connect(config[key])
            button.setLayout(layout)
            return button
        return False

    def getInfo(self, id):
        return Demeter.service('common').one(self.model, id=id)

    def update(self):
        id = self.widget.getItem()
        self.showUpdate('编辑', id)

    def delete(self):
        if hasattr(self, 'deleteCheck'):
            state = self.deleteCheck()
            if state:
                InfoBar.error(
                    title='错误提示',
                    content=state,
                    orient=Qt.Orientation.Horizontal,
                    isClosable=True,
                    position=InfoBarPosition.TOP,
                    duration=2000,
                    parent=self.window()
                )
                return
        id = self.widget.getItem()
        self.showOper('删除', id, {'state' : 2})

    def showUpdate(self, title='', id=0):
        self.id = id
        box = MessageBoxBase(self.window())
        title = SubtitleLabel(title + self.title)
        box.viewLayout.addWidget(title)

        self.showUpdateItem(id)
        for key, value in self.updateItem.items():
            box.viewLayout.addWidget(value)

        # 设置对话框的最小宽度
        box.widget.setMinimumWidth(self.box)
        box.yesButton.setText("保存")
        box.cancelButton.setText("取消")
        if box.exec():
            self.save(box)

    def showOper(self, title = '', id = 0, data = {}):
        self.id = id
        box = MessageBoxBase(self.window())
        label = BodyLabel('确定要' + title + '吗？')
        title = SubtitleLabel(title + self.title)
        box.viewLayout.addWidget(title)
        box.viewLayout.addWidget(label)

        # 设置对话框的最小宽度
        box.widget.setMinimumWidth(350)
        box.yesButton.setText("确定")
        box.cancelButton.setText("取消")
        if box.exec():
            self.up(data)

    def save(self, box):
        up = True
        data = {}
        for key, value in self.updateItem.items():
            if 'text' not in key:
                if 'check' in key:
                    key = key.replace('_check', '')
                    data[key] = value.isChecked()
                else:
                    data[key] = value.text()
                if hasattr(self, 'updateFunc') and key in self.updateFunc:
                    data[key] = self.updateFunc[key](data[key])
        if hasattr(self, 'updateDefault'):
            for key, value in self.updateDefault.items():
                data[key] = value
        if 'name' in data:
            if not data['name']:
                InfoBar.error(
                    title='错误提示',
                    content="名称不能为空",
                    orient=Qt.Orientation.Horizontal,
                    isClosable=True,
                    position=InfoBarPosition.TOP,
                    duration=2000,
                    parent=self.window()
                )
                up = False
                if box.exec():
                    return self.save(box)
            else:
                model = Demeter.model(self.model)
                model.name = data['name']
                if hasattr(self, 'updateDefault'):
                    for key, value in self.updateDefault.items():
                        setattr(model, key, value)
                model.id.nq(self.id)
                info = model.select(type='fetchone')
                if info:
                    up = False
                    InfoBar.error(
                        title='错误提示',
                        content="名称已存在",
                        orient=Qt.Orientation.Horizontal,
                        isClosable=True,
                        position=InfoBarPosition.TOP,
                        duration=2000,
                        parent=self.window()
                    )
                    if box.exec():
                        return self.save(box)
        if up:
            self.up(data)

    def up(self, data = {}):
        Demeter.service('common').update(self.model, id=self.id, data=data)
        InfoBar.success(
            title='系统提示',
            content="操作已成功",
            orient=Qt.Orientation.Horizontal,
            isClosable=True,
            position=InfoBarPosition.TOP,
            duration=2000,
            parent=self.window()
        )
        if hasattr(self, 'upFunc'):
            for key, value in data.items():
                if key in self.upFunc:
                    self.upFunc[key](data[key])
        self.initData()

    def showTip(self, title, cur, data, click):
        position = TeachingTipTailPosition.LEFT
        view = TeachingTipView(
            icon=None,
            title=title,
            content="",
            isClosable=True,
            tailPosition=position,
        )

        self.listWidget = ListWidget()

        index = 0
        stands = []
        for key, value in enumerate(data):
            if (cur['id'] == value['id']):
                index = key
            stands.append(value['name'])

        # 添加列表项
        for stand in stands:
            item = QListWidgetItem(stand)
            #item.setIcon(QIcon(':/qfluentwidgets/images/logo.png'))
            self.listWidget.addItem(item)


        self.listWidget.setCurrentRow(index)
        self.listWidget.itemClicked.connect(click)

        view.addWidget(self.listWidget)

        # 添加组件
        '''
        button = PushButton('确认选择')
        button.setFixedWidth(120)
        view.addWidget(button, align=Qt.AlignCenter)
        '''
        

        self.tip = PopupTeachingTip.make(
            target=self.toolBar.button[0],
            view=view,
            duration=-1,    # 关闭自动消失
            tailPosition=position,
            parent=self
        )
        view.closed.connect(self.tip.close)

class TableBase(TableWidget):

    def __init__(self, parent=None):
        super().__init__(parent)
        self.parent = parent
        # 设置不可编辑
        #self.setEditTriggers(QAbstractItemView.NoEditTriggers)
        self.verticalHeader().hide()
        self.setBorderRadius(8)
        self.setBorderVisible(True)

        self.horizontalHeader().setSectionResizeMode(QHeaderView.ResizeMode.Stretch)
        self.verticalHeader().setSectionResizeMode(QHeaderView.ResizeMode.ResizeToContents)

        width = parent.window().width()
        height = parent.window().height()

        self.setFixedSize(width-170, height-380)
        #self.resize(800, 300)
        self.resizeColumnsToContents()
        self.resizeRowsToContents()
        self.init()

    def initData(self, page=1, param={}):
        column = len(self.head)
        self.setColumnCount(column)
        body = Demeter.model(self.parent.model).getList(page, param)
        self.setRowCount(len(body))
        for i, info in enumerate(body):
            id = info[0]
            for j in range(column):
                info[j] = QTableWidgetItem(info[j])
                info[j].setTextAlignment(Qt.AlignmentFlag.AlignCenter)
                self.setItem(i, j, info[j])
            self.setCellWidget(i, column-1, self.parent.initButton(self.button))
        self.setHorizontalHeaderLabels(self.head)
        if Demeter.config['page']['total'] > 0:
            self.parent.initPage()

    def getItem(self, index = 0):
        item = ''
        pos = self.getPos()
        if pos:
            row = pos.row()
            column = pos.column()
            item = self.item(row, index).text()
        return item

    def getPos(self):
        button = self.sender()
        if button:
            pos = self.indexAt(button.parent().pos())
            return pos
        return False

    
class SeparatorWidget(QWidget):
    """ Seperator widget """

    def __init__(self, parent=None):
        super().__init__(parent=parent)
        self.setFixedSize(6, 16)

    def paintEvent(self, e):
        painter = QPainter(self)
        pen = QPen(1)
        pen.setCosmetic(True)
        c = QColor(255, 255, 255, 21) if isDarkTheme() else QColor(0, 0, 0, 15)
        pen.setColor(c)
        painter.setPen(pen)

        x = self.width() // 2
        painter.drawLine(x, 0, x, self.height())


class ToolBar(QWidget):
    """ Tool bar """

    def __init__(self, title, subtitle, parent=None, button=None, addButton=True):
        super().__init__(parent=parent)
        self.parent = parent
        self.button = button
        self.titleLabel = TitleLabel(title, self)
        self.subtitleLabel = CaptionLabel(subtitle, self)
        self.addButton = addButton

        if self.addButton:
            self.createButton = PushButton(
                '添加', self, FluentIcon.ADD)
        
        self.helpButton = ToolButton(FluentIcon.FEEDBACK, self)
        self.separator = SeparatorWidget(self)
        self.themeButton = ToolButton(FluentIcon.CONSTRACT, self)

        self.vBoxLayout = QVBoxLayout(self)
        self.buttonLayout = QHBoxLayout()

        self.__initWidget()

    def __initWidget(self):
        self.setFixedHeight(138)
        self.vBoxLayout.setSpacing(0)
        self.vBoxLayout.setContentsMargins(36, 22, 36, 12)
        self.vBoxLayout.addWidget(self.titleLabel)
        self.vBoxLayout.addSpacing(4)
        self.vBoxLayout.addWidget(self.subtitleLabel)
        self.vBoxLayout.addSpacing(4)
        self.vBoxLayout.addLayout(self.buttonLayout, 1)
        self.vBoxLayout.setAlignment(Qt.AlignmentFlag.AlignTop)

        self.buttonLayout.setSpacing(4)
        self.buttonLayout.setContentsMargins(0, 0, 0, 0)
        if self.addButton:
            self.buttonLayout.addWidget(self.createButton, 0, Qt.AlignmentFlag.AlignLeft)
        if self.button:
            for key, value in enumerate(self.button):
                self.button[key] = PushButton(value['name'], self, value['icon'])
                self.button[key].clicked.connect(value['click'])
                self.buttonLayout.addWidget(self.button[key], 0, Qt.AlignmentFlag.AlignLeft)
        
        self.buttonLayout.addStretch(1)
        self.buttonLayout.addWidget(self.helpButton, 0, Qt.AlignmentFlag.AlignRight)
        self.buttonLayout.addWidget(self.separator, 0, Qt.AlignmentFlag.AlignRight)
        self.buttonLayout.addWidget(self.themeButton, 0, Qt.AlignmentFlag.AlignRight)
        self.buttonLayout.setAlignment(Qt.AlignmentFlag.AlignVCenter | Qt.AlignmentFlag.AlignLeft)

        self.themeButton.installEventFilter(ToolTipFilter(self.themeButton))
        self.helpButton.installEventFilter(ToolTipFilter(self.helpButton))
        self.themeButton.setToolTip(self.tr('Toggle theme'))
        self.helpButton.setToolTip('在线教程')

        self.themeButton.clicked.connect(lambda: toggleTheme(True))
        if self.addButton:
            self.createButton.clicked.connect(self.add)
        
        self.helpButton.clicked.connect(
            lambda: QDesktopServices.openUrl(QUrl(HELP_URL)))

        self.subtitleLabel.setTextColor(QColor(96, 96, 96), QColor(216, 216, 216))

    def add(self):
        self.parent.showUpdate('添加')

        

class Card(QWidget):
    """ card """

    def __init__(self, title, widget: QWidget, sourcePath, stretch=0, parent=None):
        super().__init__(parent=parent)
        self.widget = widget
        self.stretch = stretch

        self.titleLabel = StrongBodyLabel(title, self)
        self.card = QFrame(self)

        self.sourceWidget = QFrame(self.card)
        self.sourcePath = sourcePath
        self.sourcePathLabel = BodyLabel(
            self.tr('Source code'), self.sourceWidget)
        self.linkIcon = IconWidget(FluentIcon.LINK, self.sourceWidget)

        self.vBoxLayout = QVBoxLayout(self)
        self.cardLayout = QVBoxLayout(self.card)
        self.topLayout = QHBoxLayout()
        #self.bottomLayout = QHBoxLayout(self.sourceWidget)

        self.__initWidget()

    def __initWidget(self):
        self.linkIcon.setFixedSize(16, 16)
        self.__initLayout()

        self.sourceWidget.setCursor(Qt.CursorShape.PointingHandCursor)
        self.sourceWidget.installEventFilter(self)

        self.card.setObjectName('card')
        self.sourceWidget.setObjectName('sourceWidget')

    def __initLayout(self):
        '''
        setSizeConstraint 方法用于设置布局的大小约束，它可以接受以下参数：
        QLayout.SizeConstraint.SetDefaultConstraint：默认约束，布局将根据其内容自动调整大小。
        QLayout.SizeConstraint.SetFixedSize：固定大小约束，布局的大小将保持不变。
        QLayout.SizeConstraint.SetMinimumSize：最小尺寸约束，布局的大小将不小于指定的最小尺寸。
        QLayout.SizeConstraint.SetMaximumSize：最大尺寸约束，布局的大小将不大于指定的最大尺寸。
        '''
        self.vBoxLayout.setSizeConstraint(QLayout.SizeConstraint.SetMinimumSize)
        self.cardLayout.setSizeConstraint(QLayout.SizeConstraint.SetMinimumSize)
        self.topLayout.setSizeConstraint(QLayout.SizeConstraint.SetMinimumSize)

        self.vBoxLayout.setSpacing(12)
        self.vBoxLayout.setContentsMargins(0, 0, 0, 0)
        self.topLayout.setContentsMargins(12, 12, 12, 12)
        #self.bottomLayout.setContentsMargins(18, 18, 18, 18)
        self.cardLayout.setContentsMargins(0, 0, 0, 0)

        self.vBoxLayout.addWidget(self.titleLabel, 0, Qt.AlignmentFlag.AlignTop)
        self.vBoxLayout.addWidget(self.card, 0, Qt.AlignmentFlag.AlignTop)
        self.vBoxLayout.setAlignment(Qt.AlignmentFlag.AlignTop)

        self.cardLayout.setSpacing(0)
        self.cardLayout.setAlignment(Qt.AlignmentFlag.AlignTop)
        self.cardLayout.addLayout(self.topLayout, 0)
        self.cardLayout.addWidget(self.sourceWidget, 0, Qt.AlignmentFlag.AlignBottom)

        self.widget.setParent(self.card)
        self.topLayout.addWidget(self.widget)
        if self.stretch == 0:
            self.topLayout.addStretch(1)

        self.widget.show()

        #self.bottomLayout.addWidget(self.sourcePathLabel, 0, Qt.AlignmentFlag.AlignLeft)
        #self.bottomLayout.addStretch(1)
        #self.bottomLayout.addWidget(self.linkIcon, 0, Qt.AlignmentFlag.AlignRight)
        #self.bottomLayout.setAlignment(Qt.AlignmentFlag.AlignLeft | Qt.AlignmentFlag.AlignVCenter)

    def eventFilter(self, obj, e):
        if obj is self.sourceWidget:
            if e.type() == QEvent.Type.MouseButtonRelease:
                QDesktopServices.openUrl(QUrl(self.sourcePath))

        return super().eventFilter(obj, e)

class WorkerSignals(QObject):
    finished = pyqtSignal()
    error = pyqtSignal(tuple)
    result = pyqtSignal(object)
    progress = pyqtSignal(int)  # Add progress signal


class Worker(QThread):
    def __init__(self, fn, *args, **kwargs):
        super().__init__()
        self.fn = fn
        self.args = args
        self.kwargs = kwargs
        self.signals = WorkerSignals()
        self._is_running = True

    @pyqtSlot()
    def run(self):
        try:
            # Call the function and pass a callback to report progress
            result = self.fn(self.report_progress, *self.args, **self.kwargs)
            if self._is_running:  # 检查标志，如果未停止则发出结果信号
                self.signals.result.emit(result)
        except Exception as e:
            self.signals.error.emit((e.__class__, e, e.__traceback__))
        finally:
            self.signals.finished.emit()

    def report_progress(self, progress):
        if self._is_running:  # 检查标志，如果未停止则发出进度信号
            self.signals.progress.emit(progress)

    def stop(self):
        self._is_running = False  # 设置标志为 False 以停止线程的运行

    def terminate(self):
        self._is_running = False  # 确保标志位设置为 False
        super().terminate()  # 强制终止线程
        self.clean_up()  # 执行清理

    def clean_up(self):
        # 执行任何必要的清理
        del self.fn
        del self.args
        del self.kwargs
        gc.collect()  # 强制垃圾回收


class Progress(MaskDialogBase):
    cancelSignal = pyqtSignal()

    def __init__(self, title, parent=None):
        super().__init__(parent=parent)
        self.work = False
        self.title = BodyLabel(title)
        #self.widget.setFixedSize(500, 150)
        self.vBoxLayout = QVBoxLayout(self.widget)
        self.viewLayout = QVBoxLayout()
        self.__initWidget()

    def __initWidget(self):
        self.__setQss()
        self.__initLayout()

        self.setShadowEffect(60, (0, 10), QColor(0, 0, 0, 50))
        self.setMaskColor(QColor(0, 0, 0, 76))

    def __initLayout(self):
        self._hBoxLayout.removeWidget(self.widget)
        self._hBoxLayout.addWidget(self.widget, 1, Qt.AlignmentFlag.AlignCenter)

        self.vBoxLayout.setSpacing(0)
        self.vBoxLayout.setContentsMargins(0, 0, 0, 0)
        self.vBoxLayout.addLayout(self.viewLayout, 1)

        self.viewLayout.setSpacing(12)
        self.viewLayout.setContentsMargins(24, 24, 24, 24)

        
        self.progressBar = ProgressBar()
        #ProgressBar ProgressRing IndeterminateProgressRing
        self.progressBar.setRange(0, 100)
        self.progressBar.setValue(0)
        self.viewLayout.addWidget(self.title)
        self.viewLayout.addWidget(self.progressBar)

        self.show()

    def __setQss(self):
        FluentStyleSheet.DIALOG.apply(self)

    def setValue(self, value):
        self.progressBar.setValue(value)

    def __onCancelButtonClicked(self):
        self.cancelSignal.emit()
        self.reject()
        '''
        if self.work:
            self.work.close()
        '''

    def bind(self, work):
        self.work = work