def __init__(self, parent, afefuc): super(SeleniumExportWrapper, self).__init__(parent) self.parent = parent self.form = Ui_SeleniumExport() self.afefuc = afefuc self.system = None self.browser = None self.path = None self.tc = None self.isClosed = False
class SeleniumExportWrapper(QtGui.QDialog): def __init__(self, parent, afefuc): super(SeleniumExportWrapper, self).__init__(parent) self.parent = parent self.form = Ui_SeleniumExport() self.afefuc = afefuc self.system = None self.browser = None self.path = None self.tc = None self.isClosed = False def load(self): self.form.testCaseComboBox.addItem('All test cases', self.afefuc['project'].testcases) for testcase in self.afefuc['project'].testcases.tests: self.form.testCaseComboBox.addItem(testcase.title, testcase) def show(self): self.form.setupUi(self) self.load() QtCore.QObject.connect(self.form.buttonOK, QtCore.SIGNAL(_fromUtf8("clicked()")), self.clickedOKButton) QtCore.QObject.connect(self.form.buttonCancel, QtCore.SIGNAL(_fromUtf8("clicked()")), self.clickedCancelButton) QtCore.QObject.connect(self.form.selectPathButton, QtCore.SIGNAL(_fromUtf8("clicked()")), self.clickedSelectButton) self.exec_() def clickedCancelButton(self): self.isClosed = True self.close() def clickedOKButton(self): tccb = self.form.testCaseComboBox self.tc = tccb.itemData(tccb.currentIndex()).toPyObject() bccb = self.form.browserComboBox self.browser = bccb.currentText() sccb = self.form.systemComboBox self.system = sccb.currentText() if self.path: if len(self.tc) > 0: sb = None if self.afefuc['project'].language == 'en': sb = highlighter('generated/testcases/en.xml') else: sb = highlighter('generated/testcases/pl.xml') s = selenium(sb, self) if isinstance(self.tc, TestCases): for test in self.tc.tests: s.generateCode(test, self.browser, self.system, self.path) else: s.generateCode(self.tc, self.browser, self.system, self.path) QtGui.QMessageBox.information(self, 'Message', 'Export finished.', QtGui.QMessageBox.Ok) else: QtGui.QMessageBox.information(self, 'Message', 'There is nothing to export.', QtGui.QMessageBox.Ok) else: QtGui.QMessageBox.warning(self, 'Message', 'Directory is not selected.', QtGui.QMessageBox.Ok) self.close() def clickedSelectButton(self): path = QtGui.QFileDialog.getExistingDirectory(self, 'Select directory') if path: self.path = path self.form.pathLineEdit.setText(path) def closeEvent(self, e): if self.isClosed: e.accept() else: e.ignore()