from backfill_lambda.config import Config
from backfill_lambda.jobs import (
    apple_playlists_backfill,
    dapd_apple_playlists_backfill,
    dapd_apple_playlists_backfill_2,
    dapd_spotify_playlists_backfill,
    dapd_spotify_playlists_countries_backfill,
    spotify_playlists_backfill,
)

jobs_list = {
    'dapd_spotify_playlists': dapd_spotify_playlists_backfill,
    'dapd_apple_music_playlists': dapd_apple_playlists_backfill,
    'dapd_apple_music_playlists_2': dapd_apple_playlists_backfill_2,
    'spotify_playlists': spotify_playlists_backfill,
    'apple_music_playlists': apple_playlists_backfill,
    'dapd_spotify_playlists_countries': dapd_spotify_playlists_countries_backfill,
}


config = Config()


def handler(event, context) -> None:
    report_type = event.get('report_type')
    job = jobs_list.get(report_type)

    if not job:
        raise Exception('Not implemented')

    return job(event, config)


if __name__ == '__main__':
    handler({'report_type': 'dapd_spotify_playlists_countries'}, {})