from djagitit.tableau import refresh_extracts
from loguru import logger

from ..config import QUERY_PATHS, TABLEAU_DATASOURCE_IDS_LIVE, TABLEAU_SITE
from ..db import get_rdb
from ..db.queries import QueryLoader


def run_update() -> str | None:
    logger.info("Starting live update")
    rdb = get_rdb()
    ql = QueryLoader()

    logger.info("Rebuilding live hyper table")
    status = rdb.execute(ql.load(QUERY_PATHS.make_live_hyper))
    if not status:
        logger.error("Failed to rebuild live hyper table")
        return None

    refresh_extracts(TABLEAU_DATASOURCE_IDS_LIVE, site=TABLEAU_SITE)
    logger.success("Live hyper table updated")
    return "ok"