class QmyMainWindow(QtWidgets.QMainWindow):
def __init__(self, parent=None):
super().__init__(parent)
self.ui = Ui_MainWindow()
self.ui.setupUi(self)
self.setCentralWidget(self.ui.qSplitter)
self.__ColCount = 6
self.itemModel = QStandardItemModel(5, self.__ColCount, self)
self.selectionModel = QItemSelectionModel(self.itemModel)
self.selectionModel.currentChanged.connect(self.do_curChanged)
self.__lastColumnTitle = "测井取样"
self.__lastColumnFlags = (Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled)
self.ui.qTableView.setModel(self.itemModel)
self.ui.qTableView.setSelectionModel(self.selectionModel)
oneOrMore = QAbstractItemView.ExtendedSelection
self.ui.qTableView.setSelectionMode(oneOrMore)
itemOrRow = QAbstractItemView.SelectItems
self.ui.qTableView.setSelectionBehavior(itemOrRow)
self.ui.qTableView.verticalHeader().setDefaultSectionSize(22)
self.ui.qTableView.setAlternatingRowColors(True)
self.ui.qTableView.setEnabled(False)
self.qLabel1 = QLabel("当前单元格：", self)
self.qLabel1.setMinimumWidth(180)
self.qLabel2 = QLabel("单元格内容：", self)
self.qLabel2.setMinimumWidth(150)
self.qLabel3 = QLabel("当前文件：", self)
self.ui.qStatusBar.addWidget(self.qLabel1)
self.ui.qStatusBar.addWidget(self.qLabel2)
self.ui.qStatusBar.addPermanentWidget(self.qLabel3)
def __iniModelFromStringList(self, allLines):
rowCnt = len(allLines)
self.itemModel.setRowCount(rowCnt - 1)
headerText = allLines[0].strip()
headerList = headerText.split("\t")
self.itemModel.setHorizontalHeaderLabels(headerList)
self.__lastColumnTitle = headerList[len(headerList) - 1]
lastColNo = self.__ColCount - 1
for i in range(rowCnt - 1):
lineText = allLines[i + 1].strip()
strList = lineText.split("\t")
for j in range(self.__ColCount - 1):
item = QStandardItem(strList[j])
self.itemModel.setItem(i, j, item)
item = QStandardItem(self.__lastColumnTitle)
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
if (strList[lastColNo] == "0"):
item.setCheckState(Qt.Unchecked)
else:
item.setCheckState(Qt.Checked)
self.itemModel.setItem(i, lastColNo, item)
def __setCellAlignment(self, align=Qt.AlignHCenter):
if (not self.selectionModel.hasSelection()):
return
selectedIndex = self.selectionModel.selectedIndexes()
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i]
item = self.itemModel.itemFromIndex(index)
item.setTextAlignment(align)
@pyqtSlot()
def on_qAction1_triggered(self): # 打开文件
curPath = os.getcwd()
filename, flt = QFileDialog.getOpenFileName(
self, "打开一个文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.qLabel3.setText("当前文件：" + filename)
self.ui.qPlainTextEdit.clear()
aFile = open(filename, 'r')
allLines = aFile.readlines()
aFile.close()
for strLine in allLines:
self.ui.qPlainTextEdit.appendPlainText(strLine.strip())
self.__iniModelFromStringList(allLines)
self.ui.qTableView.setEnabled(True)
self.ui.qAction2.setEnabled(True)
self.ui.qAction3.setEnabled(True)
self.ui.qAction4.setEnabled(True)
self.ui.qAction5.setEnabled(True)
self.ui.qAction6.setEnabled(True)
@pyqtSlot()
def on_qAction2_triggered(self): # 另存文件
curPath = os.getcwd()
filename, flt = QFileDialog.getSaveFileName(
self, "保存文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.on_qAction3_triggered()
aFile = open(filename, "w")
aFile.write(self.ui.qPlainTextEdit.toPlainText())
aFile.close()
@pyqtSlot()
def on_qAction4_triggered(self): # 添加行
itemList = []
for i in range(self.__ColCount - 1):
item = QStandardItem("0")
itemList.append(item)
item = QStandardItem(self.__lastColumnTitle)
item.setCheckable(True)
item.setFlags(self.__lastColumnFlags)
itemList.append(item)
self.itemModel.appendRow(itemList)
curIndex = self.itemModel.index(self.itemModel.rowCount() - 1, 0)
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot()
def on_qAction5_triggered(self): # 插入行
itemlist = []
for i in range(self.__ColCount - 1):
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle)
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
item.setCheckState(Qt.Checked)
itemlist.append(item)
curIndex = self.selectionModel.currentIndex()
self.itemModel.insertRow(curIndex.row(), itemlist)
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot()
def on_qAction6_triggered(self): # 删除行
curIndex = self.selectionModel.currentIndex()
self.itemModel.removeRow(curIndex.row())
@pyqtSlot()
def on_qAction7_triggered(self): # 居左
self.__setCellAlignment(Qt.AlignLeft | Qt.AlignVCenter)
@pyqtSlot()
def on_qAction8_triggered(self): # 居中
self.__setCellAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
@pyqtSlot()
def on_qAction9_triggered(self): # 居右
self.__setCellAlignment(Qt.AlignRight | Qt.AlignVCenter)
@pyqtSlot(bool)
def on_qAction10_triggered(self, checked): # 粗体
if (not self.selectionModel.hasSelection()):
return
selectedIndex = self.selectionModel.selectedIndexes()
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i]
item = self.itemModel.itemFromIndex(index)
font = item.font()
font.setBold(checked)
item.setFont(font)
@pyqtSlot()
def on_qAction3_triggered(self): # 模型数据
self.ui.qPlainTextEdit.clear()
lineStr = ""
for i in range(self.itemModel.columnCount() - 1):
item = self.itemModel.horizontalHeaderItem(i)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.horizontalHeaderItem(self.__ColCount - 1)
lineStr = lineStr + item.text()
self.ui.qPlainTextEdit.appendPlainText(lineStr)
for i in range(self.itemModel.rowCount()):
lineStr = ""
for j in range(self.itemModel.columnCount() - 1):
item = self.itemModel.item(i, j)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.item(i, self.__ColCount - 1)
if (item.checkState() == Qt.Checked):
lineStr = lineStr + "1"
else:
lineStr = lineStr + "0"
self.ui.qPlainTextEdit.appendPlainText(lineStr)
def do_curChanged(self, current, previous):
if (current != None):
text = "当前单元格：%d行，%d列" % (current.row(), current.column())
self.qLabel1.setText(text)
item = self.itemModel.itemFromIndex(current)
self.qLabel2.setText("单元格内容：" + item.text())
font = item.font()
self.ui.qAction10.setChecked(font.bold())
class QmyMainWindow(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent) #调用父类构造函数,创建窗体
self.ui = Ui_MainWindow() #创建Ui对象
self.ui.setupUi(self) #构造UI
self.__ColCount = 6 #列数
self.itemModel = QStandardItemModel(
10, self.__ColCount, self) #创建QStandardItemModel类型的数据模型,指定行列值
'''
setSelectionBehavior()
此属性保存视图使用的选择行为。
此属性保存选择是根据单个项目，行还是列进行的
#QItemSelectionModel()
此属性保存视图在哪种选择模式下运行。
#此属性控制用户是否可以选择一个或多个项目，并且在多个项目选择中控制选择是否必须是连续范围的项目
'''
self.selectionModel = QItemSelectionModel(self.itemModel)
self.selectionModel.currentChanged.connect(
self.do_curChanged) #单元格选择发生变化时会发射此信号
self.__lastColumnTitle = "" #设置最后一列的标题，可以是空
self.__lastColumnFlags = (Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled)
#设置tableView属性
self.ui.tableView.setModel(self.itemModel) #数据模型
self.ui.tableView.setSelectionModel(self.selectionModel) #选择模型
oneOrMore = QAbstractItemView.ExtendedSelection #选择模式->多选模式
self.ui.tableView.setSelectionMode(oneOrMore) #多选模式
itemOrRow = QAbstractItemView.SelectItems #项选择模式->单元格选择
self.ui.tableView.setSelectionBehavior(itemOrRow) #单元格选择
self.ui.tableView.verticalHeader().setDefaultSectionSize(22)
self.ui.tableView.setAlternatingRowColors(True) #交替行颜色
self.ui.tableView.setEnabled(False) #设置默认禁用tabelView
self.ui.actFontBold.setCheckable(False)
self.setCentralWidget(self.ui.splitter) #设置中心组件
# self.setCentralWidget(self.ui.tableView)
self.__buildStatusBar()
self.spinCeshen = QmyFloatSpinDelegate(0, 10000, 0, self)
self.spinLength = QmyFloatSpinDelegate(0, 6000, 2, self)
self.spinDegree = QmyFloatSpinDelegate(0, 360, 1, self)
self.ui.tableView.setItemDelegateForColumn(0, self.spinCeshen)
self.ui.tableView.setItemDelegateForColumn(1, self.spinLength)
self.ui.tableView.setItemDelegateForColumn(3, self.spinLength)
self.ui.tableView.setItemDelegateForColumn(2, self.spinDegree)
qualities = ["优", "良", "合格", "不合格"]
self.comboDelegate = QmyComboBoxDelegate(self)
self.comboDelegate.setItems(qualities, False)
self.ui.tableView.setItemDelegateForColumn(4, self.comboDelegate)
##==========自定义功能函数==========
def __buildStatusBar(self):
'''
设置状态栏ui组件
:return:
'''
self.LabCellPos = QLabel("当前单元格：", self)
self.LabCellPos.setMinimumWidth(180)
self.ui.statusBar.addWidget(self.LabCellPos)
self.LabCellText = QLabel("单元格内容：", self)
self.LabCellText.setMinimumWidth(150)
self.ui.statusBar.addWidget(self.LabCellText)
self.LabCurFile = QLabel("当前文件：", self)
self.ui.statusBar.addWidget(self.LabCurFile)
def __iniModelFromStringList(self, allLines):
rowCnt = len(allLines) #获取总行数
self.itemModel.setRowCount(rowCnt - 1) #除去表头的数据行数
headerText = allLines[0].strip() #表头去除换行符，文件呢的空格使用Tab代替
headerList = headerText.split("\t") #按照制表符转化为列表
self.itemModel.setHorizontalHeaderLabels(headerList) #设置表头标题
# print(headerList)
self.__lastColumnTitle = headerList[len(headerList) - 1] #最后一列的标题
lastColNo = self.__ColCount - 1 #最后一列的列号
for i in range(rowCnt - 1): #除去表头的数据行数
# print(i)
lineText = allLines[i + 1].strip() #去除换行符,不包括表头
strList = lineText.split("\t") #按制表符生成列表
for j in range(self.__ColCount - 1): #不包括最后一列
'''
QStandardItem是一个数据结构，他可以存储一个cell的各种信息，比如文本、图标、是否可选、字体、别景色、前景色等等。
并且QStandardItem可以有孩子和兄弟，他是为model提供数据存储的节点。
QTableView：作为表格cell时，有一个作为根节点的QStandardItem，其他节点都是QStandardItem节点的孩子节点，并且都是兄弟节点(这里暂时不考虑多列的情况)。
QTreeView：作为树节点cell时，有一个作为根节点的QStandardItem，其他节点都是他的孩子节点，但是其他节点也可以作为父节点存在(这里暂时不考虑多列的情况)。
'''
item = QStandardItem(strList[j])
self.itemModel.setItem(i, j, item)
#设置最后一行
# print(self.__lastColumnTitle)
item = QStandardItem(self.__lastColumnTitle) #将最后一行的表头
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
# print(strList[lastColNo])
if strList[lastColNo] == '0': #对比文本内的数值进行设定,类型是字符串
item.setCheckState(Qt.Unchecked)
else:
item.setCheckState(Qt.Checked)
self.itemModel.setItem(i, lastColNo, item)
def __setCellAlignment(self, align=Qt.AlignHCenter):
if not self.selectionModel.hasSelection():
return
'''
selectedIndexes()返回一个元素为QModelIndex类型的列表,包括所有被选中的单元格的模型索引
'''
selectdIndex = self.selectionModel.selectedIndexes()
# print(selectdIndex)
count = len(selectdIndex)
for i in range(count):
index = selectdIndex[i]
'''
itemFromIndex(index)返回的是模型索引为index的QStandardItem对象
'''
item = self.itemModel.itemFromIndex(index)
item.setTextAlignment(align)
##==========事件处理函数===========
##==========由connectSlotsByName()自动关联的槽函数====
@pyqtSlot()
def on_actOpenFile_triggered(self):
curPath = os.getcwd() #获取当前路径
# print(curPath)?
#flt是文件过滤器
filename, flt = QFileDialog.getOpenFileName(
self, "打开一个文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if filename == "":
return
self.LabCurFile.setText(("当前文件: " + filename)) #设置状态栏文本
self.ui.plainTextEdit.clear()
aFile = open(filename, "r")
allLines = aFile.readlines()
aFile.close()
for strLine in allLines:
self.ui.plainTextEdit.appendPlainText(
strLine.strip()) #按照行添加到plainTextEdit中
self.__iniModelFromStringList(allLines)
#设置激活状态
self.ui.tableView.setEnabled(True)
self.ui.actAppend.setEnabled(True)
self.ui.actInsert.setEnabled(True)
self.ui.actDel.setEnabled(True)
self.ui.actSaveFile.setEnabled(True)
self.ui.actModelData.setEnabled(True)
self.ui.actFontBold.setCheckable(True) #设置加粗可用
@pyqtSlot()
def on_actAppend_triggered(self):
itemlist = []
for i in range(self.__ColCount - 1): #循环一行中的各个列,不包括最后一列
item = QStandardItem("0") #添加0到数据结构中
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle) #将最后一行的表头添加入数据结构
item.setCheckable(True) #可选
item.setFlags(self.__lastColumnFlags)
itemlist.append(item) #添加到itemlist内(添加到最后一个)
self.itemModel.appendRow(itemlist)
curIndex = self.itemModel.index(self.itemModel.rowCount() - 1,
0) #获取最后一行第一个单元格的模型索引
self.selectionModel.clearSelection() #清除选择
self.selectionModel.setCurrentIndex(
curIndex,
QItemSelectionModel.Select) #设置在添加后自动选择添加行的第一个单元格(可从状态栏确认)
@pyqtSlot()
def on_actInsert_triggered(self):
'''
插入行
:return:
'''
itemlist = []
for i in range(self.__ColCount - 1):
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle)
item.setFlags(self.__lastColumnFlags)
item.setCheckable(False) #是否可以修改选择
item.setCheckState(Qt.Checked) #是否选中
itemlist.append(item)
curIndex = self.selectionModel.currentIndex() #选中项的模型索引,包括行列等其他信息
self.itemModel.insertRow(curIndex.row(), itemlist)
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot()
def on_actDel_triggered(self):
'''
删除行
:return:
'''
curIndex = self.selectionModel.currentIndex()
self.itemModel.removeRow(curIndex.row())
@pyqtSlot()
def on_actAlignLeft_triggered(self):
self.__setCellAlignment(Qt.AlignLeft | Qt.AlignVCenter)
@pyqtSlot()
def on_actAlignCenter_triggered(self):
self.__setCellAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
@pyqtSlot()
def on_actAlignRight_triggered(self):
self.__setCellAlignment(Qt.AlignRight | Qt.AlignVCenter)
@pyqtSlot(bool)
def on_actFontBold_triggered(self, checked):
print("1,checked", checked)
if not self.selectionModel.hasSelection():
return
selectIndex = self.selectionModel.selectedIndexes()
for i in range(len(selectIndex)):
index = selectIndex[i]
item = self.itemModel.itemFromIndex(index)
font = item.font()
font.setBold(checked)
item.setFont(font)
@pyqtSlot()
def on_actModelData_triggered(self):
self.ui.plainTextEdit.clear()
lineStr = ""
for i in range(self.itemModel.columnCount() - 1): #不包括最后一列的表头遍历
item = self.itemModel.horizontalHeaderItem(i) #返回行标题的一个数据项对象,i是列号
lineStr = lineStr + item.text() + "\t" #给每个项对象的文本添加Tab制表符
item = self.itemModel.horizontalHeaderItem(self.__ColCount -
1) #最后一列的表头
lineStr = lineStr + item.text()
self.ui.plainTextEdit.appendPlainText(lineStr)
for i in range(self.itemModel.rowCount()): #获取行总数,不包括表头行
lineStr = ""
for j in range(self.itemModel.columnCount() - 1): #不包括最后一列的遍历
item = self.itemModel.item(i, j) #按照行列号获取工作区内项对象
lineStr = lineStr + item.text() + "\t" #项对象文本添加制表符
item = self.itemModel.item(i, self.__ColCount - 1) #按照行号获取最后一列的项对象
if item.checkState() == Qt.Checked: #匹配状态是否为选中
lineStr = lineStr + "1"
else:
lineStr = lineStr + "0"
self.ui.plainTextEdit.appendPlainText(lineStr)
@pyqtSlot()
def on_actSaveFile_triggered(self):
'''
保存文件
:return:
'''
curPath = os.getcwd()
filename, flt = QFileDialog.getSaveFileName(
self, "保存文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if filename == "":
return
self.on_actModelData_triggered()
aFile = open(filename, "w")
aFile.write(self.ui.plainTextEdit.toPlainText())
aFile.close()
##=========自定义槽函数============
def do_curChanged(self, current, previous):
'''
设置状态栏组件显示内容
:param current:
:param previous:
:return:
'''
if current != None:
text = " 当前单元格： %d行，%d列" % (current.row() + 1,
current.column() + 1)
self.LabCellPos.setText(text)
item = self.itemModel.itemFromIndex(current)
self.LabCellText.setText("单元格内容：" + item.text())
font = item.font()
self.ui.actFontBold.setChecked(
font.bold()) #设置按钮按下,当font.Bold的值大于50式font.blod()会为True
class QmyMainWindow(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent) #调用父类构造函数，创建窗体
self.ui = Ui_MainWindow() #创建UI对象
self.ui.setupUi(self) #构造UI界面
self.setCentralWidget(self.ui.splitter)
self.__buildStatusBar()
self.COL_COUNT = 6 #常数，列数=6
self.itemModel = QStandardItemModel(5, self.COL_COUNT,
self) # 数据模型,10行6列
## headerList=["测深(m)","垂深(m)","方位(°)","总位移(m)","固井质量","测井取样"]
## self.itemModel.setHorizontalHeaderLabels(headerList) #设置表头文字
self.selectionModel = QItemSelectionModel(self.itemModel) #Item选择模型
self.selectionModel.currentChanged.connect(self.do_currentChanged)
self.__lastColumnFlags = (Qt.ItemIsSelectable | Qt.ItemIsUserCheckable
| Qt.ItemIsEnabled)
self.__lastColumnTitle = "测井取样"
#为tableView设置数据模型
self.ui.tableView.setModel(self.itemModel) #设置数据模型
self.ui.tableView.setSelectionModel(self.selectionModel) #设置选择模型
oneOrMore = QAbstractItemView.ExtendedSelection
self.ui.tableView.setSelectionMode(oneOrMore) #可多选
itemOrRow = QAbstractItemView.SelectItems
self.ui.tableView.setSelectionBehavior(itemOrRow) #单元格选择
self.ui.tableView.verticalHeader().setDefaultSectionSize(22) #缺省行高
self.ui.tableView.setEnabled(False) #先禁用tableView
## self.__resetTable(5)
#创建自定义代理组件并设置
self.spinCeShen = QmyFloatSpinDelegate(0, 10000, 0, self) #用于 测深
self.spinLength = QmyFloatSpinDelegate(0, 6000, 2, self) #垂深，总位移
self.spinDegree = QmyFloatSpinDelegate(0, 360, 1, self) #用于 方位
self.ui.tableView.setItemDelegateForColumn(0, self.spinCeShen) #测深
self.ui.tableView.setItemDelegateForColumn(1, self.spinLength) #垂深
self.ui.tableView.setItemDelegateForColumn(3, self.spinLength) #总位移
self.ui.tableView.setItemDelegateForColumn(2, self.spinDegree) #方位角
qualities = ["优", "良", "合格", "不合格"]
self.comboDelegate = QmyComboBoxDelegate(self)
self.comboDelegate.setItems(qualities, False) #不可编辑
self.ui.tableView.setItemDelegateForColumn(4,
self.comboDelegate) #固井质量
## ==============自定义功能函数============
def __buildStatusBar(self): ##构建状态栏
self.LabCellPos = QLabel("当前单元格：", self)
self.LabCellPos.setMinimumWidth(180)
self.ui.statusBar.addWidget(self.LabCellPos)
self.LabCellText = QLabel("单元格内容：", self)
self.LabCellText.setMinimumWidth(150)
self.ui.statusBar.addWidget(self.LabCellText)
self.LabCurFile = QLabel("当前文件：", self)
self.ui.statusBar.addPermanentWidget(self.LabCurFile)
## def __resetTable(self,tableRowCount): #复位数据表，必须为数值设置0，否则代理组件出错
## self.itemModel.removeRows(0,self.itemModel.rowCount()) #删除所有行
## self.itemModel.setRowCount(tableRowCount) #设置新的行数
##
## for i in range(tableRowCount): #设置最后一列
## for j in range(self.COL_COUNT-1):
## index=self.itemModel.index(i,j) #获取模型索引
## item=self.itemModel.itemFromIndex(index) #获取item
## item.setData(0,Qt.DisplayRole) #数值必须初始化为0
## item.setTextAlignment(Qt.AlignHCenter)
##
## index=self.itemModel.index(i,self.COL_COUNT-1) #获取模型索引
## item=self.itemModel.itemFromIndex(index) #获取item
## item.setCheckable(True)
## item.setData(self.__lastColumnTitle,Qt.DisplayRole)
## item.setEditable(False) #不可编辑
def __iniModelFromStringList(self, allLines): ##从字符串列表构建模型
rowCnt = len(allLines) #文本行数，第1行是标题
self.itemModel.setRowCount(rowCnt - 1) #实际数据行数
headerText = allLines[0].strip() #第1行是表头,去掉末尾的换行符 "\n"
headerList = headerText.split("\t")
self.itemModel.setHorizontalHeaderLabels(headerList)
self.__lastColumnTitle = headerList[len(headerList) -
1] # 最后一列表头的标题，即“测井取样”
lastColNo = self.COL_COUNT - 1 #最后一列的列号
for i in range(rowCnt - 1):
lineText = allLines[i + 1].strip() #一行的文字，\t分隔
strList = lineText.split("\t")
for j in range(self.COL_COUNT - 1): #不含最后一列
item = QStandardItem(strList[j])
self.itemModel.setItem(i, j, item)
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
if (strList[lastColNo] == "0"):
item.setCheckState(Qt.Unchecked)
else:
item.setCheckState(Qt.Checked)
self.itemModel.setItem(i, lastColNo, item) #设置最后一列的item
def __setCellAlignment(self, align=Qt.AlignHCenter):
if (not self.selectionModel.hasSelection()): #没有选择的项
return
selectedIndex = self.selectionModel.selectedIndexes() #列表类型
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i] #获取其中的一个模型索引
item = self.itemModel.itemFromIndex(index) #获取一个单元格的项数据对象
item.setTextAlignment(align) #设置文字对齐方式
## ==========由connectSlotsByName() 自动连接的槽函数==================
@pyqtSlot() ##“打开文件”
def on_actOpen_triggered(self):
curPath = os.getcwd() #获取当前路径
filename, flt = QFileDialog.getOpenFileName(
self, "打开一个文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.LabCurFile.setText("当前文件：" + filename)
self.ui.plainTextEdit.clear()
aFile = open(filename, 'r')
allLines = aFile.readlines() #读取所有行,list类型,每行末尾带有 \n
## for eachLine in aFile: #每次读取一行
## self.ui.plainTextEdit.appendPlainText(eachLine)
aFile.close()
for strLine in allLines:
self.ui.plainTextEdit.appendPlainText(strLine.strip())
self.__iniModelFromStringList(allLines)
self.ui.actAppend.setEnabled(True) #更新Actions的enable属性
self.ui.actInsert.setEnabled(True)
self.ui.actDelete.setEnabled(True)
self.ui.actSave.setEnabled(True)
self.ui.actModelData.setEnabled(True)
self.ui.tableView.setEnabled(True) #启用tableView
@pyqtSlot() ##保存文件
def on_actSave_triggered(self):
curPath = os.getcwd() #获取当前路径
filename, flt = QFileDialog.getSaveFileName(
self, "保存文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.on_actModelData_triggered() #更新数据到plainTextEdit
aFile = open(filename, 'w') #以写方式打开
aFile.write(self.ui.plainTextEdit.toPlainText())
aFile.close()
@pyqtSlot()
def on_actAppend_triggered(self): #在最后添加一行
itemlist = [] # 列表
for i in range(self.COL_COUNT - 1):
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setCheckable(True)
item.setFlags(self.__lastColumnFlags)
itemlist.append(item)
self.itemModel.appendRow(itemlist)
curIndex = self.itemModel.index(self.itemModel.rowCount() - 1, 0)
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot() ##插入一行
def on_actInsert_triggered(self):
itemlist = [] # 列表
for i in range(self.COL_COUNT - 1):
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
item.setCheckState(Qt.Checked)
itemlist.append(item)
curIndex = self.selectionModel.currentIndex()
#获取当前选中项的模型索引
self.itemModel.insertRow(curIndex.row(), itemlist)
#在当前行的前面插入一行
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot() ##删除当前行
def on_actDelete_triggered(self):
curIndex = self.selectionModel.currentIndex() #获取当前选择单元格的模型索引
self.itemModel.removeRow(curIndex.row()) # //删除当前行
@pyqtSlot() ##左对齐
def on_actAlignLeft_triggered(self):
self.__setCellAlignment(Qt.AlignLeft | Qt.AlignVCenter)
@pyqtSlot() ##中间对齐
def on_actAlignCenter_triggered(self):
self.__setCellAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
@pyqtSlot() ##右对齐
def on_actAlignRight_triggered(self):
self.__setCellAlignment(Qt.AlignRight | Qt.AlignVCenter)
@pyqtSlot(bool) ##字体Bold
def on_actFontBold_triggered(self, checked):
if (not self.selectionModel.hasSelection()): #没有选择的项
return
selectedIndex = self.selectionModel.selectedIndexes() #列表类型
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i] #获取其中的一个模型索引
item = self.itemModel.itemFromIndex(index) #获取一个单元格的项数据对象
font = item.font()
font.setBold(checked)
item.setFont(font)
@pyqtSlot() ##模型数据显示到plainTextEdit里
def on_actModelData_triggered(self):
self.ui.plainTextEdit.clear()
lineStr = ""
for i in range(self.itemModel.columnCount() - 1): #表头
item = self.itemModel.horizontalHeaderItem(i)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.horizontalHeaderItem(self.COL_COUNT - 1) #最后一列
lineStr = lineStr + item.text()
self.ui.plainTextEdit.appendPlainText(lineStr)
for i in range(self.itemModel.rowCount()):
lineStr = ""
for j in range(self.itemModel.columnCount() - 1): #不包括最后一列
item = self.itemModel.item(i, j)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.item(i, self.COL_COUNT - 1) #最后一列
if (item.checkState() == Qt.Checked):
lineStr = lineStr + "1"
else:
lineStr = lineStr + "0"
self.ui.plainTextEdit.appendPlainText(lineStr)
## =============自定义槽函数===============================
def do_currentChanged(self, current, previous):
if (current != None): #当前模型索引有效
self.LabCellPos.setText(
"当前单元格：%d行，%d列" %
(current.row(), current.column())) #显示模型索引的行和列号
item = self.itemModel.itemFromIndex(current) #从模型索引获得Item
self.LabCellText.setText("单元格内容：" + item.text()) #显示item的文字内容
font = item.font() #获取item的字体
self.ui.actFontBold.setChecked(font.bold()) #更新actFontBold的check状态
class QmyMainWindow(QMainWindow):
def __init__(self, parent=None):
super().__init__(parent) #调用父类构造函数，创建窗体
self.ui = Ui_MainWindow() #创建UI对象
self.ui.setupUi(self) #构造UI界面
self.setCentralWidget(self.ui.splitter)
self.__ColCount = 6 #列数=6
self.itemModel = QStandardItemModel(5, self.__ColCount,
self) # 数据模型,10行6列
self.selectionModel = QItemSelectionModel(self.itemModel) #Item选择模型
self.selectionModel.currentChanged.connect(self.do_curChanged)
self.__lastColumnTitle = "测井取样"
self.__lastColumnFlags = (Qt.ItemIsSelectable
| Qt.ItemIsUserCheckable | Qt.ItemIsEnabled)
##tableView设置
self.ui.tableView.setModel(self.itemModel) #设置数据模型
self.ui.tableView.setSelectionModel(self.selectionModel) #设置选择模型
oneOrMore = QAbstractItemView.ExtendedSelection
self.ui.tableView.setSelectionMode(oneOrMore) #可多选
itemOrRow = QAbstractItemView.SelectItems
self.ui.tableView.setSelectionBehavior(itemOrRow) #单元格选择
self.ui.tableView.verticalHeader().setDefaultSectionSize(22) #缺省行高
self.ui.tableView.setAlternatingRowColors(True) #交替行颜色
self.ui.tableView.setEnabled(False) #先禁用tableView
self.__buildStatusBar()
## ==============自定义功能函数============
def __buildStatusBar(self): ##构建状态栏
self.LabCellPos = QLabel("当前单元格：", self)
self.LabCellPos.setMinimumWidth(180)
self.ui.statusBar.addWidget(self.LabCellPos)
self.LabCellText = QLabel("单元格内容：", self)
self.LabCellText.setMinimumWidth(150)
self.ui.statusBar.addWidget(self.LabCellText)
self.LabCurFile = QLabel("当前文件：", self)
self.ui.statusBar.addPermanentWidget(self.LabCurFile)
def __iniModelFromStringList(self, allLines): ##从字符串列表构建模型
rowCnt = len(allLines) #文本行数，第1行是标题
self.itemModel.setRowCount(rowCnt - 1) #实际数据行数
headerText = allLines[0].strip() #第1行是表头,去掉末尾的换行符 "\n"
headerList = headerText.split("\t") #转换为字符串列表
self.itemModel.setHorizontalHeaderLabels(headerList) #设置表头标题
self.__lastColumnTitle = headerList[len(headerList) -
1] # 最后一列表头的标题，即“测井取样”
lastColNo = self.__ColCount - 1 #最后一列的列号
for i in range(rowCnt - 1):
lineText = allLines[i + 1].strip() #一行的文字，\t分隔
strList = lineText.split("\t") #分割为字符串列表
for j in range(self.__ColCount - 1): #不含最后一列
item = QStandardItem(strList[j])
self.itemModel.setItem(i, j, item) #设置模型的item
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
if (strList[lastColNo] == "0"):
item.setCheckState(Qt.Unchecked)
else:
item.setCheckState(Qt.Checked)
self.itemModel.setItem(i, lastColNo, item) #设置最后一列的item
def __setCellAlignment(self, align=Qt.AlignHCenter):
if (not self.selectionModel.hasSelection()): #没有选择的项
return
selectedIndex = self.selectionModel.selectedIndexes() #模型索引列表
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i] #获取其中的一个模型索引
item = self.itemModel.itemFromIndex(index) #获取一个单元格的项数据对象
item.setTextAlignment(align) #设置文字对齐方式
## ==========由connectSlotsByName() 自动连接的槽函数==================
@pyqtSlot() ##“打开文件”
def on_actOpen_triggered(self):
## curPath=QDir.currentPath() #获取当前路径
curPath = os.getcwd() #获取当前路径
filename, flt = QFileDialog.getOpenFileName(
self, "打开一个文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.LabCurFile.setText("当前文件：" + filename)
self.ui.plainTextEdit.clear()
aFile = open(filename, 'r')
allLines = aFile.readlines() #读取所有行,list类型,每行末尾带有 \n
aFile.close()
for strLine in allLines:
self.ui.plainTextEdit.appendPlainText(strLine.strip())
self.__iniModelFromStringList(allLines)
self.ui.tableView.setEnabled(True) #tableView可用
self.ui.actAppend.setEnabled(True) #更新Actions的enable属性
self.ui.actInsert.setEnabled(True)
self.ui.actDelete.setEnabled(True)
self.ui.actSave.setEnabled(True)
self.ui.actModelData.setEnabled(True)
@pyqtSlot() ##保存文件
def on_actSave_triggered(self):
## curPath=QDir.currentPath() #获取当前路径
curPath = os.getcwd() #获取当前路径
filename, flt = QFileDialog.getSaveFileName(
self, "保存文件", curPath, "井斜数据文件(*.txt);;所有文件(*.*)")
if (filename == ""):
return
self.on_actModelData_triggered() #更新数据到plainTextEdit
aFile = open(filename, 'w') #以写方式打开
aFile.write(self.ui.plainTextEdit.toPlainText())
aFile.close()
@pyqtSlot() ##在最后添加一行
def on_actAppend_triggered(self):
itemlist = [] # QStandardItem 对象列表
for i in range(self.__ColCount - 1): #不包括最后一列
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setCheckable(True)
item.setFlags(self.__lastColumnFlags)
itemlist.append(item)
self.itemModel.appendRow(itemlist) #添加一行
curIndex = self.itemModel.index(self.itemModel.rowCount() - 1, 0)
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot() ##插入一行
def on_actInsert_triggered(self):
itemlist = [] # QStandardItem 对象列表
for i in range(self.__ColCount - 1): #不包括最后一列
item = QStandardItem("0")
itemlist.append(item)
item = QStandardItem(self.__lastColumnTitle) #最后一列
item.setFlags(self.__lastColumnFlags)
item.setCheckable(True)
item.setCheckState(Qt.Checked)
itemlist.append(item)
curIndex = self.selectionModel.currentIndex()
#获取当前选中项的模型索引
self.itemModel.insertRow(curIndex.row(), itemlist)
#在当前行的前面插入一行
self.selectionModel.clearSelection()
self.selectionModel.setCurrentIndex(curIndex,
QItemSelectionModel.Select)
@pyqtSlot() ##删除当前行
def on_actDelete_triggered(self):
curIndex = self.selectionModel.currentIndex() #获取当前选择单元格的模型索引
self.itemModel.removeRow(curIndex.row()) #删除当前行
@pyqtSlot() ##左对齐
def on_actAlignLeft_triggered(self):
self.__setCellAlignment(Qt.AlignLeft | Qt.AlignVCenter)
@pyqtSlot() ##中间对齐
def on_actAlignCenter_triggered(self):
self.__setCellAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
@pyqtSlot() ##右对齐
def on_actAlignRight_triggered(self):
self.__setCellAlignment(Qt.AlignRight | Qt.AlignVCenter)
@pyqtSlot(bool) ##字体Bold
def on_actFontBold_triggered(self, checked):
if (not self.selectionModel.hasSelection()): #没有选择的项
return
selectedIndex = self.selectionModel.selectedIndexes() #模型索引列表
count = len(selectedIndex)
for i in range(count):
index = selectedIndex[i] #获取其中的一个模型索引
item = self.itemModel.itemFromIndex(index) #获取一个单元格的项数据对象
font = item.font()
font.setBold(checked)
item.setFont(font)
@pyqtSlot() ##模型数据显示到plainTextEdit里
def on_actModelData_triggered(self):
self.ui.plainTextEdit.clear()
lineStr = ""
for i in range(self.itemModel.columnCount() - 1): #表头,不含最后一列
item = self.itemModel.horizontalHeaderItem(i)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.horizontalHeaderItem(self.__ColCount - 1) #最后一列
lineStr = lineStr + item.text() #表头文字字符串
self.ui.plainTextEdit.appendPlainText(lineStr)
for i in range(self.itemModel.rowCount()):
lineStr = ""
for j in range(self.itemModel.columnCount() - 1): #不包括最后一列
item = self.itemModel.item(i, j)
lineStr = lineStr + item.text() + "\t"
item = self.itemModel.item(i, self.__ColCount - 1) #最后一列
if (item.checkState() == Qt.Checked):
lineStr = lineStr + "1"
else:
lineStr = lineStr + "0"
self.ui.plainTextEdit.appendPlainText(lineStr)
## =============自定义槽函数===============================
def do_curChanged(self, current, previous):
if (current != None): #当前模型索引有效
text = "当前单元格：%d行，%d列" % (current.row(), current.column())
self.LabCellPos.setText(text)
item = self.itemModel.itemFromIndex(current) #从模型索引获得Item
self.LabCellText.setText("单元格内容：" + item.text()) #显示item的文字内容
font = item.font() #获取item的字体
self.ui.actFontBold.setChecked(font.bold()) #更新actFontBold的check状态
