spacepaste

Paste details

reply | raw

  1.  
  2. import sys
  3. import operator
  4. from PyQt5 import QtWidgets, QtCore
  5. from PyQt5.QtCore import QAbstractTableModel, Qt, QVariant
  6. from PyQt5.QtGui import QPixmap, QIcon, QFont, QBrush, QColor
  8. def getXExpandYExpandSizePolicy():
  9. sizePolicy = QtWidgets.QSizePolicy(QtWidgets.QSizePolicy.Expanding,
  10. QtWidgets.QSizePolicy.Expanding)
  11. return sizePolicy
  13. class TableModel(QAbstractTableModel):
  14. def __init__(self, parent, datain, headerdata):
  15. QAbstractTableModel.__init__(self, parent)
  17. self.arraydata=datain
  18. self.headerdata=headerdata
  20. def rowCount(self,p):
  21. return len(self.arraydata)
  23. def columnCount(self,p):
  24. if len(self.arraydata)>0:
  25. return len(self.arraydata[0])
  26. return 0
  28. def data(self, index, role):
  29. if not index.isValid():
  30. return QVariant()
  31. elif role == Qt.BackgroundRole:
  32. if index.row()%2==0:
  33. return QBrush(QColor(220,220,240))
  34. elif role != Qt.DisplayRole:
  35. return QVariant()
  36. return QVariant(self.arraydata[index.row()][index.column()])
  39. def headerData(self, col, orientation, role):
  40. if orientation==Qt.Horizontal and role==Qt.DisplayRole:
  41. return self.headerdata[col]
  42. return None
  44. def sort(self,col,order):
  45. if col!=0:
  46. self.layoutAboutToBeChanged.emit()
  47. self.arraydata=sorted(self.arraydata,key=operator.itemgetter(col))
  48. if order==Qt.DescendingOrder:
  49. self.arraydata.reverse()
  50. self.layoutChanged.emit()
  52. class MyHeaderView(QtWidgets.QHeaderView):
  53. def __init__(self,parent):
  54. QtWidgets.QHeaderView.__init__(self,Qt.Horizontal,parent)
  55. self.sectionResized.connect(self.myresize)
  57. def myresize(self, *args):
  58. """ Resize all sections to content and user interactive """
  60. header=self
  61. #header.setSectionResizeMode(QtWidgets.QHeaderView.Stretch)
  62. #header.setSectionResizeMode(QtWidgets.QHeaderView.Interactive)
  63. total_width=0
  64. ws=[]
  65. for c in range(header.count()):
  66. wii=header.sectionSize(c)
  67. ws.append(wii)
  68. total_width+=wii
  70. if args[0]>0 or args[0]<header.count():
  71. for ii in range(args[0],header.count()):
  72. if ii==args[0]:
  73. header.resizeSection(ii,args[2])
  74. elif ii==args[0]+1:
  75. #this=ws[args[0]]
  76. nexto=ws[ii]
  77. new=nexto-(args[2]-args[1])
  78. header.resizeSection(ii,new)
  79. else:
  80. header.resizeSection(ii,ws[ii])
  82. def resizeEvent(self, event):
  83. """ Resize all sections to content and user interactive """
  85. super(QtWidgets.QHeaderView, self).resizeEvent(event)
  86. header=self
  87. header.setSectionResizeMode(1,QtWidgets.QHeaderView.Stretch)
  88. for column in range(header.count()):
  89. header.setSectionResizeMode(column, QtWidgets.QHeaderView.Stretch)
  90. width = header.sectionSize(column)
  91. print 'column',column,'width',width
  92. header.setSectionResizeMode(column, QtWidgets.QHeaderView.Interactive)
  93. header.resizeSection(column, width)
  95. width = header.sectionSize(column)
  96. print 'column',column,'width2',width
  98. header.setSectionResizeMode(QtWidgets.QHeaderView.Stretch)
  99. header.setSectionResizeMode(QtWidgets.QHeaderView.Interactive)
  101. class MainFrame(QtWidgets.QWidget):
  103. def __init__(self):
  104. super(MainFrame,self).__init__()
  105. self.initUI()
  107. def initUI(self):
  109. self.doc_table=self.createTable()
  110. hlayout=QtWidgets.QHBoxLayout()
  111. hlayout.addWidget(self.doc_table)
  112. self.setLayout(hlayout)
  113. self.show()
  115. def createTable(self):
  116. self.tabledata=[['aaa' ,' title1', True, 1999],
  117. ['bbb' ,' title2', True, 2000],
  118. ['ccc' ,' title3', False, 2001]
  119. ]
  120. header=['author', 'title', 'read', 'year']
  121. tablemodel=TableModel(self,self.tabledata,header)
  123. tv=QtWidgets.QTableView(self)
  124. #hh=MyHeaderView(self)
  125. hh=tv.horizontalHeader()
  126. #tv.setHorizontalHeader(hh)
  127. tv.setModel(tablemodel)
  128. #tv.setSizePolicy(getXExpandYExpandSizePolicy())
  129. tv.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  130. tv.setShowGrid(True)
  132. hh.setSectionsMovable(True)
  133. hh.setStretchLastSection(False)
  134. hh.setSectionResizeMode(1,QtWidgets.QHeaderView.Stretch)
  135. print hh.stretchSectionCount()
  136. tv.setSortingEnabled(True)
  138. return tv
  141. class MainWindow(QtWidgets.QMainWindow):
  143. def __init__(self):
  144. super(MainWindow,self).__init__()
  146. self.main_frame=MainFrame()
  147. self.setCentralWidget(self.main_frame)
  148. self.setGeometry(100,100,1200,900) #(x_left,y_top,w,h)
  149. self.show()
  152. if __name__=='__main__':
  154. app=QtWidgets.QApplication(sys.argv)
  155. mainwindow=MainWindow()
  156. sys.exit(app.exec_())
  157.  
Powered by spacepaste. Paste removal.