from typing import List

from src.constants.core import VENDOR_APPLE, IdType, VendorType
from src.legacy.core.sony_check import check_is_sony
from src.legacy.core.vendor import VendorBase


class AppleMusicBase(VendorBase):
    vendor_name = VENDOR_APPLE

    def check_track_is_sony(self, ids_list: List[str], id_type: IdType, market: str = None) -> List[str]:
        """Method fetch sony/non-sony data from DSP-API for batch of tracks."""

        return check_is_sony(ids_list, id_type=id_type, vendor=VendorType.APPLE)