diff --git a/src/budget/_account.py b/src/budget/_account.py index 6560054..697c665 100644 --- a/src/budget/_account.py +++ b/src/budget/_account.py @@ -15,7 +15,11 @@ class AccountTab(TabPage): # 创建顶部控件 self.accountComboBox = QComboBox() self.accountComboBox.currentIndexChanged.connect(self.onAccountIndexChanged) - + + self.accountBalanceLine = QLineEdit() + self.accountBalanceLine.setReadOnly(True) + # 默认宽度 + self.accountBalanceLine.setFixedWidth(150) self.accountModifyButton = QPushButton('Modify') self.accountAddButton = QPushButton('Add') self.accountModifyButton.clicked.connect(self.onAccountModifyClicked) @@ -24,6 +28,7 @@ class AccountTab(TabPage): # 创建控件布局 topLayout = QHBoxLayout() topLayout.addWidget(self.accountComboBox) + topLayout.addWidget(self.accountBalanceLine) topLayout.addWidget(self.accountModifyButton) topLayout.addWidget(self.accountAddButton) @@ -220,11 +225,14 @@ class AccountTab(TabPage): meta->>'description' FROM transaction \ WHERE a_id = %s ORDER BY time DESC", (a_id,)) self.transData = self.pg.fetchall() - self.pg.execute("SELECT a_id, meta ->> 'name' FROM account ORDER BY a_id") + self.pg.execute("SELECT a_id, meta ->> 'name', balance FROM account ORDER BY a_id") self.accountData = self.pg.fetchall() self.pg.execute("SELECT c_id, meta ->> 'name' FROM category ORDER BY c_id") self.categoryData = self.pg.fetchall() + # 设置余额 + self.accountBalanceLine.setText(str(self.accountData[index][2])) + # 初始化表格 self.accountTable.clear() self.accountTable.setRowCount(len(self.transData))