from typing import Callable

from update_market_vendors.constants import Vendor
from update_market_vendors.vendors.apple import get_markets as get_markets_apple
from update_market_vendors.vendors.spotify import get_markets as get_markets_spotify


_GETTERS = {
    Vendor.APPLE: get_markets_apple,
    Vendor.SPOTIFY: get_markets_spotify
}


def get_markets_by_vendor(vendor: Vendor) -> Callable:
    return _GETTERS[vendor]