#! /usr/bin/env python

import os
from tracker.logger import get_logger

logger = get_logger(__name__)

from tracker import db
db.setup_session()

artist_id = os.environ.get("WL_RT_REGRESSION_ARTIST_ID")
username = os.environ.get("WL_RT_REGRESSION_USERNAME")
source = os.environ.get("WL_RT_REGRESSION_SOURCE")

if not artist_id or not source:
    raise RuntimeError(
        "Script must execute in env with WL_RT_REGRESSION_ARTIST_ID and  WL_RT_REGRESSION_SOURCE set"
    )

valid_sources = ("sc",)

if source not in valid_sources:
    raise RuntimeError(
        "Invalid WL_RT_REGRESSION_SOURCE, {}, must be one of {}".format(source, valid_sources)
    )

logger.info("Starting regression follower download {}/{}/{}".format(source, artist_id, username))
if source == "sc":
    from tracker import soundcloud

    soundcloud.start_regression_follower_download(username, int(artist_id))