from enum import Enum __all__ = ["ServiceType", "MusicService", "FieldCopyMode", "FlagType", "get_music_service"] class ServiceType(Enum): spotify = 0 deezer = 1 youtube = 2 # soundcloud = 3 # groove = 4 apple_music = 5 class MusicService(Enum): spotify = 1 deezer = 2 youtube = 3 # soundcloud = 4 # groove = 5 apple_music = 6 class FieldCopyMode(Enum): no_update = 0 use_setting = 1 copy_source = 2 class FlagType(Enum): whitelist = 0 blacklist = 1 _SERVICE_TYPE_MUSIC_SERVICE_MAP = { ServiceType.spotify: MusicService.spotify, ServiceType.deezer: MusicService.deezer, ServiceType.youtube: MusicService.youtube, # ServiceType.soundcloud: MusicService.soundcloud, # ServiceType.groove: MusicService.groove, ServiceType.apple_music: MusicService.apple_music, } def get_music_service(service_type: ServiceType) -> MusicService: return _SERVICE_TYPE_MUSIC_SERVICE_MAP[service_type]