[PyQT5] QTableView 틀 고정(Frozen Columns)

개인적으로 QT를 좋아하는데 다만 아쉬운점은 엑셀과 유사한 Spread 컨트롤이 없다는 것입니다. 아쉬운대로 테이블 형식의 데이터는 QTableView 컨트롤을 이용하여 표시하고 있습니다.

유지보수 중인 프로젝트에서 데이터 양이 많아져 횡스크롤이 길어지게 되었습니다.

고객사에서 편의상 앞의 두 칼럼은 엑셀처럼 틀고정을 시켜달라는 요청을 해왔습니다.

하지만 QT의 QTableView 컨트롤에는 틀 고정과 같은 기능이 없습니다.

구글링하여 이에 관한 방법을 찾아보면 한 가지 자료가 검색됩니다.

검색한 자료를 프로젝트에 맞게 수정하여 올립니다.

아래 코드를 보시면 틀 고정을 하기 위한 추가적인 QTableView를 생성하는 것을 확인할 수 있습니다.

1. # -*- coding: utf-8 -*-
2. """ This is FreezeTableWidget module """
3.  
4. from PyQt5.QtCore import *
5. from PyQt5.QtGui import *
6. from PyQt5.QtWidgets import *
7.  
8.  
9. class FreezeTableWidget(QTableView):
10. def __init__(self, parent=None, fixed_col_count=1, *args):
11. QTableView.__init__(self, parent, *args)
12.  
13. self._fixed_col_count = fixed_col_count
14.  
15. self.frozenTableView = QTableView(self)
16.  
17. self.frozenTableView.verticalHeader().hide()
18. self.frozenTableView.setFocusPolicy(Qt.NoFocus)
19. self.frozenTableView.setStyleSheet('''border: none; background-color: #CCC''')
20. self.frozenTableView.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
21. self.frozenTableView.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
22. self.frozenTableView.horizontalHeader().setSectionResizeMode(QHeaderView.Fixed)
23.  
24. self.viewport().stackUnder(self.frozenTableView)
25.  
26. self.setShowGrid(True)
27.  
28. hh = self.horizontalHeader()
29. hh.setDefaultAlignment(Qt.AlignCenter)
30. hh.setStretchLastSection(True)
31.  
32. self.resizeColumnsToContents()
33.  
34. vh = self.verticalHeader()
35. vh.setDefaultSectionSize(25)
36. vh.setDefaultAlignment(Qt.AlignCenter)
37. vh.setVisible(True)
38. self.frozenTableView.verticalHeader().setDefaultSectionSize(vh.defaultSectionSize())
39.  
40. self.frozenTableView.show()
41. self.updateFrozenTableGeometry()
42.  
43. self.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
44. self.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel)
45. self.frozenTableView.setVerticalScrollMode(QAbstractItemView.ScrollPerPixel)
46.  
47. # connect the headers and scrollbars of both table view's together
48. self.horizontalHeader().sectionResized.connect(self.updateSectionWidth)
49. self.verticalHeader().sectionResized.connect(self.updateSectionHeight)
50. self.frozenTableView.verticalScrollBar().valueChanged.connect(self.verticalScrollBar().setValue)
51. self.verticalScrollBar().valueChanged.connect(self.frozenTableView.verticalScrollBar().setValue)
52.  
53. @property
54. def fixed_col_count(self):
55. return self._fixed_col_count
56.  
57. @fixed_col_count.setter
58. def fixed_col_count(self, value):
59. self._fixed_col_count = value
60.  
61. def setModel(self, model: QAbstractTableModel):
62. QTableView.setModel(self, model)
63. self.frozenTableView.setModel(model)
64. self.frozenTableView.verticalHeader().hide()
65. self.frozenTableView.setFocusPolicy(Qt.NoFocus)
66.  
67. cols = model.columnCount()
68. for col in range(cols):
69. if col not in range(self._fixed_col_count):
70. self.frozenTableView.setColumnHidden(col, True)
71. else:
72. self.frozenTableView.setColumnWidth(col, self.columnWidth(col))
73.  
74. def updateSectionWidth(self, logicalIndex, oldSize, newSize):
75. if logicalIndex in range(self._fixed_col_count):
76. self.frozenTableView.setColumnWidth(logicalIndex, newSize)
77. self.updateFrozenTableGeometry()
78.  
79. def updateSectionHeight(self, logicalIndex, oldSize, newSize):
80. self.frozenTableView.setRowHeight(logicalIndex, newSize)
81.  
82. def resizeEvent(self, event):
83. QTableView.resizeEvent(self, event)
84. self.updateFrozenTableGeometry()
85.  
86. def scrollTo(self, index, hint):
87. if index.column() >= self._fixed_col_count:
88. QTableView.scrollTo(self, index, hint)
89.  
90. def updateFrozenTableGeometry(self):
91. frozen_width = sum([self.frozenTableView.columnWidth(col) for col in range(self._fixed_col_count)])
92. if self.verticalHeader().isVisible():
93. self.frozenTableView.setGeometry(self.verticalHeader().width() + self.frameWidth(),
94. self.frameWidth(), frozen_width,
95. self.viewport().height() + self.horizontalHeader().height())
96. else:
97. self.frozenTableView.setGeometry(self.frameWidth(),
98. self.frameWidth(), frozen_width,
99. self.viewport().height() + self.horizontalHeader().height())
100.  
101. def moveCursor(self, cursorAction, modifiers):
102. current = QTableView.moveCursor(self, cursorAction, modifiers)
103. x = self.visualRect(current).topLeft().x()
104. frozen_width = sum([self.frozenTableView.columnWidth(col) for col in range(self._fixed_col_count)])
105. if cursorAction == self.MoveLeft:
106. if current.column() >= self._fixed_col_count and x < frozen_width:
107. new_value = self.horizontalScrollBar().value() + x - frozen_width
108. self.horizontalScrollBar().setValue(new_value)
109. elif current.column() < self._fixed_col_count:
110. current = self.model().index(current.row(), current.column() + 1)
111. elif cursorAction == self.MoveHome:
112. new_value = self.horizontalScrollBar().value() + x - frozen_width
113. self.horizontalScrollBar().setValue(new_value)
114. current = self.model().index(current.row(), self._fixed_col_count)
115.  
116. return current