feat(account): 初步添加账户删除功能

src/budget/_account.py

@@ -15,23 +15,144 @@ class AccountTab(TabPage):
         # 创建连接信息控件
         self.accountComboBox = QComboBox()
         
-        self.accountAddButton = QPushButton('Add New account')
+        self.accountModifyButton = QPushButton('Modify')
-        self.accountAddButton.clicked.connect(self.onaccountAddClicked)
+        self.accountAddButton = QPushButton('Add')
+        self.accountModifyButton.clicked.connect(self.onAccountModifyClicked)
+        self.accountAddButton.clicked.connect(self.onAccountAddClicked)
 
         # 创建控件布局
         accountLayout = QHBoxLayout()
         accountLayout.addWidget(self.accountComboBox)
+        accountLayout.addWidget(self.accountModifyButton)
         accountLayout.addWidget(self.accountAddButton)
 
         accountWidget = QWidget()
         self.setLayout(accountLayout)
 
-    def onaccountAddClicked(self):
 
+    def onAccountDeleteClicked(self):
+
+        # 新建对话框，询问是否删除
+        dialog = QDialog(self)
+        dialog.setWindowTitle('Delete Account')
+        dialog.resize(300, 200)
+
+        label1 = QLabel("Are you sure to delete this account?")
+        label2 = QLabel("This action cannot be undone.")
+
+        buttonBox = QDialogButtonBox(QDialogButtonBox.StandardButton.Ok | QDialogButtonBox.StandardButton.Cancel)
+        buttonBox.accepted.connect(dialog.accept)
+        buttonBox.rejected.connect(dialog.reject)
+
+        layout = QVBoxLayout()
+        layout.addWidget(label1)
+        layout.addWidget(label2)
+        layout.addWidget(buttonBox)
+
+        dialog.setLayout(layout)
+
+        # 若确认删除，则执行删除操作
+        if dialog.exec() == QDialog.DialogCode.Accepted:
+            index = self.accountComboBox.currentIndex()
+            if (index == -1):
+                QMessageBox.critical(self, 'Error', 'No account selected')
+                return
+            a_id = self.rows[index][0]
+            self.pg.execute("DELETE FROM account WHERE a_id=%s", (a_id,))
+            self.selected()
+            QMessageBox.information(self, 'Success', 'Account deleted')
+            # 关闭原来的对话框
+            self.aDialog.close()
+        
+
+
+    def onAccountModifyClicked(self):
         # 创建添加数据对话框
-        addDialog = QDialog(self)
+        self.aDialog = QDialog(self)
-        addDialog.setWindowTitle('Add New Account')
+        self.aDialog.setWindowTitle('Modify')
-        addDialog.resize(400, 300)
+        self.aDialog.resize(400, 300)
+
+        index = self.accountComboBox.currentIndex()
+        # 若为空，抛出错误
+        if (index == -1):
+            QMessageBox.critical(self, 'Error', 'No account selected')
+            return
+        
+        # 构造查询语句
+        a_id = self.rows[index][0]
+        print(a_id)
+        self.pg.execute(
+            "SELECT meta->>'name', meta->>'balance', meta->'description' FROM account WHERE a_id=%s", (a_id,))
+        data = self.pg.fetchall()
+        print(data)
+
+        # 绘制界面并填充数据
+        label1 = QLabel("Account Name:")
+        line1 = QLineEdit()
+        line1.setText(data[0][0])
+
+        label2 = QLabel("Account Balance:")
+        line2 = QLineEdit()
+        line2.setText(f'{float(data[0][1]):.2f}')
+        # 只允许输入两位小数
+        line2.setValidator(QDoubleValidator(0.00, 999999999.99, 2))
+
+        label3 = QLabel("Description:")
+        line3 = QLineEdit()
+        line3.setText(data[0][2])
+
+        button1 = QPushButton("Confirm")
+        button3 = QPushButton("Delete")
+        button2 = QPushButton("Abort")
+
+        buttonLayout = QHBoxLayout()
+        buttonLayout.addWidget(button1)
+        buttonLayout.addWidget(button3)
+        buttonLayout.addWidget(button2)
+
+        button1.clicked.connect(self.aDialog.accept)
+        button3.clicked.connect(self.onAccountDeleteClicked)
+        button2.clicked.connect(self.aDialog.reject)
+
+        layout = QFormLayout()
+        layout.addRow(label1,line1)
+        layout.addRow(label2,line2)
+        layout.addRow(label3,line3)
+        layout.addRow(buttonLayout)
+
+        self.aDialog.setLayout(layout)
+
+        # 显示添加数据对话框
+        if self.aDialog.exec() == QDialog.DialogCode.Accepted:
+            try:
+                # 获取输入数据
+                name = line1.text()
+                balance = float(line2.text())
+
+                description = line3.text()
+                if (name == ''):
+                    raise Exception('Account name cannot be empty')
+
+                # JSONB插入数据
+                data = {"name": line1.text(), "balance": balance, "description": description}
+                data = json.dumps(data)
+                print("insert data:",data)
+                
+                self.pg.execute("UPDATE account SET meta=%s WHERE a_id=%s", (data, a_id))
+
+                # 刷新表格
+                self.selected()
+
+            except Exception as e:
+                # 处理插入数据错误
+                print(e)
+                QMessageBox.critical(self, 'Error', str(e))
+
+    def onAccountAddClicked(self):
+        # 创建添加数据对话框
+        self.aDialog = QDialog(self)
+        self.aDialog.setWindowTitle('Add New Account')
+        self.aDialog.resize(400, 300)
 
         label1 = QLabel("Account Name:")
         line1 = QLineEdit()
@@ -49,8 +170,8 @@ class AccountTab(TabPage):
         buttonLayout.addWidget(button1)
         buttonLayout.addWidget(button2)
 
-        button1.clicked.connect(addDialog.accept)
+        button1.clicked.connect(self.aDialog.accept)
-        button2.clicked.connect(addDialog.reject)
+        button2.clicked.connect(self.aDialog.reject)
 
         layout = QFormLayout()
         layout.addRow(label1,line1)
@@ -58,10 +179,10 @@ class AccountTab(TabPage):
         layout.addRow(label3,line3)
         layout.addRow(buttonLayout)
 
-        addDialog.setLayout(layout)
+        self.aDialog.setLayout(layout)
 
         # 显示添加数据对话框
-        if addDialog.exec() == QDialog.DialogCode.Accepted:
+        if self.aDialog.exec() == QDialog.DialogCode.Accepted:
             try:
                 # 获取输入数据
                 name = line1.text()