from os import environ

from dotenv import load_dotenv

from src.connector import execute

with open("sql/create_oneoff_table.sql", "r") as f:
    CREATE_ONEOFF_TABLE_SQL = f.read()

with open("sql/load_from_stage.sql", "r") as f:
    LOAD_FROM_STAGE_SQL = f.read()

# Loads a oneoff investigation run into its own table. Unlike app.py, this
# does NOT append to HIVE_MODEL_RESPONSE_TRACKING — oneoff assets are not
# part of the regression set.
if __name__ == "__main__":
    load_dotenv()
    run_id = environ["RUN_ID"]
    table_name = environ["TABLE_NAME"]

    execute("SELECT 1")
    execute(CREATE_ONEOFF_TABLE_SQL.format(table_name=table_name))
    execute(LOAD_FROM_STAGE_SQL.format(run_id=run_id, table_name=table_name))