【日常收支账本】【Day05】编辑账本界面增加删除、更新记录功能——提高代码复用性

一、项目地址

https://github.com/LinFeng-BingYi/DailyAccountBook

二、新增

1. 增加删除记录功能

1.1 功能详述

• 点击删除按钮后，获取对应行的数据组成字典，用字典的键值对匹配到对应日期的记录元素；
• 接着用该字典数据冲正存款账户余额（实现思路为新增记录时的反向操作），同时删除记录元素；
• 最后再更新表格。

1.2 代码实现

    def deleteTableRow(self, triggeredBtn, tableWidget):
        print("触发了删除按钮")
        # 获取触发信号的控件所在行号
        current_row = tableWidget.indexAt(triggeredBtn.parent().pos()).row()
        if tableWidget == self.tableWidget_expense:
            const_class = expenseConst
            action = 'expense'
        elif tableWidget == self.tableWidget_income:
            const_class = incomeConst
            action = 'income'
        elif tableWidget == self.tableWidget_movement:
            const_class = movementConst
            action = 'movement'
        else:
            print('未知控件触发新增按钮！')
            return
        # 获取待删除行的数据
        old_data_dict = self.getExistTableCell(tableWidget, current_row, const_class)
        if old_data_dict is None:
            print("获取待删除数据失败！！")
            return
        print("待删除记录内容为: \n", old_data_dict)

        if self.file_processor.deleteRecord(old_data_dict, self.dateEdit.text().replace('/', ''), action):
            print("删除成功！")
            # 把删除结果写入文件
            self.file_processor.writeXMLFile(self.lineEdit_file_path.text())

            # 更新self.file_parse_result以及记录表
            records_list = self.file_parse_result[action + 's']
            records_list.pop(current_row)
            self.updateRecordTable(tableWidget, records_list, const_class)

其中处理xml文件的方法代码如下：

    def deleteRecord(self, old_record_dict, date_str, action):
        e_date = self.getSpecificDateElement(date_str)
        if isinstance(e_date, int):
            print("未找到这一天的数据！")
            return False

        if action == 'expense':
            action_path = ".//expenses"
            record_path = """.//expense[@necessity='{}'][@associatedFund='{}'][value='{}'][category='{}'][detail='{}'][describe='{}'][from='{}']""".format(
                old_record_dict['necessity'],
                old_record_dict['associatedFund'],
                old_record_dict['value'],
                old_record_dict['category'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['from'])
        elif action == 'income':
            action_path = ".//incomes"
            record_path = """.//income[@associatedFund='{}'][value='{}'][category='{}'][detail='{}'][describe='{}'][to='{}']""".format(
                old_record_dict['associatedFund'],
                old_record_dict['value'],
                old_record_dict['category'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['to'])
        elif action == 'movement':
            action_path = ".//movements"
            record_path = """.//movement[value='{}'][detail='{}'][describe='{}'][from='{}'][to='{}']""".format(
                old_record_dict['value'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['from'],
                old_record_dict['to'])
        else:
            print("未知的动账操作！！")
            return False
        # print(record_path)

        e_action = e_date.find(action_path)
        e_record = e_action.find(record_path)
        if e_record is None:
            print("未找到待删除的记录")
            return False
        e_action.remove(e_record)

        if action == 'movement':
            self.modifyBalance(old_record_dict['from'], Decimal(old_record_dict['value']))
            self.modifyBalance(old_record_dict['to'], Decimal(old_record_dict['value']) * (-1))
        else:
            self.reversalVariation(old_record_dict, e_date)
        return True

    def reversalVariation(self, change_dict, e_date):
        """
        Describe: 删除某条记录时，需要冲正原来的记录，将回退对应的存款账户数值变化、以及余额

        Args:
            change_dict: dict
                记录字典
            e_date: Element
                指定日期的day元素
        """
        e_variation = e_date.find(".//variation")
        if e_variation is None:
            print("冲正记录时异常！！该记录不存在余额变化")
            return
        if 'from' in change_dict:
            e_fund_variety = e_variation.find(".//fund[category='{}']/out".format(change_dict['from']))
            if e_fund_variety is None:
                print("冲正记录时异常！！该记录不存在余额变化")
                return
            e_fund_variety.text = str((Decimal(e_fund_variety.text) - Decimal(change_dict['value'])).quantize(Decimal('0.00')))
            self.modifyBalance(change_dict['from'], Decimal(change_dict['value']))

            self.reversalAssociatedFund(e_variation, change_dict, 'from')
        if 'to' in change_dict:
            e_fund_variety = e_variation.find(".//fund[category='{}']/in".format(change_dict['to']))
            if e_fund_variety is None:
                print("冲正记录时异常！！该记录不存在余额变化")
                return
            e_fund_variety.text = str((Decimal(e_fund_variety.text) - Decimal(change_dict['value'])).quantize(Decimal('0.00')))
            self.modifyBalance(change_dict['to'], Decimal(change_dict['value'])*(-1))

            self.reversalAssociatedFund(e_variation, change_dict, 'to')

    def reversalAssociatedFund(self, e_variation, change_dict, from_or_to):
        """
        Describe: 冲正关联账户

        Args:
            e_variation: Element
            change_dict: dict
            from_or_to: ['from', 'to']
        """
        # print(change_dict['associatedFund'])
        if 'associatedFund' in change_dict and change_dict['associatedFund'] != 'None':
            print('执行了associatedFund冲正，操作为', from_or_to)
            if e_variation.find(".//fund[category='{}']".format(change_dict['associatedFund'])) is None:
                print("冲正记录时异常！！该记录不存在关联账户余额变化")
                return
            if from_or_to == 'from':
                e_fund_variety = e_variation.find(".//fund[category='{}']/out".format(change_dict['associatedFund']))
                flag = 1
            elif from_or_to == 'to':
                e_fund_variety = e_variation.find(".//fund[category='{}']/in".format(change_dict['associatedFund']))
                flag = -1
            else:
                print('未知的收支动作！')
                return
            e_fund_variety.text = str((Decimal(e_fund_variety.text) - Decimal(change_dict['value'])).quantize(Decimal('0.00')))
            self.modifyBalance(change_dict['associatedFund'], Decimal(change_dict['value'])*flag)

2. 增加更新记录功能

2.1 功能详述

• 点击更新按钮后，获取对应行的数据组成新记录字典；
• 同时根据行号获取编辑账本界面的self.file_parse_result属性对应的旧记录字典（每次更新xml文件时，该属性都会同步）；
• 接着用旧字典数据冲正存款账户余额（实现思路为新增记录时的反向操作），再用新字典数据更新账户余额，>- 最后再更新self.file_parse_result属性。

2.2 代码实现

    def updateTableRow(self, triggeredBtn, tableWidget):
        print("触发了新增按钮")
        # 获取触发信号的控件所在行号
        current_row = tableWidget.indexAt(triggeredBtn.parent().pos()).row()
        if tableWidget == self.tableWidget_expense:
            const_class = expenseConst
            action = 'expense'
        elif tableWidget == self.tableWidget_income:
            const_class = incomeConst
            action = 'income'
        elif tableWidget == self.tableWidget_movement:
            const_class = movementConst
            action = 'movement'
        else:
            print('未知控件触发新增按钮！')
            return
        # 获取被更新的旧数据(文件解析后，记录顺序与表格顺序相同)
        old_data_dict = self.file_parse_result[action+'s'][current_row]
        print("旧记录内容为: \n", old_data_dict)
        # 获取更新后的新数据
        new_data_dict = self.getExistTableCell(tableWidget, current_row, const_class)
        if new_data_dict is None:
            print("获取更新后的数据失败！！")
            return
        print("更新后的记录内容为: \n", new_data_dict)

        if self.file_processor.updateRecord(old_data_dict, new_data_dict, self.dateEdit.text().replace('/', ''), action):
            print("更新成功！")
            # 把删除结果写入文件
            self.file_processor.writeXMLFile(self.lineEdit_file_path.text())

            # 更新self.file_parse_result
            self.file_parse_result[action+'s'][current_row] = new_data_dict
            # print(self.file_parse_result)

其中处理xml文件的方法代码如下：

    def updateRecord(self, old_record_dict, new_record_dict, date_str, action):
        e_date = self.getSpecificDateElement(date_str)
        if isinstance(e_date, int):
            print("未找到这一天的数据！")
            return False

        if action == 'expense':
            action_path = ".//expenses"
            record_path = """.//expense[@necessity='{}'][@associatedFund='{}'][value='{}'][category='{}'][detail='{}'][describe='{}'][from='{}']""".format(
                old_record_dict['necessity'],
                old_record_dict['associatedFund'],
                old_record_dict['value'],
                old_record_dict['category'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['from'])
        elif action == 'income':
            action_path = ".//incomes"
            record_path = """.//income[@associatedFund='{}'][value='{}'][category='{}'][detail='{}'][describe='{}'][to='{}']""".format(
                old_record_dict['associatedFund'],
                old_record_dict['value'],
                old_record_dict['category'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['to'])
        elif action == 'movement':
            action_path = ".//movements"
            record_path = """.//movement[value='{}'][detail='{}'][describe='{}'][from='{}'][to='{}']""".format(
                old_record_dict['value'],
                old_record_dict['detail'],
                old_record_dict['describe'],
                old_record_dict['from'],
                old_record_dict['to'])
        else:
            print("未知的动账操作！！")
            return False
        # print(record_path)

        e_action = e_date.find(action_path)
        e_record = e_action.find(record_path)
        if e_record is None:
            print("未找到待删除的记录")
            return False

        # 修改了数值则需要冲正
        if old_record_dict['value'] != new_record_dict['value']:
            # 先冲正原记录数据
            # 在用新数据修改账户变化和余额
            if action == 'movement':
                self.modifyBalance(old_record_dict['from'], Decimal(old_record_dict['value']))
                self.modifyBalance(old_record_dict['to'], Decimal(old_record_dict['value']) * (-1))
                self.modifyBalance(new_record_dict['from'], Decimal(new_record_dict['value']) * (-1))
                self.modifyBalance(new_record_dict['to'], Decimal(new_record_dict['value']))
            else:
                self.reversalVariation(old_record_dict, e_date)
                self.organizeVariation(new_record_dict, e_date)

        # 修改记录数据

        for key in new_record_dict.keys():
            if key == 'necessity':
                e_record.attrib['necessity'] = new_record_dict['necessity']
            elif key == 'associatedFund':
                e_record.attrib['associatedFund'] = new_record_dict['associatedFund']
            else:
                e_record.find(".//"+key).text = str(new_record_dict[key])
        return True

3. 优化界面控件（内置spinBox、增加按钮切换日期）

3.1 功能详述

将记录表格数值(value)列内置QSpinBox，以避免存于文件的数值小数点位不一致；增加按钮来控制日期的切换

3.2 效果展示

数值(value)列内置QSpinBox

增加按钮来控制日期的切换

3.3 代码实现

# 内置QDoubleSpinBox
spinBox = QDoubleSpinBox(decimals=2)# 设置保留小数后2位
# 设置范围，默认为[0.0, 100.0)。必须在调整数值前设置，否则大于等于100.0时会变成99.99
spinBox.setRange(0.0, 100000000.0)
spinBox.setSingleStep(0.1)          # 设置步长
spinBox.setValue(value)             # 调整数值
tableWidget.setCellWidget(current_row, keys_list.index(key), spinBox)

# 日期切换
# 利用QDate类的addDays(int)方法
self.pushButton_prev_day.clicked.connect(lambda: self.dateEdit.setDate(self.dateEdit.date().addDays(-1)))
self.pushButton_post_day.clicked.connect(lambda: self.dateEdit.setDate(self.dateEdit.date().addDays(1)))

三、开发总结

1. 提高代码复用性

最近几篇都没有技术含量，就不总结了。唯一值得一提的就是提高代码复用性。

• 对于支出、收入、转移三种动账类型，字段大量重复，于是创建表格时，将所有列的初始化集中到一个函数中，根据列名执行对应初始化方式。
• 通过设置ExpenseConst、IncomeConst、MovementConst三个具有同名常量的类，更好地支持代码复用
• 第二章删除和新增记录时修改xml文件的方法deleteRecord(old_record_dict, date_str, action)和updateRecord(self, old_record_dict, new_record_dict, date_str, action)是通过action区分动账类型，避免同一个功能需要实现三个方法