from parser_factory import ParserFactory
from journal_generator import JournalGenerator

pf = ParserFactory()
parser = pf.get_parser('农信银行')
print('农信银行解析器:', parser)
print('银行名称:', parser.get_bank_name())

flow_df = parser.parse('C:\\Users\\EDY\\Desktop\\日记账1月\\新手游农信.xls')
print('解析成功，流水数据:')
print(flow_df.head())

jg = JournalGenerator()
journal_df = jg.generate_journal_data('绵阳新手游科技技术有限公司', '农信银行', '参数表.xlsx', flow_df)
print('\n日记账数据:')
print(journal_df[['日期', '月份', '收支', '资金分类-1级', '资金分类-2级', '资金分类-3级', '对手户', '备注']].head(10))