from .article_attribute_getter import ArticleAttributeGetter
from .brand_attribute_getter import BrandAttributeGetter
from .client_name_attribute_getter import ClientNameAttributeGetter
from .color_attribute_getter import ColorAttributeGetter
from .company_name_attribute_getter import CompanyNameAttributeGetter
from .composition_attribute_getter import CompositionAttributeGetter
from .inn_attribute_getter import InnAttributeGetter
from .name_attribute_getter import NameAttributeGetter


class AttributeWriterFactory:
    @staticmethod
    def get_writer(key: str):
        match key:
            case 'name':
                return NameAttributeGetter()
            case 'article':
                return ArticleAttributeGetter()
            case 'client.name':
                return ClientNameAttributeGetter()
            case 'client.company_name':
                return CompanyNameAttributeGetter()
            case 'client.inn':
                return InnAttributeGetter()
            case 'brand':
                return BrandAttributeGetter()
            case 'color':
                return ColorAttributeGetter()
            case 'composition':
                return CompositionAttributeGetter()
            case _:
                return None