customWidget.py

import os,sys
from PyQt5.QtCore import *
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from math import *
from testWidgets import myDial


DEFAULT_HEAD = 'icons/qq.png'
DEFAULT_MSG = 'Hello is there anyone?'
DEFAULT_IMG = 'icons/img.png'
DEFAULT_MSG="Hello World,zai zhe ge shijie shang meiyou shei nenggou sheipan wo"

def checkContainChinese(s):#判断是否为英文
    for ch in s:
        if u'\u4e00' <= ch <= u'\u9fff':
            return True
    return False

def splitStringByLen(text,Len):#其中使用\n分割，因此原来的\n将会被替换，一个中文大致顶三个英文的宽度
    #reload(sys) #暂时用
    text = text.replace('\n','.')
    (myText, nLen) = ('',0)
    for s in text:
        myText += s
        nLen += 3 if checkContainChinese(s) else 1
        if nLen >= (Len - 1):
            myText += '\n'
            nLen = 0
    return myText

class BubbleText(QLabel):
    """
    文字的显示  主要是控件的大小调节，
    起初准备用QTextEdit后来发现实现起来很难控制大小和混动条！
    只能舍弃次用QLabel继承实现了，关于控件的水平大小采用控制字符数量的方法(ヘ(_ _ヘ))，
    考虑到一个中文字符的宽度大概是3倍英文字符因此出现了checkContainChinese和splitStringByLen函数
    （我也不记得哪儿抄来的方法了）。在输入调用super(BubbleText, self).__init__(myText)
    前就把字符用\n分割好来显示"""
    border = 5
    trigon = 15
    lineLen = 60 #每行的文字数量

    minH = 2 * trigon + 2 * border
    minW = 2 * trigon + 2 * border

    def __init__(self,listItem,listView,text = DEFAULT_MSG,lr = True):
        self.listItem = listItem
        self.listView = listView
        self.text = text
        #字符串长度限制
        myText = splitStringByLen(text, self.lineLen) 
        # 文字分割
        super(BubbleText, self).__init__(myText)
        self.setMinimumWidth(self.minW)
        self.setFont(QFont("Times",15,QFont.Normal))
        #self.setStyleSheet("QLabel:hover{background-color:rgba(210,240,250,255);}")        # 
        #鼠标滑过的颜色设置，这样自定义的paintEvent绘图区域就被看穿了
        self.setState(False)#设置鼠标不进入状态，方便绘图区域的颜色更新
        self.lr = lr #标志绘制左还是右
        if self.lr:
            '''为了让实现显示的图片不会在super(BubbleImage, self).paintEvent(e)时和绘制的背景气泡冲突，
            设置控件的setContentsMargins绘图范围保证图像的绘图区域。'''
            self.setContentsMargins(self.trigon*sqrt(3)/2 + 15,self.border + 10,self.border + 15,self.border + 10)
        else:
            self.setContentsMargins(self.border + 15,self.border + 10,self.trigon*sqrt(3)/2 + 15,self.border + 10)

    def paintEvent(self, e):
        size =  self.size()
        qp = QPainter()
        qp.begin(self)
        if self.lr:
            self.leftBubble(qp,size.width(),size.height())
        else:
            self.rightBubble(qp,size.width(),size.height())
        qp.end()
        super(BubbleText, self).paintEvent(e)

    def leftBubble(self,qp, w, h):
        qp.setPen(self.colorLeftE)#设置画笔颜色，绘制的矩形边缘颜色
        qp.setBrush(self.colorLeftM)#设置红色的笔刷
        middle = h*(1-0.618)
        shifty = self.trigon/2
        shiftx = self.trigon*sqrt(3)/2
        rL = QRectF(shiftx,1,w-shiftx-self.border,h-3)
        #更改为圆角矩形
        pL=QPolygonF()
        pL.append(QPointF(0,middle)) #起始点
        pL.append(QPointF(shiftx, middle + shifty)) # 第二点
        pL.append(QPointF(shiftx, middle-shifty)) #第三点
        """
        pL.append(QPointF(w - self.border, h - self.border)) #第四点
        pL.append(QPointF(w - self.border, self.border)) #第五点
        pL.append(QPointF(shiftx, self.border)) #第六点
        pL.append(QPointF(shiftx, middle - shifty)) #第七点
        """
        qp.drawPolygon(pL)
        qp.drawRoundedRect(rL,10,10)
        qp.setPen(self.colorLeftM)
        #qp.setPen(QColor("#ff0000"))
        line=QLine(shiftx,middle+shifty,shiftx,middle-shifty)
        qp.drawLine(line)
        

    def rightBubble(self, qp, w, h):
        qp.setPen(self.colorRightE)#设置画笔颜色，绘制的矩形边缘颜色
        qp.setBrush(self.colorRightM)#设置红色的笔刷
        middle = h*(1-0.618)
        shifty = self.trigon/2
        shiftx = self.trigon*sqrt(3)/2
        rL = QRectF(self.border,1,w-shiftx-self.border,h-3)
        #更改为圆角矩形
        pL=QPolygonF()
        pL.append(QPointF(w,middle)) #起始点
        pL.append(QPointF(w-shiftx, middle + shifty)) # 第二点
        pL.append(QPointF(w-shiftx, middle - shifty)) #第三点
        """
        pL.append(QPointF(w - self.border, h - self.border)) #第四点
        pL.append(QPointF(w - self.border, self.border)) #第五点
        pL.append(QPointF(shiftx, self.border)) #第六点
        pL.append(QPointF(shiftx, middle - shifty)) #第七点
        """
        qp.drawPolygon(pL)
        qp.drawRoundedRect(rL,10,10)
        qp.setPen(self.colorRightM)
        #qp.setPen(QColor("#ff0000"))
        line=QLine(w-shiftx,middle+shifty,w-shiftx,middle-shifty)
        qp.drawLine(line)

    def setState(self,mouse):
        '''鼠标进入和鼠标出时需要显示不一样的效果，主要就是更新颜色变量，然后调用update更新重绘'''
        if mouse:#鼠标进入
            #self.colorLeftM = QColor("#eaeaea")
            self.colorLeftM = QColor("#eaeaea")
            self.colorLeftE = QColor("#D6D6D6")
            self.colorRightM = QColor("#8FD648")
            self.colorRightE = QColor("#85AF65")
        else:
            self.colorLeftM = QColor("#fafafa")
            self.colorLeftE = QColor("#D6D6D6")
            self.colorRightM = QColor("#9FE658")
            self.colorRightE = QColor("#85AF65")
        self.update() #更新界面，不用执行也可以更新，但是不实时

    def enterEvent(self,e):
        # print 'mouse entered'
        self.setState(True)
    def leaveEvent(self,e):
        # print 'mouse leaved'
        self.setState(False)
    
    def contextMenuEvent(self,e): 
        ''' 右键菜单实现文本的复制和控件的删除'''
        editUser = QAction(QIcon('icons/copy.png'),u'复制',self)#第一个参数也可以给一个QIcon图标
        editUser.triggered.connect(self.copyText)
        delUser = QAction(QIcon('icons/delete.png'),u'删除',self)
        delUser.triggered.connect(self.delTextItem)#选中就会触发
        menu = QMenu()
        menu.addAction(editUser)
        #menu.addAction(delUser)
        menu.exec_(QCursor.pos())#全局位置比较好，使用e.pos()还得转换
        e.accept() #禁止弹出菜单事件传递到父控件中

    def copyText(self,b):
        # print 'msg copyed'
        cb = QApplication.clipboard()
        cb.setText(self.text)

    def delTextItem(self,b):
        # print 'msg deleted'
        self.listView.takeItem(self.listView.indexFromItem(self.listItem).row())


class TextItem(QWidget):
    '''显示文字的Widget内容，为了让消息可以删除增加listItem和list传递到文本控件'''
    def __init__(self, listItem, listView, text = DEFAULT_MSG, lr=True, head = DEFAULT_HEAD):
        super(TextItem,self).__init__()
        hbox = QHBoxLayout()
        text = BubbleText(listItem,listView,text,lr)
        head = LabelHead(head)
        head.setFixedSize(50,50)
        dial=myDial()
        if lr is not True:
            hbox.addSpacerItem(QSpacerItem(1,1,QSizePolicy.Expanding,QSizePolicy.Preferred))
            hbox.addWidget(dial)
            hbox.addWidget(text)
            hbox.addWidget(head)
        else:
            hbox.addWidget(head)
            hbox.addWidget(text)
            hbox.addWidget(dial)
            hbox.addSpacerItem(QSpacerItem(1,1,QSizePolicy.Expanding,QSizePolicy.Preferred))
            
        hbox.setContentsMargins(0,0,0,0)
        self.setLayout(hbox)
        self.setContentsMargins(0,0,0,0)

class LabelHead(QLabel):
    '''LabelHead(QLabel)  类主要是为了展示用户头像'''
    def __init__(self,addr = DEFAULT_HEAD):
        super(LabelHead,self).__init__()
        self.setScaledContents(True)
        self.setReadOnly(True)
        self.setPicture(addr)

    def setReadOnly(self,b):
        self._readOnly = bool(b)

    def setPicture(self,addr):
        '''设置图像：继承至QLabel以便可以setPixmap设置图片'''
        self._picAddr = addr
        img = QPixmap(addr)
        self.setPixmap(img)
        return True

    def getPicture(self):
        return self._picAddr


class MsgList(QListWidget):
    """消息消息列表的控件，支持增加文字消息和增加图片消息"""
    def __init__(self):
        super(MsgList, self).__init__()
        # 设置所有样式锁定
        #qssFile.open(QFile.ReadOnly|QFile.Text)
        #print(qssFile.read())
        #self.setStyleSheet(qssFile.read())

    def addTextMsg(self,sz = DEFAULT_MSG, lr = True, head = DEFAULT_HEAD):
        it = QListWidgetItem(self)
        wid = self.size().width()
        item = TextItem(it,self,sz,lr,head) #增加必须指定本list和本item用于删除item
        # item.setEnabled(False) #对象灰度显示，不能导致ITEM不可选
        it.setSizeHint(item.sizeHint())
        it.setFlags(Qt.NoItemFlags)# 设置Item不可选择
        self.addItem(it)
        self.setItemWidget(it,item)
        #self.setCurrentItem(it)
        
    def addImageMsg(self,img = DEFAULT_IMG, lr = True, head = DEFAULT_HEAD):
        it = QListWidgetItem(self)
        wid = self.size().width()
        item = ImageItem(it,self,img,lr,head) #增加必须指定本list和本item用于删除item
        # item.setEnabled(False) #对象灰度显示，不能导致ITEM不可选
        it.setSizeHint(item.sizeHint())
        it.setFlags(Qt.ItemIsEnabled)# 设置Item不可选择
        self.addItem(it)
        self.setItemWidget(it,item)
        self.setCurrentItem(it)



if __name__=='__main__':
    app=QApplication(sys.argv)
    ml=MsgList()
    ml.setMinimumSize(500,500)
    ml.addTextMsg("Hello",True)
    ml.addTextMsg("World!",False)
    ml.addTextMsg(u"昨夜小楼又东风，春心泛秋意上心头，恰似故人远来载乡愁，今夜月稀掩朦胧，低声叹呢喃望星空，恰似回首终究一场梦，轻轻叹哀怨...",True)
    ml.addTextMsg(u"With a gentle look on her face, she paused and said,她脸上带着温柔的表情，稍稍停顿了一下，便开始讲话...",False)
    ml.addTextMsg(u"With a gentle look on her face, she paused and said,她脸上带着温柔的表情，稍稍停顿了一下，便开始讲话",False)
    ml.addTextMsg(u"With a gentle look on her face, she paused and said,她脸上带着温柔的表情，稍稍停顿了一下，便开始讲话",False)
    ml.addTextMsg(u"With a gentle look on her face, she paused and said,她脸上带着温柔的表情，稍稍停顿了一下，便开始讲话",False)
    ml.addTextMsg(u"With a gentle look on her face, she paused and said,她脸上带着温柔的表情，稍稍停顿了一下，便开始讲话",False)
    ml.find
    ml.show()

    app.exec_()