【Python、Qt】使用QItemDelegate实现单元格的富文本显示+复选框功能

主打一个 折磨 坑多 陪伴。代码为Python,C++的就自己逐条语句慢慢改吧。



Python代码:

import sys
from types import MethodType
from PyQt5.QtCore import Qt,QPoint,QSize,QRect,QEvent
from PyQt5.QtGui import QStandardItemModel, QStandardItem,QTextDocument,QTextCursor
from PyQt5.QtWidgets import QTreeView,QApplication,QItemDelegate,QStyle

class RichDelegate(QItemDelegate):#使用QItemDelegate进行单元格重绘:https://blog.csdn.net/Loc_Haoclass/article/details/106528047
	__cboxSize=QSize(14,14)#复选框大小
	__cboxAlignV=Qt.AlignVCenter#竖直位置(居中)
	__cboxAlignH=Qt.AlignLeft#水平位置(左对齐)
	def __init__(self,parent=None,*,align=None,size=None):
		super().__init__(parent)
		if(align):
			self.SetCheckboxAlign(align)
		if(size):
			self.SetCheckboxSize(size)
	def SetCheckboxAlign(self,align):#设置复选框位置
		alignV=[Qt.AlignTop,Qt.AlignVCenter,Qt.AlignBottom]
		alignH=[Qt.AlignLeft,Qt.AlignHCenter,Qt.AlignRight]
		alignV=list(filter(lambda a:int(align) & int(a)!=0,alignV))
		alignH=list(filter(lambda a:int(align) & int(a)!=0,alignH))
		alignV.append(Qt.AlignVCenter)
		alignH.append(Qt.AlignLeft)
		self.__cboxAlignV=alignV[0]
		self.__cboxAlignH=alignH[0]
	def SetCheckboxSize(self,size):#设置复选框大小
		self.__cboxSize=size

	def editorEvent(self,event,model,opt,index):#处理复选框点击逻辑:https://blog.csdn.net/xbnlkdbxl/article/details/51316424
		if(event.type()==QEvent.MouseButtonRelease):#仅处理鼠标抬起事件
			if(event.button()==Qt.LeftButton):#仅处理鼠标左键
				item=index.model().itemFromIndex(index)
				if(item.isCheckable()):#仅处理复选框存在的情况
					rect_cbox=self.__GetRect_Checkbox(opt.rect)
					if(rect_cbox.contains(event.pos())):#仅复选框被点击时翻转状态
						item.setCheckState(Qt.Unchecked if item.checkState()==Qt.Checked else Qt.Checked)
						return True
		return False
	def drawCheck(self,ptr,opt,rect,state):#绘制复选框(这里直接默认绘制,有想法的可以改成其他绘制例如画个圈之类的
		super().drawCheck(ptr,opt,rect,state)#默认绘制的复选框总是正方形
	def paint(self,ptr,opt,index):
		style=opt.widget.style() if opt.widget else QApplication.style()
		style.drawControl(QStyle.CE_ItemViewItem, opt, ptr, opt.widget)#这条语句解决了行选中时背景色不变化的问题:https://blog.csdn.net/gongjianbo1992/article/details/108687172

		rect=QRect(opt.rect)
		item=index.model().itemFromIndex(index)
		if(item.isCheckable()):#绘制复选框:https://blog.csdn.net/xbnlkdbxl/article/details/51316424
			rect_cbox=self.__GetRect_Checkbox(rect)
			self.drawCheck(ptr,opt,rect_cbox,item.checkState())
			#计算剩余位置用于绘制文本内容
			if(self.__cboxAlignH==Qt.AlignRight):#只调整水平位置(应该不会有人那么异端把复选框放在单元格正中间的吧,不会吧不会吧
				rect.setRight(rect.right()-rect_cbox.width())
			else:
				rect.setLeft(rect.left()+rect_cbox.width())

		tx=index.data()
		doc=QTextDocument()
		doc.setHtml(tx)
		txDot='...'#替换为省略号
		testPoint=QPoint(rect.width(), rect.height() / 2)#获取能完整显示的字符个数:https://blog.csdn.net/eiilpux17/article/details/118461445
		pos = doc.documentLayout().hitTest(testPoint, Qt.ExactHit)
		if(pos!=-1 and pos !=len(doc.toPlainText())):#不能完全显示的情况下进行字符替换
			docDot=QTextDocument()
			docDot.setHtml(txDot)
			docDot.setDocumentMargin(0)#发现调用该语句后doc.size取值恢复正常:https://cloud.tencent.com/developer/ask/sof/105271901
			testPoint=QPoint(rect.width()-docDot.size().width(), rect.height() / 2)
			pos = doc.documentLayout().hitTest(testPoint, Qt.ExactHit)
			if(pos==-1):
				pos=0
			cursor=QTextCursor(doc)
			cursor.setPosition(pos)
			# cursor.insertText(txDot)
			cursor.movePosition(QTextCursor.End, QTextCursor.KeepAnchor)
			cursor.insertText(txDot,cursor.block().charFormat())
		LT=rect.bottomLeft()
		LT.setY(LT.y()-doc.size().height())
		# LT=rect.topLeft()
		ptr.save()
		ptr.translate(LT)
		ptr.setClipRect(rect.translated(-rect.topLeft()))
		# doc.setDefaultTextOption(doc.defaultTextOption())
		doc.drawContents(ptr)
		ptr.restore()

	def sizeHint(self,opt,index):#设置行高函数:https://blog.csdn.net/Lutx/article/details/6641353
		tx=index.data()
		doc=QTextDocument()
		doc.setHtml(tx)
		size=doc.size()
		return QSize(size.width(),size.height())

	def __GetRect_Checkbox(self,rect):#返回复选框确切位置
		alignV=[Qt.AlignTop,Qt.AlignVCenter,Qt.AlignBottom]
		alignH=[Qt.AlignLeft,Qt.AlignHCenter,Qt.AlignRight]
		posV=[rect.top(),rect.bottom()]
		posH=[rect.left(),rect.right()]
		for nape in [[self.__cboxAlignV,alignV,posV,self.__cboxSize.height()],
					[self.__cboxAlignH,alignH,posH,self.__cboxSize.width()]]:
			align,alignLst,pos,width=nape
			index=alignLst.index(align)
			if(index==0):#靠左/靠上
				pos[1]=pos[0]+width
			elif(index==1):#居中
				pos[0]=pos[0]+int((pos[1]-pos[0]-width)/2)
				pos[1]=pos[0]+width
			elif(index==2):#靠右/靠下
				pos[0]=pos[1]-width
		return QRect(posH[0],posV[0],posH[1]-posH[0],posV[1]-posV[0])

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

	tv=QTreeView()
	tv.setModel(QStandardItemModel(tv))
	model=tv.model()
	model.appendRow([QStandardItem(d) for d in [' R3 ']])
	model.appendRow([QStandardItem(d) for d in [' R5 ']])
	model.appendRow([QStandardItem(d) for d in [' R7 ']])
	model.appendRow([QStandardItem(d) for d in [' R50px ','B70px']])
	model.appendRow([QStandardItem(d) for d in ['bbbppp
SSSIIIUUU

R20pt'
]]) model.item(1,0).setCheckable(True) model.item(3,1).setCheckable(True) model.item(4,0).setCheckable(True) rich_1=RichDelegate() rich_2=RichDelegate(align=Qt.AlignBottom|Qt.AlignRight)#复选框右对齐是什么邪道行为,太怪了(感觉除了左居中以外的对齐都是邪道 tv.setItemDelegateForRow(0,rich_1) tv.setItemDelegateForRow(1,rich_1) tv.setItemDelegateForRow(2,rich_1) tv.setItemDelegateForRow(3,rich_1) tv.setItemDelegateForRow(4,rich_2) tv.show() sys.exit(app.exec())

运行结果:

【Python、Qt】使用QItemDelegate实现单元格的富文本显示+复选框功能_第1张图片


补充:

1、我的代码仅完成富文本显示,像是往单元格里塞入按钮、下拉列表亦或是其他控件不在本篇讨论范围之内,有需要的可以参考[CSDN]QStyledItemDelegate单元格数据渲染与编辑、[51CTO]QTableWidget使用setCellWidget设置控件居中显示或是自行搜索其他文章
2、复选框的绘制样式甚至可以自定义,像是画成圆圈或是其他东西,又或是嫌黑色不好看改成紫色绿色啥的,只不过得自己实现就是了,重绘仅需QPainter倒少了挺多麻烦(只不过还是挺麻烦的所以没这需求就没必要自找麻烦


参考资料:

  • 使用QItemDelegate显示富文本:[CSDN]https://blog.csdn.net/Loc_Haoclass/article/details/106528047
  • QItemDelegate实现CheckBox复选框功能:[CSDN]https://blog.csdn.net/xbnlkdbxl/article/details/51316424
  • QStyledItemDelegate单元格数据渲染与编辑:[CSDN]https://blog.csdn.net/gongjianbo1992/article/details/108687172
  • 精确获取字符完整显示个数:[CSDN]https://blog.csdn.net/eiilpux17/article/details/118461445
  • QItemDelegate单元格设置行高:[CSDN]https://blog.csdn.net/Lutx/article/details/6641353

本文发布于CSDN,未经个人同意不得私自转载:https://blog.csdn.net/weixin_44733774/article/details/133838003

你可能感兴趣的:(python,qt)