def __init__(self, tableName, parent=None):
super().__init__(parent)
self.model = QSqlTableModel(self)
self.model.setTable(tableName)
self.model.setEditStrategy(QSqlTableModel.OnManualSubmit)
self.model.select()
self.model.setHeaderData(0, Qt.Horizontal, self.tr("ID"))
self.model.setHeaderData(1, Qt.Horizontal, self.tr("First name"))
self.model.setHeaderData(2, Qt.Horizontal, self.tr("Last name"))
view = QTableView()
view.setModel(self.model)
view.resizeColumnsToContents()
self.submitButton = QPushButton(self.tr("Submit"))
self.submitButton.setDefault(True)
self.revertButton = QPushButton(self.tr("&Revert"))
self.quitButton = QPushButton(self.tr("Quit"))
self.buttonBox = QDialogButtonBox(Qt.Vertical)
self.buttonBox.addButton(self.submitButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.revertButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.quitButton, QDialogButtonBox.RejectRole)
self.submitButton.clicked.connect(self.submit)
self.revertButton.clicked.connect(self.model.revertAll)
self.quitButton.clicked.connect(self.close)
mainLayout = QHBoxLayout()
mainLayout.addWidget(view)
mainLayout.addWidget(self.buttonBox)
self.setLayout(mainLayout)
self.setWindowTitle(self.tr("Cached Table"))
class TableEditor(QWidget):
def __init__(self, tableName, parent=None):
super().__init__(parent)
self.model = QSqlTableModel(self)
self.model.setTable(tableName)
self.model.setEditStrategy(QSqlTableModel.OnManualSubmit)
self.model.select()
self.model.setHeaderData(0, Qt.Horizontal, self.tr("ID"))
self.model.setHeaderData(1, Qt.Horizontal, self.tr("First name"))
self.model.setHeaderData(2, Qt.Horizontal, self.tr("Last name"))
view = QTableView()
view.setModel(self.model)
view.resizeColumnsToContents()
self.submitButton = QPushButton(self.tr("Submit"))
self.submitButton.setDefault(True)
self.revertButton = QPushButton(self.tr("&Revert"))
self.quitButton = QPushButton(self.tr("Quit"))
self.buttonBox = QDialogButtonBox(Qt.Vertical)
self.buttonBox.addButton(self.submitButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.revertButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.quitButton, QDialogButtonBox.RejectRole)
self.submitButton.clicked.connect(self.submit)
self.revertButton.clicked.connect(self.model.revertAll)
self.quitButton.clicked.connect(self.close)
mainLayout = QHBoxLayout()
mainLayout.addWidget(view)
mainLayout.addWidget(self.buttonBox)
self.setLayout(mainLayout)
self.setWindowTitle(self.tr("Cached Table"))
def submit(self):
self.model.database().transaction()
if self.model.submitAll():
self.model.database().commit()
else:
self.model.database().rollback()
QMessageBox.warning(
self,
self.tr("Cached Table"),
self.tr(
"The database reported an error: {}".format(
self.model.lastError().text()
)
),
)
def ui(self):
super(JointsLabelDialog, self).ui()
name_label = label.BaseLabel('Specify label sides', parent=self)
left_side_label = label.BaseLabel('Left Side: ', parent=self)
self._left_side_line = lineedit.BaseLineEdit(parent=self)
right_side_label = label.BaseLabel('Right Side: ', parent=self)
self._right_side_line = lineedit.BaseLineEdit(parent=self)
info_widget = message.BaseMessage(
'You can disable auto joint label in tool\n button options (right click on it', parent=self).info()
grid_layout = layouts.GridLayout()
grid_layout.addWidget(name_label, 0, 0, 1, 2)
grid_layout.addWidget(left_side_label, 1, 0, Qt.AlignRight)
grid_layout.addWidget(self._left_side_line, 1, 1)
grid_layout.addWidget(right_side_label, 2, 0, Qt.AlignRight)
grid_layout.addWidget(self._right_side_line, 2, 1)
self._button_box = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
self.main_layout.addLayout(grid_layout)
self.main_layout.addWidget(dividers.Divider())
self.main_layout.addWidget(self._button_box)
self.main_layout.addWidget(dividers.Divider())
self.main_layout.addWidget(info_widget)
def createButtons(self):
self.closeButton = QPushButton(self.tr("&Close"))
self.revertButton = QPushButton(self.tr("&Revert"))
self.submitButton = QPushButton(self.tr("&Submit"))
self.closeButton.setDefault(True)
self.closeButton.clicked.connect(self.close)
self.revertButton.clicked.connect(self.revert)
self.submitButton.clicked.connect(self.submit)
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.addButton(self.submitButton,
QDialogButtonBox.AcceptRole)
self.buttonBox.addButton(self.revertButton, QDialogButtonBox.ResetRole)
self.buttonBox.addButton(self.closeButton, QDialogButtonBox.RejectRole)
def __init__(self, settings, parent=None, description=None, pstvOnly=True):
super(ConfigDialog, self).__init__(parent)
self.setWindowTitle("Configure source")
self.layout = QGridLayout(self)
row = 0
self.widgets = {}
self.settings = settings
# Settings should be a dictionary
for name, val in settings.items():
label = QLabel(self)
label.setText(str(name))
self.layout.addWidget(label, row, 0, 1, 1)
# Check the type of each setting, and create widgets accordingly
if isinstance(val, str):
# A string of some kind
widget = QLineEdit(self)
widget.setText(val)
elif isinstance(val, list):
# A list of alternative values, first is selected
print("List: ", name)
continue
elif isinstance(val, bool):
widget = QCheckBox(self)
if val:
widget.setCheckState(Qt.CheckState.Checked)
else:
widget.setCheckState(Qt.CheckState.Unchecked)
elif isinstance(val, int):
widget = QLineEdit(self)
widget.setInputMask("9000000")
widget.setText(str(val).strip())
elif isinstance(val, float):
widget = QLineEdit(self)
if pstvOnly:
widget.setInputMask("0.000")
widget.setText(str(val).strip())
else:
print("Ignoring: " + name)
continue
widget.config = name
self.widgets[name] = widget
self.layout.addWidget(widget, row, 1, 1, 1)
row += 1
# Add OK and Cancel buttons
buttonbox = QDialogButtonBox(QDialogButtonBox.Ok
| QDialogButtonBox.Cancel)
buttonbox.accepted.connect(self.getValues)
buttonbox.rejected.connect(self.reject)
self.layout.addWidget(buttonbox, row, 1, 2, 1)
def __init__(self, parent: QWidget = None) -> None:
super().__init__(parent)
self.tcpServer = QTcpServer()
self.tcpClient = QTcpSocket()
self.tcpServerConnection: QTcpSocket = None
self.bytesToWrite = 0
self.bytesWritten = 0
self.bytesReceived = 0
self.clientProgressBar = QProgressBar()
self.clientStatusLabel = QLabel(self.tr("Client ready"))
self.serverProgressBar = QProgressBar()
self.serverStatusLabel = QLabel(self.tr("Server ready"))
self.startButton = QPushButton(self.tr("&Start"))
self.quitButton = QPushButton(self.tr("&Quit"))
self.buttonBox = QDialogButtonBox()
self.buttonBox.addButton(self.startButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.quitButton, QDialogButtonBox.RejectRole)
self.startButton.clicked.connect(self.start)
self.quitButton.clicked.connect(self.close)
self.tcpServer.newConnection.connect(self.acceptConnection)
self.tcpClient.connected.connect(self.startTransfer)
self.tcpClient.bytesWritten.connect(self.updateClientProgress)
self.tcpClient.error.connect(self.displayError)
mainLayout = QVBoxLayout(self)
mainLayout.addWidget(self.clientProgressBar)
mainLayout.addWidget(self.clientStatusLabel)
mainLayout.addWidget(self.serverProgressBar)
mainLayout.addWidget(self.serverStatusLabel)
mainLayout.addStretch(1)
mainLayout.addSpacing(10)
mainLayout.addWidget(self.buttonBox)
def __init__(self, font, textColor, parent=None):
super(TextEditDialog, self).__init__(parent)
self.setWindowFlags(QtCore.Qt.Window | QtCore.Qt.FramelessWindowHint)
self.resize(QtCore.QSize(400, 300))
self.layout = QVBoxLayout(self)
self.layout.setContentsMargins(2, 2, 2, 2)
self.te = TextEditingField()
self.te.accepted.connect(self.onAccept)
self._font = QtGui.QFont(font)
self.te.setTextColor(textColor)
self.layout.addWidget(self.te)
self.buttons = QDialogButtonBox(
QDialogButtonBox.Ok | QDialogButtonBox.Cancel, self)
self.buttons.accepted.connect(self.onAccept)
self.buttons.rejected.connect(self.onReject)
self.layout.addWidget(self.buttons)
self._result = None
def __init__(self, parent: QWidget = None):
super().__init__(parent)
self._in = QDataStream()
self.blockSize = 0
self.currentFortune = ""
self.hostLineEdit = QLineEdit("fortune")
self.getFortuneButton = QPushButton(self.tr("Get Fortune"))
self.statusLabel = QLabel(
self.tr(
"This examples requires that you run the Local Fortune Server example as well."
)
)
self.socket = QLocalSocket()
self.setWindowFlags(self.windowFlags() & ~Qt.WindowContextHelpButtonHint)
hostLabel = QLabel(self.tr("&Server name:"))
hostLabel.setBuddy(self.hostLineEdit)
self.statusLabel.setWordWrap(True)
self.getFortuneButton.setDefault(True)
quitButton = QPushButton(self.tr("Quit"))
buttonBox = QDialogButtonBox()
buttonBox.addButton(self.getFortuneButton, QDialogButtonBox.ActionRole)
buttonBox.addButton(quitButton, QDialogButtonBox.RejectRole)
self._in.setDevice(self.socket)
self._in.setVersion(QDataStream.Qt_5_10)
self.hostLineEdit.textChanged.connect(self.enableGetFortuneButton)
self.getFortuneButton.clicked.connect(self.requestNewFortune)
quitButton.clicked.connect(self.close)
self.socket.readyRead.connect(self.readFortune)
self.socket.errorOccurred.connect(self.displayError)
mainLayout = QGridLayout(self)
mainLayout.addWidget(hostLabel, 0, 0)
mainLayout.addWidget(self.hostLineEdit, 0, 1)
mainLayout.addWidget(self.statusLabel, 2, 0, 1, 2)
mainLayout.addWidget(buttonBox, 3, 0, 1, 2)
self.setWindowTitle(QGuiApplication.applicationDisplayName())
self.hostLineEdit.setFocus()
def __init__(self,
name='messageBox',
width=None,
height=None,
enable_input_edit=False,
enable_dont_show_checkbox=False,
parent=None):
super(MessageBox, self).__init__(parent=parent)
self._frame = None
self._animation = None
self._dont_show_checkbox = False
self._clicked_button = None
self._clicked_standard_button = None
self.setMinimumWidth(width or self.MAX_WIDTH)
self.setMinimumHeight(height or self.MAX_HEIGHT)
self.setObjectName(name)
self.setWindowFlags(self.windowFlags() | Qt.FramelessWindowHint)
self.setAttribute(Qt.WA_TranslucentBackground)
# self.setStyleSheet('background-color: rgb(68, 68, 68, 255);')
parent = self.parent()
self._frame = None
if parent and parent != dcc.get_main_window():
parent.installEventFilter(self)
self._frame = QFrame(parent)
self._frame.setStyleSheet(
'background-color: rgba(25, 25, 25, 150);')
self._frame.setObjectName('messageBoxFrame')
self._frame.show()
self.setParent(self._frame)
self.main_layout = layouts.VerticalLayout(spacing=0,
margins=(0, 0, 0, 0))
self.setLayout(self.main_layout)
self._header = QFrame(self)
self._header.setFixedHeight(46)
self._header.setObjectName('messageBoxHeaderFrame')
self._header.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Fixed)
self._icon = label.BaseLabel(parent=self._header)
self._icon.hide()
self._icon.setFixedHeight(32)
self._icon.setFixedHeight(32)
self._icon.setScaledContents(True)
self._icon.setAlignment(Qt.AlignTop)
self._icon.setSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
self._title = label.BaseLabel(parent=self._header)
self._title.setObjectName('messageBoxHeaderLabel')
self._title.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding)
hlayout = layouts.HorizontalLayout(spacing=10, margins=(15, 7, 15, 10))
hlayout.addWidget(self._icon)
hlayout.addWidget(self._title)
self._header.setLayout(hlayout)
body_layout = layouts.VerticalLayout()
self._body = QFrame(self)
self._body.setObjectName('messageBoxBody')
self._body.setLayout(body_layout)
self._message = label.BaseLabel(parent=self._body)
self._message.setWordWrap(True)
self._message.setMinimumHeight(15)
self._message.setAlignment(Qt.AlignLeft)
self._message.setTextInteractionFlags(Qt.TextSelectableByMouse)
self._message.setSizePolicy(QSizePolicy.Expanding,
QSizePolicy.Expanding)
body_layout.addWidget(self._message)
body_layout.setContentsMargins(15, 15, 15, 15)
if enable_input_edit:
self._input_edit = lineedit.BaseLineEdit(parent=self._body)
self._input_edit.setObjectName('messageBoxInputEdit')
self._input_edit.setMinimumHeight(32)
self._input_edit.setFocus()
body_layout.addStretch(1)
body_layout.addWidget(self._input_edit)
body_layout.addStretch(10)
if enable_dont_show_checkbox:
msg = 'Do not show this message again'
self._dont_show_checkbox = checkbox.BaseCheckBox(msg,
parent=self._body)
body_layout.addStretch(10)
body_layout.addWidget(self._dont_show_checkbox)
body_layout.addStretch(2)
self._button_box = QDialogButtonBox(None, Qt.Horizontal, self)
self._button_box.clicked.connect(self._on_clicked)
self._button_box.accepted.connect(self._on_accept)
self._button_box.rejected.connect(self._on_reject)
self.main_layout.addWidget(self._header)
self.main_layout.addWidget(self._body)
body_layout.addWidget(self._button_box)
self.updateGeometry()
class Dialog(QDialog):
def __init__(self, parent: QWidget = None) -> None:
super().__init__(parent)
self.tcpServer = QTcpServer()
self.tcpClient = QTcpSocket()
self.tcpServerConnection: QTcpSocket = None
self.bytesToWrite = 0
self.bytesWritten = 0
self.bytesReceived = 0
self.clientProgressBar = QProgressBar()
self.clientStatusLabel = QLabel(self.tr("Client ready"))
self.serverProgressBar = QProgressBar()
self.serverStatusLabel = QLabel(self.tr("Server ready"))
self.startButton = QPushButton(self.tr("&Start"))
self.quitButton = QPushButton(self.tr("&Quit"))
self.buttonBox = QDialogButtonBox()
self.buttonBox.addButton(self.startButton, QDialogButtonBox.ActionRole)
self.buttonBox.addButton(self.quitButton, QDialogButtonBox.RejectRole)
self.startButton.clicked.connect(self.start)
self.quitButton.clicked.connect(self.close)
self.tcpServer.newConnection.connect(self.acceptConnection)
self.tcpClient.connected.connect(self.startTransfer)
self.tcpClient.bytesWritten.connect(self.updateClientProgress)
self.tcpClient.error.connect(self.displayError)
mainLayout = QVBoxLayout(self)
mainLayout.addWidget(self.clientProgressBar)
mainLayout.addWidget(self.clientStatusLabel)
mainLayout.addWidget(self.serverProgressBar)
mainLayout.addWidget(self.serverStatusLabel)
mainLayout.addStretch(1)
mainLayout.addSpacing(10)
mainLayout.addWidget(self.buttonBox)
@Slot()
def start(self):
self.startButton.setEnabled(False)
QGuiApplication.setOverrideCursor(Qt.WaitCursor)
self.bytesWritten = 0
self.bytesReceived = 0
while not self.tcpServer.isListening() and not self.tcpServer.listen():
ret = QMessageBox.critical(
self,
self.tr("Loopback"),
self.tr(
"Unable to start the test: %s" % (self.tcpServer.errorString())
),
QMessageBox.Retry | QMessageBox.Cancel,
)
if ret == QMessageBox.Cancel:
return
self.serverStatusLabel.setText(self.tr("Listening"))
self.clientStatusLabel.setText(self.tr("Connecting"))
self.tcpClient.connectToHost(
QHostAddress.LocalHost, self.tcpServer.serverPort()
)
@Slot()
def acceptConnection(self):
self.tcpServerConnection = self.tcpServer.nextPendingConnection()
if not self.tcpServerConnection:
self.serverStatusLabel.setText(
self.tr("Error: got invalid pending connection!")
)
return
self.tcpServerConnection.readyRead.connect(self.updateServerProgress)
self.tcpServerConnection.error.connect(self.displayError)
self.tcpServerConnection.disconnected.connect(
self.tcpServerConnection.deleteLater
)
self.serverStatusLabel.setText(self.tr("Accepted connection"))
self.tcpServer.close()
@Slot()
def startTransfer(self):
# called when the TCP client connected to the loopback server
self.bytesToWrite = TOTAL_BYTES - int(
self.tcpClient.write(QByteArray(PAYLOAD_SIZE, "@"))
)
self.clientStatusLabel.setText(self.tr("Connected"))
@Slot()
def updateServerProgress(self):
self.bytesReceived += int(self.tcpServerConnection.bytesAvailable())
self.tcpServerConnection.readAll()
self.serverProgressBar.setMaximum(TOTAL_BYTES)
self.serverProgressBar.setValue(self.bytesReceived)
self.serverStatusLabel.setText(
self.tr("Received %dMB" % (self.bytesReceived / (1024 * 1024),))
)
if self.bytesReceived == TOTAL_BYTES:
self.tcpServerConnection.close()
self.startButton.setEnabled(True)
QGuiApplication.restoreOverrideCursor()
@Slot("qint64")
def updateClientProgress(self, numBytes):
self.bytesWritten += int(numBytes)
if self.bytesToWrite > 0 and self.tcpClient.bytesToWrite() <= 4 * PAYLOAD_SIZE:
self.bytesToWrite -= self.tcpClient.write(
QByteArray(min(self.bytesToWrite, PAYLOAD_SIZE), "@")
)
self.clientProgressBar.setMaximum(TOTAL_BYTES)
self.clientProgressBar.setValue(self.bytesWritten)
self.clientStatusLabel.setText(
self.tr("Sent %dMB" % (self.bytesWritten / (1024 * 1024),))
)
@Slot(QAbstractSocket.SocketError)
def displayError(self, socketError):
if socketError == QTcpSocket.RemoteHostClosedError:
return
QMessageBox.information(
self,
self.tr("Network error"),
self.tr(
"The following error occurred: {}.".format(self.tcpClient.errorString())
),
)
self.tcpClient.close()
self.tcpServer.close()
self.clientProgressBar.reset()
self.serverProgressBar.reset()
self.clientStatusLabel.setText(self.tr("Client ready"))
self.serverStatusLabel.setText(self.tr("Server ready"))
self.startButton.setEnabled(True)
QGuiApplication.restoreOverrideCursor()
class InformationWindow(QDialog):
imageChanged = Signal(int, str)
def __init__(self, id_, items, parent=None):
super().__init__(parent)
itemLabel = QLabel(self.tr("Item: "))
descriptionLabel = QLabel(self.tr("Description: "))
imageFileLabel = QLabel(self.tr("Image file: "))
self.createButtons()
self.itemText = QLabel()
self.descriptionEditor = QTextEdit()
self.imageFileEditor = QComboBox()
self.imageFileEditor.setModel(items.relationModel(1))
self.imageFileEditor.setModelColumn(
items.relationModel(1).fieldIndex("file"))
self.mapper = QDataWidgetMapper(self)
self.mapper.setModel(items)
self.mapper.setSubmitPolicy(QDataWidgetMapper.ManualSubmit)
self.mapper.setItemDelegate(QSqlRelationalDelegate(self.mapper))
self.mapper.addMapping(self.imageFileEditor, 1)
self.mapper.addMapping(self.itemText, 2, b"text")
self.mapper.addMapping(self.descriptionEditor, 3)
self.mapper.setCurrentIndex(id_)
self.descriptionEditor.textChanged.connect(self.enableButtons)
self.imageFileEditor.currentIndexChanged.connect(self.enableButtons)
formLayout = QFormLayout()
formLayout.addRow(itemLabel, self.itemText)
formLayout.addRow(imageFileLabel, self.imageFileEditor)
formLayout.addRow(descriptionLabel, self.descriptionEditor)
layout = QVBoxLayout()
layout.addLayout(formLayout)
layout.addWidget(self.buttonBox)
self.setLayout(layout)
self.itemId = id_
self.displayedImage = self.imageFileEditor.currentText()
self.setWindowFlags(Qt.Window)
self.enableButtons(False)
self.setWindowTitle(self.itemText.text())
def id(self):
return self.itemId
@Slot()
def revert(self):
self.mapper.revert()
self.enableButtons(False)
@Slot()
def submit(self):
newImage = self.imageFileEditor.currentText()
if self.displayedImage != newImage:
self.displayedImage = newImage
self.imageChanged.emit(self.itemId, newImage)
self.mapper.submit()
self.mapper.setCurrentIndex(self.itemId)
self.enableButtons(False)
def createButtons(self):
self.closeButton = QPushButton(self.tr("&Close"))
self.revertButton = QPushButton(self.tr("&Revert"))
self.submitButton = QPushButton(self.tr("&Submit"))
self.closeButton.setDefault(True)
self.closeButton.clicked.connect(self.close)
self.revertButton.clicked.connect(self.revert)
self.submitButton.clicked.connect(self.submit)
self.buttonBox = QDialogButtonBox(self)
self.buttonBox.addButton(self.submitButton,
QDialogButtonBox.AcceptRole)
self.buttonBox.addButton(self.revertButton, QDialogButtonBox.ResetRole)
self.buttonBox.addButton(self.closeButton, QDialogButtonBox.RejectRole)
@Slot()
@Slot(bool)
def enableButtons(self, enable=True):
self.revertButton.setEnabled(enable)
self.submitButton.setEnabled(enable)