rijizhang/parser_factory.py

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


class ParserFactory:
    """银行流水解析器工厂类"""
    
    def __init__(self):
        self.parsers = {
            "建设银行": CCBFlowParser(),
            "中国银行": BOCFlowParser(),
            "农业银行": ABCFlowParser(),
            "微众银行": WeChatFlowParser()
        }
    
    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())