def startDrag(self, index):
"""start a drag operation with a PandasCellPayload on defined index.
Args:
index (QModelIndex): model index you want to start the drag operation.
"""
if not index.isValid():
return
dataFrame = self.model().dataFrame()
# get all infos from dataFrame
dfindex = dataFrame.iloc[[index.row()]].index
columnName = dataFrame.columns[index.column()]
dtype = dataFrame[columnName].dtype
value = dataFrame[columnName][dfindex]
# create the mime data
mimePayload = PandasCellPayload(dfindex, columnName, value, dtype,
hex(id(self.model())))
mimeData = MimeData()
mimeData.setData(mimePayload)
# create the drag icon and start drag operation
drag = QtGui.QDrag(self)
drag.setMimeData(mimeData)
pixmap = QtGui.QPixmap(":/icons/insert-table.png")
drag.setHotSpot(QtCore.QPoint(pixmap.width() / 3, pixmap.height() / 3))
drag.setPixmap(pixmap)
result = drag.start(Qt.MoveAction)
def setDataFrame(self, dataFrame):
self.df = dataFrame
dataModel = DataFrameModel()
dataModel.setDataFrame(self.df)
self.dataModel = dataModel
self.dataListView.setModel(dataModel)
self.dataTableView.setViewModel(dataModel)
self.dataComboBox.setModel(dataModel)
# self.dataTableView.resizeColumnsToContents()
# create a simple item model for our choosing combobox
columnModel = QtGui.QStandardItemModel()
for column in self.df.columns:
columnModel.appendRow(QtGui.QStandardItem(column))
self.chooseColumnComboBox.setModel(columnModel)
self.tableViewColumnDtypes.setModel(dataModel.columnDtypeModel())
self.tableViewColumnDtypes.horizontalHeader().setDefaultSectionSize(
200)
self.tableViewColumnDtypes.setItemDelegateForColumn(
1, DtypeComboDelegate(self.tableViewColumnDtypes))
dataModel.changingDtypeFailed.connect(self.changeColumnValue)
def initUI(self):
self.setGeometry(100, 100, 300, 300)
self.vlayout = QtGui.QVBoxLayout(self)
self.imgContainer = QtGui.QLabel(self)
img = QtGui.QPixmap(':/europe.png')
self.imgContainer.setPixmap(img)
size = img.size()
self.imgContainer.resize(size.width(), self.height())
self.vlayout.addWidget(self.imgContainer)
self.vlayout.addWidget(QtGui.QLabel('FOOO', self))
threads = []
worker1 = ExampleWorker('foo', 10)
worker2 = ExampleWorker('bar', 13)
worker3 = ExampleWorker('spam', 25)
workers = [worker1, worker2, worker3]
for worker in workers:
thread = createThread(self, worker)
threads.append(thread)
worker.finished.connect(self.debugPrint)
self.pgFrame = OverlayProgressWidget(self.imgContainer,
workers=workers)
for t in threads:
t.start()
def initUi(self):
self.sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.MinimumExpanding, QtGui.QSizePolicy.Expanding)
self._pbHeight = 30
self.setMinimumWidth(self._width)
#self.setMaximumWidth(self._width)
self.setMinimumHeight(self._minHeight)
self.glayout = QtGui.QGridLayout(self)
self.totalProgressBar = QtGui.QProgressBar(self)
self.totalProgressBar.setMinimumHeight(self._pbHeight)
self.totalProgressBar.setMaximumHeight(self._pbHeight)
self.toggleButton = QtGui.QPushButton('Details', self)
self.toggleButton.setCheckable(True)
self.toggleButton.toggled.connect(self.showDetails)
self.glayout.addWidget(self.totalProgressBar, 0, 0, 1, 1)
self.glayout.addWidget(self.toggleButton, 0, 1, 1, 1)
#styleSheet = """.QProgressBar {
#border: none;
#border-radius: 3px;
#text-align: center;
#background-color: rgba(37, 37, 37, 50%);
#color: white;
#margin: 1px;
#border-bottom-left-radius:5px;
#border-top-left-radius:5px;
#}
#.QProgressBar::chunk {
#background-color: #05B8CC;
#border-radius: 3px;
#}
#.OverlayProgressWidget {
#background-color: white;
#}
#"""
## set stylesheet for all progressbars in this widget
#self.setStyleSheet(styleSheet)
parent = self.parent()
xAnchor = parent.width() - self._width - self._margin
yAnchor = self._margin
self.setGeometry(xAnchor, yAnchor, self._width, self._minHeight)
def _addProgressBar(self, worker):
progressBar = QtGui.QProgressBar(self)
progressBar.setMinimumHeight(self._pbHeight - 5)
progressBar.setMaximumHeight(self._pbHeight - 5)
label = QtGui.QLabel(worker.name, self)
if not self.toggleButton.isChecked():
progressBar.hide()
label.hide()
row = self._addedBars + 1
self.glayout.addWidget(progressBar, row, 0, 1, 1)
self.glayout.addWidget(label, row, 1, 1, 1)
self._addedBars += 1
self._detailProgressBars.append((progressBar, label))
worker.progressChanged.connect(progressBar.setValue)
worker.progressChanged.connect(self.calculateTotalProgress)
worker.progressChanged.connect(self.debugProgressChanged)
def initUi(self):
self.setWindowTitle(self.tr('Remove Attributes'))
self.setModal(True)
self.resize(366, 274)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Expanding)
self.setSizePolicy(sizePolicy)
self.gridLayout = QtGui.QGridLayout(self)
self.dialogHeading = QtGui.QLabel(
self.tr('Select the attribute column(s) which shall be removed'),
self)
self.listView = QtGui.QListView(self)
model = QtGui.QStandardItemModel()
for column in self.columns:
item = QtGui.QStandardItem(column)
model.appendRow(item)
self.listView.setModel(model)
self.listView.setSelectionMode(QtGui.QListView.MultiSelection)
self.buttonBox = QtGui.QDialogButtonBox(self)
self.buttonBox.setOrientation(QtCore.Qt.Horizontal)
self.buttonBox.setStandardButtons(QtGui.QDialogButtonBox.Cancel
| QtGui.QDialogButtonBox.Ok)
self.gridLayout.addWidget(self.dialogHeading, 0, 0, 1, 1)
self.gridLayout.addWidget(self.listView, 1, 0, 1, 1)
self.gridLayout.addWidget(self.buttonBox, 2, 0, 1, 1)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
def test_tableViewMissing(self, widgetClass, qtbot, model, exception,
exceptionContains):
widget = widgetClass()
qtbot.addWidget(widget)
with pytest.raises(exception) as excinfo:
if model:
widget.setModel(QtGui.QStandardItemModel())
createDelegate('foo', 'bar', widget)
assert exceptionContains in str(excinfo.value)
def test_input(self, qtbot):
widget = QtGui.QLineEdit()
widget.setValidator(DelimiterValidator())
qtbot.addWidget(widget)
widget.show()
qtbot.keyPress(widget, ' ')
assert widget.text() == ''
qtbot.keyPress(widget, 'a')
assert widget.text() == 'a'
def createEditor(self, parent, option, index):
"""Returns the widget used to edit the item specified by index for editing. The parent widget and style option are used to control how the editor widget appears.
Args:
parent (QWidget): parent widget.
option (QStyleOptionViewItem): controls how editor widget appears.
index (QModelIndex): model data index.
"""
editor = QtGui.QLineEdit(parent)
return editor
def display_query(selcols):
## setup a new empty model
model1 = DataFrameModel()
## setup an application and create a table view widget
app = QtGui.QApplication([])
widget1 = DataTableWidget()
widget1.resize(1600, 800)
widget1.show()
## asign the created model
widget1.setViewModel(model1)
## fill the model with data
model1.setDataFrame(selcols)
## start the app"""
app.exec_()
return
def createEditor(self, parent, option, index):
"""Returns the widget used to edit the item specified by index for editing. The parent widget and style option are used to control how the editor widget appears.
Args:
parent (QWidget): parent widget.
option (QStyleOptionViewItem): controls how editor widget appears.
index (QModelIndex): model data index.
"""
editor = QtGui.QDoubleSpinBox(parent)
try:
editor.setMinimum(self.minimum)
editor.setMaximum(self.maximum)
editor.setSingleStep(self.singleStep)
editor.setDecimals(self.decimals)
except TypeError, err:
# initiate the spinbox with default values.
pass
def test_setDelegates(self, qtbot, tableView, index, value, singleStep):
dlg = createDelegate(numpy.dtype(value), 0, tableView)
assert dlg is not None
data = pandas.DataFrame([value], columns=['A'])
data['A'] = data['A'].astype(value.dtype)
model = tableView.model()
model.setDataFrame(data)
for i, delegate in enumerate([dlg]):
assert tableView.itemDelegateForColumn(i) == delegate
option = QtGui.QStyleOptionViewItem()
option.rect = QtCore.QRect(0, 0, 100, 100)
editor = delegate.createEditor(tableView, option, index)
delegate.setEditorData(editor, index)
assert editor.value() == index.data()
delegate.setModelData(editor, model, index)
delegate.updateEditorGeometry(editor, option, index)
dtype = value.dtype
if dtype in DataFrameModel._intDtypes:
info = numpy.iinfo(dtype)
assert isinstance(delegate, BigIntSpinboxDelegate)
elif dtype in DataFrameModel._floatDtypes:
info = numpy.finfo(dtype)
assert isinstance(delegate, CustomDoubleSpinboxDelegate)
assert delegate.decimals == DataFrameModel._float_precisions[
str(value.dtype)]
assert delegate.maximum == info.max
assert editor.maximum() == info.max
assert delegate.minimum == info.min
assert editor.minimum() == info.min
assert delegate.singleStep == singleStep
assert editor.singleStep() == singleStep
def clickEvent(index):
assert index.isValid()
tableView.clicked.connect(clickEvent)
with qtbot.waitSignal(tableView.clicked) as blocker:
qtbot.mouseClick(tableView.viewport(),
QtCore.Qt.LeftButton,
pos=QtCore.QPoint(10, 10))
assert blocker.signal_triggered
def __init__(self, parent=None):
"""the __init__ method.
Args:
parent (QObject): defaults to None. If parent is 0, the new widget becomes a window.
If parent is another widget, this widget becomes a child window inside parent.
The new widget is deleted when its parent is deleted.
"""
super(BigIntSpinbox, self).__init__(parent)
self._singleStep = 1
self._minimum = -18446744073709551616
self._maximum = 18446744073709551615
rx = QtCore.QRegExp("[0-9]\\d{0,20}")
validator = QtGui.QRegExpValidator(rx, self)
self._lineEdit = QtWidgets.QLineEdit(self)
self._lineEdit.setText('0')
self._lineEdit.setValidator(validator)
self.setLineEdit(self._lineEdit)
def test_editing(self, dataframe, qtbot):
model = ColumnDtypeModel(dataFrame=dataframe)
model.setEditable(True)
tableView = QtGui.QTableView()
qtbot.addWidget(tableView)
tableView.setModel(model)
delegate = DtypeComboDelegate(tableView)
tableView.setItemDelegateForColumn(1, delegate)
tableView.show()
index = model.index(0, 1)
preedit_data = index.data(DTYPE_ROLE)
tableView.edit(index)
editor = tableView.findChildren(QtGui.QComboBox)[0]
selectedIndex = editor.currentIndex()
editor.setCurrentIndex(selectedIndex+1)
postedit_data = index.data(DTYPE_ROLE)
assert preedit_data != postedit_data
def createEditor(self, parent, option, index):
"""Creates an Editor Widget for the given index.
Enables the user to manipulate the displayed data in place. An editor
is created, which performs the change.
The widget used will be a `QComboBox` with all available datatypes in the
`pandas` project.
Args:
parent (QtCore.QWidget): Defines the parent for the created editor.
option (QtGui.QStyleOptionViewItem): contains all the information
that QStyle functions need to draw the items.
index (QtCore.QModelIndex): The item/index which shall be edited.
Returns:
QtGui.QWidget: he widget used to edit the item specified by index
for editing.
"""
combo = QtGui.QComboBox(parent)
combo.addItems(SupportedDtypes.names())
combo.currentIndexChanged.connect(self.currentIndexChanged)
return combo
def test_editing(self, dataFrame, qtbot):
model = DataFrameModel(dataFrame)
tableView = QtGui.QTableView()
qtbot.addWidget(tableView)
tableView.setModel(model)
delegate = TextDelegate(tableView)
createDelegate(numpy.dtype('O'), 0, tableView)
tableView.show()
index = model.index(0, 0)
preedit_data = index.data()
assert not model.editable
model.enableEditing(True)
tableView.edit(index)
editor = tableView.findChildren(QtGui.QLineEdit)[0]
qtbot.keyPress(editor, QtCore.Qt.Key_F)
qtbot.keyPress(editor, QtCore.Qt.Key_Enter)
QtGui.QApplication.processEvents()
with qtbot.waitSignal(timeout=100):
assert index.data(QtCore.Qt.DisplayRole) == 'f'
def _initUI(self):
"""Creates the inital layout with all subwidgets.
The layout is a `QHBoxLayout`. Each time a radio button is
selected or unselected, a slot
`DelimiterSelectionWidget._delimiter` is called.
Furthermore the `QLineEdit` widget has a custom regex validator
`DelimiterValidator` enabled.
"""
#layout = QtGui.QHBoxLayout(self)
self.semicolonRadioButton = QtGui.QRadioButton(u'Semicolon')
self.commaRadioButton = QtGui.QRadioButton(u'Comma')
self.tabRadioButton = QtGui.QRadioButton(u'Tab')
self.otherRadioButton = QtGui.QRadioButton(u'Other')
self.semicolonRadioButton.setChecked(True)
self.otherSeparatorLineEdit = QtGui.QLineEdit(self)
self.otherSeparatorLineEdit.setEnabled(False)
self.semicolonRadioButton.toggled.connect(self._delimiter)
self.commaRadioButton.toggled.connect(self._delimiter)
self.tabRadioButton.toggled.connect(self._delimiter)
self.otherRadioButton.toggled.connect(self._enableLine)
self.otherSeparatorLineEdit.textChanged.connect(
lambda: self._delimiter(True))
self.otherSeparatorLineEdit.setValidator(DelimiterValidator(self))
currentLayout = self.layout()
# unset and delete the current layout in order to set a new one
if currentLayout is not None:
del currentLayout
layout = QtGui.QHBoxLayout()
layout.addWidget(self.semicolonRadioButton)
layout.addWidget(self.commaRadioButton)
layout.addWidget(self.tabRadioButton)
layout.addWidget(self.otherRadioButton)
layout.addWidget(self.otherSeparatorLineEdit)
self.setLayout(layout)
def _showMessageBox(text):
errorbox = QtGui.QMessageBox()
errorbox.setText(text)
errorbox.exec_()
def _initUI(self):
"""Initiates the user interface with a grid layout and several widgets.
"""
self.setModal(self._modal)
self.setWindowTitle(self._windowTitle)
layout = QtWidgets.QGridLayout()
self._filenameLabel = QtWidgets.QLabel(u'Output File', self)
self._filenameLineEdit = QtWidgets.QLineEdit(self)
chooseFileButtonIcon = QtGui.QIcon(
QtGui.QPixmap(':/icons/document-save-as.png'))
self._chooseFileAction = QtWidgets.QAction(self)
self._chooseFileAction.setIcon(chooseFileButtonIcon)
self._chooseFileAction.triggered.connect(self._createFile)
self._chooseFileButton = QtWidgets.QToolButton(self)
self._chooseFileButton.setDefaultAction(self._chooseFileAction)
layout.addWidget(self._filenameLabel, 0, 0)
layout.addWidget(self._filenameLineEdit, 0, 1, 1, 2)
layout.addWidget(self._chooseFileButton, 0, 3)
self._encodingLabel = QtWidgets.QLabel(u'File Encoding', self)
encoding_names = map(lambda x: x.upper(),
sorted(list(set(_encodings.viewvalues()))))
self._encodingComboBox = QtWidgets.QComboBox(self)
self._encodingComboBox.addItems(encoding_names)
self._idx = encoding_names.index('UTF_8')
self._encodingComboBox.setCurrentIndex(self._idx)
#self._encodingComboBox.activated.connect(self._updateEncoding)
layout.addWidget(self._encodingLabel, 1, 0)
layout.addWidget(self._encodingComboBox, 1, 1, 1, 1)
self._hasHeaderLabel = QtWidgets.QLabel(u'Header Available?', self)
self._headerCheckBox = QtWidgets.QCheckBox(self)
#self._headerCheckBox.toggled.connect(self._updateHeader)
layout.addWidget(self._hasHeaderLabel, 2, 0)
layout.addWidget(self._headerCheckBox, 2, 1)
self._delimiterLabel = QtWidgets.QLabel(u'Column Delimiter', self)
self._delimiterBox = DelimiterSelectionWidget(self)
layout.addWidget(self._delimiterLabel, 3, 0)
layout.addWidget(self._delimiterBox, 3, 1, 1, 3)
self._exportButton = QtWidgets.QPushButton(u'Export Data', self)
self._cancelButton = QtWidgets.QPushButton(u'Cancel', self)
self._buttonBox = QtWidgets.QDialogButtonBox(self)
self._buttonBox.addButton(self._exportButton,
QtWidgets.QDialogButtonBox.AcceptRole)
self._buttonBox.addButton(self._cancelButton,
QtWidgets.QDialogButtonBox.RejectRole)
self._buttonBox.accepted.connect(self.accepted)
self._buttonBox.rejected.connect(self.rejected)
layout.addWidget(self._buttonBox, 5, 2, 1, 2)
self._exportButton.setDefault(False)
self._filenameLineEdit.setFocus()
self._statusBar = QtWidgets.QStatusBar(self)
self._statusBar.setSizeGripEnabled(False)
layout.addWidget(self._statusBar, 4, 0, 1, 4)
self.setLayout(layout)
def _initUI(self):
"""Initiates the user interface with a grid layout and several widgets.
"""
self.setModal(self._modal)
self.setWindowTitle(self._windowTitle)
layout = QtWidgets.QGridLayout()
self._filenameLabel = QtWidgets.QLabel(u'Choose File', self)
self._filenameLineEdit = QtWidgets.QLineEdit(self)
self._filenameLineEdit.textEdited.connect(self._updateFilename)
chooseFileButtonIcon = QtGui.QIcon(
QtGui.QPixmap(':/icons/document-open.png'))
self._chooseFileAction = QtWidgets.QAction(self)
self._chooseFileAction.setIcon(chooseFileButtonIcon)
self._chooseFileAction.triggered.connect(self._openFile)
self._chooseFileButton = QtWidgets.QToolButton(self)
self._chooseFileButton.setDefaultAction(self._chooseFileAction)
layout.addWidget(self._filenameLabel, 0, 0)
layout.addWidget(self._filenameLineEdit, 0, 1, 1, 2)
layout.addWidget(self._chooseFileButton, 0, 3)
self._encodingLabel = QtWidgets.QLabel(u'File Encoding', self)
encoding_names = map(lambda x: x.upper(),
sorted(list(set(_encodings.viewvalues()))))
self._encodingComboBox = QtWidgets.QComboBox(self)
self._encodingComboBox.addItems(encoding_names)
self._encodingComboBox.activated.connect(self._updateEncoding)
layout.addWidget(self._encodingLabel, 1, 0)
layout.addWidget(self._encodingComboBox, 1, 1, 1, 1)
self._hasHeaderLabel = QtWidgets.QLabel(u'Header Available?', self)
self._headerCheckBox = QtWidgets.QCheckBox(self)
self._headerCheckBox.toggled.connect(self._updateHeader)
layout.addWidget(self._hasHeaderLabel, 2, 0)
layout.addWidget(self._headerCheckBox, 2, 1)
self._delimiterLabel = QtWidgets.QLabel(u'Column Delimiter', self)
self._delimiterBox = DelimiterSelectionWidget(self)
self._delimiter = self._delimiterBox.currentSelected()
self._delimiterBox.delimiter.connect(self._updateDelimiter)
layout.addWidget(self._delimiterLabel, 3, 0)
layout.addWidget(self._delimiterBox, 3, 1, 1, 3)
self._tabWidget = QtWidgets.QTabWidget(self)
self._previewTableView = QtWidgets.QTableView(self)
self._datatypeTableView = QtWidgets.QTableView(self)
self._tabWidget.addTab(self._previewTableView, u'Preview')
self._tabWidget.addTab(self._datatypeTableView, u'Change Column Types')
layout.addWidget(self._tabWidget, 4, 0, 3, 4)
self._datatypeTableView.horizontalHeader().setDefaultSectionSize(200)
self._datatypeTableView.setItemDelegateForColumn(
1, DtypeComboDelegate(self._datatypeTableView))
self._loadButton = QtWidgets.QPushButton(u'Load Data', self)
#self.loadButton.setAutoDefault(False)
self._cancelButton = QtWidgets.QPushButton(u'Cancel', self)
# self.cancelButton.setDefault(False)
# self.cancelButton.setAutoDefault(True)
self._buttonBox = QtWidgets.QDialogButtonBox(self)
self._buttonBox.addButton(self._loadButton,
QtWidgets.QDialogButtonBox.AcceptRole)
self._buttonBox.addButton(self._cancelButton,
QtWidgets.QDialogButtonBox.RejectRole)
self._buttonBox.accepted.connect(self.accepted)
self._buttonBox.rejected.connect(self.rejected)
layout.addWidget(self._buttonBox, 9, 2, 1, 2)
self._loadButton.setDefault(False)
self._filenameLineEdit.setFocus()
self._statusBar = QtWidgets.QStatusBar(self)
self._statusBar.setSizeGripEnabled(False)
layout.addWidget(self._statusBar, 8, 0, 1, 4)
self.setLayout(layout)
def initUi(self):
"""Initalizes the Uuser Interface with all sub widgets.
"""
self.gridLayout = QtWidgets.QGridLayout(self)
self.buttonFrame = QtWidgets.QFrame(self)
self.buttonFrame.setMinimumSize(QtCore.QSize(250, 50))
self.buttonFrame.setMaximumSize(QtCore.QSize(250, 50))
self.buttonFrame.setFrameShape(QtWidgets.QFrame.NoFrame)
self.buttonFrameLayout = QtWidgets.QGridLayout(self.buttonFrame)
self.buttonFrameLayout.setContentsMargins(0, 6, 0, 6)
self.editButton = QtWidgets.QToolButton(self.buttonFrame)
self.editButton.setObjectName('editbutton')
icon = QtGui.QIcon(QtGui.QPixmap(_fromUtf8(':/icons/document-edit.png')))
self.editButton.setIcon(icon)
self.addColumnButton = QtWidgets.QToolButton(self.buttonFrame)
self.addColumnButton.setObjectName('addcolumnbutton')
icon = QtGui.QIcon(QtGui.QPixmap(_fromUtf8(':/icons/edit-table-insert-column-right.png')))
self.addColumnButton.setIcon(icon)
self.addRowButton = QtWidgets.QToolButton(self.buttonFrame)
self.addRowButton.setObjectName('addrowbutton')
icon = QtGui.QIcon(QtGui.QPixmap(_fromUtf8(':/icons/edit-table-insert-row-below.png')))
self.addRowButton.setIcon(icon)
self.removeColumnButton = QtWidgets.QToolButton(self.buttonFrame)
self.removeColumnButton.setObjectName('removecolumnbutton')
icon = QtGui.QIcon(QtGui.QPixmap(_fromUtf8(':/icons/edit-table-delete-column.png')))
self.removeColumnButton.setIcon(icon)
self.removeRowButton = QtWidgets.QToolButton(self.buttonFrame)
self.removeRowButton.setObjectName('removerowbutton')
icon = QtGui.QIcon(QtGui.QPixmap(_fromUtf8(':/icons/edit-table-delete-row.png')))
self.removeRowButton.setIcon(icon)
self.buttons = [self.editButton, self.addColumnButton, self.addRowButton, self.removeColumnButton, self.removeRowButton]
for index, button in enumerate(self.buttons):
button.setMinimumSize(QtCore.QSize(36, 36))
button.setMaximumSize(QtCore.QSize(36, 36))
button.setIconSize(QtCore.QSize(36, 36))
button.setCheckable(True)
self.buttonFrameLayout.addWidget(button, 0, index, 1, 1)
for button in self.buttons[1:]:
button.setEnabled(False)
self.tableView = QtWidgets.QTableView(self)
self.tableView.setAlternatingRowColors(True)
self.tableView.setSortingEnabled(True)
self.gridLayout.addWidget(self.buttonFrame, 0, 0, 1, 1)
self.gridLayout.addWidget(self.tableView, 1, 0, 1, 1)
self.editButton.toggled.connect(self.enableEditing)
self.addColumnButton.toggled.connect(self.showAddColumnDialog)
self.addRowButton.toggled.connect(self.addRow)
self.removeRowButton.toggled.connect(self.removeRow)
self.removeColumnButton.toggled.connect(self.showRemoveColumnDialog)
def __init__(self, parent=None):
super(TestWidget, self).__init__(parent)
self.resize(1680, 756)
self.move(0, 0)
self.df = pandas.DataFrame()
self.dataModel = None
# init the data view's
self.dataTableView = DataTableWidget(self)
# self.dataTableView.setSortingEnabled(True)
# self.dataTableView.setAlternatingRowColors(True)
self.dataListView = QtGui.QListView(self)
self.dataListView.setAlternatingRowColors(True)
self.dataComboBox = QtGui.QComboBox(self)
# make combobox to choose the model column for dataComboBox and dataListView
self.chooseColumnComboBox = QtGui.QComboBox(self)
self.buttonCsvData = QtGui.QPushButton("load csv data")
self.buttonRandomData = QtGui.QPushButton("load random data")
importDialog = CSVImportDialog(self)
importDialog.load.connect(self.updateModel)
self.buttonCsvData.clicked.connect(lambda: importDialog.show())
self.buttonRandomData.clicked.connect(
lambda: self.setDataFrame(getRandomData(rows=100, columns=100)))
self.exportDialog = CSVExportDialog(self)
self.buttonCSVExport = QtGui.QPushButton("export to csv")
self.buttonCSVExport.clicked.connect(self._exportModel)
self.buttonLayout = QtGui.QHBoxLayout()
self.buttonLayout.addWidget(self.buttonCsvData)
self.buttonLayout.addWidget(self.buttonCSVExport)
self.buttonLayout.addWidget(self.buttonRandomData)
self.mainLayout = QtGui.QVBoxLayout()
self.setLayout(self.mainLayout)
self.mainLayout.addLayout(self.buttonLayout)
self.mainLayout.addWidget(self.dataTableView)
self.spinbox = QtGui.QSpinBox()
self.mainLayout.addWidget(self.spinbox)
self.spinbox.setMaximum(99999999999)
self.spinbox.setValue(99999999999)
self.rightLayout = QtGui.QVBoxLayout()
self.chooseColumLayout = QtGui.QHBoxLayout()
self.mainLayout.addLayout(self.rightLayout)
self.rightLayout.addLayout(self.chooseColumLayout)
self.chooseColumLayout.addWidget(QtGui.QLabel("Choose column:"))
self.chooseColumLayout.addWidget(self.chooseColumnComboBox)
self.rightLayout.addWidget(self.dataListView)
self.rightLayout.addWidget(self.dataComboBox)
self.tableViewColumnDtypes = QtGui.QTableView(self)
self.rightLayout.addWidget(QtGui.QLabel('dtypes'))
self.rightLayout.addWidget(self.tableViewColumnDtypes)
self.buttonGoToColumn = QtGui.QPushButton("go to column")
self.rightLayout.addWidget(self.buttonGoToColumn)
self.buttonGoToColumn.clicked.connect(self.goToColumn)
self.buttonSetFilter = QtGui.QPushButton("set filter")
self.rightLayout.addWidget(self.buttonSetFilter)
self.buttonSetFilter.clicked.connect(self.setFilter)
self.buttonClearFilter = QtGui.QPushButton("clear filter")
self.rightLayout.addWidget(self.buttonClearFilter)
self.buttonClearFilter.clicked.connect(self.clearFilter)
self.lineEditFilterCondition = QtGui.QLineEdit("freeSearch('am')")
self.rightLayout.addWidget(self.lineEditFilterCondition)
self.chooseColumnComboBox.currentIndexChanged.connect(
self.setModelColumn)
self.dataListView.mouseReleaseEvent = self.mouseReleaseEvent
self.dropLineEdit = DropLineEdit("drop data from table here", self)
self.rightLayout.addWidget(self.dropLineEdit)
self.dropWidget = ComplexDropWidget(self)
self.dropWidget.dropRecieved.connect(self.processDataDrops)
self.rightLayout.addWidget(self.dropWidget)
def initUi(self):
"""Initalizes the Uuser Interface with all sub widgets.
"""
self.gridLayout = QtGui.QGridLayout(self)
self.gridLayout.setContentsMargins(0, 0, 0, 0)
self.buttonFrame = QtGui.QFrame(self)
#self.buttonFrame.setMinimumSize(QtCore.QSize(250, 50))
#self.buttonFrame.setMaximumSize(QtCore.QSize(250, 50))
self.buttonFrame.setFrameShape(QtGui.QFrame.NoFrame)
spacerItemButton = QtGui.QSpacerItem(40, 20,
QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum)
self.buttonFrameLayout = QtGui.QGridLayout(self.buttonFrame)
self.buttonFrameLayout.setContentsMargins(0, 0, 0, 0)
self.editButton = QtGui.QToolButton(self.buttonFrame)
self.editButton.setObjectName('editbutton')
self.editButton.setText(self.tr(u'edit'))
self.editButton.setToolTip(self.tr(u'toggle editing mode'))
icon = QtGui.QIcon(
QtGui.QPixmap(_fromUtf8(':/icons/document-edit.png')))
self.editButton.setIcon(icon)
self.addColumnButton = QtGui.QToolButton(self.buttonFrame)
self.addColumnButton.setObjectName('addcolumnbutton')
self.addColumnButton.setText(self.tr(u'+col'))
self.addColumnButton.setToolTip(self.tr(u'add new column'))
icon = QtGui.QIcon(
QtGui.QPixmap(
_fromUtf8(':/icons/edit-table-insert-column-right.png')))
self.addColumnButton.setIcon(icon)
self.addRowButton = QtGui.QToolButton(self.buttonFrame)
self.addRowButton.setObjectName('addrowbutton')
self.addRowButton.setText(self.tr(u'+row'))
self.addRowButton.setToolTip(self.tr(u'add new row'))
icon = QtGui.QIcon(
QtGui.QPixmap(
_fromUtf8(':/icons/edit-table-insert-row-below.png')))
self.addRowButton.setIcon(icon)
self.removeColumnButton = QtGui.QToolButton(self.buttonFrame)
self.removeColumnButton.setObjectName('removecolumnbutton')
self.removeColumnButton.setText(self.tr(u'-col'))
self.removeColumnButton.setToolTip(self.tr(u'remove a column'))
icon = QtGui.QIcon(
QtGui.QPixmap(_fromUtf8(':/icons/edit-table-delete-column.png')))
self.removeColumnButton.setIcon(icon)
self.removeRowButton = QtGui.QToolButton(self.buttonFrame)
self.removeRowButton.setObjectName('removerowbutton')
self.removeRowButton.setText(self.tr(u'-row'))
self.removeRowButton.setToolTip(self.tr(u'remove selected rows'))
icon = QtGui.QIcon(
QtGui.QPixmap(_fromUtf8(':/icons/edit-table-delete-row.png')))
self.removeRowButton.setIcon(icon)
self.buttons = [
self.editButton, self.addColumnButton, self.addRowButton,
self.removeColumnButton, self.removeRowButton
]
for index, button in enumerate(self.buttons):
button.setMinimumSize(self._iconSize)
button.setMaximumSize(self._iconSize)
button.setIconSize(self._iconSize)
button.setCheckable(True)
self.buttonFrameLayout.addWidget(button, 0, index, 1, 1)
self.buttonFrameLayout.addItem(spacerItemButton, 0, index + 1, 1, 1)
for button in self.buttons[1:]:
button.setEnabled(False)
#self.tableView = QtGui.QTableView(self)
self.tableView = DragTable(self)
self.tableView.setAlternatingRowColors(True)
self.tableView.setSortingEnabled(True)
self.gridLayout.addWidget(self.buttonFrame, 0, 0, 1, 1)
self.gridLayout.addWidget(self.tableView, 1, 0, 1, 1)
self.editButton.toggled.connect(self.enableEditing)
self.addColumnButton.toggled.connect(self.showAddColumnDialog)
self.addRowButton.toggled.connect(self.addRow)
self.removeRowButton.toggled.connect(self.removeRow)
self.removeColumnButton.toggled.connect(self.showRemoveColumnDialog)
def goToColumn(self):
print "go to column 7"
index = self.dataTableView.view().model().index(7, 0)
self.dataTableView.view().setCurrentIndex(index)
def changeColumnValue(self, columnName, index, dtype):
print "failed to change", columnName, "to", dtype
print index.data(), index.isValid()
self.dataTableView.view().setCurrentIndex(index)
def setFilter(self):
#filterIndex = eval(self.lineEditFilterCondition.text())
search = DataSearch("Test", self.lineEditFilterCondition.text())
self.dataTableView.view().model().setFilter(search)
#raise NotImplementedError
def clearFilter(self):
self.dataTableView.view().model().clearFilter()
if __name__ == '__main__':
app = QtGui.QApplication(sys.argv)
widget = TestWidget()
widget.show()
widget.setDataFrame(getCsvData())
#widget.setDataFrame( getRandomData(2, 2) )
app.exec_()
raise ValueError, "Test Test ä"
def exception2():
raise ValueError, u"Test Test ä"
def exception3():
raise ValueError, u"Test Test"
def exception4():
raise ValueError, "Test Test"
app = QtGui.QApplication([])
sys.excepthook = excepthook
widget = QtGui.QPushButton("raise exceptions")
widget.move(100, 100)
widget.resize(100, 100)
widget.show()
widget.clicked.connect(exception)
widget.clicked.connect(exception2)
widget.clicked.connect(exception3)
widget.clicked.connect(exception4)
app.exec_()
#@pytest.fixture()
#def overwriteExcepthook():
#sys.excepthook = excepthook
def initUi(self):
self.setModal(True)
self.resize(303, 168)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Fixed)
self.setSizePolicy(sizePolicy)
self.verticalLayout = QtGui.QVBoxLayout(self)
self.dialogHeading = QtGui.QLabel(
self.tr('Add a new attribute column'), self)
self.gridLayout = QtGui.QGridLayout()
self.columnNameLineEdit = QtGui.QLineEdit(self)
self.columnNameLabel = QtGui.QLabel(self.tr('Name'), self)
self.dataTypeComboBox = QtGui.QComboBox(self)
self.dataTypeComboBox.addItems(SupportedDtypes.names())
self.columnTypeLabel = QtGui.QLabel(self.tr('Type'), self)
self.defaultValueLineEdit = QtGui.QLineEdit(self)
self.lineEditValidator = DefaultValueValidator(self)
self.defaultValueLineEdit.setValidator(self.lineEditValidator)
self.defaultValueLabel = QtGui.QLabel(self.tr('Inital Value(s)'), self)
self.gridLayout.addWidget(self.columnNameLabel, 0, 0, 1, 1)
self.gridLayout.addWidget(self.columnNameLineEdit, 0, 1, 1, 1)
self.gridLayout.addWidget(self.columnTypeLabel, 1, 0, 1, 1)
self.gridLayout.addWidget(self.dataTypeComboBox, 1, 1, 1, 1)
self.gridLayout.addWidget(self.defaultValueLabel, 2, 0, 1, 1)
self.gridLayout.addWidget(self.defaultValueLineEdit, 2, 1, 1, 1)
self.buttonBox = QtGui.QDialogButtonBox(self)
self.buttonBox.setOrientation(QtCore.Qt.Horizontal)
self.buttonBox.setStandardButtons(QtGui.QDialogButtonBox.Cancel
| QtGui.QDialogButtonBox.Ok)
self.verticalLayout.addWidget(self.dialogHeading)
self.verticalLayout.addLayout(self.gridLayout)
self.verticalLayout.addWidget(self.buttonBox)
self.buttonBox.accepted.connect(self.accept)
self.buttonBox.rejected.connect(self.reject)
self.dataTypeComboBox.currentIndexChanged.connect(
self.updateValidatorDtype)
self.updateValidatorDtype(self.dataTypeComboBox.currentIndex())
class TestCustomDelegates(object):
@pytest.fixture
def emptyTableView(self, qtbot):
widget = DemoTableView()
qtbot.addWidget(widget)
return widget
@pytest.fixture
def dataFrame(self):
return pandas.DataFrame(['abc'], columns=['A'])
@pytest.fixture
def model(self, dataFrame):
return DataFrameModel(dataFrame)
@pytest.fixture
def tableView(self, model, emptyTableView):
emptyTableView.setModel(model)
return emptyTableView
@pytest.fixture
def index(self, model):
index = model.index(0, 0)
assert index.isValid()
return index
@pytest.mark.parametrize(
"widgetClass, model, exception, exceptionContains", [
(QtGui.QWidget, None, AttributeError, "has no attribute 'model'"),
(DemoTableView, None, ValueError,
"no model set for the current view"),
(DemoTableView, QtGui.QStandardItemModel(), TypeError,
'model is not of type DataFrameModel'),
])
def test_tableViewMissing(self, widgetClass, qtbot, model, exception,
exceptionContains):
widget = widgetClass()
qtbot.addWidget(widget)
with pytest.raises(exception) as excinfo:
if model:
widget.setModel(QtGui.QStandardItemModel())
createDelegate('foo', 'bar', widget)
assert exceptionContains in str(excinfo.value)
@pytest.mark.parametrize(
"value, singleStep",
[
(numpy.int8(1), 1),
(numpy.int16(1), 1),
(numpy.int32(1), 1),
(numpy.int64(1), 1),
(numpy.uint8(1), 1),
(numpy.uint16(1), 1),
(numpy.uint32(1), 1),
(numpy.uint64(1), 1),
(numpy.float16(1.11111), 0.1),
(numpy.float32(1.11111111), 0.1),
(numpy.float64(1.1111111111111111), 0.1),
#(numpy.float128(1.11111111111111111111), 0.1),
])
def test_setDelegates(self, qtbot, tableView, index, value, singleStep):
dlg = createDelegate(numpy.dtype(value), 0, tableView)
assert dlg is not None
data = pandas.DataFrame([value], columns=['A'])
data['A'] = data['A'].astype(value.dtype)
model = tableView.model()
model.setDataFrame(data)
for i, delegate in enumerate([dlg]):
assert tableView.itemDelegateForColumn(i) == delegate
option = QtGui.QStyleOptionViewItem()
option.rect = QtCore.QRect(0, 0, 100, 100)
editor = delegate.createEditor(tableView, option, index)
delegate.setEditorData(editor, index)
assert editor.value() == index.data()
delegate.setModelData(editor, model, index)
delegate.updateEditorGeometry(editor, option, index)
dtype = value.dtype
if dtype in DataFrameModel._intDtypes:
info = numpy.iinfo(dtype)
assert isinstance(delegate, BigIntSpinboxDelegate)
elif dtype in DataFrameModel._floatDtypes:
info = numpy.finfo(dtype)
assert isinstance(delegate, CustomDoubleSpinboxDelegate)
assert delegate.decimals == DataFrameModel._float_precisions[
str(value.dtype)]
assert delegate.maximum == info.max
assert editor.maximum() == info.max
assert delegate.minimum == info.min
assert editor.minimum() == info.min
assert delegate.singleStep == singleStep
assert editor.singleStep() == singleStep
def clickEvent(index):
assert index.isValid()
tableView.clicked.connect(clickEvent)
with qtbot.waitSignal(tableView.clicked) as blocker:
qtbot.mouseClick(tableView.viewport(),
QtCore.Qt.LeftButton,
pos=QtCore.QPoint(10, 10))
assert blocker.signal_triggered
except UnicodeEncodeError, e:
excValueStr = unicode(excValue)
errmsg = u'{0}: \n{1}'.format(excType, excValueStr)
sections = [u'\n', separator, timeString, separator, errmsg, separator, tbinfo]
msg = u'\n'.join(sections)
try:
f = codecs.open(logFile, "a+", encoding='utf-8')
f.write(msg)
f.close()
except IOError, e:
msgbox(u"unable to write to {0}".format(logFile), u"Writing error")
# always show an error message
try:
if not _isQAppRunning():
app = QtGui.QApplication([])
_showMessageBox(unicode(notice) + unicode(msg))
except:
msgbox(unicode(notice) + unicode(msg), u"Error")
def _isQAppRunning():
if QtGui.QApplication.instance() is None:
return False
else:
return True
def _showMessageBox(text):
errorbox = QtGui.QMessageBox()
errorbox.setText(text)
errorbox.exec_()
