from tracker.bangers import run_frunches
from tracker.logger import get_logger
from tracker.unicorn import artist_ingestion

from pycommon.database import db as cdb

logger = get_logger(__name__)

cdb.setup_randh_session()
logger.info("Starting update_viral_charts")


include_daily_streams = True
logger.info(f"Ingesting charts, include_daily_streams={include_daily_streams}")
try:
    artist_ingestion.process_all_viral(include_daily_streams, force=True)
except artist_ingestion.ListenerImportFailure as e:
    logger.error(str(e))
    logger.warn("Failed to import all listeners, proceeding without.")

logger.info(f"Viral charts updating.  Creating frunch sheet.")
run_frunches(
    ["spotifyViral"],
    include_previous=False,
    store_chart_in_db=True,
    title_prefix="Spotify Viral",
)