import os
import sys
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import Qt

from utils.CreateToDo import CreateToDo
from utils.QSSLoader import QSSLoader
from view.ImportantView import Important
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):
    def __init__(self):
        super(MainWidget, self).__init__()
        self.user_label = User()
        # 主布局，左右两侧
        layout = QHBoxLayout()
        # 子布局,左边的部分
        self.sub_layout = QVBoxLayout()
        self.sub_layout.spacing()
        self.sub_layout.addWidget(self.user_label, 2)
        # 左侧列表
        self.system_listWidget = QListWidget(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.CreateNewItem)

        # self.add_item_label.setPixmap(QPixmap())
        self.sub_layout.addWidget(self.add_list_action, 1)
        # 禁止双击可编辑
        # self.custom.setEditTriggers(QAbstractItemView.NoEditTriggers)
        # 右键菜单
        # self.custom.setContextMenuPolicy(Qt.CustomContextMenu)
        # self.custom.setObjectName('custom')
        # self.custom.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.one_day = SelfListWidgetItem('我的一天', 0, os.getcwd() + '/../images/sun.svg')
        self.system_listWidget.setCurrentRow(1)
        # self.one_day.setSelected(True)
        self.important_item = SelfListWidgetItem('重要', 0, os.getcwd() + '/../images/star.svg')

        self.system_listWidget.currentRowChanged.connect(self.stackedWidget.setCurrentIndex)

        self.setLayout(layout)
        self.initUI()

    def initUI(self):
        # 系统默认的两个 item
        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)
        # self.system_listWidget.addScrollBarWidget(False)

        # 测试添加下面的
        # self.two_day = SelfListWidgetItem('我的一天', 1, os.getcwd() + '/../images/sun.svg')
        # self.custom.addItem(self.two_day)
        # self.custom.setItemWidget(self.two_day, self.two_day.widget)
        # self.custom.

    # todo 仔细研究右键菜单

    def myListWidgetContext(self, position):
        # 设置右键菜单
        pop_menu = QMenu(self)

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

        copy_action = QAction("复制分组", self)
        del_action = QAction("删除分组", self)

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

        rename_action.triggered.connect(self.RenameItem)
        copy_action.triggered.connect(self.CreateNewItem)
        del_action.triggered.connect(self.DeleteItem)
        pop_menu.exec_(self.custom.mapToGlobal(position))

    # 创建新的分组
    def CreateNewItem(self):
        # 创建一个没有名字的item
        new_list = '新建列表'
        create_todo = CreateToDo('ToDoList', new_list)
        item = SelfListWidgetItem(new_list)
        item.setTextAlignment(Qt.AlignCenter)
        # 使得item是可以编辑的.
        item.setFlags(item.flags() | Qt.ItemIsEditable)
        self.system_listWidget.addItem(item)
        todo_list = ToDoList(new_list)
        self.stackedWidget.addWidget(todo_list)
        # 创建后就可以编辑item,用户自己起名字.
        # self.system_listWidget.editItem(item)
        self.system_listWidget.setItemWidget(item, item.widget)

    # 删除分组
    def DeleteItem(self):
        self.custom.takeItem(self.custom.currentRow())

    # 重命名分组
    def RenameItem(self):
        curRow = self.custom.currentRow()
        item = self.custom.item(curRow)
        item.setFlags(item.flags() | Qt.ItemIsEditable)
        self.custom.editItem(item)
        self.custom.itemChanged.connect(lambda: self.ChangeItem(item))

    def ChangeItem(self, item):
        print("test")


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)


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

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