多哥 发表于 2023-10-11 23:02:42

【日常收支账本】【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
            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='{}']""".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='{}']""".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""".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/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/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".format(change_dict['associatedFund'])) is None:
                print("冲正记录时异常!!该记录不存在关联账户余额变化")
                return
            if from_or_to == 'from':
                e_fund_variety = e_variation.find(".//fund/out".format(change_dict['associatedFund']))
                flag = 1
            elif from_or_to == 'to':
                e_fund_variety = e_variation.find(".//fund/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
      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 = 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='{}']""".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='{}']""".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""".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)
      return True3. 优化界面控件(内置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区分动账类型,避免同一个功能需要实现三个方法

来源:https://www.cnblogs.com/LinfengBingyi/p/17756104.html
免责声明:由于采集信息均来自互联网,如果侵犯了您的权益,请联系我们【E-Mail:cb@itdo.tech】 我们会及时删除侵权内容,谢谢合作!
页: [1]
查看完整版本: 【日常收支账本】【Day05】编辑账本界面增加删除、更新记录功能——提高代