import sys

from PyQt5.Qt import *
from PyQt5.QtCore import *
from PyQt5.QtWidgets import *

try:
    from .CAvatar import CAvatar
except:
    from CAvatar import CAvatar

Stylesheet = """
QWidget{
    background: rgb(238,244,249);
}
"""
Stylesheet_hover = """
QWidget,QLabel{
    background: rgb(230, 235, 240);
}
"""
Stylesheet_clicked = """
QWidget,QLabel{
    background: rgb(230, 235, 240);
}
"""


class QListWidgetItemWidget(QWidget):
    def __init__(self):
        super().__init__()
        self.is_selected = False

    def leaveEvent(self, e):  # 鼠标离开label
        if self.is_selected:
            return
        self.setStyleSheet(Stylesheet)

    def enterEvent(self, e):  # 鼠标移入label
        self.setStyleSheet(Stylesheet_hover)


# 自定义的item 继承自QListWidgetItem
class ContactQListWidgetItem(QListWidgetItem):
    def __init__(self, name, url, img_bytes=None):
        super().__init__()
        self.is_select = False
        # 自定义item中的widget 用来显示自定义的内容
        self.widget = QListWidgetItemWidget()
        # 用来显示name
        self.nameLabel = QLabel(self.widget)
        self.nameLabel.setText(name)
        # 用来显示avator(图像)
        self.avatorLabel = CAvatar(parent=self.widget, shape=CAvatar.Rectangle, size=QSize(30, 30),
                                   url=url, img_bytes=img_bytes)
        # 设置布局用来对nameLabel和avatorLabel进行布局
        hbox = QHBoxLayout(self.widget)
        self.checkBox = QCheckBox(self.widget)
        self.checkBox.clicked.connect(self.select)
        hbox.addWidget(self.checkBox)
        hbox.addWidget(self.avatorLabel)
        hbox.addWidget(self.nameLabel)
        hbox.addStretch(1)
        # 设置widget的布局
        self.widget.setLayout(hbox)
        self.widget.setStyleSheet(Stylesheet)
        # 设置自定义的QListWidgetItem的sizeHint，不然无法显示
        self.setSizeHint(self.widget.sizeHint())
        sizePolicy = QSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
        sizePolicy.setHorizontalStretch(0)
        sizePolicy.setVerticalStretch(0)
        sizePolicy.setHeightForWidth(self.widget.sizePolicy().hasHeightForWidth())
        self.widget.setSizePolicy(sizePolicy)

    def select(self):
        """
        设置选择后的事件
        @return:
        """
        self.widget.is_selected = True
        self.is_select = not self.is_select
        # print('选择',self.is_select)
        self.checkBox.setChecked(self.is_select)
        # self.widget.setStyleSheet(Stylesheet_clicked)

    def force_select(self):
        self.is_select = True
        self.checkBox.setChecked(self.is_select)

    def force_dis_select(self):
        self.is_select = False
        self.checkBox.setChecked(self.is_select)

    def dis_select(self):
        """
        设置取消选择的事件
        @return:
        """
        self.widget.is_selected = False
        self.widget.setStyleSheet(Stylesheet)


if __name__ == "__main__":
    app = QApplication(sys.argv)

    # 主窗口
    w = QWidget()
    w.setWindowTitle("QListWindow")
    # 新建QListWidget
    listWidget = QListWidget(w)
    listWidget.resize(300, 300)

    # 新建两个自定义的QListWidgetItem(customQListWidgetItem)
    item1 = ContactQListWidgetItem("鲤鱼王", "liyuwang.jpg")
    item2 = ContactQListWidgetItem("可达鸭", "kedaya.jpg")

    # 在listWidget中加入两个自定义的item
    listWidget.addItem(item1)
    listWidget.setItemWidget(item1, item1.widget)
    listWidget.addItem(item2)
    listWidget.setItemWidget(item2, item2.widget)

    # 绑定点击槽函数 点击显示对应item中的name
    listWidget.itemClicked.connect(lambda item: item.select())

    w.show()
    sys.exit(app.exec_())