forked from HeLiangHIT/pyqtChat
-
Notifications
You must be signed in to change notification settings - Fork 0
/
msgList.py
423 lines (361 loc) · 17.3 KB
/
msgList.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date : 2016-12-08 10:56:55
# @Author : He Liang ([email protected])
# @Link : https://github.com/HeLiangHIT
import os,sys
from PyQt4.QtCore import *
from PyQt4.QtGui import *
from math import *
DEFAULT_HEAD = 'icons/qq.png'
DEFAULT_MSG = 'Hello is there anyone?'
DEFAULT_IMG = 'icons/img.png'
def checkContainChinese(s):#判断是否为英文
for ch in s.decode('utf-8',errors = 'ignore'):
if u'\u4e00' <= ch <= u'\u9fff':
return True
return False
def splitStringByLen(text,Len):#其中使用\n分割，因此原来的\n将会被替换，一个中文大致顶三个英文的宽度
reload(sys) #暂时用
sys.setdefaultencoding('utf-8')
text = text.replace('\n','.')
(myText, nLen) = ('',0)
for s in text.decode('utf-8',errors = 'ignore'):
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 = 20
lineLen = 40 #每行的文字数量
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",20,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 + 3,self.border + 3,self.border + 3,self.border + 3)
else:
self.setContentsMargins(self.border + 3,self.border + 3,self.trigon*sqrt(3)/2 + 3,self.border + 3)
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/2
shifty = self.trigon/2
shiftx = self.trigon*sqrt(3)/2
pL = QPolygonF()
pL.append(QPointF(0,middle)) #起始点
pL.append(QPointF(shiftx, middle + shifty)) # 第二点
pL.append(QPointF(shiftx, h - self.border)) #第三点
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)
def rightBubble(self, qp, w, h):
qp.setPen(self.colorRightE)#设置画笔颜色，绘制的矩形边缘颜色
qp.setBrush(self.colorRightM)#设置红色的笔刷
middle = h/2
shifty = self.trigon/2
shiftx = self.trigon*sqrt(3)/2
pL = QPolygonF()
pL.append(QPointF(w,middle)) #起始点
pL.append(QPointF(w - shiftx, middle + shifty)) # 第二点
pL.append(QPointF(w - shiftx, h - self.border)) #第三点
pL.append(QPointF(self.border, h - self.border)) #第四点
pL.append(QPointF(self.border, self.border)) #第五点
pL.append(QPointF(w - shiftx, self.border)) #第六点
pL.append(QPointF(w - shiftx, middle - shifty)) #第七点
qp.drawPolygon(pL)
def setState(self,mouse):
'''鼠标进入和鼠标出时需要显示不一样的效果，主要就是更新颜色变量，然后调用update更新重绘'''
if mouse:#鼠标进入
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 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 = unicode(addr)
img = QPixmap(addr)
self.setPixmap(img)
return True
def getPicture(self):
return self._picAddr
# ref : http://stackoverflow.com/questions/18047427/pyqt-sms-bubble-widget
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)
if lr is not True:
hbox.addSpacerItem(QSpacerItem(1,1,QSizePolicy.Expanding,QSizePolicy.Preferred))
hbox.addWidget(text)
hbox.addWidget(head)
else:
hbox.addWidget(head)
hbox.addWidget(text)
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 BubbleImage(QLabel):
"""绘制背景气泡的控件"""
border = 5
trigon = 20
lineLen = 40 #每行的文字数量
minH = 2 * trigon + 2 * border
minW = 2 * trigon + 2 * border
def __init__(self,listItem,listView,img = DEFAULT_IMG,lr = True,maxWidth = 500):#图片显示的最长宽度
self.listItem = listItem
self.listView = listView
self.img = img
super(BubbleImage, self).__init__()
self.setMinimumWidth(self.minW)
# self.setStyleSheet("QLabel:hover{background-color:rgba(210,240,250,255);}")#鼠标滑过的颜色设置，这样自定义的paintEvent绘图区域就被看穿了
self.setState(False)#设置鼠标不进入状态，方便绘图区域的颜色更新
self.lr = lr #标志绘制左还是右
if self.lr:
self.setContentsMargins(self.trigon*sqrt(3)/2 + 3,self.border + 3,self.border + 3,self.border + 3)
else:
self.setContentsMargins(self.border + 3,self.border + 3,self.trigon*sqrt(3)/2 + 3,self.border + 3)
self.setScaledContents(True)
if not os.path.exists(img):
img = DEFAULT_IMG
pic = QPixmap(img)
self.wid = pic.size().width() if pic.size().width()<maxWidth else maxWidth
nPic = pic.scaledToWidth(self.wid) #高度自动计算，以便保持比例
self.setPixmap(nPic)
if img.endswith('gif'):
self.movie = QMovie(self)
self.movie.setFileName(img)
self.movie.setCacheMode(QMovie.CacheNone)
self.movie.frameChanged.connect(self.animate)
self.movie.start()
@pyqtSlot(int)
def animate(self,index):
pic = self.movie.currentPixmap()
nPic = pic.scaledToWidth(self.wid)
self.setPixmap(nPic)
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(BubbleImage, self).paintEvent(e)
def leftBubble(self,qp, w, h):
qp.setPen(self.colorLeftE)#设置画笔颜色，绘制的矩形边缘颜色
qp.setBrush(self.colorLeftM)#设置红色的笔刷
middle = h/2
shifty = self.trigon/2
shiftx = self.trigon*sqrt(3)/2
pL = QPolygonF()
pL.append(QPointF(0,middle)) #起始点
pL.append(QPointF(shiftx, middle + shifty)) # 第二点
pL.append(QPointF(shiftx, h - self.border)) #第三点
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)
def rightBubble(self, qp, w, h):
qp.setPen(self.colorRightE)#设置画笔颜色，绘制的矩形边缘颜色
qp.setBrush(self.colorRightM)#设置红色的笔刷
middle = h/2
shifty = self.trigon/2
shiftx = self.trigon*sqrt(3)/2
pL = QPolygonF()
pL.append(QPointF(w,middle)) #起始点
pL.append(QPointF(w - shiftx, middle + shifty)) # 第二点
pL.append(QPointF(w - shiftx, h - self.border)) #第三点
pL.append(QPointF(self.border, h - self.border)) #第四点
pL.append(QPointF(self.border, self.border)) #第五点
pL.append(QPointF(w - shiftx, self.border)) #第六点
pL.append(QPointF(w - shiftx, middle - shifty)) #第七点
qp.drawPolygon(pL)
def setState(self,mouse):
'''鼠标进入和鼠标出时需要显示不一样的效果，主要就是更新颜色变量，然后调用update更新重绘'''
if mouse:#鼠标进入
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.copyImage)
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 copyImage(self,b):
# print 'msg copyed'
cb = QApplication.clipboard()
cb.setImage(QImage(self.img))
def delTextItem(self,b):
# print 'msg deleted'
self.listView.takeItem(self.listView.indexFromItem(self.listItem).row())
def mouseDoubleClickEvent(self,e):
from PIL import Image
im = Image.open(self.img)
im.show()
class ImageItem(QWidget):
'''显示文字的Widget内容，为了让消息可以删除增加listItem和list传递到文本控件'''
def __init__(self, listItem, listView, img = DEFAULT_MSG, lr=True, head = DEFAULT_HEAD):
super(ImageItem,self).__init__()
hbox = QHBoxLayout()
img = BubbleImage(listItem,listView,img,lr)
head = LabelHead(head)
head.setFixedSize(50,50)
if lr is not True:
hbox.addSpacerItem(QSpacerItem(1,1,QSizePolicy.Expanding,QSizePolicy.Preferred))
hbox.addWidget(img)
hbox.addWidget(head)
else:
hbox.addWidget(head)
hbox.addWidget(img)
hbox.addSpacerItem(QSpacerItem(1,1,QSizePolicy.Expanding,QSizePolicy.Preferred))
hbox.setContentsMargins(0,0,0,0)
self.setLayout(hbox)
self.setContentsMargins(0,0,0,0)
'''
设置样式参考: http://blog.csdn.net/taiyang1987912/article/details/40979309
"QListWidget{background-color:rgba(0,240,250,255);color:#19649F;border:0px solid gray;padding:0px -2px 5px 5px;}"
"QListWidget::item{width:94px;height:35px;border:0px solid gray;background-color:transparent;padding:-1px;color:#000000}"
"QListView::item:!enabled{background-image:url(:/handleMenu_clusters_error.png);background:#ceaf01;color:#FF0000}"
"QListWidget::item:hover{background-image:url(:/handleMenu_lixt_bg_hover);color:#FFFFFF;border-width:0;}"
"QListWidget::item:selected{background-image:url(:/handleMenu_lixt_bg_selected.png);}"
'''
class MsgList(QListWidget):
"""消息消息列表的控件，支持增加文字消息和增加图片消息"""
def __init__(self):
super(MsgList, self).__init__()
# 设置所有样式锁定
self.setStyleSheet(
"QListWidget::item{border:0px solid gray;background-color:transparent;padding:0px;color:transparent}"
"QListView::item:!enabled{background-color:transparent;color:transparent;border:0px solid gray;padding:0px 0px 0px 0px;}"
"QListWidget::item:hover{background-color:transparent;color:transparent;border:0px solid gray;padding:0px 0px 0px 0px;}"
"QListWidget::item:selected{background-color:transparent;color:transparent;border:0px solid gray;padding:0px 0px 0px 0px;}")
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.ItemIsEnabled)# 设置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.addImageMsg('ref/bq.gif',True)
ml.addImageMsg('ref/mt.gif',False)
ml.show()
sys.exit(app.exec_())