def addFile(self):
linetexts = QFileDialog.getOpenFileNames(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*.ctf *.xml *.ang)"))[0]
for line in linetexts:
self.uselistmodel.addRow(line)
self.toggleButtons()
def selectFile(self, fileLink=None):
if fileLink == False:
linetext = QFileDialog.getOpenFileName(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*.*)"))[0]
else:
linetext = fileLink
if linetext != "":
self.setWindowTitle(linetext)
self.ui.dataFileLineEdit.setText(linetext)
self.ui.dataTypeText.setText(linetext.split(".")[1].upper())
if self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser") != -1:
headerDict = {}
self.ui.fileParserCombo.setCurrentIndex(
self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser"))
if linetext.split(".")[1].upper() == "CTF":
headerDict = ctf.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "ANG":
headerDict = ang.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "XML":
print("XML Parser used")
self.createtablemodel = TableModelC(headerDict, self)
self.filterModel.displayed = []
self.filterModel.setSourceModel(self.createtablemodel)
self.filterModel.fileType.append(linetext)
self.createTableSearchFilterModel = QSortFilterProxyModel(self)
self.createTableSearchFilterModel.setSourceModel(self.filterModel)
self.createTableSearchFilterModel.setFilterKeyColumn(1)
self.createTableSearchFilterModel.setDynamicSortFilter(True)
self.ui.metadataTableView.setModel(
self.createTableSearchFilterModel)
self.treeModel = TreeModel(["Available File Metadata"], headerDict,
self.createtablemodel)
if len(self.customInputs) != 0:
for i in range(len(self.customInputs)):
self.createtablemodel.beginInsertRows(
self.createtablemodel.index(
len(self.createtablemodel.metadataList), 0), i, i)
self.createtablemodel.metadataList.append(
self.customInputs[i])
self.createtablemodel.endInsertRows()
self.createTreeSearchFilterModel = QSortFilterProxyModel(self)
self.createTreeSearchFilterModel.setSourceModel(self.treeModel)
self.createTreeSearchFilterModel.setFilterKeyColumn(0)
self.createTreeSearchFilterModel.setDynamicSortFilter(True)
self.createTreeSearchFilterModel.setRecursiveFilteringEnabled(True)
self.ui.metadataTreeView.setModel(self.createTreeSearchFilterModel)
self.treeModel.checkChanged.connect(self.filterModel.checkList)
self.createtrashDelegate.pressed.connect(self.handleRemoveCreate)
self.toggleButtons()
return True
def restoreTemplate(self):
#Clear data on create side
self.createtablemodel.metadataList = []
self.createtablemodel.hiddenList = []
self.filterModel.displayed = []
#open template
linetext = QFileDialog.getOpenFileName(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*.ez)"))[0]
if linetext != "":
infile = open(linetext, "r")
infile.readline()
fileType = infile.readline()
self.fileType = json.loads(fileType)
infile.readline()
newList = infile.readline()
newList = json.loads(newList)
newDict = infile.readline()
newDict = json.loads(newDict)
self.ui.dataTypeText.setText(self.fileType[0][-3:].upper())
self.ui.fileParserCombo.setCurrentIndex(
self.ui.fileParserCombo.findText(
self.fileType[0][-3:].upper() + " Parser"))
self.ui.dataFileLineEdit.setText(self.fileType[0])
self.createtablemodel = TableModelC(newDict, self)
self.filterModel.displayed = []
self.filterModel.setSourceModel(self.createtablemodel)
self.filterModel.fileType = self.fileType
self.createTableSearchFilterModel = QSortFilterProxyModel(self)
self.createTableSearchFilterModel.setSourceModel(self.filterModel)
self.createTableSearchFilterModel.setFilterKeyColumn(1)
self.createTableSearchFilterModel.setDynamicSortFilter(True)
self.ui.metadataTableView.setModel(
self.createTableSearchFilterModel)
self.treeModel = TreeModelR(["Available File Metadata"], newDict,
self.createtablemodel, newList,
self.filterModel)
for i in range(len(newList)):
if "Custom Input" in newList[i]["Source"]:
self.numCustoms += 1
self.customInputs.append(newList[i])
self.createtablemodel.beginInsertRows(
self.createtablemodel.index(
len(self.createtablemodel.metadataList), 0), i, i)
self.createtablemodel.metadataList.append(newList[i])
self.createtablemodel.endInsertRows()
self.createTreeSearchFilterModel = QSortFilterProxyModel(self)
self.createTreeSearchFilterModel.setSourceModel(self.treeModel)
self.createTreeSearchFilterModel.setFilterKeyColumn(0)
self.createTreeSearchFilterModel.setDynamicSortFilter(True)
self.createTreeSearchFilterModel.setRecursiveFilteringEnabled(True)
self.ui.metadataTreeView.setModel(self.createTreeSearchFilterModel)
self.treeModel.checkChanged.connect(self.filterModel.checkList)
self.createtrashDelegate.pressed.connect(self.handleRemoveCreate)
self.ui.TabWidget.setCurrentIndex(0)
def extractFile(self, fileLink=False):
if fileLink == False:
linetext = QFileDialog.getOpenFileName(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*" + self.fileType + ")"))[0]
else:
linetext = fileLink
if linetext != "":
self.setWindowTitle(linetext)
self.ui.dataTypeText.setText(linetext.split(".")[1].upper())
self.ui.otherDataFileLineEdit.setText(linetext)
if self.ui.addMetadataFileCheckBox.checkState() == Qt.Checked:
self.uselistmodel.removeAllRows()
self.uselistmodel.addRow(linetext)
self.toggleButtons()
if self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser") != -1:
headerDict = {}
self.ui.fileParserCombo.setCurrentIndex(
self.ui.fileParserCombo.findText(
linetext.split(".")[1].upper() + " Parser"))
if linetext.split(".")[1].upper() == "CTF":
headerDict = ctf.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "ANG":
headerDict = ang.parse_header_as_dict(linetext)
elif linetext.split(".")[1].upper() == "XML":
print("XML Parser used")
if self.templatedata:
self.usetablemodel.metadataList = []
self.usefilterModel = FilterModelU(self)
self.usefilterModel.setSourceModel(self.usetablemodel)
self.unusedTreeModel = TreeModelU(["Available File Metadata"],
headerDict,
self.usetablemodel,
self.editableKeys)
self.templatelist = []
self.templatesources = []
for i in range(len(self.templatedata)):
self.templatelist.append(self.templatedata[i])
if "Custom Input" not in self.templatedata[i]["Source"]:
self.templatesources.append("/".join(
self.templatedata[i]['Source'].split("/")[1:]))
else:
self.usetablemodel.addExistingRow(self.templatedata[i])
self.usesearchFilterModel = QSortFilterProxyModel(self)
self.usesearchFilterModel.setSourceModel(self.usefilterModel)
self.usesearchFilterModel.setFilterKeyColumn(0)
self.usesearchFilterModel.setDynamicSortFilter(True)
self.ui.useTemplateTableView.setModel(
self.usesearchFilterModel)
self.toggleButtons()
def openPackage(self):
linetext = QFileDialog.getOpenFileName(
self, self.tr("Select File"),
QStandardPaths.displayName(QStandardPaths.HomeLocation),
self.tr("Files (*.ezpak)"))[0]
if linetext != "":
infile = open(linetext, "r")
self.loadTemplateFile(infile.readline()[1:-2])
extractedFile = infile.readline()[1:-2]
fileList = infile.readline()
fileList = json.loads(fileList)
self.fileList = fileList[:]
self.uselistmodel = ListModel(self, self.usetablemodel,
self.fileList)
self.ui.useTemplateListView.setModel(self.uselistmodel)
self.extractFile(extractedFile)
self.uselistmodel.removeAllRows()
for i in range(len(fileList)):
self.uselistmodel.addRow(fileList[i])
self.toggleButtons()
infile.close()
self.ui.TabWidget.setCurrentIndex(1)
