189 8069 5689
from PyQt5.QtWidgets import (QWidget,QTableWidget,QHBoxLayout,QVBoxLayout,QGridLayout,QGroupBox,QLineEdit,QLabel,QPushButton,
QHeaderView,QTableWidgetItem,QMessageBox)
import os,re
#定义站点信息窗口
class stcd_window(QWidget):
def __init__(self):
super().__init__()
self.stcd_tablewidget = QTableWidget(0, 3)
self.stcd_stcd = QLineEdit()
self.stcd_rtu = QLineEdit()
self.stcd_name = QLineEdit()
self.list_stcd = []
self.stcd_pik_path = './/DATA//STCDinfo.pik'
self.stcd_txt_path = './/DATA//站点信息.txt'
self.table_clicked_old = -1 #用来记录行号
self.initUI()
def initUI(self):
#初始化窗口组件
self.stcd_tablewidget.setHorizontalHeaderLabels(['站点编码', 'RTU编码', '站点名称'])
self.stcd_tablewidget.setEditTriggers(QTableWidget.NoEditTriggers)
table_group = QGroupBox('站点')
table_vbox = QVBoxLayout()
table_vbox.addWidget(self.stcd_tablewidget)
table_group.setLayout(table_vbox)
stcd_group = QGroupBox('站点信息配置')
stcd_grid = QGridLayout()
stcd_label = QLabel('站点编码:')
rtu_label = QLabel('RTU编码:')
name_label = QLabel('站点名称:')
delete_button = QPushButton('删除')
input_button = QPushButton('保存')
delete_button.clicked.connect(self.delete_stcd_info)
input_button.clicked.connect(self.save_stcd_info)
self.stcd_tablewidget.cellClicked[int, int].connect(self.cell_clicked)
stcd_grid.addWidget(stcd_label, 0, 0)
stcd_grid.addWidget(self.stcd_stcd, 0, 1)
stcd_grid.addWidget(rtu_label, 1, 0)
stcd_grid.addWidget(self.stcd_rtu, 1, 1)
stcd_grid.addWidget(name_label, 2, 0)
stcd_grid.addWidget(self.stcd_name, 2, 1)
button_grid = QGridLayout()
button_grid.addWidget(delete_button, 0, 0)
button_grid.addWidget(input_button, 0, 1)
stcd_vbox = QVBoxLayout()
stcd_vbox.addLayout(stcd_grid)
stcd_vbox.addLayout(button_grid)
stcd_group.setLayout(stcd_vbox)
hbox = QHBoxLayout()
hbox.addWidget(table_group)
hbox.addWidget(stcd_group)
self.setLayout(hbox)
self.get_list_stcd()
self.initTable()
#初始化站点信息表单
def get_list_stcd(self):
try:
if not os.path.exists(self.stcd_txt_path):
with open(self.stcd_txt_path, 'w') as stcd_txt_file:
stcd_txt_file.write('站点编码;RTU站点;站点名称\n')
return
with open(self.stcd_txt_path, 'r') as stcd_txt_file:
stcd_txt_file.readline()
self.list_stcd.clear()
txt_line = stcd_txt_file.readline()
while txt_line:
txt_line = txt_line.strip('\n')
txt_line = txt_line.replace('；', ';')
x = re.findall(r';', txt_line)
if len(x) >1:
txt_line = txt_line.replace('—', '-')
line = txt_line.strip().split(';', 2)
if not line[0].strip() == '':
self.list_stcd.append([line[0].strip(), line[1].strip(), line[2].strip()])
txt_line = stcd_txt_file.readline()
try:
os.remove(self.stcd_txt_path)
except Exception as error1:
return
with open(self.stcd_txt_path, 'a') as stcd_txt_file:
stcd_txt_file.write('站点编码;RTU站点;站点名称\n')
for line in self.list_stcd:
stcd_txt_file.write(line[0] + ';' + line[1] + ';' + line[2] + '\n')
except Exception as error:
return
#初始化列表
def initTable(self):
x = 0
for line in self.list_stcd:
self.stcd_tablewidget.insertRow(x)
self.stcd_tablewidget.setItem(x, 0, QTableWidgetItem(line[0]))
self.stcd_tablewidget.setItem(x, 1, QTableWidgetItem(line[1]))
self.stcd_tablewidget.setItem(x, 2, QTableWidgetItem(line[2]))
x = x + 1
#表格被选择
def cell_clicked(self, x, y):
if self.table_clicked_old == x:
return
self.table_clicked_old = x
self.stcd_stcd.setText(self.stcd_tablewidget.item(x, 0).text())
self.stcd_rtu.setText(self.stcd_tablewidget.item(x, 1).text())
self.stcd_name.setText(self.stcd_tablewidget.item(x, 2).text())
#保存按钮
def save_stcd_info(self):
if self.stcd_stcd.text().strip() == '' or self.stcd_rtu.text().strip() == '' or self.stcd_name.text().strip() == '':
QMessageBox.warning(self,'注意','站点编码,RTU编码,站点名称\n不能为空.')
return
x = 0
while x < len(self.list_stcd):
if self.stcd_stcd.text().strip() == self.list_stcd[x][0]:
flag = 0
if self.stcd_rtu.text().strip() != self.list_stcd[x][1]:
del self.list_stcd[x][1]
self.list_stcd[x].insert(1,self.stcd_rtu.text().strip())
self.stcd_tablewidget.setItem(x,1,QTableWidgetItem(self.stcd_rtu.text().strip()))
flag = 1
if self.stcd_name.text().strip() != self.list_stcd[x][2]:
del self.list_stcd[x][2]
self.list_stcd[x].insert(2, self.stcd_name.text().strip())
self.stcd_tablewidget.setItem(x, 2, QTableWidgetItem(self.stcd_name.text().strip()))
flag = 2
if flag > 0:
os.remove(self.stcd_txt_path)
with open(self.stcd_txt_path, 'a') as r_file:
r_file.write('站点编码;RTU站点;站点名称\n')
for line in self.list_stcd:
r_file.write(line[0] + ';' + line[1] + ';' + line[2] + '\n')
r_file.flush()
return
x = x + 1
self.stcd_tablewidget.insertRow(len(self.list_stcd))
self.stcd_tablewidget.setItem(len(self.list_stcd), 0, QTableWidgetItem(self.stcd_stcd.text().strip()))
self.stcd_tablewidget.setItem(len(self.list_stcd), 1, QTableWidgetItem(self.stcd_rtu.text().strip()))
self.stcd_tablewidget.setItem(len(self.list_stcd), 2, QTableWidgetItem(self.stcd_name.text().strip()))
self.list_stcd.append([self.stcd_stcd.text().strip(), self.stcd_rtu.text().strip(), self.stcd_name.text().strip()])
with open(self.stcd_txt_path,'a') as file:
file.write(self.stcd_stcd.text().strip() + ';' + self.stcd_rtu.text().strip() + ';' + self.stcd_name.text().strip() + '\n')
self.table_clicked_old = -1
#删除按钮
def delete_stcd_info(self):
if self.table_clicked_old == -1:
QMessageBox.warning(self,'提示','请先选择要删除的站点,再删除.')
return
self.stcd_tablewidget.removeRow(self.table_clicked_old)
del self.list_stcd[self.table_clicked_old]
os.remove(self.stcd_txt_path)
with open(self.stcd_txt_path, 'a') as de_file:
de_file.write('站点编码;RTU站点;站点名称\n')
for line in self.list_stcd:
de_file.write(line[0] + ';' + line[1] + ';' + line[2] + '\n')
de_file.flush()
self.stcd_stcd.setText('')
self.stcd_rtu.setText('')
self.stcd_name.setText('')
self.table_clicked_old = -1