from PyQt5.QtGui import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *

from config import items


class MyTreeWidget(QTreeWidget):
    def __init__(self, parent=None):
        super().__init__(parent=parent)
        self.mainwindow = parent
        self.setDragEnabled(True)
        # 选中不显示虚线
        # self.setEditTriggers(QAbstractItemView.NoEditTriggers)
        self.setFocusPolicy(Qt.NoFocus)


class FuncTreeWidget(QTreeWidget):
    def __init__(self, parent=None):
        super(FuncTreeWidget, self).__init__(parent)

        self.mainwindow = parent
        # 根节点
        root1 = QTreeWidgetItem(self)
        root1.setText(0, '重构')  # 0代表第一列，即Key列
        root1.setIcon(0, QIcon('icons/color.png'))  # 为节点设置图标
        self.setColumnWidth(0, 200)  # 第一列列宽设为200
        self.header().hide()
        self.itemClicked.connect(self.function)

        # 添加子节点1
        child1 = QTreeWidgetItem(root1)
        child1.setText(0, '裁剪')
        child1.setIcon(0, QIcon('icons/color.png'))

        root2 = QTreeWidgetItem(self)
        root2.setText(0, '调整')  # 第一列Key为 子节点1
        root2.setIcon(0, QIcon('icons/color.png'))

        child4 = QTreeWidgetItem(root2)
        child4.setText(0, '光效')
        child4.setIcon(0, QIcon('icons/color.png'))

        child5 = QTreeWidgetItem(root2)
        child5.setText(0, '色彩')
        child5.setIcon(0, QIcon('icons/color.png'))

        child6 = QTreeWidgetItem(root2)
        child6.setText(0, 'HSL')
        child6.setIcon(0, QIcon('icons/color.png'))

        child8 = QTreeWidgetItem(root2)
        child8.setText(0, '质感')
        child8.setIcon(0, QIcon('icons/color.png'))

        child10 = QTreeWidgetItem(root2)
        child10.setText(0, '特效')
        child10.setIcon(0, QIcon('icons/color.png'))

        root3 = QTreeWidgetItem(self)
        root3.setText(0, '风格迁移')
        root3.setIcon(0, QIcon('icons/color.png'))

        root4 = QTreeWidgetItem(self)
        root4.setText(0, '其他操作')
        root4.setIcon(0, QIcon('icons/color.png'))

        # 默认所有节点都处于展开状态
        # self.expandAll()

    def function(self):
        func_item = self.currentItem()
        self.mainwindow.update_dock(func_item.text(0))