from apollo_utils.core.utils.ext_enum import ExtEnum, extend_enum


class DISTRIBUTORS(ExtEnum):
    AWAL = "awal"
    ORCHARD = "theorchard"
    SME = "sme"


@extend_enum(DISTRIBUTORS)
class DISTRIBUTORS_FILTERS(ExtEnum):
    OTHER = "other"


DISTRIBUTORS_ORDER = (DISTRIBUTORS.SME, DISTRIBUTORS.ORCHARD, DISTRIBUTORS.AWAL)  # desc


# options for scenarios.distributors.is_sony scenarios
class IsSonyInclude:
    ALBUM_LIST = "album_list"
    ALBUM_UPC_MAPPING = "album_upc_mapping"
    TRACK_LIST = "track_list"
    TRACK_UPC_MAPPING = "track_upc_mapping"


IS_SONY_INCLUDE_DEFAULTS = {
    IsSonyInclude.ALBUM_LIST: list,
    IsSonyInclude.ALBUM_UPC_MAPPING: dict,
    IsSonyInclude.TRACK_LIST: list,
    IsSonyInclude.TRACK_UPC_MAPPING: dict,
}