# coding:utf-8
from .html_preprocess import HtmlHelper
from typing import AnyStr
from .attach_preprocess import AttachHelper

class Preprocess(object):
    """
    预处理接口
    """

    def __init__(self):
        self.__pool = dict(html=HtmlHelper, attach=AttachHelper)

    def get_preprocess(self, category_name: AnyStr):
        """
        获取处理接口
        :param category_name: 处理接口名称
        :return: 处理接口对象
        """
        if category_name in self.__pool:
            return self.__pool[category_name]()
        else:
            raise ValueError(f"Preprocess doesn't have {category_name}")


__all__ = [Preprocess]