from parser_factory import ParserFactory
from journal_generator import JournalGenerator

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

try:
    flow_df = parser.parse('C:\\Users\\EDY\\Desktop\\日记账1月\\新手游农信.xls')
    if flow_df is not None:
        print('解析成功，流水数据:')
        print(flow_df[['交易时间', '对方户名', '摘要', '收入', '支出', '余额']].head(10))
        print('\n数据形状:', flow_df.shape)
        
        jg = JournalGenerator()
        journal_df = jg.generate_journal_data('绵阳新手游科技技术有限公司', '农信银行', '参数表.xlsx', flow_df)
        if journal_df is not None:
            print('\n日记账数据形状:', journal_df.shape)
            print('\n前10行日记账:')
            print(journal_df[['日期', '月份', '收支', '资金分类-1级', '资金分类-2级', '资金分类-3级', '对手户', '备注']].head(10).to_string())
        else:
            print('日记账生成失败')
    else:
        print('流水解析失败')
except Exception as e:
    print('发生错误:', str(e))
    import traceback
    traceback.print_exc()