pyqt-todolist/view/ToDoListView.py

import os
import time

from PyQt5.QtGui import QFont, QPixmap, QIcon
from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QListWidget, QLabel, QPushButton, QMenu, QAction, \
    QLineEdit, QDialog
from PyQt5.QtCore import Qt, pyqtSignal, QPropertyAnimation, QRect, QEasingCurve

from utils.BasicUtils import get_todo, change_value, load_value, remove_todo_list
from utils.CreateToDo import CreateToDo
from view.AddToDoLabel import AddToDoAction
from view.ToDoItem import ToDoItem


class ToDoList(QWidget):
    """
    :param list_name: 列表名，用于显示标题
    :param uid : 列表的uid，用于加载Todo和创建时传递参数
    """

    # 测试更新的信号发送
    update_signal = pyqtSignal()

    # todo : 实现刷新功能，重新加载list_name ，把list_name设置为全局变量
    def __init__(self, list_name, uid):
        super(ToDoList, self).__init__()
        self.setObjectName('todolist')
        self.setAttribute(Qt.WA_StyledBackground)

        self.setStyleSheet('#todolist{background-image:url("../images/10.jpg");border-radius: 10px;}')

        # 布局样式：最外面一个大的垂直布局，里面的最上面标题是一个水平布局，水平布局显示的是一个垂直布局和一个右边的选项，用来更换背景
        # 布局样式：继承QWidget,里面一个垂直布局，最上面一个QWidget用来设置头部布局
        #         头部QWidget里面一个水平布局，水平布局里面一个QWidget（方便设置整体位置），然后是一个
        layout = QVBoxLayout()
        head_layout = QHBoxLayout()

        head_widget = QWidget(self)

        # head_widget.setMinimumHeight(175)

        self.todo_list = QListWidget()
        # 关闭滚动条
        self.todo_list.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
        self.todo_list.setObjectName('todo_list')

        # 添加ToDo的标签
        self.add_todo = AddToDoAction()
        self.add_todo.line_edit.returnPressed.connect(lambda: self.line_edit_add(self.add_todo.line_edit.text()))

        self.add_todo.setAlignment(Qt.AlignCenter)
        head_widget.setAttribute(Qt.WA_StyledBackground)

        head_widget.setLayout(head_layout)
        head_layout.setAlignment(Qt.AlignCenter)

        layout.addWidget(head_widget, 2, Qt.AlignTop)

        layout.addWidget(self.todo_list, 6, Qt.AlignCenter | Qt.AlignTop)
        self.head_label = QLabel(list_name)
        self.head_label.setObjectName('head_label')
        font = QFont()
        font.setPointSize(32)
        self.head_label.setFont(font)

        layout.addStretch(1)

        head_layout.addWidget(self.head_label, 8)
        layout.addStretch(1)
        layout.addWidget(self.add_todo, 2, Qt.AlignHCenter)

        self.menu_image = QPushButton()
        self.menu_image.setObjectName('menu_image')
        self.menu_image.setFixedSize(30, 30)
        # self.menu_image.setScaledContents(True)
        pic_btn = QPixmap(os.getcwd() + "/../images/menu.png")
        self.menu_image.setIcon(QIcon(pic_btn))

        head_layout.addWidget(self.menu_image, 2)
        head_layout.addStretch(1)

        self.setLayout(layout)
        # self.menu_image.clicked.connect(self.menu_action)

        self.button_menu = QMenu(self)
        # 背景透明
        self.button_menu.setAttribute(Qt.WA_TranslucentBackground)
        # 无边框、去掉自带阴影
        self.button_menu.setWindowFlags(
            self.button_menu.windowFlags() | Qt.FramelessWindowHint | Qt.NoDropShadowWindowHint)

        self.menu_image.setMenu(self.button_menu)

        refresh_label = QAction(self)
        refresh_label.setText('刷新')
        refresh_label.setIcon(QIcon('../images/refresh.svg'))
        self.button_menu.addAction(refresh_label)
        self.button_menu.addSeparator()
        refresh_label.triggered.connect(lambda: self.reload_name(uid))
        self.button_menu.addAction(refresh_label)

        theme_label = QAction(self)
        theme_label.setText('主题')
        theme_label.setDisabled(True)
        self.button_menu.addAction(theme_label)
        self.button_menu.addSeparator()

        label_action1 = QAction(self)
        label_action1.setIcon(QIcon('../images/1.jpg'))
        label_action1.setText('背景1')
        label_action1.triggered.connect(lambda: self.menu_action(1))
        self.button_menu.addAction(label_action1)

        label_action2 = QAction(self)
        label_action2.setIcon(QIcon('../images/2.jpg'))
        label_action2.setText('背景2')
        label_action2.triggered.connect(lambda: self.menu_action(2))
        self.button_menu.addAction(label_action2)

        label_action3 = QAction(self)
        label_action3.setIcon(QIcon('../images/3.jpg'))
        label_action3.setText('背景3')
        label_action3.triggered.connect(lambda: self.menu_action(3))
        self.button_menu.addAction(label_action3)

        label_action4 = QAction(self)
        label_action4.setIcon(QIcon('../images/4.jpg'))
        label_action4.setText('背景4')
        label_action4.triggered.connect(lambda: self.menu_action(4))
        self.button_menu.addAction(label_action4)

        label_action5 = QAction(self)
        label_action5.setIcon(QIcon('../images/5.jpg'))
        label_action5.setText('背景5')
        label_action5.triggered.connect(lambda: self.menu_action(5))
        self.button_menu.addAction(label_action5)
        self.uid = uid

        # self.update_signal.connect(self.refresh_action)

        self.refresh_action()
        # 右键菜单
        self.todo_list.setContextMenuPolicy(Qt.CustomContextMenu)
        self.todo_list.customContextMenuRequested.connect(self.todo_list_context)
        self.load_theme()

    def load_theme(self):
        theme = str(load_value(self.uid, 'Theme'))
        self.setStyleSheet('#todolist{background-image:url("../images/' + theme + '.jpg");'
                                                                                  'background-position: center;    border-radius: 10px;'
                                                                                  '}')

    def menu_action(self, count):
        # 点击按钮
        count = str(count)
        # self.menu_label = ChangeTheme()
        change_value(self.uid, 'Theme', count)
        self.setStyleSheet('#todolist{background-image:url("../images/' + count + '.jpg");'
                                                                                  'background-position: center;    border-radius: 10px;'
                                                                                  '}')
        # print('选中：' + count)

    def load_todo(self, name, uid):
        self.todo_item = ToDoItem(name, uid)
        self.todo_item.mark_icon.clicked.connect(lambda: self.mark_done(uid))
        self.todo_item.important_button.clicked.connect(lambda: self.mark_important(uid))

        self.todo_list.addItem(self.todo_item)
        self.todo_list.setItemWidget(self.todo_item, self.todo_item.widget)
        self.update_signal.emit()

    def reload_name(self, uid):
        label_name = load_value(uid, 'title')
        self.head_label.setText(label_name)
        self.refresh_action()

    def refresh_action(self):
        # self.my_day_list[]
        self.todo_list.clear()
        # print(self.my_day_list)
        load_todo = get_todo(self.uid)
        # print('<ToDoListView>', load_todo)
        if load_todo:
            for load in load_todo:
                self.load_todo(load[0], load[1])

    def line_edit_add(self, name):
        if name:
            self.add_todo.todo_name.emit(name)
            # print('添加的新ToDo：' + name)

            create_todo = CreateToDo('Todo', name, self.uid)
            todo_item = ToDoItem(name, create_todo)

            self.todo_list.addItem(todo_item)
            self.todo_list.setItemWidget(todo_item, todo_item.widget)

            self.add_todo.line_edit.clear()
            self.refresh_action()
            self.update_signal.emit()
            # self.initUI()

    def todo_list_context(self, position):
        # 设置右键菜单
        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)

        mark_myday = QAction(QIcon(QPixmap(os.getcwd() + '/../images/make_sun.svg')), u"设置为我的一天", self)
        mark_important = QAction(QIcon(QPixmap(os.getcwd() + '/../images/star.svg')), u"设置为重要", self)

        del_action = QAction("删除", self)

        # 查看右键时是否在item上面,如果不在.就不显示删除和修改.
        # pop_menu.addAction(rename_action)
        if self.todo_list.itemAt(position):
            pop_menu.addAction(mark_myday)
            pop_menu.addAction(mark_important)
            pop_menu.addAction(del_action)
            pop_menu.addAction(rename_action)

        mark_myday.triggered.connect(lambda: self.make_action('MyDay'))
        rename_action.triggered.connect(self.rename_item)
        del_action.triggered.connect(self.delete_item)
        mark_important.triggered.connect(lambda: self.make_action('Important'))
        pop_menu.exec_(self.todo_list.mapToGlobal(position))

    def make_action(self, make_type):
        # print(self.todo_list.currentRow())
        load_todo = get_todo(self.uid)
        uid = load_todo[self.todo_list.currentRow()][1]
        if make_type == 'MyDay':
            change_value(uid, 'isMyDay', True)
        elif make_type == 'Important':
            change_value(uid, 'isImportant', True)

        self.update_signal.emit()

    # 删除分组
    def delete_item(self):
        # print(self.load_list[self.system_listWidget.currentRow() - 2][1])
        loadtodo = get_todo(self.uid)
        # load_todo = get_todo(my_day_list[self.todo_list.currentRow()][1])
        uid = loadtodo[self.todo_list.currentRow()][1]
        remove_todo_list(uid)
        self.todo_list.takeItem(self.todo_list.currentRow())
        self.update_signal.emit()

    # 重命名分组

    def rename_item(self):
        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)
        if item_name:
            load_todo = get_todo(self.uid)
            uid = load_todo[self.todo_list.currentRow()][1]
            change_value(uid, 'title', item_name)
            # 重命名发送信号
            self.refresh_action()
            self.update_signal.emit()
            self.dialog.close()

    def mark_done(self, uid):
        # print('uid:',uid)
        self.todo_item.mark_icon.setIcon(QIcon(os.getcwd() + '/../images/finished.svg'))
        # print('mark done:', uid)
        change_value(uid, 'done', True)
        time.sleep(0.5)
        self.todo_list.takeItem(self.todo_list.currentRow())
        # self.refresh_action()
        self.update_signal.emit()
        self.refresh_action()

    def mark_important(self, uid):
        change_value(uid, 'isImportant', True)
        self.update_signal.emit()
        pass