import os
import sys
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import Qt
from utils.QSSLoader import QSSLoader
from view.ImportantView import Important
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('我的一天', 1, 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
        item = SelfListWidgetItem('新建列表')
        item.setTextAlignment(Qt.AlignCenter)
        # 使得item是可以编辑的.
        item.setFlags(item.flags() | Qt.ItemIsEditable)
        self.system_listWidget.addItem(item)
        # 创建后就可以编辑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_())