rijizhang/parser_factory.py

from ccb_parser import CCBFlowParser
from boc_parser import BOCFlowParser
from abc_parser import ABCFlowParser
from wechat_parser import WeChatFlowParser
from nongxin_parser import NongxinFlowParser


class ParserFactory:
    """银行流水解析器工厂类"""
    
    def __init__(self):
        self.parsers = {
                "建设银行": CCBFlowParser(),
                "中国银行": BOCFlowParser(),
                "农业银行": ABCFlowParser(),
                "微众银行": WeChatFlowParser(),
                "农信银行": NongxinFlowParser(),
                "农信银行芙蓉支行": NongxinFlowParser()
            }
    
    def get_parser(self, bank_name):
        """根据银行名称获取对应的解析器"""
        for key, parser in self.parsers.items():
            if key in bank_name:
                return parser
        return None
    
    def add_parser(self, bank_name, parser):
        """添加新的银行解析器"""
        self.parsers[bank_name] = parser
    
    def get_supported_banks(self):
        """获取支持的银行列表"""
        return list(self.parsers.keys())