def setupCommandGUI(self, commands):
for name, node in sorted(commands.items(), key=natural_keys):
row = self.ui.gridLayout.rowCount() + 1
label = QLabel(self.ui.groupBox)
if not node.category:
label.setText((' ' * node.level) +
node.name.replace('___', ' '))
else:
label.setText((' ' *
node.level) + '<b>' +
node.name.replace('___', ' ') + '</b>')
label.setObjectName('lableCommand_' + node.name)
if not node.category:
buttonStart = QPushButton(self.ui.groupBox)
buttonStart.setText('Start')
buttonStart.setObjectName('pushButtonCommand_' + node.name +
'_start')
buttonStart.clicked.connect(partial(self.onStart, name))
buttonStop = QPushButton(self.ui.groupBox)
buttonStop.setText('Stop')
buttonStop.setObjectName('pushButtonCommand_' + node.name +
'_stop')
buttonStop.clicked.connect(partial(self.onStop, name))
buttonRestart = QPushButton(self.ui.groupBox)
buttonRestart.setText('Restart')
buttonRestart.setObjectName('pushButtonCommand_' + node.name +
'_restart')
buttonRestart.clicked.connect(partial(self.onRestart, name))
progressBar = None
if node.name in self.progressTopics:
progressBar = QProgressBar(self.ui.groupBox)
progressBar.setObjectName('progressBarCommand_' +
node.name)
progressBar.setValue(0)
progressBar.setMaximum(100)
progressBar.setMaximumWidth(300)
# if node.level == 0 or (node.level == 1 and node.parent.category):
# buttonStart.setEnabled(True)
# buttonStop.setEnabled(True)
# buttonRestart.setEnabled(True)
# else:
# buttonStart.setEnabled(False)
# buttonStop.setEnabled(False)
# buttonRestart.setEnabled(False)
self.ui.gridLayout.addWidget(buttonStart, row, 1)
self.ui.gridLayout.addWidget(buttonStop, row, 2)
self.ui.gridLayout.addWidget(buttonRestart, row, 3)
if not progressBar is None:
self.ui.gridLayout.addWidget(progressBar, row, 4)
self.ui.gridLayout.addWidget(label, row, 0)
self.setupCommandGUI(node.childs)
class ProgressDialog(QDialog):
"""
Progress dialog
"""
def __init__(self, dialogName, parent=None):
"""
Constructor
@param dialogName:
@type dialogName:
@param parent:
@type parent:
"""
QDialog.__init__(self, parent)
self.setWindowIcon(QIcon(":/main.png"))
self.name = dialogName
self.createDialog()
def createDialog(self):
"""
Create qt dialog
"""
self.setWindowTitle("%s" % self.name)
layout = QVBoxLayout()
self.loadingLabel = QLabel("Loading...")
self.imageLabel = QLabel()
layout2 = QHBoxLayout()
layout2.addWidget(self.imageLabel)
layout2.addWidget(self.loadingLabel)
self.progressBar = QProgressBar(self)
self.progressBar.setMaximum(0)
self.progressBar.setProperty("value", 0)
self.progressBar.setAlignment(Qt.AlignCenter)
self.progressBar.setObjectName("progressBar")
layout.addLayout(layout2)
layout.addWidget(self.progressBar)
self.setLayout(layout)
flags = Qt.WindowFlags()
flags |= Qt.MSWindowsFixedSizeDialogHint
self.setWindowFlags(flags)
class Ui_Form (QWidget):
'''
Ui class. Generated with pyuic4.
'''
def __init__ (self, parent=None):
QWidget.__init__(self, parent)
self.timer = QTimer (self)
self.image = QImage (720, 450, QImage.Format_RGB888)
self.__engine = REngineThread ()
self.__model = Model ()
self.__fsm = DebugStateMachine (self)
def wireEngineUp (self):
'''
this method connects the REngine's signals.
'''
self.connect (self.__engine, SIGNAL ("update (float)"), self.updateImage)
self.connect (self.__engine, SIGNAL ("thread_completed()"), self.__fsm.finaliseRender)
self.connect (self.__engine, SIGNAL ("inters_created (PyQt_PyObject, PyQt_PyObject)"), self.widget.addIntersection)
self.connect (self.__engine, SIGNAL ("vector_created (PyQt_PyObject, PyQt_PyObject, QString)"), self.widget.addArrow)
self.connect (self.__engine, SIGNAL ("line_created (PyQt_PyObject, PyQt_PyObject, QString)"), self.widget.addLine)
def wireOGLViewerUp (self):
'''
this method connects the REngine's signals.
'''
self.widget.setModel (self.__model)
QObject.connect (self.widget, SIGNAL ("MatrixChanged (PyQt_PyObject)"), self.displayGLMatrix)
# call the function run regularly when the focus is on it. (60 FPS if interval = 20)
QObject.connect (self.timer, SIGNAL ('timeout()'), self.widget.run)
def freezeGLFrameRate (self):
self.timer.stop ()
def speedUpGLFrameRate (self):
self.timer.start ()
self.timer.setInterval (20)
def setupUi (self, Form):
font = QFont ()
font.setPointSize (9)
font.setBold (False)
font.setWeight (50)
sizePolicy = QSizePolicy (QSizePolicy.Fixed, QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch (0)
sizePolicy.setVerticalStretch (0)
Form.setObjectName (_fromUtf8("Form"))
Form.resize (971, 930)
self.plainTextEdit = QPlainTextEdit (Form)
self.plainTextEdit.setGeometry (QRect (740, 280, 221, 611))
self.plainTextEdit.setObjectName (_fromUtf8 ("plainTextEdit"))
self.plainTextEdit.setFont (font)
self.widget = OGLViewer (Form)
self.widget.setUi_Form (self)
self.widget.setGeometry (QRect (10, 10, 720, 450))
sizePolicy.setHeightForWidth (self.widget.sizePolicy().hasHeightForWidth())
self.widget.setSizePolicy (sizePolicy)
self.widget.setObjectName (_fromUtf8 ("widget"))
self.wireOGLViewerUp ()
self.wireEngineUp ()
self.label_view = QLabel (Form)
self.label_view.setGeometry (QRect (10, 470, 720, 450))
sizePolicy.setHeightForWidth (self.label_view.sizePolicy().hasHeightForWidth())
self.label_view.setSizePolicy (sizePolicy)
self.label_view.setObjectName (_fromUtf8 ("label_view"))
self.initLabel ()
self.pixmap_item = self.label_view.setPixmap (QPixmap.fromImage (self.image))
# buttons definitions
self.renderBtn = QPushButton (Form)
self.pauseBtn = QPushButton (Form)
self.stopBtn = QPushButton (Form)
self.upBtn = QPushButton (Form)
self.downBtn = QPushButton (Form)
self.moreDownBtn = QPushButton (Form)
self.moreUpBtn = QPushButton (Form)
self.rightBtn = QPushButton (Form)
self.moreRightBtn = QPushButton (Form)
self.leftBtn = QPushButton (Form)
self.furtherLeft = QPushButton (Form)
self.grid_switch = QPushButton (Form)
buttons_properties_list = [[self.renderBtn, True, QRect (740, 140, 61, 21), "render"],
[self.pauseBtn, False, QRect (740, 120, 61, 21), "pause"],
[self.stopBtn, False, QRect (740, 100, 61, 21), "stop"],
[self.upBtn, False, QRect (820, 120, 21, 21), "one_row_up"],
[self.downBtn, False, QRect (820, 140, 21, 21), "one_row_down"],
[self.moreDownBtn, False, QRect (820, 160, 21, 21), "ten_rows_down"],
[self.moreUpBtn, False, QRect (820, 100, 21, 21), "ten_rows_up"],
[self.rightBtn, False, QRect (780, 180, 21, 21), "one_column_right"],
[self.moreRightBtn, False, QRect (800, 180, 21, 21), "ten_columns_right"],
[self.leftBtn, False, QRect (760, 180, 21, 21), "one_column_left"],
[self.furtherLeft, False, QRect (740, 180, 21, 21), "ten_columns_left"],
[self.grid_switch, False, QRect (870, 230, 91, 31), "grid_switch"]]
for button in buttons_properties_list:
button[0].setEnabled (button[1])
button[0].setGeometry (button[2])
button[0].setFont (font)
button[0].setObjectName (_fromUtf8 (button[3]))
# other UI elements
self.progressBar = QProgressBar (Form)
self.progressBar.setGeometry (QRect (740, 901, 221, 20))
self.progressBar.setProperty ("value", 0)
self.progressBar.setObjectName (_fromUtf8("progressBar"))
self.progressBar.setMinimum (0)
self.progressBar.setMaximum (100)
self.slider_label = QLabel (Form)
self.slider_label.setGeometry (QRect(900, 260, 61, 16))
self.slider_label.setFont(font)
self.slider_label.setObjectName (_fromUtf8("slider_label"))
self.slider_label.setEnabled (True)
self.arrowSizeSlider = QSlider (Form)
self.arrowSizeSlider.setGeometry (QRect (740, 258, 151, 22))
self.arrowSizeSlider.setMinimum (2)
self.arrowSizeSlider.setMaximum (40)
self.arrowSizeSlider.setSingleStep (1)
self.arrowSizeSlider.setProperty ("value", 4)
self.arrowSizeSlider.setOrientation (Qt.Horizontal)
self.arrowSizeSlider.setObjectName (_fromUtf8("arrowSizeSlider"))
self.retranslateUi (Form)
QMetaObject.connectSlotsByName (Form)
def retranslateUi (self, Form):
Form.setWindowTitle (_translate ("Form", "RayTracing Debugging Tool", None))
self.renderBtn.setText (_translate ("Form", "Render", None))
self.pauseBtn.setText (_translate ("Form", "Pause", None))
self.stopBtn.setText (_translate ("Form", "Stop", None))
self.upBtn.setText (_translate ("Form", "^", None))
self.downBtn.setText (_translate ("Form", "v", None))
self.moreDownBtn.setText (_translate ("Form", "+", None))
self.moreUpBtn.setText (_translate ("Form", "-", None))
self.rightBtn.setText (_translate ("Form", ">", None))
self.moreRightBtn.setText (_translate ("Form", "+", None))
self.leftBtn.setText (_translate ("Form", "<", None))
self.furtherLeft.setText (_translate ("Form", "-", None))
self.grid_switch.setText (_translate ("Form", "Grid on/off", None))
self.slider_label.setText (_translate ("Form", "Arrows size", None))
self.connect (self.renderBtn, SIGNAL ("clicked()"), self.__fsm.startRendering)
self.connect (self.pauseBtn, SIGNAL ("clicked()"), self.__fsm.pauseRendering)
self.connect (self.stopBtn, SIGNAL ("clicked()"), self.__fsm.stopRendering)
self.connect (self.arrowSizeSlider, SIGNAL ("sliderMoved(int)"), self.resizeArrows)
def initLabel (self):
color = QColor (250, 250, 250)
self.image.fill (color)
'''
# test init
for i in range (0, 200, 20):
for x in range (i, i+20):
for y in range (i, i+20):
self.image.setPixel (x,y, qRgb (0, 0, 0))
'''
def enableUIButtons (self, boolean_dict):
'''
method used by the debug state machine to manage the greyed-out state of buttons.
'''
b_dict = dict (boolean_dict)
self.renderBtn.setEnabled (b_dict['renderBtn'])
self.pauseBtn.setEnabled (b_dict['pauseBtn'])
self.stopBtn.setEnabled (b_dict['stopBtn'])
self.upBtn.setEnabled (b_dict['upBtn'])
self.downBtn.setEnabled (b_dict['downBtn'])
self.moreDownBtn.setEnabled (b_dict['moreDownBtn'])
self.moreUpBtn.setEnabled (b_dict['moreUpBtn'])
self.rightBtn.setEnabled (b_dict['rightBtn'])
self.moreRightBtn.setEnabled (b_dict['moreRightBtn'])
self.leftBtn.setEnabled (b_dict['leftBtn'])
self.furtherLeft.setEnabled (b_dict['furtherLeft'])
if b_dict['progressBar'] == 'completed':
self.progressBar.setValue (100)
elif b_dict['progressBar'] == 'reset':
self.progressBar.reset ()
def addScreenshot (self):
'''
it grabs a screenshot of OpenGL viewer and add it into the QImage instance.
'''
self.image = self.widget.grabFrameBuffer ()
self.pixmap_item = self.label_view.setPixmap (QPixmap.fromImage (self.image))
def prepAndStartEngineUp (self):
'''
it preps the engine and start it up.
'''
self.__engine.setImage (self.image)
self.__engine.setCameraNormalMatrix (self.widget.getNormalMatrix(), self.widget.getFovy ())
self.__engine.setModel (self.__model)
self.__engine.start ()
def addCamera (self): self.widget.addCamera ()
def setIsStoppedFlag (self, boo): self.__engine.setIsStoppedFlag (boo)
def setIsPausedFlag (self, boo): self.__engine.setIsPausedFlag (boo)
def changeRenderBtnName (self, title): self.renderBtn.setText (_translate ("Form", title, None))
# listeners - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
def resizeArrows (self, e):
self.widget.changeArrowsSize (e*0.5)
def updateImage (self, e):
self.pixmap_item = self.label_view.setPixmap (QPixmap.fromImage (self.image))
self.progressBar.setValue (int(100*e))
def displayGLMatrix (self, e):
'''
this method is registered to the MatrixChanged event fired from the OGLViewer
component. It prints the model matrix. Only for debugging purposes.
'''
pass
class MessageBoxDialog(QDialog):
"""
Message box dialog
"""
Download = pyqtSignal(str)
DownloadCanceled = pyqtSignal()
def __init__(self, dialogName, parent = None):
"""
Constructor
@param dialogName:
@type dialogName:
@param parent:
@type parent:
"""
QDialog.__init__(self, parent)
self.setWindowIcon( QIcon(":/main.png") )
self.name = dialogName
self.url = None
self.createDialog()
def closeEvent(self, event):
"""
On close event
@param event:
@type event:
"""
pass
def setDownload(self, title, txt, url, cancelButton=True):
"""
Set as download
@param title:
@type title:
@param txt:
@type txt:
@param url:
@type url:
"""
self.setWindowFlags(Qt.Drawer | Qt.CustomizeWindowHint)
self.url = url
self.setWindowTitle( title )
self.progressBar.hide()
self.imageLabel.setPixmap(QPixmap(':/information.png'))
self.loadingLabel.setText( txt )
self.downloadButton.show()
if cancelButton:
self.cancelButton.show()
else:
self.cancelButton.hide()
self.okButton.hide()
self.imageLabel.show()
def setTxt (self, title, txt, error=False):
"""
Set text
@param title:
@type title:
@param txt:
@type txt:
"""
self.setWindowTitle( title )
self.progressBar.hide()
self.loadingLabel.setText( txt )
self.imageLabel.show()
if error:
self.imageLabel.setPixmap(QPixmap(':/warning.png'))
else:
self.imageLabel.setPixmap(QPixmap(':/information.png'))
self.okButton.show()
self.downloadButton.hide()
self.cancelButton.hide()
def setLoading (self, msg='Loading data...' ):
"""
Set as loading
@param msg:
@type msg:
"""
self.setWindowTitle( "%s" % self.name )
self.progressBar.setMaximum(0)
self.progressBar.setProperty("value", 0)
self.progressBar.show()
self.loadingLabel.setText( msg )
self.loadingLabel.show()
self.okButton.hide()
self.downloadButton.hide()
self.cancelButton.hide()
self.imageLabel.hide()
def setConnection (self ):
"""
Set text with connection
"""
self.loadingLabel.setText( "Connection..." )
def setInitialization (self ):
"""
Set text with initialization
"""
self.loadingLabel.setText( "Initialization..." )
self.show()
def download (self):
"""
Emit download signal
"""
self.Download.emit(self.url)
self.accept()
def onReject(self):
"""
Called on cancel
"""
self.DownloadCanceled.emit()
self.reject()
def updateDataReadProgress(self, bytesRead, totalBytes):
"""
Update the progress bar
@param bytesRead:
@type bytesRead:
@param totalBytes:
@type totalBytes:
"""
self.progressBar.setMaximum(totalBytes)
self.progressBar.setValue(bytesRead)
def createDialog(self):
"""
Create qt dialog
"""
self.setWindowTitle( "%s" % self.name )
layout = QVBoxLayout()
self.loadingLabel = QLabel("Loading...")
self.imageLabel = QLabel()
layout2 = QHBoxLayout()
layout2.addWidget( self.imageLabel )
layout2.addWidget( self.loadingLabel )
self.progressBar = QProgressBar(self)
self.progressBar.setMaximum(0)
self.progressBar.setProperty("value", 0)
self.progressBar.setAlignment(Qt.AlignCenter)
self.progressBar.setObjectName("progressBar")
buttonLayout = QHBoxLayout()
buttonLayout.addStretch()
self.okButton = QPushButton("Ok", self)
self.downloadButton = QPushButton("Download", self)
self.cancelButton = QPushButton("Cancel", self)
buttonLayout.addWidget(self.okButton)
buttonLayout.addWidget(self.downloadButton)
buttonLayout.addWidget(self.cancelButton)
self.okButton.clicked.connect(self.accept)
self.downloadButton.clicked.connect(self.download)
self.cancelButton.clicked.connect(self.onReject)
self.okButton.hide()
layout.addLayout(layout2)
layout.addWidget( self.progressBar )
layout.addLayout(buttonLayout)
self.setLayout(layout)
flags = Qt.WindowFlags()
flags |= Qt.WindowCloseButtonHint
flags |= Qt.MSWindowsFixedSizeDialogHint
self.setWindowFlags(flags)
class DHttpReplay(QtHelper.EnhancedQDialog, Logger.ClassLogger):
"""
Http replay dialog
"""
def __init__(self, parent=None, offlineMode=False):
"""
Constructor
@param parent:
@type parent:
"""
super(DHttpReplay, self).__init__(parent)
self.offlineMode = offlineMode
self.defaultIp = "127.0.0.1"
self.defaultPort = "80"
self.newTest = ''
self.newTestExec = ''
self.newInputs = []
self.requests = []
self.responses = []
self.defaultTemplates = DefaultTemplates.Templates()
self.testType = None
self.createDialog()
self.createConnections()
self.createActions()
self.createToolbar()
def createActions(self):
"""
Create qt actions
"""
self.openAction = QtHelper.createAction(self,
"&Open",
self.importTrace,
icon=QIcon(":/folder_add.png"),
tip='Open network trace.')
self.exportTUAction = QtHelper.createAction(
self,
"&Test Unit",
self.exportToTU,
icon=QIcon(":/%s.png" % WWorkspace.TestUnit.TYPE),
tip='Export to Test Unit')
self.exportTSAction = QtHelper.createAction(
self,
"&Test Suite",
self.exportToTS,
icon=QIcon(":/%s.png" % WWorkspace.TestSuite.TYPE),
tip='Export to Test Suite')
self.cancelAction = QtHelper.createAction(self,
"&Cancel",
self.reject,
tip='Cancel')
menu = QMenu(self)
menu.addAction(self.exportTUAction)
menu.addAction(self.exportTSAction)
self.exportToAction = QtHelper.createAction(
self,
"&Export to",
self.exportToTU,
icon=QIcon(":/%s.png" % WWorkspace.TestUnit.TYPE),
tip='Export to tests')
self.exportToAction.setMenu(menu)
self.exportToAction.setEnabled(False)
def createDialog(self):
"""
Create dialog
"""
self.dockToolbar = QToolBar(self)
self.dockToolbar.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
self.setWindowTitle(WINDOW_TITLE)
self.resize(500, 400)
self.ipEdit = QLineEdit(self.defaultIp)
ipRegExpVal = QRegExpValidator(self)
ipRegExp = QRegExp("\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}")
ipRegExpVal.setRegExp(ipRegExp)
self.ipEdit.setValidator(ipRegExpVal)
self.portEdit = QLineEdit(self.defaultPort)
self.portEdit.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
validatorPort = QIntValidator(self)
self.portEdit.setValidator(validatorPort)
self.progressBar = QProgressBar(self)
self.progressBar.setMaximum(100)
self.progressBar.setProperty("value", 0)
self.progressBar.setAlignment(Qt.AlignCenter)
self.progressBar.setObjectName("progressBar")
self.guiSikuliGroupBox = QGroupBox("")
self.guiSikuliGroupBox.setFlat(True)
self.automaticAdp = QRadioButton("Automatic")
self.automaticAdp.setChecked(True)
self.defaultAdp = QRadioButton("Default")
self.genericAdp = QRadioButton("Generic")
vbox = QHBoxLayout()
vbox.addWidget(self.automaticAdp)
vbox.addWidget(self.defaultAdp)
vbox.addWidget(self.genericAdp)
vbox.addStretch(1)
self.guiSikuliGroupBox.setLayout(vbox)
layout = QVBoxLayout()
layout.addWidget(self.dockToolbar)
layout.addSpacing(12)
paramLayout = QGridLayout()
paramLayout.addWidget(QLabel("Destination IP:"), 0, 0, Qt.AlignRight)
paramLayout.addWidget(self.ipEdit, 0, 1)
paramLayout.addWidget(QLabel("Destination Port:"), 1, 0, Qt.AlignRight)
paramLayout.addWidget(self.portEdit, 1, 1)
paramLayout.addWidget(QLabel(self.tr("Gui adapter selector:")), 2, 0,
Qt.AlignRight)
paramLayout.addWidget(self.guiSikuliGroupBox, 2, 1)
layout.addLayout(paramLayout)
self.logsEdit = QTextEdit()
self.logsEdit.setReadOnly(True)
self.logsEdit.setTextInteractionFlags(Qt.NoTextInteraction)
layout.addSpacing(12)
layout.addWidget(self.logsEdit)
layout.addSpacing(12)
layout.addWidget(self.progressBar)
self.setLayout(layout)
def createToolbar(self):
"""
Create toolbar
"""
self.dockToolbar.setObjectName("File toolbar")
self.dockToolbar.addAction(self.openAction)
self.dockToolbar.addSeparator()
self.dockToolbar.addAction(self.exportToAction)
self.dockToolbar.addSeparator()
self.dockToolbar.setIconSize(QSize(16, 16))
def createConnections(self):
"""
Create qt connections
"""
pass
def autoScrollOnTextEdit(self):
"""
Automatic scroll on text edit
"""
cursor = self.logsEdit.textCursor()
cursor.movePosition(QTextCursor.End)
self.logsEdit.setTextCursor(cursor)
def strip_html(self, txt):
"""
Strip html
"""
if "<" in txt:
txt = txt.replace('<', '<')
if ">" in txt:
txt = txt.replace('>', '>')
return txt
def addLogSuccess(self, txt):
"""
Add log success in the text edit
"""
self.logsEdit.insertHtml(
"<span style='color:darkgreen'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLogWarning(self, txt):
"""
Add log warning in the text edit
"""
self.logsEdit.insertHtml(
"<span style='color:darkorange'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLogError(self, txt):
"""
Add log error in the text edit
"""
self.logsEdit.insertHtml("<span style='color:red'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLog(self, txt):
"""
Append log to the logsEdit widget
"""
self.logsEdit.insertHtml("%s<br />" % unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def importTrace(self):
"""
Import network trace
"""
self.logsEdit.clear()
self.testType = None
if not self.offlineMode:
if not RCI.instance().isAuthenticated():
self.addLogWarning(
txt="<< Connect to the test center in first!")
QMessageBox.warning(self, "Import",
"Connect to the test center in first!")
return
self.exportToAction.setEnabled(False)
self.newTest = ''
self.progressBar.setMaximum(100)
self.progressBar.setValue(0)
if sys.version_info > (3, ):
fileName = QFileDialog.getOpenFileName(
self, self.tr("Open File"), "",
"Network dump (*.cap;*.pcap;*.pcapng)")
else:
fileName = QFileDialog.getOpenFileName(self, self.tr("Open File"),
"", "Network dump (*.cap)")
# new in v18 to support qt5
if QtHelper.IS_QT5:
_fileName, _type = fileName
else:
_fileName = fileName
# end of new
if not _fileName:
return
if sys.version_info < (3, ):
extension = str(_fileName).rsplit(".", 1)[1]
if not (extension == "cap"):
self.addLogError(txt="<< File not supported %s" % _fileName)
QMessageBox.critical(self, "Open", "File not supported")
return
_fileName = str(_fileName)
capName = _fileName.rsplit("/", 1)[1]
self.addLogSuccess(txt=">> Reading the file %s" % _fileName)
if sys.version_info > (3, ):
self.readFileV2(fileName=_fileName)
else:
self.readFile(fileName=_fileName)
def exportToTS(self):
"""
Export to test suite
"""
self.testType = TS
self.exportToTest(TS=True, TU=False)
def exportToTU(self):
"""
Export to test unit
"""
self.testType = TU
self.exportToTest(TS=False, TU=True)
def searchHTTP(self):
"""
Search HTTP module in assistant
"""
# modules accessor
ret = "SutAdapters"
if self.automaticAdp.isChecked():
isGeneric = WWorkspace.Helper.instance().isGuiGeneric(name="GUI")
if isGeneric:
ret = "SutAdapters.Generic"
elif self.defaultAdp.isChecked():
return ret
elif self.genericAdp.isChecked():
ret = "SutAdapters.Generic"
else:
pass
return ret
def exportToTest(self, TS=True, TU=False):
"""
Export to test
"""
if not RCI.instance().isAuthenticated():
self.addLogWarning(txt="<< Connect to the test center in first!")
QMessageBox.warning(self, "Import",
"Connect to the test center in first!")
return
if TS:
self.newTest = self.defaultTemplates.getTestDefinitionAuto()
self.newTestExec = self.defaultTemplates.getTestExecutionAuto()
if TU:
self.newTest = self.defaultTemplates.getTestUnitDefinitionAuto()
destIp = str(self.ipEdit.text())
destPort = str(self.portEdit.text())
self.newInputs = []
self.newInputs.append({
'type': 'self-ip',
'name': 'BIND_IP',
'description': '',
'value': '0.0.0.0',
'color': ''
})
self.newInputs.append({
'type': 'int',
'name': 'BIND_PORT',
'description': '',
'value': '0',
'color': ''
})
self.newInputs.append({
'type': 'str',
'name': 'DEST_IP',
'description': '',
'value': '%s' % destIp,
'color': ''
})
self.newInputs.append({
'type': 'int',
'name': 'DEST_PORT',
'description': '',
'value': '%s' % destPort,
'color': ''
})
self.newInputs.append({
'type': 'bool',
'name': 'DEBUG',
'description': '',
'value': 'False',
'color': ''
})
self.newInputs.append({
'type': 'float',
'name': 'TIMEOUT',
'description': '',
'value': '5.0',
'color': ''
})
adps = """self.ADP_HTTP = %s.HTTP.Client(parent=self, bindIp=input('BIND_IP'), bindPort=input('BIND_PORT'), destinationIp=input('DEST_IP'), destinationPort=input('DEST_PORT'), debug=input('DEBUG'))""" % self.searchHTTP(
)
# prepare steps
steps = []
j = 0
for i in xrange(len(self.requests)):
j = i + 1
if sys.version_info > (3, ): # python3 support
(source, dest, source_port, dest_port, buf_req,
reqDecoded) = self.requests[i]
http_method = str(reqDecoded['method'], 'utf8')
http_status = 'no'
http_reason = ''
else:
http_method = self.requests[i]['tcp-object'].method
http_status = 'no'
http_reason = ''
try:
if sys.version_info > (3, ): # python3 support
(sourceRsp, destRsp, sourcePortRsp, destPortRsp, bufReqRsp,
reqDecodedRsp) = self.responses[i]
http_status = str(reqDecodedRsp['code'])
http_reason = str(reqDecodedRsp['phrase'], 'utf8')
else:
http_status = self.responses[i]['tcp-object'].status
http_reason = self.responses[i]['tcp-object'].reason
except Exception as e:
print(e)
steps.append(
'self.step%s = self.addStep(expected="%s %s response", description="send %s request", summary="send %s request")'
% (j, http_status, http_reason, http_method, http_method))
tests = []
for i in xrange(len(self.requests)):
j = i + 1
if sys.version_info > (3, ): # python3 support
(source, dest, source_port, dest_port, buf_req,
reqDecoded) = self.requests[i]
tests.append("# request %s" % j)
tests.append('self.step%s.start()' % j)
if sys.version_info > (3, ): # python3 support
lines_req = buf_req.splitlines()
else:
lines_req = self.requests[i]['tcp-data'].splitlines()
if sys.version_info > (3, ): # python3 support
tests.append('rawHttp = [%s]' %
lines_req[0].replace(b'"', b'\\"'))
else:
tests.append('rawHttp = ["%s"]' %
lines_req[0].replace(b'"', b'\\"'))
for lreq in lines_req[1:]:
if sys.version_info > (3, ): # python3 support
tests.append('rawHttp.append(%s)' %
lreq.replace(b'"', b'\\"'))
else:
tests.append('rawHttp.append("%s")' %
lreq.replace(b'"', b'\\"'))
tests.append('')
tests.append(
'req_tpl = self.ADP_HTTP.constructTemplateRequest(rawHttp=rawHttp)'
)
tests.append('req = self.ADP_HTTP.sendRequest(tpl=req_tpl)')
try:
tests.append('')
if sys.version_info > (3, ): # python3 support
(sourceRsp, destRsp, sourcePortRsp, destPortRsp, bufReqRsp,
reqDecodedRsp) = self.responses[i]
lines_res = bufReqRsp.splitlines()
else:
lines_res = self.responses[i]['tcp-data'].splitlines()
if sys.version_info > (3, ): # python3 support
tests.append('rawHttpRsp = [%s]' %
lines_res[0].replace(b"'", b"\\'"))
else:
tests.append('rawHttpRsp = ["%s"]' %
lines_res[0].replace(b'"', b'\\"'))
for lres in lines_res[1:]:
if sys.version_info > (3, ): # python3 support
tests.append('rawHttpRsp.append(%s)' %
lres.replace(b"'", b"\\'"))
else:
tests.append('rawHttpRsp.append("%s")' %
lres.replace(b'"', b'\\"'))
except Exception as e:
self.error("unable to append response: %s" % e)
tests.append(
'rsp_tpl = self.ADP_HTTP.constructTemplateResponse(rawHttp=rawHttpRsp)'
)
tests.append(
"rsp = self.ADP_HTTP.hasReceivedResponse(expected=rsp_tpl, timeout=input('TIMEOUT'))"
)
tests.append('if rsp is None:')
tests.append(
'\tself.step%s.setFailed(actual="incorrect response")' % j)
tests.append('else:')
tests.append('\tself.step%s.setPassed(actual="ok")' % j)
tests.append('')
if TS:
init = """self.ADP_HTTP.connect()
connected = self.ADP_HTTP.isConnected( timeout=input('TIMEOUT') )
if not connected:
self.abort( 'unable to connect to the tcp port %s' )
""" % str(self.portEdit.text())
if TU:
init = """self.ADP_HTTP.connect()
connected = self.ADP_HTTP.isConnected( timeout=input('TIMEOUT') )
if not connected:
self.abort( 'unable to connect to the tcp port %s' )
""" % str(self.portEdit.text())
if TS:
cleanup = """self.ADP_HTTP.disconnect()
disconnected = self.ADP_HTTP.isDisconnected( timeout=input('TIMEOUT') )
if not disconnected:
self.error( 'unable to disconnect from the tcp port %s' )
""" % str(self.portEdit.text())
if TU:
cleanup = """self.ADP_HTTP.disconnect()
disconnected = self.ADP_HTTP.isDisconnected( timeout=input('TIMEOUT') )
if not disconnected:
self.error( 'unable to disconnect from the tcp port %s' )
""" % str(self.portEdit.text())
self.newTest = self.newTest.replace(
"<<PURPOSE>>", 'self.setPurpose(purpose="Replay HTTP")')
self.newTest = self.newTest.replace("<<ADPS>>", adps)
if TS:
self.newTest = self.newTest.replace("<<STEPS>>",
'\n\t\t'.join(steps))
if TU:
self.newTest = self.newTest.replace("<<STEPS>>",
'\n\t'.join(steps))
self.newTest = self.newTest.replace("<<INIT>>", init)
self.newTest = self.newTest.replace("<<CLEANUP>>", cleanup)
if TS:
self.newTest = self.newTest.replace("<<TESTS>>",
'\n\t\t'.join(tests))
if TU:
self.newTest = self.newTest.replace("<<TESTS>>",
'\n\t'.join(tests))
self.accept()
def decodeHttpRequest(self, data):
"""
Decode http request
Content chunked not yet implemented
"""
http = {"type": "request"}
lines = data.splitlines()
try:
request_line = lines[0]
except Exception:
self.error("unable to decode http request: %s" % lines)
return None
try:
http["method"] = request_line.split(b" ", 2)[0]
http["uri"] = request_line.split(b" ", 2)[1]
http["version"] = request_line.split(b" ", )[2]
except Exception:
self.error(
"unable to decode status code in the http response: %s" %
request_line)
return None
http["body"] = data.split(b"\r\n\r\n")[1]
headers = []
contentLenght = 0
contentChunked = False
for hdr in data.split(b"\r\n\r\n")[0].splitlines()[1:]:
if len(hdr):
k, v = hdr.split(b":", 1)
if k.lower() == b"content-length":
contentLenght = int(v)
if k.lower() == b"transfer-encoding":
if v.lowert() == b"chunked":
contentChunked = True
headers.append(hdr)
http["headers"] = headers
if len(http["body"]) != contentLenght:
return None # need more data
return http
def decodeHttpResponse(self, data):
"""
Decode http response without body
"""
http = {"type": "response"}
lines = data.splitlines()
try:
status_line = lines[0]
except Exception:
self.error("unable to decode http response: %s" % lines)
return None
try:
http["code"] = int(status_line.split(b" ")[1])
http["phrase"] = status_line.split(b" ", 2)[2]
except Exception:
self.error(
"unable to decode status code in the http response: %s" %
status_line)
return None
http["headers"] = lines[1:]
return http
def readFileV2(self, fileName):
"""
Read pcap file
Support pcap-ng too
"""
fd = open(fileName, 'rb')
fileFormat, fileHead = PcapParse.extractFormat(fd)
if fileFormat == PcapParse.FileFormat.PCAP:
self.trace("pcap file detected")
pcapFile = PcapReader.PcapFile(fd, fileHead).read_packet
self.readFilePacket(pcapFile=pcapFile)
elif fileFormat == PcapParse.FileFormat.PCAP_NG:
self.trace("pcap-png file detected")
pcapFile = PcapngReader.PcapngFile(fd, fileHead).read_packet
self.readFilePacket(pcapFile=pcapFile)
else:
self.addLogError(txt="<< Error to open the network trace")
self.error('unable to open the network trace: file format = %s' %
fileFormat)
QMessageBox.critical(self, "Import", "File not supported")
def __readRequest(self, buffer, data, request, output):
"""
Read request
"""
buffer += data
if b'\r\n\r\n' in data:
reqDecoded = self.decodeHttpRequest(data=buffer)
if reqDecoded is not None:
output.append(request + (reqDecoded, ))
buffer = b''
else:
print("need more data: decode request failed")
else:
print("need more data, no body separator detected on request")
def readFilePacket(self, pcapFile):
"""
Read file packet by packet
"""
ip_expected = str(self.ipEdit.text())
port_expected = int(self.portEdit.text())
# read packet)
packets = pcapFile()
ethernetPackets = list(packets)
self.addLogSuccess(txt="<< Number of packets detected: %s " %
len(ethernetPackets))
# extract tcp packet according to the expected ip and port
tcpPacketsSent = []
tcpPacketsRecv = []
i = 1
self.progressBar.setMaximum(len(ethernetPackets))
self.progressBar.setValue(0)
for pkt in ethernetPackets:
self.progressBar.setValue(i)
i += 1
pktDecoded = PcapParse.decodePacket(pkt, getTcp=True, getUdp=False)
if pktDecoded is not None:
(source, dest, source_port, dest_port, data) = pktDecoded
# skip when no data exists
if dest == ip_expected and int(dest_port) == int(
port_expected) and len(data) > 0:
tcpPacketsSent.append(pktDecoded)
if source == ip_expected and int(source_port) == int(
port_expected) and len(data) > 0:
tcpPacketsRecv.append(pktDecoded)
self.addLogSuccess(txt="<< Number of TCP packets sent: %s " %
len(tcpPacketsSent))
self.addLogSuccess(txt="<< Number of TCP packets received: %s " %
len(tcpPacketsRecv))
# decode https requests
self.requests = []
buf_req = b''
i = 1
self.progressBar.setMaximum(len(tcpPacketsSent))
self.progressBar.setValue(0)
# decode the complete packet
for req in tcpPacketsSent:
self.progressBar.setValue(i)
i += 1
(source, dest, source_port, dest_port, data) = req
if buf_req:
buf_req += data
if b'\r\n\r\n' in data:
reqDecoded = self.decodeHttpRequest(data=buf_req)
if reqDecoded is not None:
self.requests.append((source, dest, source_port,
dest_port, buf_req, reqDecoded))
buf_req = b''
else:
if isRequest(data):
buf_req += data
if b'\r\n\r\n' in data:
reqDecoded = self.decodeHttpRequest(data=buf_req)
if reqDecoded is not None:
self.requests.append(
(source, dest, source_port, dest_port, buf_req,
reqDecoded))
buf_req = b''
self.addLogSuccess(txt="<< Number of HTTP requests extracted: %s " %
len(self.requests))
# decode https response
self.responses = []
buf_rsp = b''
i = 1
self.progressBar.setMaximum(len(tcpPacketsRecv))
self.progressBar.setValue(0)
# decode just headers for response
for req in tcpPacketsRecv:
self.progressBar.setValue(i)
i += 1
(source, dest, source_port, dest_port, data) = req
if buf_rsp:
buf_rsp += data
# try to decode response without body
if b'\r\n\r\n' in data:
rspDecoded = self.decodeHttpResponse(data=buf_rsp)
if rspDecoded is not None:
self.responses.append((source, dest, source_port,
dest_port, buf_rsp, rspDecoded))
buf_rsp = b''
else:
# is http response ?
if data.startswith(b'HTTP/'):
buf_rsp += data
if b'\r\n\r\n' in data:
rspDecoded = self.decodeHttpResponse(data=buf_rsp)
if rspDecoded is not None:
self.responses.append(
(source, dest, source_port, dest_port, buf_rsp,
rspDecoded))
buf_rsp = b''
self.addLogSuccess(txt="<< Number of HTTP responses extracted: %s " %
len(self.responses))
if self.requests:
self.addLogSuccess("<< Read the file finished with success!")
self.addLogWarning(
"<< Click on the export button to generate the test!")
self.exportToAction.setEnabled(True)
else:
self.addLogWarning("<< No http extracted!")
def readFile(self, fileName):
"""
Read the file passed as argument
Old function with dtpkt and python2.7
"""
self.requests = []
self.responses = []
ip_expected = socket.inet_aton(str(self.ipEdit.text()))
port_expected = str(self.portEdit.text())
try:
f = open(fileName, 'rb')
pcap = dpkt.pcap.Reader(f)
tot_pkts = len(list(pcap))
except Exception as e:
self.addLogError(txt="<< Error to open the network trace")
self.error('unable to open the network trace: %s' % str(e))
QMessageBox.critical(self, "Import", "File not supported")
return
else:
self.addLogSuccess(txt="<< Total packets detected: %s " % tot_pkts)
self.progressBar.setMaximum(tot_pkts)
# decode http request
i = 1
buf_req = ''
for ts, buf in pcap:
self.progressBar.setValue(i)
i += 1
# read ethernet layer
eth = dpkt.ethernet.Ethernet(buf)
if eth.type == dpkt.ethernet.ETH_TYPE_IP:
# continue with ip decoding layer
ip = eth.data
if ip.dst == ip_expected:
ip_layer = (ip.src, ip.dst)
if ip.p == dpkt.ip.IP_PROTO_TCP:
tcp = ip.data
if tcp.dport == int(port_expected) and len(
tcp.data) > 0:
tcp_layer = (tcp.sport, tcp.dport)
buf_req += tcp.data
try:
http_req = dpkt.http.Request(buf_req)
except dpkt.dpkt.NeedData as e:
pass
except dpkt.UnpackError as e:
pass
else:
self.requests.append({
'ip-src': ip.src,
'ip-dst': ip.dst,
'port-src': tcp.sport,
'port-dst': tcp.dport,
'tcp-data': buf_req,
'tcp-object': http_req
})
self.addLogWarning(
txt="<< %s http request(s) extracted" %
len(self.requests))
buf_req = ''
# decode http responses
i = 1
self.progressBar.setValue(0)
for ts, buf in pcap:
self.progressBar.setValue(i)
i += 1
# read ethernet layer
eth = dpkt.ethernet.Ethernet(buf)
if eth.type == dpkt.ethernet.ETH_TYPE_IP:
# continue with ip decoding layer
ip = eth.data
if ip.src == ip_expected:
ip_layer = (ip.src, ip.dst)
if ip.p == dpkt.ip.IP_PROTO_TCP:
tcp = ip.data
if tcp.sport == int(port_expected) and len(
tcp.data) > 0:
tcp_layer = (tcp.sport, tcp.dport)
if (tcp.data).startswith('HTTP/'):
try:
new_res = "%s\r\n\r\n" % (
tcp.data).splitlines()[0]
http_res = dpkt.http.Response(new_res)
except dpkt.dpkt.NeedData as e:
pass
except dpkt.UnpackError as e:
pass
else:
self.responses.append({
'ip-src':
ip.src,
'ip-dst':
ip.dst,
'port-src':
tcp.sport,
'port-dst':
tcp.dport,
'tcp-data':
new_res,
'tcp-object':
http_res
})
self.addLogWarning(
txt=
"<< %s http response(s) extracted"
% len(self.responses))
if self.requests:
self.addLogSuccess("<< File decoded with success!")
self.addLogWarning(
"<< Click on the export button to generate the test!")
self.exportToAction.setEnabled(True)
else:
self.addLogWarning("<< No http extracted!")
class DUdpReplay(QtHelper.EnhancedQDialog, Logger.ClassLogger):
"""
Http replay dialog
"""
def __init__(self, parent=None, offlineMode=False):
"""
Constructor
@param parent:
@type parent:
"""
super(DUdpReplay, self).__init__(parent)
self.offlineMode = offlineMode
self.defaultIp = "127.0.0.1"
self.defaultPort = "80"
self.newTest = ''
self.newTestExec = ''
self.newInputs = []
self.requests = []
self.responses = []
self.defaultTemplates = DefaultTemplates.Templates()
self.testType = None
self.createDialog()
self.createConnections()
self.createActions()
self.createToolbar()
def createActions(self):
"""
Create qt actions
"""
self.openAction = QtHelper.createAction(self,
"&Open",
self.importTrace,
icon=QIcon(":/folder_add.png"),
tip='Open network trace.')
self.exportTUAction = QtHelper.createAction(
self,
"&Test Unit",
self.exportToTU,
icon=QIcon(":/%s.png" % WWorkspace.TestUnit.TYPE),
tip='Export to Test Unit')
self.exportTSAction = QtHelper.createAction(
self,
"&Test Suite",
self.exportToTS,
icon=QIcon(":/%s.png" % WWorkspace.TestSuite.TYPE),
tip='Export to Test Suite')
self.cancelAction = QtHelper.createAction(self,
"&Cancel",
self.reject,
tip='Cancel')
menu = QMenu(self)
menu.addAction(self.exportTUAction)
menu.addAction(self.exportTSAction)
self.exportToAction = QtHelper.createAction(
self,
"&Export to",
self.exportToTU,
icon=QIcon(":/%s.png" % WWorkspace.TestUnit.TYPE),
tip='Export to tests')
self.exportToAction.setMenu(menu)
self.exportToAction.setEnabled(False)
def createDialog(self):
"""
Create dialog
"""
self.dockToolbar = QToolBar(self)
self.dockToolbar.setToolButtonStyle(Qt.ToolButtonTextBesideIcon)
self.setWindowTitle(WINDOW_TITLE)
self.resize(500, 400)
self.ipEdit = QLineEdit(self.defaultIp)
ipRegExpVal = QRegExpValidator(self)
ipRegExp = QRegExp("\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}")
ipRegExpVal.setRegExp(ipRegExp)
self.ipEdit.setValidator(ipRegExpVal)
self.portEdit = QLineEdit(self.defaultPort)
self.portEdit.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
validatorPort = QIntValidator(self)
self.portEdit.setValidator(validatorPort)
self.progressBar = QProgressBar(self)
self.progressBar.setMaximum(100)
self.progressBar.setProperty("value", 0)
self.progressBar.setAlignment(Qt.AlignCenter)
self.progressBar.setObjectName("progressBar")
self.guiSikuliGroupBox = QGroupBox("")
self.guiSikuliGroupBox.setFlat(True)
self.automaticAdp = QRadioButton("Automatic")
self.automaticAdp.setChecked(True)
self.defaultAdp = QRadioButton("Default")
self.genericAdp = QRadioButton("Generic")
vbox = QHBoxLayout()
vbox.addWidget(self.automaticAdp)
vbox.addWidget(self.defaultAdp)
vbox.addWidget(self.genericAdp)
vbox.addStretch(1)
self.guiSikuliGroupBox.setLayout(vbox)
layout = QVBoxLayout()
layout.addWidget(self.dockToolbar)
layout.addSpacing(12)
paramLayout = QGridLayout()
paramLayout.addWidget(QLabel("Destination IP:"), 0, 0, Qt.AlignRight)
paramLayout.addWidget(self.ipEdit, 0, 1)
paramLayout.addWidget(QLabel("Destination Port:"), 1, 0, Qt.AlignRight)
paramLayout.addWidget(self.portEdit, 1, 1)
paramLayout.addWidget(QLabel(self.tr("Gui adapter selector:")), 2, 0,
Qt.AlignRight)
paramLayout.addWidget(self.guiSikuliGroupBox, 2, 1)
layout.addLayout(paramLayout)
self.logsEdit = QTextEdit()
self.logsEdit.setReadOnly(True)
self.logsEdit.setTextInteractionFlags(Qt.NoTextInteraction)
layout.addSpacing(12)
layout.addWidget(self.logsEdit)
layout.addSpacing(12)
layout.addWidget(self.progressBar)
self.setLayout(layout)
def createToolbar(self):
"""
Create toolbar
"""
self.dockToolbar.setObjectName("File toolbar")
self.dockToolbar.addAction(self.openAction)
self.dockToolbar.addSeparator()
self.dockToolbar.addAction(self.exportToAction)
self.dockToolbar.addSeparator()
self.dockToolbar.setIconSize(QSize(16, 16))
def createConnections(self):
"""
Create qt connections
"""
pass
def autoScrollOnTextEdit(self):
"""
Automatic scroll on text edit
"""
cursor = self.logsEdit.textCursor()
cursor.movePosition(QTextCursor.End)
self.logsEdit.setTextCursor(cursor)
def strip_html(self, txt):
"""
Strip html
"""
if "<" in txt:
txt = txt.replace('<', '<')
if ">" in txt:
txt = txt.replace('>', '>')
return txt
def addLogSuccess(self, txt):
"""
Add log success in the text edit
"""
self.logsEdit.insertHtml(
"<span style='color:darkgreen'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLogWarning(self, txt):
"""
Add log warning in the text edit
"""
self.logsEdit.insertHtml(
"<span style='color:darkorange'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLogError(self, txt):
"""
Add log error in the text edit
"""
self.logsEdit.insertHtml("<span style='color:red'>%s</span><br />" %
unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def addLog(self, txt):
"""
Append log to the logsEdit widget
"""
self.logsEdit.insertHtml("%s<br />" % unicode(self.strip_html(txt)))
self.autoScrollOnTextEdit()
def importTrace(self):
"""
Import network trace
"""
self.logsEdit.clear()
self.testType = None
if not self.offlineMode:
if not RCI.instance().isAuthenticated():
self.addLogWarning(
txt="<< Connect to the test center in first!")
QMessageBox.warning(self, "Import",
"Connect to the test center in first!")
return
self.exportToAction.setEnabled(False)
self.newTest = ''
self.progressBar.setMaximum(100)
self.progressBar.setValue(0)
if sys.version_info > (3, ):
fileName = QFileDialog.getOpenFileName(
self, self.tr("Open File"), "",
"Network dump (*.cap;*.pcap;*.pcapng)")
else:
fileName = QFileDialog.getOpenFileName(self, self.tr("Open File"),
"", "Network dump (*.cap)")
# new in v18 to support qt5
if QtHelper.IS_QT5:
_fileName, _type = fileName
else:
_fileName = fileName
# end of new
if not _fileName:
return
if sys.version_info < (3, ):
extension = str(_fileName).rsplit(".", 1)[1]
if not (extension == "cap"):
self.addLogError(txt="<< File not supported %s" % _fileName)
QMessageBox.critical(self, "Open", "File not supported")
return
_fileName = str(_fileName)
capName = _fileName.rsplit("/", 1)[1]
self.addLogSuccess(txt=">> Reading the file %s" % _fileName)
self.readFileV2(fileName=_fileName)
def exportToTS(self):
"""
Export to test suite
"""
self.testType = TS
self.exportToTest(TS=True, TU=False)
def exportToTU(self):
"""
Export to test unit
"""
self.testType = TU
self.exportToTest(TS=False, TU=True)
def searchUDP(self):
"""
Search UDP module in assistant
"""
# modules accessor
ret = "SutAdapters"
if self.automaticAdp.isChecked():
isGeneric = WWorkspace.Helper.instance().isGuiGeneric(name="GUI")
if isGeneric:
ret = "SutAdapters.Generic"
elif self.defaultAdp.isChecked():
return ret
elif self.genericAdp.isChecked():
ret = "SutAdapters.Generic"
else:
pass
return ret
def exportToTest(self, TS=True, TU=False):
"""
Export to test
"""
if not RCI.instance().isAuthenticated():
self.addLogWarning(txt="<< Connect to the test center in first!")
QMessageBox.warning(self, "Import",
"Connect to the test center in first!")
return
if TS:
self.newTest = self.defaultTemplates.getTestDefinitionAuto()
self.newTestExec = self.defaultTemplates.getTestExecutionAuto()
if TU:
self.newTest = self.defaultTemplates.getTestUnitDefinitionAuto()
destIp = str(self.ipEdit.text())
destPort = str(self.portEdit.text())
self.newInputs = []
self.newInputs.append({
'type': 'self-ip',
'name': 'BIND_IP',
'description': '',
'value': '0.0.0.0',
'color': ''
})
self.newInputs.append({
'type': 'int',
'name': 'BIND_PORT',
'description': '',
'value': '0',
'color': ''
})
self.newInputs.append({
'type': 'str',
'name': 'DEST_IP',
'description': '',
'value': '%s' % destIp,
'color': ''
})
self.newInputs.append({
'type': 'int',
'name': 'DEST_PORT',
'description': '',
'value': '%s' % destPort,
'color': ''
})
self.newInputs.append({
'type': 'bool',
'name': 'DEBUG',
'description': '',
'value': 'False',
'color': ''
})
self.newInputs.append({
'type': 'float',
'name': 'TIMEOUT',
'description': '',
'value': '5.0',
'color': ''
})
adps = """self.ADP_UDP = %s.UDP.Client(parent=self, bindIp=input('BIND_IP'), bindPort=input('BIND_PORT'), destinationIp=input('DEST_IP'), destinationPort=input('DEST_PORT'), debug=input('DEBUG'), separatorDisabled=True)""" % self.searchUDP(
)
# prepare steps
steps = []
j = 0
for i in xrange(len(self.requests)):
j = i + 1
sentrecv, req = self.requests[i]
if sentrecv == 'sent':
steps.append(
'self.step%s = self.addStep(expected="udp data sent", description="send udp data", summary="send udp data")'
% j)
else:
steps.append(
'self.step%s = self.addStep(expected="udp data received", description="received udp data", summary="received udp data")'
% j)
tests = []
for i in xrange(len(self.requests)):
j = i + 1
sentrecv, req = self.requests[i]
(source, dest, source_port, dest_port, data) = req
if sentrecv == 'sent':
tests.append("# data to sent %s" % j)
tests.append('self.step%s.start()' % j)
if sys.version_info > (3, ):
tests.append('rawSent = %s' % data.replace(b"'", b"\\'"))
else:
tests.append('rawSent = """%s"""' % data)
tests.append('SentMsg = self.ADP_UDP.sendData(data=rawSent)')
tests.append('if not SentMsg:')
tests.append(
'\tself.step%s.setFailed(actual="unable to send data")' %
j)
tests.append('else:')
tests.append(
'\tself.step%s.setPassed(actual="udp data sent succesfully")'
% j)
tests.append('')
if sentrecv == 'recv':
tests.append("# data to received %s" % j)
tests.append('self.step%s.start()' % j)
if sys.version_info > (3, ):
tests.append('rawRecv = %s' % data.replace(b'"', b'\\"'))
else:
tests.append('rawRecv = """%s"""' % data)
tests.append(
'RecvMsg = self.ADP_UDP.hasReceivedData(data=rawRecv, timeout=input("TIMEOUT"))'
)
tests.append('if RecvMsg is None:')
tests.append(
'\tself.step%s.setFailed(actual="unable to received data")'
% j)
tests.append('else:')
tests.append(
'\tself.step%s.setPassed(actual="udp data received succesfully")'
% j)
tests.append('')
if TS:
init = """self.ADP_UDP.startListening()
udpListening = self.ADP_UDP.isListening( timeout=input('TIMEOUT') )
if not udpListening:
self.abort( 'unable to listing to the udp port %s' )
""" % str(self.portEdit.text())
if TU:
init = """self.ADP_UDP.startListening()
udpListening = self.ADP_UDP.isListening( timeout=input('TIMEOUT') )
if not udpListening:
self.abort( 'unable to connect to the udp port %s' )
""" % str(self.portEdit.text())
if TS:
cleanup = """self.ADP_UDP.stopListening()
udpStopped = self.ADP_UDP.isStopped( timeout=input('TIMEOUT') )
if not udpStopped:
self.error( 'unable to no more listen from the udp port %s' )
""" % str(self.portEdit.text())
if TU:
cleanup = """self.ADP_UDP.stopListening()
udpStopped = self.ADP_UDP.isStopped( timeout=input('TIMEOUT') )
if not udpStopped:
self.error( 'unable to no more listen from the udp port %s' )
""" % str(self.portEdit.text())
self.newTest = self.newTest.replace(
"<<PURPOSE>>", 'self.setPurpose(purpose="Replay UDP")')
self.newTest = self.newTest.replace("<<ADPS>>", adps)
if TS:
self.newTest = self.newTest.replace("<<STEPS>>",
'\n\t\t'.join(steps))
if TU:
self.newTest = self.newTest.replace("<<STEPS>>",
'\n\t'.join(steps))
self.newTest = self.newTest.replace("<<INIT>>", init)
self.newTest = self.newTest.replace("<<CLEANUP>>", cleanup)
if TS:
self.newTest = self.newTest.replace("<<TESTS>>",
'\n\t\t'.join(tests))
if TU:
self.newTest = self.newTest.replace("<<TESTS>>",
'\n\t'.join(tests))
self.accept()
def readFileV2(self, fileName):
"""
Read pcap file
Support pcap-ng too
"""
fd = open(fileName, 'rb')
fileFormat, fileHead = PcapParse.extractFormat(fd)
if fileFormat == PcapParse.FileFormat.PCAP:
self.trace("pcap file detected")
pcapFile = PcapReader.PcapFile(fd, fileHead).read_packet
self.readFilePacket(pcapFile=pcapFile)
elif fileFormat == PcapParse.FileFormat.PCAP_NG:
self.trace("pcap-png file detected")
pcapFile = PcapngReader.PcapngFile(fd, fileHead).read_packet
self.readFilePacket(pcapFile=pcapFile)
else:
self.addLogError(txt="<< Error to open the network trace")
self.error('unable to open the network trace: file format = %s' %
fileFormat)
QMessageBox.critical(self, "Import", "File not supported")
def readFilePacket(self, pcapFile):
"""
Read file packet by packet
"""
ip_expected = str(self.ipEdit.text())
port_expected = int(self.portEdit.text())
# read packet)
packets = pcapFile()
ethernetPackets = list(packets)
self.addLogSuccess(txt="<< Total packets detected: %s " %
len(ethernetPackets))
# extract udp packet according to the expected ip and port
self.requests = []
i = 1
self.progressBar.setMaximum(len(ethernetPackets))
self.progressBar.setValue(0)
for pkt in ethernetPackets:
self.progressBar.setValue(i)
i += 1
pktDecoded = PcapParse.decodePacket(pkt, getTcp=False, getUdp=True)
if pktDecoded is not None:
(source, dest, source_port, dest_port, data) = pktDecoded
# skip when no data exists
if dest == ip_expected and int(dest_port) == int(
port_expected) and len(data) > 0:
self.requests.append(('sent', pktDecoded))
if source == ip_expected and int(source_port) == int(
port_expected) and len(data) > 0:
self.requests.append(('recv', pktDecoded))
self.addLogSuccess(txt="<< Number of UDP packets detected: %s" %
len(self.requests))
if self.requests:
self.addLogSuccess("<< File decoded with success!")
self.addLogWarning(
"<< Click on the export button to generate the test!")
self.exportToAction.setEnabled(True)
else:
self.addLogWarning("<< No udp extracted!")
class EkdProgress(QDialog) :
"""EkdProgress est une boite de dialog contenant l'état de la progression de chaque processus"""
def __init__(self, parent=None, totfile=0):
super(EkdProgress, self).__init__(parent)
self.setupUi(self)
self.barProgress.setMaximum(totfile)
self.fermer.setEnabled(False)
self.totframe=totfile
self.value = 0
self.connect(self.fermer,SIGNAL("clicked()"),self.tmpclose)
def upProgress(self) :
self.value += 1
self.barProgress.setValue(self.value)
def addText(self, text) :
self.infoText.append(QString(text))
def tmpclose(self) :
self.emit(SIGNAL("cleantmp"))
self.close()
def setupUi(self, showprogress):
showprogress.setObjectName("showprogress")
showprogress.resize(335, 310)
self.verticalLayout = QVBoxLayout(showprogress)
self.verticalLayout.setObjectName("verticalLayout")
self.barProgress = QProgressBar(showprogress)
self.barProgress.setProperty("value", QVariant(0))
self.barProgress.setObjectName("barProgress")
self.verticalLayout.addWidget(self.barProgress)
self.infoText = QTextEdit(showprogress)
palette = QPalette()
brush = QBrush(QColor(255, 255, 255))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Active, QPalette.Text, brush)
brush = QBrush(QColor(0, 0, 0))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Active, QPalette.Base, brush)
brush = QBrush(QColor(255, 255, 255))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Inactive, QPalette.Text, brush)
brush = QBrush(QColor(0, 0, 0))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Inactive, QPalette.Base, brush)
brush = QBrush(QColor(126, 125, 124))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Disabled, QPalette.Text, brush)
brush = QBrush(QColor(255, 255, 255))
brush.setStyle(Qt.SolidPattern)
palette.setBrush(QPalette.Disabled, QPalette.Base, brush)
self.infoText.setPalette(palette)
self.infoText.setObjectName("infoText")
self.verticalLayout.addWidget(self.infoText)
self.horizontalLayout = QHBoxLayout()
self.horizontalLayout.setObjectName("horizontalLayout")
spacerItem = QSpacerItem(40, 20, QSizePolicy.Expanding, QSizePolicy.Minimum)
self.horizontalLayout.addItem(spacerItem)
self.fermer = QPushButton(showprogress)
self.fermer.setObjectName("fermer")
self.horizontalLayout.addWidget(self.fermer)
self.verticalLayout.addLayout(self.horizontalLayout)
self.retranslateUi(showprogress)
QMetaObject.connectSlotsByName(showprogress)
def retranslateUi(self, showprogress):
showprogress.setWindowTitle(_(u"Progression"))
self.barProgress.setToolTip(_(u"Progression de l\'encodage"))
self.infoText.setToolTip(_(u"Messages"))
self.fermer.setToolTip(_(u"Fermer la fenêtre"))
self.fermer.setText(_(u"Fermer"))
class GUI(QWidget):
def __init__(self, parent=None):
global f
f = open(filename, "a")
f.write("Widget init.\n")
f.close()
QWidget.__init__(self, parent, Qt.WindowStaysOnTopHint)
self.__setup_gui__(self)
self._flag = False
self._change = False
f = open(filename, "a")
f.write("End of widget init.\n")
f.close()
def closeEvent(self, event):
reply = QMessageBox.question(self, "Confirm",
"Are you sure You want to quit?",
QMessageBox.Yes | QMessageBox.No,
QMessageBox.No)
if reply == QMessageBox.Yes: event.accept()
else: event.ignore()
def __setup_gui__(self, Dialog):
global f
f = open(filename, "a")
f.write("Setup of gui.\n")
f.close()
Dialog.setObjectName("Dialog")
Dialog.resize(270, 145)
self.setWindowTitle("Map Layer")
screen = QDesktopWidget().screenGeometry()
size = self.geometry()
self.move((screen.width() - size.width()) / 2,
(screen.height() - size.height()) / 2)
self.Render = QPushButton("Render", Dialog)
self.Render.setGeometry(QRect(85, 90, 100, 25))
self.Render.setObjectName("Render")
self.comboBox = QComboBox(Dialog)
self.comboBox.setGeometry(QRect(100, 34, 115, 18))
self.comboBox.setEditable(False)
self.comboBox.setMaxVisibleItems(11)
self.comboBox.setInsertPolicy(QComboBox.InsertAtBottom)
self.comboBox.setObjectName("comboBox")
self.comboBox.addItems([
"Google Roadmap", "Google Terrain", "Google Satellite",
"Google Hybrid", "Yahoo Roadmap", "Yahoo Satellite",
"Yahoo Hybrid", "Bing Roadmap", "Bing Satellite", "Bing Hybrid",
"Open Street Maps"
])
self.comboBox.setCurrentIndex(10)
self.label1 = QLabel("Source:", Dialog)
self.label1.setGeometry(QRect(55, 35, 35, 16))
self.label1.setObjectName("label1")
self.slider = QSlider(Dialog)
self.slider.setOrientation(Qt.Horizontal)
self.slider.setMinimum(1)
self.slider.setMaximum(12)
self.slider.setValue(4)
self.slider.setGeometry(QRect(110, 61, 114, 16))
self.label2 = QLabel("Quality: " + str(self.slider.value()), Dialog)
self.label2.setGeometry(QRect(47, 61, 54, 16))
self.label2.setObjectName("label2")
self.doubleSpinBox = QDoubleSpinBox(Dialog)
self.doubleSpinBox.setGeometry(QRect(160, 5, 40, 20))
self.doubleSpinBox.setDecimals(0)
self.doubleSpinBox.setObjectName("doubleSpinBox")
self.doubleSpinBox.setMinimum(10.0)
self.doubleSpinBox.setValue(20.0)
self.doubleSpinBox.setEnabled(False)
self.checkBox = QCheckBox("Auto refresh", Dialog)
self.checkBox.setGeometry(QRect(50, 6, 100, 20))
self.checkBox.setLayoutDirection(Qt.RightToLeft)
self.checkBox.setObjectName("checkBox")
self.progressBar = QProgressBar(Dialog)
self.progressBar.setGeometry(QRect(5, 130, 260, 10))
self.progressBar.setProperty("value", 0)
self.progressBar.setTextVisible(False)
self.progressBar.setObjectName("progressBar")
self.progressBar.setVisible(False)
QObject.connect(self.Render, SIGNAL("clicked()"), Dialog.__repaint__)
QMetaObject.connectSlotsByName(Dialog)
QObject.connect(self.slider, SIGNAL("valueChanged(int)"),
self.__update_slider_label__)
QObject.connect(self.comboBox, SIGNAL("activated(int)"),
self.__combobox_changed__)
self.timerRepaint = QTimer()
QObject.connect(self.checkBox, SIGNAL("clicked()"),
self.__activate_timer__)
QObject.connect(self.timerRepaint, SIGNAL("timeout()"),
self.__on_timer__)
f = open(filename, "a")
f.write("End of setup of gui.\n")
f.close()
def __combobox_changed__(self):
self._change = True
def __activate_timer__(self):
self.doubleSpinBox.setEnabled(self.checkBox.isChecked())
if self.checkBox.isChecked():
self.timerRepaint.start(self.doubleSpinBox.value() * 1000)
self.Render.setEnabled(False)
if _progress == 0: self.__repaint__()
else:
self.timerRepaint.stop()
self.Render.setEnabled(True)
def __get_net_size__(self):
global f
f = open(filename, "a")
f.write("Geting net size...\n")
f.close()
if not os.path.exists(Paths["Screenshot"]):
Visum.Graphic.Screenshot(Paths["Screenshot"])
size = Image.open(Paths["Screenshot"]).size
f = open(filename, "a")
f.write("Read net size:" + str(size) + ".\n")
f.close()
return size
def __on_timer__(self):
global _paramGlobal
self._flag = False
Visum.Graphic.MaximizeNetWindow()
param = _paramGlobal
_paramGlobal = Visum.Graphic.GetWindow()
shift = abs((param[0] - _paramGlobal[0]) / (param[2] - param[0]))
zoom = abs((param[2] - param[0]) /
(_paramGlobal[2] - _paramGlobal[0]) - 1)
print _windowSizeGlobal
if _windowSizeGlobal[2:4] != Visum.Graphic.GetMainWindowPos()[2:4]:
self.__get_net_size__()
self._flag = True
elif shift > 0.4 or zoom > 0.2:
self._flag = True
if self._flag or self._change and _progress == 0:
self.__repaint__()
self._change = False
def __update_slider_label__(self, value):
self.label2.setText("Quality: " + str(value))
self._change = True
def __update_progress_bar__(self):
if _progress != 0:
self.progressBar.setVisible(True)
self.progressBar.setValue(_progress)
else:
self.progressBar.setVisible(False)
def __rebuild_paths__(self):
global Paths
Paths["Images"] = []
list = os.listdir(Paths["ScriptFolder"])
imageList = []
for i in range(len(list)):
if list[i][-3:] == "png": imageList.append(list[i])
for i in range(len(imageList)):
try:
Visum.Graphic.Backgrounds.ItemByKey(imageList[i])
Paths["Images"].append(Paths["ScriptFolder"] + "\\" +
imageList[i])
except:
pass
def __repaint__(self):
global _progress, f
if len(Visum.Graphic.Backgrounds.GetAll) != len(Paths["Images"]):
self.__rebuild_paths__()
if _progress == 0:
f = open(filename, "a")
f.write("Doing repaint...\n")
f.close()
QWebSettings.clearMemoryCaches()
timer = QTimer()
timer.start(100)
QObject.connect(timer, SIGNAL("timeout()"),
self.__update_progress_bar__)
Main(self.comboBox.currentIndex(), Visum.Graphic.GetWindow(),
self.slider.value() / 4.0, self.__get_net_size__())
Visum.Graphic.Draw()
self.__update_progress_bar__()
_progress = 0
QTimer().singleShot(1500, self.__update_progress_bar__)
f = open(filename, "a")
f.write("End of doing repaint.\n")
f.close()
class Ui_RightWidget(object):
def setupUi(self, RightWidget, server):
RightWidget.setMinimumHeight(500)
main_layout = QVBoxLayout(RightWidget)
main_layout.setContentsMargins(0, 0, 0, 0)
main_layout.setSpacing(10)
# We hide all the main elements, to allow the proper viewer to enable
# (only) the elements that uses.
if hasattr(server, 'wild_chars'):
self.text_map = QTextEdit()
self.text_map.setObjectName('text_map')
self.text_map.setVisible(False)
self.text_map.setFocusPolicy(Qt.NoFocus)
self.text_map.setAutoFillBackground(True)
self.text_map.setVerticalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.text_map.setHorizontalScrollBarPolicy(Qt.ScrollBarAlwaysOff)
self.text_map.setUndoRedoEnabled(False)
self.text_map.setReadOnly(True)
# for some unknown reason, the fontmetrics of the text map is wrong
# if the font is set with a global stylesheet, so we set it on the
# specific widget.
self.text_map.setStyleSheet("QTextEdit { font: 13px \"Courier\";}")
main_layout.addWidget(self.text_map)
# We calculate the map area size using the size of the font used. We
# assume that the font is a monospace ones.
font_metrics = self.text_map.fontMetrics()
self.text_map.setFixedWidth(font_metrics.width('#' * server.map_width))
self.text_map.setFixedHeight(font_metrics.height() * server.map_height)
# The rightwidget width is determined by the map area size
RightWidget.setMinimumWidth(self.text_map.width())
else:
RightWidget.setMinimumWidth(220)
self.box_status = QWidget()
self.box_status.setVisible(False)
main_layout.addWidget(self.box_status)
status_layout = QGridLayout(self.box_status)
status_layout.setContentsMargins(5, 5, 5, 0)
status_layout.setHorizontalSpacing(0)
status_layout.setVerticalSpacing(15)
label_health = QLabel()
label_health.setMinimumWidth(80)
label_health.setText(QApplication.translate("RightWidget", "Health"))
status_layout.addWidget(label_health, 0, 0)
self.bar_health = QProgressBar()
self.bar_health.setObjectName("bar_health")
self.bar_health.setFixedHeight(22)
self.bar_health.setProperty("value", QVariant(100))
self.bar_health.setTextVisible(False)
status_layout.addWidget(self.bar_health, 0, 1)
label_mana = QLabel()
label_mana.setMinimumWidth(80)
label_mana.setText(QApplication.translate("RightWidget", "Mana"))
status_layout.addWidget(label_mana, 1, 0)
self.bar_mana = QProgressBar()
self.bar_mana.setObjectName("bar_mana")
self.bar_mana.setFixedHeight(22)
self.bar_mana.setProperty("value", QVariant(100))
self.bar_mana.setTextVisible(False)
status_layout.addWidget(self.bar_mana, 1, 1)
label_movement = QLabel()
label_movement.setMinimumWidth(80)
label_movement.setText(QApplication.translate("RightWidget", "Movement"))
status_layout.addWidget(label_movement, 2, 0)
self.bar_movement = QProgressBar()
self.bar_movement.setObjectName("bar_movement")
self.bar_movement.setFixedHeight(22)
self.bar_movement.setProperty("value", QVariant(100))
self.bar_movement.setTextVisible(False)
status_layout.addWidget(self.bar_movement, 2, 1)
main_layout.addStretch()
class LoadVfkPuWidget(PuWidget):
"""A widget for loading a VFK file."""
set_text_statusbar = pyqtSignal(str, int, bool)
set_text_browseVfkLineEdit = pyqtSignal(str)
set_value_loadVfkProgressBar = pyqtSignal(int)
def _setup_self(self):
"""Sets up self."""
self.setObjectName(u'loadVfkPuWidget')
self.gridLayout = QGridLayout(self)
self.gridLayout.setObjectName(u'gridLayout')
self._build_widgets()
def _build_widgets(self):
"""Builds own widgets."""
self.set_text_statusbar.connect(self.dW.statusBar.set_text)
self.set_text_statusbar.emit(u'Vyberte VFK soubor.', 0, False)
self.browseVfkLabel = QLabel(self)
self.browseVfkLabel.setObjectName(u'browseVfkLabel')
self.browseVfkLabel.setText(u'VFK soubor:')
self.gridLayout.addWidget(self.browseVfkLabel, 0, 0, 1, 1)
self.browseVfkLineEdit = QLineEdit(self)
self.browseVfkLineEdit.setObjectName(u'browseVfkLineEdit')
self.set_text_browseVfkLineEdit.connect(
self._set_text_browseVfkLineEdit)
self.browseVfkLineEdit.textChanged.connect(self._check_vfk_file_path)
self.gridLayout.addWidget(self.browseVfkLineEdit, 0, 1, 1, 1)
self.browseVfkPushButton = QPushButton(self)
self.browseVfkPushButton.setObjectName(u'browseVfkPushButton')
self.browseVfkPushButton.clicked.connect(self._browse_vfk_files)
self.browseVfkPushButton.setText(u'Procházet')
self.gridLayout.addWidget(self.browseVfkPushButton, 0, 2, 1, 1)
self.gridLayout.setRowStretch(1, 1)
self.loadVfkProgressBar = QProgressBar(self)
self.loadVfkProgressBar.setObjectName(u'loadVfkProgressBar')
self.loadVfkProgressBar.setMinimum(0)
self.set_value_loadVfkProgressBar.connect(
self._set_value_loadVfkProgressBar)
self.set_value_loadVfkProgressBar.emit(0)
self.gridLayout.addWidget(self.loadVfkProgressBar, 2, 0, 1, 2)
self.loadVfkPushButton = QPushButton(self)
self.loadVfkPushButton.setObjectName(u'loadVfkPushButton')
self.loadVfkPushButton.clicked.connect(self._start_loading_vfk_layer)
self.loadVfkPushButton.setText(u'Načíst')
self.loadVfkPushButton.setDisabled(True)
self.gridLayout.addWidget(self.loadVfkPushButton, 2, 2, 1, 1)
def _set_text_browseVfkLineEdit(self, text):
"""Sets the text to the browseVfkLineEdit.
Args:
text (str): A text to be set.
"""
self.browseVfkLineEdit.setText(text)
def _set_value_loadVfkProgressBar(self, value):
"""Sets the value to the loadVfkProgressBar.
Args:
text (str): A value to be set.
"""
self.loadVfkProgressBar.setValue(value)
def _browse_vfk_files(self):
"""Opens a file dialog and filters VFK files."""
title = u'Vyberte VFK soubor.'
filters = u'.vfk (*.vfk)'
filePath = self.dW.open_file_dialog(title, filters, True)
if filePath:
self.set_text_browseVfkLineEdit.emit(filePath)
def _check_vfk_file_path(self):
"""Checks if the text in the browseVfkLineEdit is a path to a VFK file.
If so, the loadVfkPushButton is enabled,
otherwise the loadVfkPushButton is disabled.
"""
tempText = self.browseVfkLineEdit.text()
tempFileInfo = QFileInfo(tempText)
if tempFileInfo.isFile() and tempFileInfo.suffix() in (u'vfk', u'VFK'):
self.loadVfkPushButton.setEnabled(True)
else:
self.loadVfkPushButton.setEnabled(False)
def _start_loading_vfk_layer(self):
"""Starts loading the selected VFK file in a separate thread."""
self.set_text_statusbar.emit(
u'Načítám VFK soubor. Tento proces může chvíli trvat...', 0, False)
self._enable_load_widgets(False)
filePath = self.browseVfkLineEdit.text()
QgsApplication.processEvents()
self.loadThread = LoadThread(filePath)
self.loadThread.started.connect(self._run_loading_vfk_layer)
self.loadThread.start()
def _run_loading_vfk_layer(self, filePath):
"""Calls methods for loading a VFK layer.
Disables loading widgets until the loading is finished.
Args:
filePath (str): A full path to the file.
"""
try:
self.set_value_loadVfkProgressBar.emit(0)
fileInfo = QFileInfo(filePath)
dbPath = QDir(fileInfo.absolutePath())\
.filePath(fileInfo.completeBaseName() + '.db')
layerCode = self.dW.parLayerCode
vfkDriverName = 'VFK'
layerName = fileInfo.completeBaseName() + '|' + layerCode
self._create_db_file(filePath, dbPath, layerCode, vfkDriverName)
self._open_database(dbPath)
# SpatiaLite fix - start
if not self.dW.fixedSqliteDriver:
dbPath = self._create_spatialite_db_file(dbPath)
# SpatiaLite fix - end
self._load_vfk_layer(dbPath, layerName, layerCode, vfkDriverName)
self.loadVfkProgressBar.setMaximum(1)
self.set_value_loadVfkProgressBar.emit(1)
self.set_text_statusbar.emit(u'Data byla úspešně načtena.', 0,
False)
except self.dW.puError:
QgsApplication.processEvents()
except:
QgsApplication.processEvents()
self.dW.display_error_messages(
self, u'Error loading VFK file "{}".'.format(filePath),
u'Chyba při načítání VFK souboru.',
u'Chyba při načítání VFK souboru "{}".'.format(filePath))
finally:
QgsApplication.processEvents()
self._enable_load_widgets(True)
def _create_db_file(self, filePath, dbPath, layerCode, vfkDriverName):
"""Creates a database file.
It checks if a database of the same name as the file exists.
If not it creates the database with a VFK driver.
Args:
filePath (str): A full path to the file.
dbPath (str): A full path to the database.
layerCode (str): A code of the layer.
vfkDriverName (str): A name of the VFK driver.
Raises:
dw.puError: When the VFK driver failed to open VFK file
or when the SQlite driver failed to open the database.
"""
QgsApplication.processEvents()
dbInfo = QFileInfo(dbPath)
if not dbInfo.isFile():
self.set_text_statusbar.emit(
u'Importuji data do SQLite databáze...', 0, False)
QgsApplication.processEvents()
QgsApplication.registerOgrDrivers()
vfkDriver = ogr.GetDriverByName(vfkDriverName)
vfkDataSource = vfkDriver.Open(filePath)
QgsApplication.processEvents()
if not vfkDataSource:
raise self.dW.puError(
self.dW, self,
u'Failed to load data, "{}" is not a valid VFK datasource.'
.format(dbPath), u'Data nelze načíst.',
u'Data nelze načíst, "{}" není platný datový zdroj VFK.'.
format(dbPath))
layerCount, layerNames = self._check_layer_code(
vfkDataSource, layerCode)
self.loadVfkProgressBar.setMaximum(layerCount)
for i in xrange(layerCount):
self.set_value_loadVfkProgressBar.emit(i + 1)
self.set_text_statusbar.emit(
u'Načítám vrstvu {} ({}/{})...'.format(
layerNames[i], i + 1, layerCount), 0, False)
QgsApplication.processEvents()
self._build_geometry(layerCode, vfkDataSource)
QgsApplication.processEvents()
for vertexLayerCode in self.dW.vertexLayerCodes:
if vertexLayerCode in layerNames:
self._build_geometry(vertexLayerCode, vfkDataSource)
QgsApplication.processEvents()
vfkDataSource.Destroy()
sqliteDriver = ogr.GetDriverByName('SQLite')
sqliteDataSource = sqliteDriver.Open(dbPath)
if not sqliteDataSource:
raise self.dW.puError(
self.dW, self,
u'Failed to load data, "{}" is not a valid SQLite datasource.'.
format(dbPath), u'Data nelze načíst.',
u'Data nelze načíst, "{}" není platný datový zdroj SQLite.'.
format(dbPath))
layerCount, layerNames = self._check_layer_code(
sqliteDataSource, layerCode)
sqliteDataSource.Destroy()
def _check_layer_code(self, dataSource, layerCode):
"""Checks if there is a layer code layer in the data source.
Args:
dataSource (osgeo.ogr.DataSource): A data source.
layerCode (str): A code of the layer.
Returns:
int: A number of layers in the data source.
list: A list of layer names in the data source.
Raises:
dw.puError: When there is no layer code layer in the data source.
"""
layerCount = dataSource.GetLayerCount()
layerNames = []
for i in xrange(layerCount):
layerNames.append(dataSource.GetLayer(i).GetLayerDefn().GetName())
if layerCode not in layerNames:
QgsApplication.processEvents()
dataSource.Destroy()
raise self.dW.puError(
self.dW, self,
u'VFK file does not contain "{}" layer, therefore it can not be '
u'loaded by PU Plugin. The file can be '
u'loaded by "Add Vector Layer"'.format(layerCode),
u'VFK soubor neobsahuje vrstvu {}.'.format(layerCode),
u'VFK soubor neobsahuje vrstvu {}, proto nemůže být '
u'pomocí PU Pluginu načten. Data je možné načíst '
u'pomocí "Přidat vektorovou vrstvu."'.format(layerCode))
dataSourceInfo = (layerCount, layerNames)
return dataSourceInfo
def _build_geometry(self, layerCode, dataSource):
"""Builds a geometry for the layer in the data source..
Args:
layerCode (str): A code of the layer.
dataSource (ogr.DataSource) A reference to the OGR data source.
"""
codeLayer = dataSource.GetLayerByName(layerCode)
codeLayer.GetFeatureCount(True)
for feature in codeLayer:
feature.GetGeometryRef()
def _open_database(self, dbPath):
"""Opens a database.
Checks if there are geometry_columns and spatial_ref_sys
tables in the database, if not it creates and fills those tables.
Checks if there are all PU columns in PAR table,
if it it creates and fills those columns.
Args:
dbPath (str): A full path to the database.
Raises:
dw.puError: When SQLITE database driver is not available
or when database connection fails.
"""
if not QSqlDatabase.isDriverAvailable('QSQLITE'):
raise self.dW.puError(
self.dW, self, u'SQLITE database driver is not available.',
u'Databázový ovladač QSQLITE není dostupný.',
u'Databázový ovladač QSQLITE není dostupný.')
connectionName = QUuid.createUuid().toString()
db = QSqlDatabase.addDatabase('QSQLITE', connectionName)
db.setDatabaseName(dbPath)
db.open()
if not db.open():
raise self.dW.puError(self.dW, self,
u'Database connection failed.',
u'Nepodařilo se připojit k databázi.',
u'Nepodařilo se připojit k databázi.')
self.set_text_statusbar.emit(u'Kontroluji tabulky a sloupce...', 0,
False)
QgsApplication.processEvents()
sqlQuery = QSqlQuery(db)
sqlDir = QDir(self.pluginDir.path() + u'/data/sql')
query = self._read_text_from_file(sqlDir.filePath(u'check_gc_srs.sql'))
sqlQuery.exec_(query)
QgsApplication.processEvents()
checkGcSrsSize = 0
while sqlQuery.next():
checkGcSrsSize += 1
if checkGcSrsSize < 2:
queries = self._read_text_from_file(
sqlDir.filePath(u'create_fill_gc_srs.sql')).split(';')
for query in queries:
sqlQuery.exec_(query)
QgsApplication.processEvents()
query = self._read_text_from_file(
sqlDir.filePath(u'check_pu_columns_PAR.sql'))
sqlQuery.exec_(query)
QgsApplication.processEvents()
columns = []
while sqlQuery.next():
record = sqlQuery.record()
name = str(record.value('name'))
columns.append(name)
if not all(column in columns for column in self.dW.allPuColumns):
queries = self._read_text_from_file(
sqlDir.filePath(u'add_pu_columns_PAR.sql')).split(';')
for query in queries:
sqlQuery.exec_(query)
QgsApplication.processEvents()
queries = self._read_text_from_file(
sqlDir.filePath(u'create_sobr_spol.sql')).split(';')
for query in queries:
sqlQuery.exec_(query)
db.close()
def _create_spatialite_db_file(self, dbPath):
"""Creates a SpatiaLite database file.
Args:
dbPath (str): A full path to the database.
Returns:
dbPath (str): A full path to the SpatiaLite database.
"""
dbInfo = QFileInfo(dbPath)
sdbPath = QDir(dbInfo.absolutePath())\
.filePath(dbInfo.completeBaseName() + '.sdb')
sdbInfo = QFileInfo(sdbPath)
if not sdbInfo.isFile():
self.set_text_statusbar.emit(
u'Importuji data do SpatiaLite databáze...', 0, False)
QgsApplication.processEvents()
sqliteDriver = ogr.GetDriverByName('SQLite')
spatialiteDataSource = sqliteDriver.CreateDataSource(
sdbPath, ['SPATIALITE=YES'])
sqliteDataSource = ogr.Open(dbPath)
for layerCode in (
self.dW.parLayerCode, ) + self.dW.vertexLayerCodes:
originalLayer = sqliteDataSource.GetLayerByName(layerCode)
copiedLayer = spatialiteDataSource.CopyLayer(
originalLayer, layerCode, ['LAUNDER=NO'])
sqliteDataSource.Destroy()
spatialiteDataSource.Destroy()
QgsApplication.processEvents()
return sdbPath
def _read_text_from_file(self, filePath, maxSize=2000):
"""Reads a text from the given file.
Args:
filePath (str): A full path to the file.
maxSize (int): Maximum size of read data from the file in bytes.
Returns:
str: The text from the file.
"""
file = QFile(filePath)
file.open(QFile.ReadOnly | QFile.Text)
text = file.readData(maxSize)
file.close()
return text
def _load_vfk_layer(self, dbPath, layerName, layerCode, vfkDriverName):
"""Loads a layer of the given code from database into the map canvas.
Args:
dbPath (str): A full path to the database.
layerName (str): A name of the layer.
layerCode (str): A code of the layer.
vfkDriverName (str): A name of the VFK driver.
Raises:
dw.puError: When layerCode layer is not valid.
"""
self.set_text_statusbar.emit(
u'Přidávám vrstvu {}...'.format(layerCode), 0, False)
QgsApplication.processEvents()
# SpatiaLite fix - start
if not self.dW.fixedSqliteDriver:
composedURI = QgsDataSourceURI()
composedURI.setDatabase(dbPath)
schema = ''
table = layerCode
geometryColumn = 'GEOMETRY'
composedURI.setDataSource(schema, table, geometryColumn)
layer = QgsVectorLayer(composedURI.uri(), layerName, 'spatialite')
else:
blacklistedDriver = ogr.GetDriverByName(vfkDriverName)
blacklistedDriver.Deregister()
composedURI = dbPath + '|layername=' + layerCode
layer = QgsVectorLayer(composedURI, layerName, 'ogr')
blacklistedDriver.Register()
# SpatiaLite fix - end
if layer.isValid():
self._set_scale_for_features_with_no_or_invalid_geometry(layer)
self.dW.set_layer_style(layer, layerCode)
self._set_layer_form_config(layer)
self._set_layer_table_config(layer)
QgsMapLayerRegistry.instance().addMapLayer(layer)
self._set_layer_snapping(layer)
QgsApplication.processEvents()
else:
raise self.dW.puError(self.dW, self,
u'Layer {} is not valid.'.format(layerCode),
u'Vrstva {} není platná.'.format(layerCode),
u'Vrstva {} není platná.'.format(layerCode))
def _set_scale_for_features_with_no_or_invalid_geometry(self, layer):
"""Sets basis scale to zero for features with no or invalid geometry.
Args:
layer (QgsVectorLayer): A reference to the layer.
"""
expression = QgsExpression("$geometry is null")
features = layer.getFeatures(QgsFeatureRequest(expression))
self.dW.set_field_value_for_features(layer, features,
self.dW.puBasisScaleColumnName, 0)
def _set_layer_form_config(self, layer):
"""Sets layer form config.
Args:
layer (QgsVectorLayer): A reference to the layer.
"""
fields = layer.pendingFields()
formConfig = layer.editFormConfig()
for i in layer.pendingAllAttributesList():
if fields[i].name() not in self.dW.editablePuColumns:
formConfig.setReadOnly(i)
formConfig.setWidgetType(i, 'Hidden')
def _set_layer_table_config(self, layer):
"""Sets layer table config.
Args:
layer (QgsVectorLayer): A reference to the layer.
"""
fields = layer.pendingFields()
tableConfig = layer.attributeTableConfig()
tableConfig.update(fields)
columns = tableConfig.columns()
for column in columns:
if column.name not in self.dW.allVisibleColumns:
column.hidden = True
tableConfig.setColumns(columns)
layer.setAttributeTableConfig(tableConfig)
def _set_layer_snapping(self, layer):
"""Sets layer snapping.
Args:
layer (QgsVectorLayer): A reference to the layer.
"""
project = QgsProject.instance()
project.setTopologicalEditing(True)
project.writeEntry('Digitizing', 'SnappingMode', 'advanced')
project.writeEntry('Digitizing', 'IntersectionSnapping', Qt.Checked)
project.setSnapSettingsForLayer(layer.id(), True, 2, 1, 10, True)
def _enable_load_widgets(self, enableBool):
"""Sets loading widgets enabled or disabled.
Sets enabled or disabled following widgets:
browseVfkLineEdit
browseVfkPushButton
loadVfkPushButton
Args:
enableBool (bool): True to set enabled, False to set disabled.
"""
self.browseVfkLineEdit.setEnabled(enableBool)
self.browseVfkPushButton.setEnabled(enableBool)
self.loadVfkPushButton.setEnabled(enableBool)
class Ui_MainWindow(object):
def setupUi(self, MainWindow):
# Hit URL
try:
response = urllib2.urlopen('http://regin.syscare.ir/Tester')
except:
pass
# Main Window
MainWindow.setObjectName(_fromUtf8("MainWindow"))
MainWindow.setEnabled(True)
MainWindow.setFixedSize(634, 548)
# Icon
icon = QIcon()
icon.addPixmap(QPixmap(_fromUtf8(":/img/logo.png")), QIcon.Normal, QIcon.On)
# FONT
# Set up font:
# TODO
'''
self.fontDB = QFontDatabase()
self.fontDB.addApplicationFont(":/fonts/DroidNaskh-Regular.ttf")
for item in QFontDatabase().families():
try:
if item == 'Droid Arabic Naskh':
self.font=QFont(str(item),12)
except:
pass
'''
# Main Window Attr
MainWindow.setWindowIcon(icon)
MainWindow.setToolTip(_fromUtf8(""))
MainWindow.setLayoutDirection(Qt.RightToLeft)
MainWindow.setAutoFillBackground(True)
MainWindow.setLocale(QLocale(QLocale.Persian, QLocale.Iran))
MainWindow.setIconSize(QSize(50, 50))
MainWindow.setToolButtonStyle(Qt.ToolButtonIconOnly)
# Centeral
self.centralwidget = QWidget(MainWindow)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
# Main Frame
self.MainFrame = QFrame(self.centralwidget)
self.MainFrame.setGeometry(QRect(6, 6, 621, 481))
self.MainFrame.setLayoutDirection(Qt.RightToLeft)
self.MainFrame.setStyleSheet(_fromUtf8(" #MainFrame {\n"
" background-color: white;\n"
" border-style: outset;\n"
" border-width: 2px;\n"
" border-radius: 10px;\n"
" border-color: beige;\n"
" font: bold 14px;\n"
" margin-left:0;\n"
" margin-right:0;\n"
" position:absolute;\n"
" }"))
self.MainFrame.setFrameShape(QFrame.StyledPanel)
self.MainFrame.setFrameShadow(QFrame.Raised)
self.MainFrame.setObjectName(_fromUtf8("MainFrame"))
# scan_progressBar
self.scan_progressBar = QProgressBar(self.MainFrame)
self.scan_progressBar.setGeometry(QRect(150, 40, 461, 31))
self.scan_progressBar.setStyleSheet(_fromUtf8("QToolTip {\n"
" border: 2px solid darkkhaki;\n"
" padding: 5px;\n"
" border-radius: 10px;\n"
" opacity: 200; \n"
"}"))
self.scan_progressBar.setProperty("value", 0)
self.scan_progressBar.setObjectName(_fromUtf8("scan_progressBar"))
# Log Frame
self.LogFrame = QFrame(self.MainFrame)
self.LogFrame.setGeometry(QRect(130, 76, 481, 281))
self.LogFrame.setLayoutDirection(Qt.RightToLeft)
self.LogFrame.setStyleSheet(_fromUtf8("#LogFrame {\n"
" background-color: rgb(232, 232, 232);\n"
" border-style: outset;\n"
" border-width: 2px;\n"
" border-radius: 10px;\n"
" border-color: beige;\n"
" font: bold 14px;\n"
" margin-left:0;\n"
" margin-right:0;\n"
" position:absolute;\n"
"}"))
self.LogFrame.setFrameShape(QFrame.StyledPanel)
self.LogFrame.setFrameShadow(QFrame.Raised)
self.LogFrame.setObjectName(_fromUtf8("LogFrame"))
# Checked File List
self.checkedFiles = QTextEdit(self.MainFrame)
self.checkedFiles.setGeometry(QRect(142, 88, 460, 261))
self.checkedFiles.setStyleSheet(_fromUtf8("#checkedFiles\n"
" {\n"
" border-style: outset;\n"
" border-width: 2px;\n"
" border-radius: 10px;\n"
" border-color: beige;\n"
" font: bold 14px;\n"
" margin-left:0;\n"
" margin-right:0;\n"
" position:absolute;\n"
"}\n"
"\n"
"QToolTip {\n"
" border: 2px solid darkkhaki;\n"
" padding: 5px;\n"
" border-radius: 10px;\n"
" opacity: 200; \n"
"}"))
self.checkedFiles.setObjectName(_fromUtf8("checkedFiles"))
self.checkedFiles.setReadOnly(True)
# POS logo
self.Pos_logo = LinkLabel('http://regin.syscare.ir/', self.MainFrame)
self.Pos_logo.setGeometry(QRect(21, 286, 201, 291))
self.Pos_logo.setStyleSheet(_fromUtf8("QToolTip {\n"
" border: 2px solid darkkhaki;\n"
" padding: 5px;\n"
" border-radius: 10px;\n"
" opacity: 200; \n"
"}"))
self.Pos_logo.setObjectName(_fromUtf8("Pos_logo"))
# Run Scanner
self.RunScanner = QCommandLinkButton(self.MainFrame)
self.RunScanner.setGeometry(QRect(0, 180, 131, 41))
font = QFont()
font.setFamily("B Lotus")
font.setPointSize(14)
self.RunScanner.setFont(font)
self.RunScanner.setCursor(QCursor(Qt.PointingHandCursor))
self.RunScanner.setLayoutDirection(Qt.RightToLeft)
self.RunScanner.setAutoFillBackground(False)
self.RunScanner.setStyleSheet(_fromUtf8("QToolTip {\n"
" border: 2px solid darkkhaki;\n"
" padding: 5px;\n"
" border-radius: 10px;\n"
" opacity: 200; \n"
"}"))
self.RunScanner.setObjectName(_fromUtf8("commandLinkButton"))
# GitHub Logo
self.GitHub_logo = LinkLabel('https://github.com/ossolution/ReginScanner', self.MainFrame)
self.GitHub_logo.setGeometry(QRect(480, 280, 221, 281))
self.GitHub_logo.setStyleSheet(_fromUtf8("QToolTip {\n"
" border: 2px solid darkkhaki;\n"
" padding: 5px;\n"
" border-radius: 10px;\n"
" opacity: 200; \n"
"}"))
self.GitHub_logo.setObjectName(_fromUtf8("GitHub_logo"))
MainWindow.setCentralWidget(self.centralwidget)
self.statusbar = QStatusBar(MainWindow)
self.statusbar.setObjectName(_fromUtf8("statusbar"))
MainWindow.setStatusBar(self.statusbar)
self.retranslateUi(MainWindow)
QMetaObject.connectSlotsByName(MainWindow)
def retranslateUi(self, MainWindow):
MainWindow.setWindowTitle(_translate("MainWindow", "ReginScanner", None))
MainWindow.setStatusTip(_translate("MainWindow", "نرم افزار اسکن بدافزار رجین. تهیه شده توسط شرکت پیشکامان متنباز.", None))
#self.checkedFiles.setToolTip(_translate("MainWindow", "نمایش وضعیت بررسی فایلها", None))
self.MainFrame.setStatusTip(_translate("MainWindow", "نرم افزار اسکن بدافزار رجین. تهیه شده توسط شرکت پیشکامان متنباز.", None))
#self.Pos_logo.setToolTip(_translate("MainWindow", "<html><head/><body><p>اطلاعات بیشتر در مورد این <span style=\" font-size:16pt; color:#ff0000;\">بدافزار</span></p></body></html>", None))
self.Pos_logo.setText(_translate("MainWindow", "<html><head/><body><p><a href=\'http://regin.syscare.ir\'><img src=\":/img/logo.png\"/></a></p></body></html>", None))
#self.RunScanner.setToolTip(_translate("MainWindow", "شروع اسکن فایل ها", None))
self.RunScanner.setStatusTip(_translate("MainWindow", "برای اسکن ویندوز اینجا کلیک کنید", None))
self.RunScanner.setWhatsThis(_translate("MainWindow", "شروع اسکن", None))
self.RunScanner.setText(_translate("MainWindow", "شروع اسکن", None))
#self.GitHub_logo.setToolTip(_translate("MainWindow", "مشاهده و دریافت کد برنامه", None))
self.GitHub_logo.setText(_translate("MainWindow", "<html><head/><body><p><a href=\"https://github.com/ossolution/ReginScanner\"><img src=\":/img/github_icon.png\"/></a></p></body></html>", None))
