import yaml
import utils


def create_playlistname_exclusions_table():
    """ Creates the playlist exclusions table if it deosn't exist."""

    sql = """
        CREATE OR REPLACE TABLE intelligence.prod.playlistname_exclusions
        (playlistname VARCHAR);
        """

    utils.execute_sql(sql)


def insert_playlistname_exclusions():
    """Inserts Playlist exclusion words based on playlistname_exclusions.yml"""

    sql = """INSERT INTO intelligence.prod.playlistname_exclusions VALUES"""

    with open("playlistname_exclusions.yml", "r") as stream:
        data_loaded = yaml.safe_load(stream)

        for name in data_loaded["playlist_names"]:
            sql_new = sql + " ('{}')".format(name)

            utils.execute_sql(sql_new)


if __name__ == "__main__":

    create_playlistname_exclusions_table()
    insert_playlistname_exclusions()