def createTable(self, header, tabledata):
"""creates a table"""
gridlayout2 = QGridLayout(self)
gridlayout2.setObjectName("gridlayout2")
tableView = QTableView(self)
tableView.setObjectName("tableView")
gridlayout2.addWidget(tableView, 0, 0, 1, 1)
self.__popUp = QMenu(tableView)
self.__copyAction = QAction(self.tr("Copy data"), tableView)
self.connect(self.__copyAction, SIGNAL("triggered()"), self.copy)
self.__popUp.addAction(self.__copyAction)
tableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.connect(tableView, SIGNAL('customContextMenuRequested(QPoint)'),
self.popUpMenu)
# set the table model
tm = TableModel(tabledata, header, self)
tableView.setModel(tm)
tableView.setAlternatingRowColors(True)
# set the minimum size
self.setMinimumSize(400, 300)
# hide grid
tableView.setShowGrid(True)
# set the font
#font = QFont("Courier New", 12)
#self.tableView.setFont(font)
# hide vertical header
vh = tableView.verticalHeader()
vh.setVisible(True)
# set horizontal header properties
hh = tableView.horizontalHeader()
hh.setStretchLastSection(True)
# set column width to fit contents
tableView.resizeColumnsToContents()
tableView.setSortingEnabled(True)
tableView.sortByColumn(0, Qt.AscendingOrder)
# set row height
nrows = len(tabledata)
for row in xrange(nrows):
tableView.setRowHeight(row, 18)
class MobileWidget(QWidget):
"""
Mobile widget
"""
RefreshScreen = pyqtSignal()
RefreshAutomatic = pyqtSignal(bool)
TapOn = pyqtSignal(int, int)
def __init__(self, parent=None):
"""
Constructor
"""
super(MobileWidget, self).__init__(parent)
self.origWidth = 0
self.origHeight = 0
self.imagePath = None
self.createActions()
self.createWidget()
self.createToolbar()
self.center()
def createActions(self):
"""
Create qt actions
"""
self.refreshAction = QtHelper.createAction(self,
self.tr("&Refresh"),
self.refreshScreen,
icon=None)
self.refreshAction.setEnabled(False)
self.copyAction = QtHelper.createAction(self,
self.tr("&Copy"),
self.copyItem,
icon=None)
def createWidget(self):
"""
Create qt widget
"""
self.screenResolutionLabel = QLabel(self)
self.screenTapLabel = QLabel(self)
mobileLayout = QVBoxLayout()
self.mobileDockToolbar = QToolBar(self)
self.mobileDockToolbar.setStyleSheet("QToolBar { border: 0px }")
self.mobileDockToolbar.setToolButtonStyle(Qt.ToolButtonTextUnderIcon)
self.mobileImageLabel = QLabel(self)
self.mobileImageLabel.setMouseTracking(True)
self.mobileImageLabel.installEventFilter(self)
self.mobileImageLabel.setScaledContents(True)
self.mobileImageLabel.mousePressEvent = self.pixelSelect
self.refreshCheckbox = QCheckBox("Automatic Refresh", self)
self.refreshCheckbox.setEnabled(False)
self.refreshCheckbox.stateChanged.connect(self.onRefreshChanged)
self.clickCheckbox = QCheckBox("Enable Tap", self)
self.clickCheckbox.setEnabled(False)
self.model = DomModel(QDomDocument(), self)
self.mobileTreeView = QTreeView(self)
self.mobileTreeView.setMinimumWidth(300)
self.mobileTreeView.setModel(self.model)
self.mobileTreeView.clicked.connect(self.onTreeViewClicked)
header = ["Attribute", "Value"]
self.tableModel = MyTableModel(self, [], header)
self.mobileTableView = QTableView(self)
self.mobileTableView.setSelectionMode(
QAbstractItemView.SingleSelection)
self.mobileTableView.setModel(self.tableModel)
self.mobileTableView.setContextMenuPolicy(Qt.CustomContextMenu)
self.mobileTableView.customContextMenuRequested.connect(
self.onContextMenuEvent)
self.mobileTableView.setMinimumWidth(300)
mobileViewLayout = QHBoxLayout()
mobileViewLayout.addWidget(self.mobileImageLabel)
mobileViewLayout.addWidget(self.mobileTreeView)
mobileViewLayout.addWidget(self.mobileTableView)
mobileLayout.addWidget(self.mobileDockToolbar)
mobileLayout.addLayout(mobileViewLayout)
self.setLayout(mobileLayout)
def createToolbar(self):
"""
Create qt toolbar
"""
self.mobileDockToolbar.setObjectName("Toolbar")
self.mobileDockToolbar.addWidget(self.refreshCheckbox)
self.mobileDockToolbar.addWidget(self.clickCheckbox)
self.mobileDockToolbar.addSeparator()
self.mobileDockToolbar.addAction(self.refreshAction)
self.mobileDockToolbar.addSeparator()
self.mobileDockToolbar.addWidget(self.screenResolutionLabel)
self.mobileDockToolbar.addSeparator()
self.mobileDockToolbar.addWidget(self.screenTapLabel)
self.mobileDockToolbar.addSeparator()
self.mobileDockToolbar.setIconSize(QSize(16, 16))
def center(self):
"""
Center the dialog
"""
qr = self.frameGeometry()
cp = QDesktopWidget().availableGeometry().center()
qr.moveCenter(cp)
self.move(qr.topLeft())
def eventFilter(self, srcEvent, event):
"""
On event filtering
"""
if srcEvent == self.mobileImageLabel:
if event.type() == QEvent.MouseMove:
x = event.pos().x()
y = event.pos().y()
pixmap = self.mobileImageLabel.pixmap()
if pixmap is not None:
x_scaled = int((self.origWidth * x) / pixmap.width())
y_scaled = int((self.origHeight * y) / pixmap.height())
self.mobileImageLabel.setToolTip("%sx%s" %
(x_scaled, y_scaled))
return False
def onContextMenuEvent(self, event):
"""
On context menu event
"""
menu = QMenu(self)
menu.addAction(self.copyAction)
menu.popup(QCursor.pos())
def copyItem(self):
"""
Copy the item
"""
indexes = self.mobileTableView.selectedIndexes()
if len(indexes):
data = self.tableModel.mylist[indexes[0].row()][
indexes[0].column()]
clipboard = QApplication.clipboard()
clipboard.setText(data)
def onTreeViewClicked(self, qindex):
"""
On click in the treeview
"""
item = qindex.internalPointer()
attributes = []
node = item.node()
attributeMap = node.attributes()
nodeName = node.nodeName()
bounds_str = None
for i in range(0, attributeMap.count()):
attribute = attributeMap.item(i)
attributes.append((attribute.nodeName(), attribute.nodeValue()))
if attribute.nodeName() == 'bounds':
bounds_str = attribute.nodeValue()
self.tableModel.mylist = attributes
if sys.version_info > (3, ):
self.tableModel.beginResetModel()
self.tableModel.endResetModel()
else:
self.tableModel.reset()
self.mobileTableView.resizeColumnsToContents()
self.mobileTableView.resizeRowsToContents()
# redraw image with rectangle
if bounds_str is not None:
xy = bounds_str.split('][')[0].split('[')[1]
wh = bounds_str.split('][')[1].split(']')[0]
x, y = xy.split(',')
w, h = wh.split(',')
# get label size
pixmap = self.mobileImageLabel.pixmap()
xlabel = pixmap.width()
ylabel = pixmap.height()
# resize the rectangle
y_scaled = (pixmap.height() * int(y)) / self.origHeight
x_scaled = (pixmap.width() * int(x)) / self.origWidth
h_scaled = (pixmap.height() * (int(h) - int(y))) / self.origHeight
w_scaled = (pixmap.width() * (int(w) - int(x))) / self.origWidth
# finally reload
self.reloadScreen(x=int(x_scaled),
y=int(y_scaled),
w=int(w_scaled),
h=int(h_scaled))
def onDeviceReady(self):
"""
On device ready
"""
self.refreshAction.setEnabled(True)
self.refreshCheckbox.setEnabled(True)
self.clickCheckbox.setEnabled(True)
def refreshScreen(self):
"""
Refresh the screen
"""
self.RefreshScreen.emit()
def onRefreshChanged(self, state):
"""
On refresh changed
"""
if state == Qt.Checked:
self.RefreshAutomatic.emit(True)
else:
self.RefreshAutomatic.emit(False)
def pixelSelect(self, event):
"""
Select pixel to click
"""
position = QPoint(event.pos().x(), event.pos().y())
x = event.pos().x()
y = event.pos().y()
pixmap = self.mobileImageLabel.pixmap()
x_scaled = int((self.origWidth * x) / pixmap.width())
y_scaled = int((self.origHeight * y) / pixmap.height())
self.screenTapLabel.setText("Tap on (%s,%s)" % (x_scaled, y_scaled))
if self.clickCheckbox.isChecked():
self.TapOn.emit(x_scaled, y_scaled)
def drawRectangle(self, x=0, y=0, w=0, h=0):
"""
Draw a rectangle
"""
self.mobileImageLabel.update()
pixmap = self.mobileImageLabel.pixmap()
if pixmap is not None:
p = QPainter(pixmap)
pen = QPen(Qt.red, 2, Qt.SolidLine)
p.setPen(pen)
p.drawRect(x, y, w, h)
p.end()
def reloadScreen(self, x, y, w, h):
"""
Reload the screen
"""
if self.imagePath is not None:
self.updateScreen(filename=self.imagePath,
xmlPath='',
x=x,
y=y,
w=w,
h=h,
reloadMode=True)
def updateScreen(self,
filename,
xmlPath,
x=0,
y=0,
w=0,
h=0,
reloadMode=False):
"""
Update the screen
"""
self.imagePath = filename
if not reloadMode:
self.tableModel.mylist = []
self.tableModel.beginResetModel()
self.tableModel.endResetModel()
pixmap = QPixmap(filename)
if pixmap is not None:
self.origWidth = pixmap.width()
self.origHeight = pixmap.height()
self.screenResolutionLabel.setText(
"Resolution=%sx%s" % (self.origWidth, self.origHeight))
#portrait
if self.origWidth < self.origHeight:
pixmap = pixmap.scaledToHeight(Settings.getInt(
'MobileAndroid', 'resolution-screen-height'),
mode=Qt.SmoothTransformation)
self.mobileImageLabel.setPixmap(pixmap)
else:
pixmap = pixmap.scaledToWidth(Settings.getInt(
'MobileAndroid', 'resolution-screen-width'),
mode=Qt.SmoothTransformation)
self.mobileImageLabel.setPixmap(pixmap)
self.drawRectangle(x=x, y=y, w=w, h=h)
self.resize(pixmap.width(), pixmap.height())
# convert xml to dict
if len(xmlPath):
f = QFile(xmlPath)
if f.open(QIODevice.ReadOnly):
document = QDomDocument()
if document.setContent(f):
newModel = DomModel(document, self)
self.mobileTreeView.setModel(newModel)
self.mobileTreeView.expandAll()
self.mobileTreeView.resizeColumnToContents(0)
f.close()
class DictValueDialog(QtHelper.EnhancedQDialog, Logger.ClassLogger):
"""
Dict dialog
"""
def __init__(self, parent, testParams, variables, advancedMode=False ):
"""
Operator to fill parameter description
@param dataArgs:
@type dataArgs:
@param parent:
@type parent:
"""
super(DictValueDialog, self).__init__(parent)
self.advancedMode = advancedMode
self.testParams = testParams
self.variables = variables
self.createDialog()
self.createConnections()
self.createActions()
def createDialog (self):
"""
Create qt dialog
"""
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.setStyleSheet( """QDialogButtonBox {
dialogbuttonbox-buttons-have-icons: 1;
dialog-ok-icon: url(:/ok.png);
dialog-cancel-icon: url(:/test-close-black.png);
}""")
self.buttonBox.setStandardButtons(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
mainLayout = QVBoxLayout()
self.dictTable = QTableView(self)
self.model = DictTableModel(self, advancedMode=self.advancedMode)
self.dictTable.setModel(self.model)
self.dictTable.setFrameShape(QFrame.StyledPanel)
self.dictTable.setShowGrid(True)
self.dictTable.setGridStyle (Qt.DotLine)
self.dictTable.setSelectionMode(QAbstractItemView.ExtendedSelection)
self.dictTable.setSelectionBehavior(QAbstractItemView.SelectRows)
self.dictTable.setContextMenuPolicy(Qt.CustomContextMenu)
self.dictTable.verticalHeader().setVisible(False)
self.dictTable.horizontalHeader().setHighlightSections(False)
self.dictTable.setHorizontalScrollMode(QAbstractItemView.ScrollPerPixel)
self.dictTable.horizontalHeader().setStretchLastSection(True)
self.dictTable.setColumnWidth(COL_KEY, 200)
# delegate item on advanced mode
if self.advancedMode:
self.dictTable.setItemDelegateForColumn( COL_KEY, ItemComboDelegate(self, COL_KEY) )
self.dictTable.setItemDelegateForColumn( COL_VALUE, ItemComboDelegate(self, COL_VALUE) )
mainLayout.addWidget(self.dictTable)
mainLayout.addWidget(self.buttonBox)
self.setLayout(mainLayout)
self.setWindowTitle(self.tr("Dict configuration"))
self.setMinimumWidth(500)
self.center()
def getInputs(self):
"""
Get test inputs
"""
return self.testParams.parameters.table().model.getData()
def getOutputs(self):
"""
Get test outputs
"""
return self.testParams.parametersOutput.table().model.getData()
def createConnections (self):
"""
Create qt connections
"""
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.dictTable.customContextMenuRequested.connect(self.onPopupMenu)
def getValue(self):
"""
Return value
"""
return self.model.getData()
def createActions (self):
"""
Actions defined:
* add
* del
"""
self.addAction = QtHelper.createAction(self, self.tr("&Add"), self.addKey, icon = QIcon(":/test-parameter-add.png"),
tip = self.tr('Add a new key') )
self.delAction = QtHelper.createAction(self, self.tr("&Delete"), self.delKey, icon = QIcon(":/test-parameter-del.png"),
tip = self.tr('Delete the selected key') )
self.delAllAction = QtHelper.createAction(self, self.tr("&Delete All"), self.delKeys, icon = QIcon(":/test-parameter-del.png"),
tip = self.tr('Delete the selected key') )
def addKey(self):
"""
Add key
"""
self.delAllAction.setEnabled(True)
index = self.dictTable.currentIndex()
if not index.isValid():
row = self.model.rowCount()
else:
row = index.row()
data = self.model.getData()
if self.advancedMode:
tpl = { 'key': { 'operator': '', 'value': '', 'type': '' }, 'value': { 'operator': '', 'value': '', 'type': '' } }
else:
tpl = { 'key': 'my key', 'value': 'my value' }
# add data to model
data.insert(row + 1, tpl)
self.model.reset()
def clear (self):
"""
clear contents
"""
self.model.setDataModel( [] )
def delKey(self):
"""
Delete key
"""
# get selected proxy indexes
indexes = self.dictTable.selectedIndexes()
if not indexes:
return
if indexes:
answer = QMessageBox.question(self, self.tr("Remove"), self.tr("Do you want to remove selected key?"),
QMessageBox.Yes | QMessageBox.No)
if answer == QMessageBox.Yes:
self.removeValues(indexes)
def removeValues(self, indexes):
"""
Remove values from data
@param indexes:
@type indexes:
"""
if not indexes:
return
# extract name, name are unique
datas = self.model.getData()
allNames = []
# remove duplicate index
cleanIndexes = {}
for index in indexes:
if index.row() not in cleanIndexes:
cleanIndexes[index.row()] = index
#for cleanIndex in cleanIndexes.keys():
for cleanIndex in list(cleanIndexes.keys()): # for python3 support
allNames.append( datas[cleanIndex]['key'] )
self.trace('Key to remove: %s' % allNames)
for paramName in allNames:
self.removeValue( paramName=paramName )
def removeValue(self, paramName):
"""
Remove one parameter according to the name passed on argument
"""
datas = self.model.getData()
i = None
for i in xrange(len(datas)):
if datas[i]['key'] == paramName:
break
if i is not None:
param = datas.pop( i )
del param
self.model.reset()
def delKeys(self):
"""
Clear all keys
"""
reply = QMessageBox.question(self,
self.tr("Clear all keys"), self.tr("Are you sure ?"),
QMessageBox.Yes | QMessageBox.No
)
if reply == QMessageBox.Yes:
data = self.model.getData()
try:
for i in xrange(len(data)):
data.pop()
except Exception as e:
pass
self.model.reset()
self.delAllAction.setEnabled(False)
def onPopupMenu(self, pos):
"""
Display menu on right click
@param pos:
@type pos:
"""
self.menu = QMenu(self.dictTable)
index = self.dictTable.currentIndex()
indexes = self.dictTable.selectedIndexes()
if not indexes:
self.menu.addAction( self.delAction )
self.menu.addAction( self.addAction )
self.menu.addSeparator()
self.menu.addAction( self.delAllAction )
self.menu.addSeparator()
else:
self.menu.addAction( self.delAction )
self.menu.addAction( self.addAction )
self.menu.addSeparator()
self.menu.addAction( self.delAllAction )
self.menu.addSeparator()
self.menu.popup( self.mapToGlobal(pos) )
def loadData(self, data):
"""
Load data
"""
if isinstance(data, dict):
data = [data]
self.model.setDataModel( data )
