class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# Chart
self.chart_view = QChartView()
self.chart_view.setRenderHint(QPainter.Antialiasing)
# Right
self.description = QLineEdit()
self.quantity = QLineEdit()
self.add = QPushButton("Add")
self.clear = QPushButton("Clear")
self.quit = QPushButton("Quit")
self.plot = QPushButton("Plot")
# Disabling 'Add' button
self.add.setEnabled(False)
self.right = QVBoxLayout()
self.right.setContentsMargins(10, 10, 10, 10)
self.right_top = QGridLayout()
self.right_top.addWidget(QLabel("Description"), 0, 0, 1, 1)
self.right_top.addWidget(self.description, 0, 1, 1, 3)
self.right_top.addWidget(QLabel("Quantity"), 1, 0, 1, 1)
self.right_top.addWidget(self.quantity, 1, 1, 1, 1)
self.right_top.addWidget(self.add, 1, 2, 1, 2)
self.right.addLayout(self.right_top)
self.right.addWidget(self.chart_view)
self.right_bottom = QGridLayout()
self.right_bottom.addWidget(self.plot, 0, 0, 1, 2)
self.right_bottom.addWidget(self.clear, 1, 0)
self.right_bottom.addWidget(self.quit, 1, 1)
self.right.addLayout(self.right_bottom)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
self.layout.addLayout(self.right)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Signals and Slots
self.add.clicked.connect(self.add_element)
self.quit.clicked.connect(self.quit_application)
self.plot.clicked.connect(self.plot_data)
self.clear.clicked.connect(self.clear_table)
self.description.textChanged[str].connect(self.check_disable)
self.quantity.textChanged[str].connect(self.check_disable)
@Slot()
def add_element(self):
des = self.description.text()
qty = self.quantity.text()
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(des))
self.table.setItem(self.items, 1, QTableWidgetItem(qty))
self.description.setText("")
self.quantity.setText("")
self.items += 1
@Slot()
def check_disable(self, s):
if not self.description.text() or not self.quantity.text():
self.add.setEnabled(False)
else:
self.add.setEnabled(True)
@Slot()
def plot_data(self):
# Get table information
series = QBarSeries()
for i in range(self.table.rowCount()):
text = self.table.item(i, 0).text()
number = float(self.table.item(i, 1).text())
bar_set = QBarSet(text)
bar_set.append(number)
series.append(bar_set)
chart = QChart()
chart.addSeries(series)
chart.legend().setAlignment(Qt.AlignLeft)
self.chart_view.setChart(chart)
@Slot()
def quit_application(self):
QApplication.quit()
@Slot()
def clear_table(self):
self.table.setRowCount(0)
self.items = 0
class Widget(QWidget):
def __init__(self):
QWidget.__init__(self)
self.items = 0
# Example data
self._data = {
"Water": 24.5,
"Electricity": 55.1,
"Rent": 850.0,
"Supermarket": 230.4,
"Internet": 29.99,
"Spätkauf": 21.85,
"BVG Ticket": 60.0,
"Coffee": 22.45,
"Meetup": 0.0
}
# Left
self.table = QTableWidget()
self.table.setColumnCount(2)
self.table.setHorizontalHeaderLabels(["Description", "Quantity"])
self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
# Chart
self.chart_view = QChartView()
self.chart_view.setRenderHint(QPainter.Antialiasing)
# Right
self.description = QLineEdit()
self.quantity = QLineEdit()
self.add = QPushButton("Add")
self.clear = QPushButton("Clear")
self.quit = QPushButton("Quit")
self.plot = QPushButton("Plot")
# Disabling 'Add' button
self.add.setEnabled(False)
self.right = QVBoxLayout()
self.right.setContentsMargins(10, 10, 10, 10)
self.right.addWidget(QLabel("Description"))
self.right.addWidget(self.description)
self.right.addWidget(QLabel("Quantity"))
self.right.addWidget(self.quantity)
self.right.addWidget(self.add)
self.right.addWidget(self.plot)
self.right.addWidget(self.chart_view)
self.right.addWidget(self.clear)
self.right.addWidget(self.quit)
# QWidget Layout
self.layout = QHBoxLayout()
self.layout.addWidget(self.table)
self.layout.addLayout(self.right)
# Set the layout to the QWidget
self.setLayout(self.layout)
# Signals and Slots
self.add.clicked.connect(self.add_element)
self.quit.clicked.connect(self.quit_application)
self.plot.clicked.connect(self.plot_data)
self.clear.clicked.connect(self.clear_table)
self.description.textChanged[str].connect(self.check_disable)
self.quantity.textChanged[str].connect(self.check_disable)
# Fill example data
self.fill_table()
@Slot()
def add_element(self):
des = self.description.text()
qty = self.quantity.text()
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(des))
self.table.setItem(self.items, 1, QTableWidgetItem(qty))
self.description.setText("")
self.quantity.setText("")
self.items += 1
@Slot()
def check_disable(self, s):
if not self.description.text() or not self.quantity.text():
self.add.setEnabled(False)
else:
self.add.setEnabled(True)
@Slot()
def plot_data(self):
# Get table information
series = QPieSeries()
for i in range(self.table.rowCount()):
text = self.table.item(i, 0).text()
number = float(self.table.item(i, 1).text())
series.append(text, number)
chart = QChart()
chart.addSeries(series)
chart.legend().setAlignment(Qt.AlignLeft)
self.chart_view.setChart(chart)
@Slot()
def quit_application(self):
QApplication.quit()
def fill_table(self, data=None):
data = self._data if not data else data
for desc, price in data.items():
self.table.insertRow(self.items)
self.table.setItem(self.items, 0, QTableWidgetItem(desc))
self.table.setItem(self.items, 1, QTableWidgetItem(str(price)))
self.items += 1
@Slot()
def clear_table(self):
self.table.setRowCount(0)
self.items = 0
