|
|
import re
|
|
|
|
|
|
import xlrd
|
|
|
from PyQt5.QtCore import QThread, pyqtSignal, QMutex
|
|
|
|
|
|
from DcsUi.Config.getData import getListData
|
|
|
from utils.WorkModels import NetworkConfig, PointModel
|
|
|
from utils.core import MainWindowConfig
|
|
|
|
|
|
qmut = QMutex()
|
|
|
|
|
|
|
|
|
class myQThreading(QThread):
|
|
|
sinOut = pyqtSignal(str)
|
|
|
|
|
|
def __init__(self, path):
|
|
|
super().__init__()
|
|
|
self.working = True
|
|
|
self.path = path
|
|
|
|
|
|
def __del__(self):
|
|
|
self.working = False
|
|
|
self.wait()
|
|
|
|
|
|
def run(self) -> None:
|
|
|
qmut.lock()
|
|
|
name = self.path.replace(re.search('.*/', self.path).group(), '')
|
|
|
self.sinOut.emit('正在解析Excel：%s\n' % name)
|
|
|
workbook = xlrd.open_workbook(self.path)
|
|
|
network_table = workbook.sheet_by_name('SLOT')
|
|
|
ni_pxi_table = workbook.sheet_by_name('PXI')
|
|
|
|
|
|
self.load_dev_table(network_table)
|
|
|
self.load_var_table(ni_pxi_table)
|
|
|
|
|
|
self.sinOut.emit('导入Excel完成\n')
|
|
|
MainWindowConfig.IOMapping.setupvalue()
|
|
|
getListData.create_group('allVariables')
|
|
|
|
|
|
self.exec_()
|
|
|
qmut.unlock()
|
|
|
|
|
|
def load_dev_table(self, sheet):
|
|
|
bulk = []
|
|
|
dic = {
|
|
|
'slot': 'slot',
|
|
|
'protocol': 'protocol',
|
|
|
'ip': 'ip',
|
|
|
'port': 'port',
|
|
|
'desc': 'description'
|
|
|
}
|
|
|
for rowx in range(1, sheet.nrows):
|
|
|
row_data = sheet.row_values(rowx)
|
|
|
item_dict = {}
|
|
|
for index, key in enumerate(sheet.row_values(0)):
|
|
|
if key not in dic:
|
|
|
continue
|
|
|
if key == 'ip':
|
|
|
self.ip = row_data[index]
|
|
|
continue
|
|
|
if key == 'port':
|
|
|
self.port = row_data[index]
|
|
|
continue
|
|
|
item_dict[dic[key]] = row_data[index]
|
|
|
item_dict['uri'] = self.ip + ':' + str(int(self.port))
|
|
|
bulk.append(item_dict)
|
|
|
|
|
|
self.sinOut.emit('正在清空设备表\n')
|
|
|
NetworkConfig.delete().execute()
|
|
|
self.sinOut.emit('正在插入设备表\n')
|
|
|
NetworkConfig.insert_many(bulk).execute()
|
|
|
self.sinOut.emit('导入设备表完成\n')
|
|
|
|
|
|
def load_var_table(self, sheet):
|
|
|
bulk = []
|
|
|
for rowx in range(1, sheet.nrows):
|
|
|
row_data = sheet.row_values(rowx)
|
|
|
item_dict = {}
|
|
|
for index, key in enumerate(sheet.row_values(0)):
|
|
|
if key in ('rlo', 'rhi', 'elo', 'ehi', 'offset'):
|
|
|
value = row_data[index]
|
|
|
try:
|
|
|
value = float(value)
|
|
|
except (TypeError, ValueError):
|
|
|
value = None
|
|
|
item_dict[key] = value
|
|
|
continue
|
|
|
else:
|
|
|
item_dict[key] = row_data[index]
|
|
|
bulk.append(item_dict)
|
|
|
self.sinOut.emit('正在清空变量表\n')
|
|
|
PointModel.delete().execute()
|
|
|
chunk = []
|
|
|
for x in bulk:
|
|
|
chunk.append(x)
|
|
|
if len(chunk) < 50:
|
|
|
continue
|
|
|
self.sinOut.emit('正在插入变量表\n')
|
|
|
PointModel.insert_many(chunk).execute()
|
|
|
chunk.clear()
|
|
|
self.sinOut.emit('正在插入变量表\n')
|
|
|
PointModel.insert_many(chunk).execute()
|
|
|
chunk.clear()
|
|
|
bulk.clear()
|
|
|
self.sinOut.emit('导入变量表完成\n')
|