test_nongxin_v2.py 1.1 KB

1234567891011121314151617181920212223242526272829
  1. from parser_factory import ParserFactory
  2. from journal_generator import JournalGenerator
  3. pf = ParserFactory()
  4. parser = pf.get_parser('农信银行')
  5. print('农信银行解析器:', parser)
  6. print('银行名称:', parser.get_bank_name())
  7. try:
  8. flow_df = parser.parse('C:\\Users\\EDY\\Desktop\\日记账1月\\新手游农信.xls')
  9. if flow_df is not None:
  10. print('解析成功,流水数据:')
  11. print(flow_df.head())
  12. print('\n数据形状:', flow_df.shape)
  13. jg = JournalGenerator()
  14. journal_df = jg.generate_journal_data('绵阳新手游科技技术有限公司', '农信银行', '参数表.xlsx', flow_df)
  15. if journal_df is not None:
  16. print('\n日记账数据:')
  17. print(journal_df[['日期', '月份', '收支', '资金分类-1级', '资金分类-2级', '资金分类-3级', '对手户', '备注']].head(10))
  18. else:
  19. print('日记账生成失败')
  20. else:
  21. print('流水解析失败')
  22. except Exception as e:
  23. print('发生错误:', str(e))
  24. import traceback
  25. traceback.print_exc()