diff --git a/src/budget/_account.py b/src/budget/_account.py index dbd128f..60d46ae 100644 --- a/src/budget/_account.py +++ b/src/budget/_account.py @@ -94,6 +94,8 @@ class AccountTab(TabPage): data = self.pg.fetchall() print(data) + oldBalance = float(data[0][1]) + # 绘制界面并填充数据 accountNameLabel = QLabel("Account Name:") accountNameLine = QLineEdit() @@ -150,6 +152,14 @@ class AccountTab(TabPage): self.pg.execute("UPDATE account SET meta=%s WHERE a_id=%s", (data, a_id)) + if oldBalance != balance: + # 修改账户时,在transaction表中添加一条记录 + data = {"type":"modify", "description": "Account Modify"} + data = json.dumps(data) + self.pg.execute( + "INSERT INTO transaction (a_id, time, amount, meta) VALUES (%s, now(), %s,%s)", (a_id, balance - oldBalance, data,)) + + # 刷新表格 self.selected()