pyqt-todolist/main/main.py

import os
import sys
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import Qt, pyqtSignal, QRect
from utils.BasicUtils import get_todo_list, remove_todo_list, change_value, read_ini, set_exit_status
from utils.CreateConfigure import CreateConfigure
from utils.CreateToDo import CreateToDo
from utils.QSSLoader import QSSLoader
from view.ImportantView import Important
from view.MyListWidget import MyListWidget
from view.ToDoListView import ToDoList
from view.UserLabel import User
from view.SelfListWidgetItem import SelfListWidgetItem
from view.AddListLabel import AddListAction
from view.MyDayView import MyDay


class MainWidget(QWidget):
    update_signal = pyqtSignal()
    # 重载UI信号
    reload_signal = pyqtSignal()

    def __init__(self):
        super(MainWidget, self).__init__()
        self.user_label = User()
        self.create_config = CreateConfigure()
        # 主布局，左右两侧
        layout = QHBoxLayout()
        # 子布局,左边的部分
        self.sub_layout = QVBoxLayout()
        self.sub_layout.spacing()
        self.sub_layout.addWidget(self.user_label, 2)
        # 左侧列表
        self.system_listWidget = MyListWidget(self)
        self.system_listWidget.setFrameShape(QListWidget.NoFrame)
        self.sub_layout.addWidget(self.system_listWidget, 15)

        # 添加最下边的QLabel
        self.add_list_action = AddListAction()
        self.add_list_action.create_todo_list.connect(self.create_item)

        # self.add_item_label.setPixmap(QPixmap())
        self.sub_layout.addWidget(self.add_list_action, 1)
        # 禁止双击可编辑
        self.system_listWidget.setEditTriggers(QAbstractItemView.NoEditTriggers)
        # 右键菜单
        self.system_listWidget.setContextMenuPolicy(Qt.CustomContextMenu)
        self.system_listWidget.setObjectName('custom')
        self.system_listWidget.customContextMenuRequested.connect(self.myListWidgetContext)

        # 右侧层叠窗口

        self.my_day = MyDay()
        self.important = Important()
        self.stackedWidget = QStackedWidget(self)
        self.stackedWidget.addWidget(self.my_day)
        self.stackedWidget.addWidget(self.important)

        layout.addLayout(self.sub_layout, 2)

        layout.addWidget(self.stackedWidget, 6)

        self.system_listWidget.currentRowChanged.connect(self.stackedWidget.setCurrentIndex)
        self.load_list = get_todo_list()

        self.setLayout(layout)
        self.user_label.sync_signal.connect(self.refresh_action)
        self.reload_signal.connect(self.refresh_action)
        self.initUI()
        # self.update_signal.connect(self.initUI)

    def initUI(self):
        self.important.update_signal.connect(self.my_day.refresh_action)
        self.my_day.update_signal.connect(self.important.refresh_action)
        # 先清除列表，方便后面调用
        self.system_listWidget.clear()
        # 系统默认的两个 item
        self.one_day = SelfListWidgetItem('我的一天', 0, None, os.getcwd() + '/../images/sun.svg')
        self.system_listWidget.setCurrentRow(1)

        self.important_item = SelfListWidgetItem('重要', 0, None, os.getcwd() + '/../images/star.svg')

        self.system_listWidget.addItem(self.one_day)
        self.system_listWidget.setItemWidget(self.one_day, self.one_day.widget)

        self.system_listWidget.addItem(self.important_item)
        self.system_listWidget.setItemWidget(self.important_item, self.important_item.widget)
        # todo ： 通过信号修改显示的标题名字
        # self.system_listWidget.change_list_name.connect(self.change_list)
        # 加载列表
        # todo：bug 第一个为空会导致无法添加
        if self.load_list:
            for load in self.load_list:
                self.load_item(load[0], load[1], load[3])

    def refresh_action(self):
        self.load_list = get_todo_list()
        self.initUI()

    def myListWidgetContext(self, position):
        if self.system_listWidget.currentRow() > 1:

            # 设置右键菜单
            pop_menu = QMenu(self)
            pop_menu.setObjectName('lift_menu')
            pop_menu.setAttribute(Qt.WA_TranslucentBackground)
            # 无边框、去掉自带阴影
            pop_menu.setWindowFlags(
                pop_menu.windowFlags() | Qt.FramelessWindowHint | Qt.NoDropShadowWindowHint)

            rename_action = QAction(u'重命名', self)

            # copy_action = QAction("复制分组", self)
            del_action = QAction("删除分组", self)
            change_icon = QMenu('修改图标', pop_menu)
            # 背景透明
            change_icon.setAttribute(Qt.WA_TranslucentBackground)
            # 无边框、去掉自带阴影
            change_icon.setWindowFlags(
                change_icon.windowFlags() | Qt.FramelessWindowHint | Qt.NoDropShadowWindowHint)

            # 二级菜单
            action1 = QAction(QIcon(os.getcwd() + '/../images/icon/星宿.svg'), "星宿", self)
            action2 = QAction(QIcon(os.getcwd() + '/../images/icon/度假.svg'), "度假", self)
            action3 = QAction(QIcon(os.getcwd() + '/../images/icon/冰淇淋.svg'), "美食", self)
            action4 = QAction(QIcon(os.getcwd() + '/../images/icon/学习.svg'), "学习", self)
            action5 = QAction(QIcon(os.getcwd() + '/../images/icon/工作.svg'), "工作", self)
            action6 = QAction(QIcon(os.getcwd() + '/../images/icon/购物.svg'), "购物", self)
            change_icon.addAction(action1)
            change_icon.addAction(action2)
            change_icon.addAction(action3)
            change_icon.addAction(action4)
            change_icon.addAction(action5)
            change_icon.addAction(action6)

            action1.triggered.connect(lambda: self.change_icon('星宿'))
            action2.triggered.connect(lambda: self.change_icon('度假'))
            action3.triggered.connect(lambda: self.change_icon('冰淇淋'))
            action4.triggered.connect(lambda: self.change_icon('学习'))
            action5.triggered.connect(lambda: self.change_icon('工作'))
            action6.triggered.connect(lambda: self.change_icon('购物'))

            # 查看右键时是否在item上面,如果不在.就不显示删除和修改.

            if self.system_listWidget.itemAt(position):
                pop_menu.addAction(del_action)
                pop_menu.addAction(rename_action)
                # pop_menu.addAction(change_icon)
                pop_menu.addMenu(change_icon)

            rename_action.triggered.connect(self.rename_item)
            # copy_action.triggered.connect(self.create_item)
            del_action.triggered.connect(self.delete_item)
            pop_menu.exec_(self.system_listWidget.mapToGlobal(position))

    def change_icon(self, icon):
        # print(icon)
        # load_list = get_todo_list()
        # print(self.load_list)
        self.load_list[self.system_listWidget.currentRow() - 2][3] = icon
        # print(self.load_list[self.system_listWidget.currentRow() - 2][3])
        change_value(self.load_list[self.system_listWidget.currentRow() - 2][1], 'icon', icon)
        self.initUI()

    # 加载左侧列表

    def load_item(self, list_name, uid, icon=None):
        # print('加载列表')
        # self.stackedWidget.currentChanged.connect(self.change)
        item = SelfListWidgetItem(list_name, 0, uid, os.getcwd() + '/../images/icon/' + icon + '.svg')
        item.setTextAlignment(Qt.AlignCenter)
        # 使得item是可以编辑的.
        item.setFlags(item.flags() | Qt.ItemIsEditable)
        self.system_listWidget.addItem(item)
        todo_list = ToDoList(list_name, uid)
        # 信号测试成功！！
        # 更新 重要 内容
        todo_list.update_signal.connect(self.important.refresh_action)
        todo_list.update_signal.connect(self.my_day.refresh_action)
        # todo_list.update_signal.connect(self.initUI)

        # 删除分组时可用
        self.important.update_signal.connect(todo_list.refresh_action)
        # self.important.update_signal.connect(self.my_day.refresh_action)
        self.my_day.update_signal.connect(todo_list.refresh_action)
        # self.my_day.update_signal.connect(self.important.refresh_action)

        self.update_signal.connect(todo_list.refresh_action)
        # 创建后就可以编辑item,用户自己起名字.
        # self.system_listWidget.editItem(item)
        self.stackedWidget.addWidget(todo_list)

        self.system_listWidget.setItemWidget(item, item.widget)

    # 点击创建新的分组
    def create_item(self, new_list='新建列表', uid=None):

        if not uid:
            create_todo = CreateToDo('ToDoList', new_list)
            self.load_list.append(create_todo)
            self.load_item(self.load_list[-1][0],self.load_list[-1][1],self.load_list[-1][3])

    # 删除分组
    def delete_item(self):
        # print(self.load_list[self.system_listWidget.currentRow() - 2][1])
        index = self.system_listWidget.currentRow() - 2
        # print(self.load_list)
        # print('index:',index)
        # print('delete:', self.load_list[index])
        remove_todo_list(self.load_list[index][1])
        del self.load_list[index]
        # print('result:', self.load_list)
        print('<main>(delete_item)', self.stackedWidget.indexOf(self.stackedWidget.currentWidget()))
        self.stackedWidget.removeWidget(self.stackedWidget.currentWidget())
        self.system_listWidget.takeItem(self.system_listWidget.currentRow())

    # 重命名分组

    def rename_item(self):
        # curRow = self.todo_list.currentRow()
        # item = self.todo_list.item(curRow)

        self.dialog = QDialog()
        self.dialog.setWindowTitle('重命名')
        self.dialog_layout = QVBoxLayout()
        self.dialog.resize(300, 200)
        self.dialog.setLayout(self.dialog_layout)
        line_edit = QLineEdit()
        # line_edit.setObjectName('rename_line')
        line_edit.setStyleSheet('border: 1px solid;background:transparent;min-width:400px;')
        line_edit.setPlaceholderText('请输入新的名字')
        self.dialog_layout.addWidget(line_edit)
        confirm_button = QPushButton('确认修改')
        self.dialog_layout.addWidget(confirm_button)
        confirm_button.setStyleSheet('background-color:#bcbcbc;border-radius:5px;')
        confirm_button.clicked.connect(lambda: self.change_name(line_edit.text()))

        self.dialog.exec_()

    def change_name(self, item_name):
        # print(item_name)
        index = self.system_listWidget.currentRow() - 2
        if item_name:
            self.load_list[index][0] = item_name
            # print(self.load_list)
            # print(self.load_list[index][1])
            change_value(self.load_list[index][1], 'title', item_name)

            self.update_signal.emit()
            # 重命名发送信号
            self.initUI()
            # self.update_signal.emit()
            self.dialog.close()


class MainWindow(QWidget):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.resize(1150, 850)
        self.setWindowTitle("待办事项")
        # 主布局，用来显示主页面，设置页面等
        self.layout = QStackedLayout(self)

        self.main_widget = MainWidget()
        # self.main_widget.resize(50, 200)
        self.layout.addWidget(self.main_widget)

        self.tray_icon = QSystemTrayIcon(self)
        self.tray_icon.setIcon(QIcon(os.getcwd() + '/../images/todo_info.svg'))
        # # self.style().standardIcon(QStyle.SP_ComputerIcon))
        #
        # self.checkbox = QCheckBox('最小化')
        # layout.addWidget(self.checkbox)

        '''
            Define and add steps to work with the system tray icon
            show - show window
            hide - hide window
            exit - exit from application
        '''
        show_action = QAction("显示", self)
        quit_action = QAction("退出", self)
        hide_action = QAction("隐藏", self)
        show_action.triggered.connect(self.show)
        hide_action.triggered.connect(self.hide)
        quit_action.triggered.connect(QApplication.instance().quit)
        tray_menu = QMenu()
        tray_menu.addAction(show_action)
        tray_menu.addAction(hide_action)
        tray_menu.addAction(quit_action)
        self.tray_icon.setContextMenu(tray_menu)
        self.tray_icon.show()

        # self.exit_widget = QWidget(self)
        #
        # self.exit_widget.setLayout(layout)

        # 关闭窗口时弹出确认消息

    def closeEvent(self, event):
        # 创建一个消息盒子（提示框）
        quitMsgBox = QMessageBox()

        # 设置气泡在屏幕上的位置，水平居中，垂直屏幕80%位置
        # desktop = QApplication.desktop()
        # quitMsgBox.setGeometry(QRect(int(desktop.width() / 2 - 75), int(desktop.height() * 0.5), 400, 300))
        self.save_config = QCheckBox('记住选择')
        self.save_config.setChecked(True)
        quitMsgBox.setCheckBox(self.save_config)
        # self.save_config.stateChanged.connect(self.save_exit)
        quitMsgBox.resize(300, 200)
        # 设置提示框的标题
        quitMsgBox.setWindowTitle('确认窗口')
        # 设置提示框的内容
        quitMsgBox.setText('你确定退出吗？')
        # 创建两个点击的按钮，修改文本显示内容
        buttonY = QPushButton('退出程序')

        buttonN = QPushButton('最小化到托盘')
        # 将两个按钮加到这个消息盒子中去，并指定yes和no的功能
        style = QSSLoader.read_qss_file(os.getcwd() + '/../resource/closeDialog.qss')
        buttonY.setStyleSheet(style)
        buttonN.setStyleSheet(style)

        quitMsgBox.addButton(buttonY, QMessageBox.YesRole)

        quitMsgBox.addButton(buttonN, QMessageBox.NoRole)
        exit_status = read_ini('System', 'exitstatus')
        if exit_status == 'None':
            # self.center(quitMsgBox)
            quitMsgBox.exec_()
            # 判断返回值，如果点击的是Yes按钮，我们就关闭组件和应用，否则就忽略关闭事件
            if quitMsgBox.clickedButton() == buttonY:
                if self.save_config.isChecked():
                    set_exit_status('Exit')
                event.accept()
            else:
                if self.save_config.isChecked():
                    set_exit_status('Min')
                self.hide()
                event.ignore()
        elif exit_status == 'Exit':
            event.accept()
        elif exit_status == 'Min':
            # print('Min')
            self.hide()
            event.ignore()


if __name__ == "__main__":
    app = QApplication(sys.argv)
    main = MainWindow()
    main.setWindowTitle('PyQtToDoList')
    # print(os.path.abspath('../'))
    style_sheet = QSSLoader.read_qss_file(os.getcwd() + '/../resource/current.qss')
    main.setStyleSheet(style_sheet)

    app.setWindowIcon(QIcon(os.path.abspath('../') + '/images/todo_info.svg'))
    main.show()
    sys.exit(app.exec_())