from engine.core.feeds.amazon_seller import AmazonSellerFeedGenerator
from engine.core.feeds.base import BaseFeedGenerator
from engine.core.feeds.google_merchant import GoogleMerchantFeedGenerator
from engine.core.feeds.yandex_market import YandexMarketFeedGenerator
from engine.core.feeds.yandex_products import YandexProductsFeedGenerator

FEED_GENERATORS: dict[str, type[BaseFeedGenerator]] = {
    "google_merchant": GoogleMerchantFeedGenerator,
    "yandex_market": YandexMarketFeedGenerator,
    "yandex_products": YandexProductsFeedGenerator,
    "amazon_seller": AmazonSellerFeedGenerator,
}

__all__ = [
    "BaseFeedGenerator",
    "GoogleMerchantFeedGenerator",
    "YandexMarketFeedGenerator",
    "YandexProductsFeedGenerator",
    "AmazonSellerFeedGenerator",
    "FEED_GENERATORS",
]